import asyncio
from omnicoreagent import OmniCoreAgent

async def main():
    agent = OmniCoreAgent(
        name="my_agent",
        system_instruction="You are a helpful assistant.",
        model_config={"provider": "openai", "model": "gpt-4o"}
    )

    result = await agent.run("What is the capital of France?")

    print(result["response"])       # The agent's text reply
    print(result["session_id"])     # Session used for this run
    print(result.get("metric"))     # Token/timing metric when provider usage is available

    await agent.cleanup()

asyncio.run(main())

# First interaction — agent learns the user's name
await agent.run("My name is Abiola.", session_id="user_42")

# Later interaction — agent remembers
result = await agent.run("What's my name?", session_id="user_42")
# → "Your name is Abiola."

history = await agent.get_history(session_id="user_42")
for message in history:
    print(f"{message['role']}: {message['content'][:80]}")

await agent.clear_session_history(session_id="user_42")

from omnicoreagent import OmniCoreAgent, MemoryRouter

agent = OmniCoreAgent(
    name="persistent_agent",
    system_instruction="You are a helpful assistant.",
    model_config={"provider": "openai", "model": "gpt-4o"},
    memory_router=MemoryRouter("redis")  # or "sql", "mongodb", "in_memory"
)

agent = OmniCoreAgent(
    name="tool_agent",
    system_instruction="You can manage files and search the web.",
    model_config={"provider": "openai", "model": "gpt-4o"},
    mcp_tools=[
        {
            "name": "filesystem",
            "transport_type": "stdio",
            "command": "npx",
            "args": ["-y", "@modelcontextprotocol/server-filesystem", "/tmp"]
        }
    ]
)

await agent.connect_mcp_servers()
result = await agent.run("List all files in /tmp")

from omnicoreagent import OmniCoreAgent, ToolRegistry

tools = ToolRegistry()

@tools.register_tool("get_weather")
def get_weather(city: str) -> dict:
    """Get current weather for a city."""
    return {"city": city, "temp": "22°C", "condition": "Sunny"}

agent = OmniCoreAgent(
    name="weather_agent",
    system_instruction="You help with weather queries.",
    model_config={"provider": "openai", "model": "gpt-4o"},
    local_tools=tools
)

agent = OmniCoreAgent(
    name="streaming_agent",
    system_instruction="You are a helpful assistant.",
    model_config={"provider": "openai", "model": "gpt-4o"}
)

result = await agent.run("What can you do?")
session_id = result["session_id"]
run_id = result["run_id"]
trace_id = result["trace_id"]

events = await agent.get_telemetry_events_after(
    cursor=None,
    session_id=session_id,
    run_id=run_id,
)
for event in events:
    print(event.event_type, event.model_dump())

exact_trace = await agent.get_trace(trace_id)
latest_session_trace = await agent.get_latest_trace(session_id)
normalized_trace = await agent.get_trace(trace_id, normalize=True)

from omnicoreagent.exceptions import OmniCoreAgentError

try:
    result = await agent.run("Analyze this dataset", session_id="user_1")
except OmniCoreAgentError as e:
    print(f"Agent runtime error: {e.message}")
except Exception as e:
    print(f"Unexpected failure: {e}")