Skip to main content

Overview

MongoStorage provides a MongoDB-based storage backend. Ideal for document-based applications, flexible schemas, and horizontally scalable deployments.

Install

Install the MongoDB storage optional dependency group:
uv pip install "upsonic[mongo-storage]"

Basic Usage

from upsonic import Agent, Task
from upsonic.storage.memory import Memory
from upsonic.storage.mongo import MongoStorage

storage = MongoStorage(
    db_url="mongodb://localhost:27017",
    db_name="agent_memory"
)

memory = Memory(
    storage=storage,
    session_id="session_001",
    user_id="user_123",
    full_session_memory=True,
    summary_memory=True,
    user_analysis_memory=True,
    model="openai/gpt-4o"
)

agent = Agent("openai/gpt-4o", memory=memory)

result1 = agent.do(Task("My name is Alice"))
result2 = agent.do(Task("What's my name?"))
print(result2)  # "Your name is Alice"

Parameters

ParameterTypeDefaultDescription
db_urlstr | NoneNoneMongoDB connection URL
db_clientMongoClient | NoneNonePre-existing pymongo client
db_namestr | None"upsonic"Database name
session_collectionstr | None"upsonic_sessions"Sessions collection name
user_memory_collectionstr | None"upsonic_user_memories"User memory collection name
idstr | Noneauto-generatedStorage instance ID

Storage Type

MongoStorage is a synchronous storage implementation using pymongo driver.