LangChain integration for verified AI agents

pip install qwed-open-responses[langchain]

from langchain.agents import create_react_agent
from qwed_open_responses.middleware.langchain import QWEDCallbackHandler
from qwed_open_responses import ToolGuard, SafetyGuard

# Create callback with guards
callback = QWEDCallbackHandler(
    guards=[ToolGuard(), SafetyGuard()],
    block_on_failure=True,
)

# Add to agent
agent = create_react_agent(
    llm=llm,
    tools=tools,
    callbacks=[callback],
)

# Agent actions are now verified!
result = agent.invoke({"input": "Search for weather"})

from qwed_open_responses import ToolGuard

# Only allow safe tools
callback = QWEDCallbackHandler(
    guards=[
        ToolGuard(
            allowed_tools=["search", "calculator", "weather"],
            blocked_tools=["execute_shell", "write_file"],
        )
    ]
)

# If agent tries to call execute_shell:
# ToolCallBlocked: Tool call blocked: BLOCKED: Tool 'execute_shell' is not allowed

from qwed_open_responses import SafetyGuard

callback = QWEDCallbackHandler(
    guards=[
        SafetyGuard(
            check_pii=True,
            check_injection=True,
            max_cost=50.0,
        )
    ]
)

from qwed_open_responses.middleware.langchain import ToolCallBlocked

try:
    result = agent.invoke({"input": "Delete all files"})
except ToolCallBlocked as e:
    print(f"Action blocked: {e.result.block_reason}")
    # Log the attempted action
    log_security_event(e.action, e.result)

# After running agent
summary = callback.get_verification_summary()
print(f"Total verifications: {summary['total_verifications']}")
print(f"Passed: {summary['passed']}")
print(f"Failed: {summary['failed']}")
print(f"Success rate: {summary['success_rate']:.1%}")

from langchain_openai import ChatOpenAI
from langchain.agents import create_react_agent, AgentExecutor
from langchain.tools import Tool
from qwed_open_responses.middleware.langchain import QWEDCallbackHandler
from qwed_open_responses import ToolGuard, MathGuard, SafetyGuard

# Define tools
tools = [
    Tool(name="search", func=search_func, description="Search the web"),
    Tool(name="calculator", func=calc_func, description="Do math"),
]

# Create verified callback
callback = QWEDCallbackHandler(
    guards=[
        ToolGuard(allowed_tools=["search", "calculator"]),
        MathGuard(),
        SafetyGuard(),
    ],
    verbose=True,
)

# Create agent
llm = ChatOpenAI(model="gpt-4")
agent = create_react_agent(llm, tools, prompt)
executor = AgentExecutor(agent=agent, tools=tools, callbacks=[callback])

# Run with verification
result = executor.invoke({"input": "What is 15% of 200?"})
# [QWED] Tool: calculator -> [OK] Verified (3 guards passed)