from omnicoreagent import OmniCoreAgent

agent = OmniCoreAgent(
    name="monitored_agent",
    system_instruction="You are a helpful assistant.",
    model_config={"provider": "openai", "model": "gpt-4o"},
    agent_config={
        "request_limit": 100,
        "total_tokens_limit": 50000,
    },
)

# After running queries...
metrics = await agent.get_metrics()
print(f"Total requests: {metrics['total_requests']}")
print(f"Total tokens: {metrics['total_tokens']}")
print(f"Request tokens: {metrics['total_request_tokens']}")
print(f"Response tokens: {metrics['total_response_tokens']}")

agent = OmniCoreAgent(
    name="protected_agent",
    system_instruction="You are a customer service agent.",
    model_config={"provider": "openai", "model": "gpt-4o"},
    agent_config={
        "guardrail_config": {
            "strict_mode": True,  # Block suspicious inputs
        },
    },
)

agent_config={
    "request_limit": 100,        # Max requests per session
    "total_tokens_limit": 50000, # Max tokens before stopping
    "max_steps": 10,             # Max reasoning loops
}