1from kern.agent import Agent
2from kern.db.base import SessionType
3from kern.db.postgres import PostgresDb
4from kern.models.meta import Llama
5from rich.pretty import pprint
6
7# Setup the database
8db_url = "postgresql+psycopg://ai:ai@localhost:5532/ai"
9db = PostgresDb(db_url=db_url)
10
11agent = Agent(
12 model=Llama(id="Llama-4-Maverick-17B-128E-Instruct-FP8"),
13 user_id="test_user",
14 session_id="test_session",
15 # Pass the database to the Agent
16 db=db,
17 # Enable user memories
18 update_memory_on_run=True,
19 # Enable session summaries
20 enable_session_summaries=True,
21 # Show debug logs so, you can see the memory being created
22 debug_mode=True,
23)
24
25# -*- Share personal information
26agent.print_response("My name is John Billings", stream=True)
27
28# -*- Print memories and session summary
29if agent.db:
30 pprint(agent.get_user_memories(user_id="test_user"))
31 pprint(
32 agent.db.get_session(
33 session_id="test_session", session_type=SessionType.AGENT
34 ).summary # type: ignore
35 )
36
37# -*- Share personal information
38agent.print_response("I live in NYC", stream=True)
39# -*- Print memories and session summary
40if agent.db:
41 pprint(agent.db.get_user_memories(user_id="test_user"))
42 pprint(
43 agent.db.get_session(
44 session_id="test_session", session_type=SessionType.AGENT
45 ).summary # type: ignore
46 )
47
48# Ask about the conversation
49agent.print_response(
50 "What have we been talking about, do you know my name?", stream=True
51)
