MongoDB for Agent

Kern supports using MongoDB as a storage backend for Agents using the MongoDb class.

Usage

You need to provide either db_url or client. The following example uses db_url.

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

1from kern.agent import Agent
2from kern.db.mongo import MongoDb
3from kern.tools.hackernews import HackerNewsTools
4
5# MongoDB connection settings
6db_url = "mongodb://localhost:27017"
7
8db = MongoDb(db_url=db_url)
9
10agent = Agent(
11    db=db,
12    tools=[HackerNewsTools()],
13    add_history_to_context=True,
14)
15agent.print_response("How many people live in Canada?")
16agent.print_response("What is their national anthem called?")

Params

Parameter	Type	Default	Description
`id`	`Optional[str]`	-	The ID of the database instance. UUID by default.
`db_client`	`Optional[MongoClient]`	-	The MongoDB client to use.
`db_name`	`Optional[str]`	-	The name of the database to use.
`db_url`	`Optional[str]`	-	The database URL to connect to.
`session_collection`	`Optional[str]`	-	Name of the collection to store sessions.
`memory_collection`	`Optional[str]`	-	Name of the collection to store memories.
`metrics_collection`	`Optional[str]`	-	Name of the collection to store metrics.
`eval_collection`	`Optional[str]`	-	Name of the collection to store evaluation runs.
`knowledge_collection`	`Optional[str]`	-	Name of the collection to store knowledge documents.
`traces_collection`	`Optional[str]`	-	Name of the collection to store traces.
`spans_collection`	`Optional[str]`	-	Name of the collection to store spans.