Trustworthy AI starts here

import os
from deepagents import create_deep_agent
from langchain.chat_models import init_chat_model
from openbox_deepagent import create_openbox_middleware  # Added import

# Create OpenBox middleware
middleware = create_openbox_middleware(
    api_url=os.getenv("OPENBOX_URL"),
    api_key=os.getenv("OPENBOX_API_KEY"),
    agent_name="ResearchBot",

)

agent = create_deep_agent(
    model=init_chat_model("openai:gpt-4o-mini"),
    tools=[search_web, write_report, export_data],
    subagents=[
        {"name": "researcher", "tools": [search_web]},
        {"name": "writer", "tools": [write_report]},
    ],
    middleware=[middleware],  # Added middleware
)

result = await agent.ainvoke(
    {"messages": [{"role": "user", "content": "Research AI safety"}]},
    config={"configurable": {"thread_id": "session-001"}},
)

import os
from langgraph.graph import StateGraph, START, END, MessagesState
from openbox_langgraph import create_openbox_graph_handler  # Added import

graph = StateGraph(MessagesState)
graph.add_node("agent", call_model)
graph.add_node("tools", tool_node)
graph.add_edge(START, "agent")
graph.add_conditional_edges("agent", should_continue, {"tools": "tools", END: END})
graph.add_edge("tools", "agent")

app = graph.compile()

# Wrap with OpenBox governance
governed = create_openbox_graph_handler(
    graph=app,
    api_url=os.getenv("OPENBOX_URL"),
    api_key=os.getenv("OPENBOX_API_KEY"),
    agent_name="MyAgent",
)

result = await governed.ainvoke({"messages": [("user", "Hello")]})

import os
import asyncio
from temporalio.client import Client
from temporalio.worker import Worker
from openbox.plugin import OpenBoxPlugin  # Add OpenBox
from your_workflows import YourWorkflow
from your_activities import your_activity

async def main():
    client = await Client.connect("localhost:7233")

    worker = Worker(
        client,
        task_queue="agent-task-queue",
        workflows=[YourWorkflow],
        activities=[your_activity],
        # Add OpenBox plugin
        plugins=[OpenBoxPlugin(
            openbox_url=os.getenv("OPENBOX_URL"),
            openbox_api_key=os.getenv("OPENBOX_API_KEY"),
        )],
    )

    await worker.run()

asyncio.run(main())