Async MongoDB

Use MongoDB asynchronously for agent session storage.

Kern supports using MongoDB asynchronously, with the AsyncMongoDb class.

Tip

v2 Migration Support: If you're upgrading from Kern v1, MongoDB is fully supported in the v2 migration script. See the migration guide for details.

Usage

1from kern.agent import Agent
2from kern.db.mongo import AsyncMongoDb
3
4# MongoDB connection settings
5db_url = "mongodb://localhost:27017"
6
7db = AsyncMongoDb(db_url=db_url)
8
9# Setup your Agent with the Database
10agent = Agent(db=db)

Run MongoDB

Install docker desktop and run MongoDB on port 27017 using:

1docker run -d \
2 --name local-mongo \
3 -p 27017:27017 \
4 -e MONGO_INITDB_ROOT_USERNAME=mongoadmin \
5 -e MONGO_INITDB_ROOT_PASSWORD=secret \
6 mongo

Params

ParameterTypeDefaultDescription
idOptional[str]-The ID of the database instance. UUID by default.
db_clientOptional[MongoClient]-The MongoDB client to use.
db_nameOptional[str]-The name of the database to use.
db_urlOptional[str]-The database URL to connect to.
session_collectionOptional[str]-Name of the collection to store sessions.
memory_collectionOptional[str]-Name of the collection to store memories.
metrics_collectionOptional[str]-Name of the collection to store metrics.
eval_collectionOptional[str]-Name of the collection to store evaluation runs.
knowledge_collectionOptional[str]-Name of the collection to store knowledge documents.
traces_collectionOptional[str]-Name of the collection to store traces.
spans_collectionOptional[str]-Name of the collection to store spans.