Multi-User, Multi-Session Chat Concurrently

1"""
2In this example, we have 3 users and 4 sessions.
3
4User 1 has 2 sessions.
5User 2 has 1 session.
6User 3 has 1 session.
7"""
8import asyncio
9
10from kern.agent import Agent
11from kern.db.postgres import PostgresDb
12from kern.models.openai import OpenAIResponses
13
14db_url = "postgresql+psycopg://ai:ai@localhost:5532/ai"
15
16db = PostgresDb(db_url=db_url)
17
18user_1_id = "user_1@example.com"
19user_2_id = "user_2@example.com"
20user_3_id = "user_3@example.com"
21
22user_1_session_1_id = "user_1_session_1"
23user_1_session_2_id = "user_1_session_2"
24user_2_session_1_id = "user_2_session_1"
25user_3_session_1_id = "user_3_session_1"
26
27chat_agent = Agent(
28    model=OpenAIResponses(id="gpt-5.2"),
29    db=db,
30    update_memory_on_run=True,
31)
32
33
34async def user_1_conversation():
35    """Handle conversation with user 1 across multiple sessions"""
36    # User 1 - Session 1
37    await chat_agent.arun(
38        "My name is Mark Gonzales and I like anime and video games.",
39        user_id=user_1_id,
40        session_id=user_1_session_1_id,
41    )
42    await chat_agent.arun(
43        "I also enjoy reading manga and playing video games.",
44        user_id=user_1_id,
45        session_id=user_1_session_1_id,
46    )
47
48    # User 1 - Session 2
49    await chat_agent.arun(
50        "I'm going to the movies tonight.",
51        user_id=user_1_id,
52        session_id=user_1_session_2_id,
53    )
54
55    # Continue the conversation in session 1
56    await chat_agent.arun(
57        "What do you suggest I do this weekend?",
58        user_id=user_1_id,
59        session_id=user_1_session_1_id,
60    )
61
62    print("User 1 Done")
63
64
65async def user_2_conversation():
66    """Handle conversation with user 2"""
67    await chat_agent.arun(
68        "Hi my name is John Doe.", user_id=user_2_id, session_id=user_2_session_1_id
69    )
70    await chat_agent.arun(
71        "I'm planning to hike this weekend.",
72        user_id=user_2_id,
73        session_id=user_2_session_1_id,
74    )
75    print("User 2 Done")
76
77
78async def user_3_conversation():
79    """Handle conversation with user 3"""
80    await chat_agent.arun(
81        "Hi my name is Jane Smith.", user_id=user_3_id, session_id=user_3_session_1_id
82    )
83    await chat_agent.arun(
84        "I'm going to the gym tomorrow.",
85        user_id=user_3_id,
86        session_id=user_3_session_1_id,
87    )
88    print("User 3 Done")
89
90
91async def run_concurrent_chat_agent():
92    """Run all user conversations concurrently"""
93    await asyncio.gather(
94        user_1_conversation(), user_2_conversation(), user_3_conversation()
95    )
96
97
98if __name__ == "__main__":
99    # Run all conversations concurrently
100    asyncio.run(run_concurrent_chat_agent())
101
102    user_1_memories = chat_agent.get_user_memories(user_id=user_1_id)
103    print("User 1's memories:")
104    assert user_1_memories is not None
105    for i, m in enumerate(user_1_memories):
106        print(f"{i}: {m.memory}")
107
108    user_2_memories = chat_agent.get_user_memories(user_id=user_2_id)
109    print("User 2's memories:")
110    assert user_2_memories is not None
111    for i, m in enumerate(user_2_memories):
112        print(f"{i}: {m.memory}")
113
114    user_3_memories = chat_agent.get_user_memories(user_id=user_3_id)
115    print("User 3's memories:")
116    assert user_3_memories is not None
117    for i, m in enumerate(user_3_memories):
118        print(f"{i}: {m.memory}")