MySQL for Agent

Kern supports using MySQL as a storage backend for Agents using the MySQLDb class.

Usage

Run MySQL

Install docker desktop and run MySQL on port 3306 using:

1docker run -d \
2  --name mysql \
3  -e MYSQL_ROOT_PASSWORD=ai \
4  -e MYSQL_DATABASE=ai \
5  -e MYSQL_USER=ai \
6  -e MYSQL_PASSWORD=ai \
7  -p 3306:3306 \
8  -d mysql:8

1from kern.agent import Agent
2from kern.db.mysql import MySQLDb
3
4db_url = "mysql+pymysql://ai:ai@localhost:3306/ai"
5
6db = MySQLDb(db_url=db_url)
7
8agent = Agent(
9    db=db,
10    add_history_to_context=True,
11)
12agent.print_response("How many people live in Canada?")
13agent.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_engine`	`Optional[Engine]`	-	The SQLAlchemy database engine to use.
`db_schema`	`Optional[str]`	-	The database schema to use.
`db_url`	`Optional[str]`	-	The database URL to connect to.
`session_table`	`Optional[str]`	-	Name of the table to store Agent, Team and Workflow sessions.
`memory_table`	`Optional[str]`	-	Name of the table to store memories.
`metrics_table`	`Optional[str]`	-	Name of the table to store metrics.
`eval_table`	`Optional[str]`	-	Name of the table to store evaluation runs data.
`knowledge_table`	`Optional[str]`	-	Name of the table to store knowledge content.
`traces_table`	`Optional[str]`	-	Name of the table to store traces.
`spans_table`	`Optional[str]`	-	Name of the table to store spans.