Merge pull request #98 from brnaba-aws/main

Fixed in-memory
awslabs · Nov 22, 2024 · f2a9b65 · f2a9b65
2 parents 9e34777 + 439340a
commit f2a9b65
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/src/multi_agent_orchestrator/storage/in_memory_chat_storage.py b/python/src/multi_agent_orchestrator/storage/in_memory_chat_storage.py
@@ -59,8 +59,9 @@ async def fetch_all_chats(
             if stored_user_id == user_id and stored_session_id == session_id:
                 for message in messages:
                     new_content = message.content if message.content else []
+
                     if len(new_content) > 0 and message.role == "assistant":
-                        new_content[0]['text'] = f"[{agent_id}] {new_content[0]['text']}"
+                        new_content = [{'text':f"[{agent_id}] {new_content[0]['text']}"}]
                     all_messages.append(TimestampedMessage(
                         role=message.role,
                         content=new_content,