From 64b4c9ea94a43a573a332ca877d7a2960009470e Mon Sep 17 00:00:00 2001 From: a <139073260+Azeem-2@users.noreply.github.com> Date: Sun, 31 Aug 2025 17:52:35 +0500 Subject: [PATCH] Update memory functions to use new API version --- 01_ai_agents_first/22_memory_management/mem0_memory/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/01_ai_agents_first/22_memory_management/mem0_memory/main.py b/01_ai_agents_first/22_memory_management/mem0_memory/main.py index e6553cbf..10d669f3 100644 --- a/01_ai_agents_first/22_memory_management/mem0_memory/main.py +++ b/01_ai_agents_first/22_memory_management/mem0_memory/main.py @@ -32,7 +32,7 @@ def add_memory(query: str, user_id: str) -> str: """Add a new memory for the user""" try: - result = mem0.add([{"role": "user", "content": query}], user_id=user_id) + result = mem0.add([{"role": "user", "content": query}], user_id=user_id, version="v2", output_format="v1.1") return f"Memory added: {result}" except Exception as e: return f"Error adding memory: {str(e)}" @@ -42,7 +42,7 @@ def add_memory(query: str, user_id: str) -> str: def search_memory(query: str, user_id: str) -> str: """Search through past conversations and memories""" try: - memories = mem0.search(query, user_id=user_id, limit=3) + memories = mem0.search(query, user_id=user_id, limit=3, version="v2", output_format="v1.1") if memories and isinstance(memories, list) and len(memories) > 0: return "\n".join([f"- {mem['memory']}" for mem in memories])