import asyncio
from omnicoreagent import OmniCoreAgent, ToolRegistry

tools = ToolRegistry()

# 32 record tools, and one that answers the question.
for area in ["billing", "shipping", "inventory", "crm", "hr", "payroll", "support", "analytics"]:
    for action in ["create", "update", "delete", "export"]:
        def make(area=area, action=action):
            def tool(record_id: str) -> str:
                return f"{action} {area} {record_id}: ok"
            tool.__doc__ = f"{action.title()} a {area} record."
            return tool
        tools.register_tool(f"{action}_{area}_record")(make())

@tools.register_tool("convert_currency")
def convert_currency(amount: float, from_currency: str, to_currency: str) -> dict:
    """Convert an amount of money from one currency to another at today's rate."""
    rate = {("USD", "EUR"): 0.92}.get((from_currency, to_currency), 1.0)
    return {"amount": round(amount * rate, 2), "currency": to_currency}

async def main():
    agent = OmniCoreAgent(
        name="ops",
        system_instruction="Use tools to answer. Answer in plain text, in one sentence.",
        model_config={"provider": "openai", "model": "gpt-5.4-mini"},
        local_tools=tools,
        agent_config={"enable_advanced_tool_use": True},
    )
    result = await agent.run("How much is 250 US dollars in euros?")
    print(result["response"])

    trajectory = await agent.get_trajectory(result["trace_id"])
    for step in trajectory["steps"]:
        for call in step["tool_calls"]:
            print("step", step["step"], "->", call["tool_name"], call["arguments"], call["outcome"])
    await agent.cleanup()

asyncio.run(main())

250 US dollars is 230 euros.
step 1 -> tools_retriever {'query': 'currency conversion USD to EUR current rate 250 US dollars in euros'} success
step 2 -> convert_currency {'amount': 250, 'from_currency': 'USD', 'to_currency': 'EUR'} success