@@ -81,6 +81,32 @@ def test_save_conversation(conversation_service):
8181 assert saved_conv .last_updated .tzinfo is not None # Should be timezone-aware
8282
8383
84+ def test_update_existing_conversation (conversation_service ):
85+ # Create and save initial conversation
86+ initial_conv = Conversation (
87+ conversation_id = "test-id" ,
88+ conversation_name = "Initial Name" ,
89+ messages = [],
90+ last_updated = datetime .now (timezone .utc ),
91+ )
92+ conversation_service .save_conversation (initial_conv )
93+
94+ # Update the conversation with new name
95+ updated_conv = Conversation (
96+ conversation_id = "test-id" ,
97+ conversation_name = "Updated Name" ,
98+ messages = [],
99+ last_updated = datetime .now (timezone .utc ),
100+ )
101+ conversation_service .save_conversation (updated_conv )
102+
103+ # Retrieve and verify the update
104+ saved_conv = conversation_service .get_conversation ("test-id" )
105+ assert saved_conv is not None
106+ assert saved_conv .conversation_id == "test-id"
107+ assert saved_conv .conversation_name == "Updated Name"
108+
109+
84110def test_generate_name (conversation_service ):
85111 # Generate name
86112 name = conversation_service .generate_name ("Hello, how are you?" )
0 commit comments