import asyncio
from omnicoreagent import OmniCoreAgent, MemoryRouter

async def main():
    agent = OmniCoreAgent(
        name="memory_agent",
        system_instruction="Remember useful details inside the active session.",
        model_config={"provider": "openai", "model": "gpt-4o"},
        memory_router=MemoryRouter("in_memory"),
    )

    await agent.run("My project is called Atlas.", session_id="user_123")
    result = await agent.run("What is my project called?", session_id="user_123")
    print(result["response"])
    await agent.cleanup()

asyncio.run(main())

pip install "omnicoreagent[redis]"
pip install "omnicoreagent[mongodb]"
pip install "omnicoreagent[postgres]"

from omnicoreagent import OmniCoreAgent, MemoryRouter

# Start with Redis
agent = OmniCoreAgent(
    name="my_agent",
    system_instruction="Remember useful details inside the active session.",
    memory_router=MemoryRouter("redis"),
    model_config={"provider": "openai", "model": "gpt-4o"}
)

# Switch at runtime — no restart needed!
await agent.switch_memory_store("mongodb")     # Switch to MongoDB
await agent.switch_memory_store("sql")         # Switch to SQL database storage
await agent.switch_memory_store("in_memory")   # Switch to in-memory
await agent.switch_memory_store("redis")       # Back to Redis

from omnicoreagent import OmniCoreAgent, MemoryRouter

memory_router = MemoryRouter("redis")

memory_config = {
    "mode": "sliding_window",  # or "token_budget"
    "value": 10,  # Keep last 10 messages, or max tokens in token_budget mode
    "summary": {
        "enabled": True,
        "retention_policy": "keep"  # Options: "keep" or "delete"
    }
}

agent = OmniCoreAgent(
    name="summarizing_agent",
    memory_router=memory_router,
    model_config={"provider": "openai", "model": "gpt-4o"},
    agent_config={"memory_config": memory_config},
)