From 07d2f1108102885e7dce5cb7a90488acbbdff5ac Mon Sep 17 00:00:00 2001 From: Dev Khant Date: Fri, 7 Mar 2025 13:12:00 +0530 Subject: [PATCH] Catch json error for new_memories_with_action (#2324) --- mem0/memory/main.py | 20 ++++++++++++++------ 1 file changed, 14 insertions(+), 6 deletions(-) diff --git a/mem0/memory/main.py b/mem0/memory/main.py index d3f2d74a..6bb1ac54 100644 --- a/mem0/memory/main.py +++ b/mem0/memory/main.py @@ -194,13 +194,21 @@ class Memory(MemoryBase): function_calling_prompt = get_update_memory_messages(retrieved_old_memory, new_retrieved_facts) - new_memories_with_actions = self.llm.generate_response( - messages=[{"role": "user", "content": function_calling_prompt}], - response_format={"type": "json_object"}, - ) + try: + new_memories_with_actions = self.llm.generate_response( + messages=[{"role": "user", "content": function_calling_prompt}], + response_format={"type": "json_object"}, + ) + except Exception as e: + logging.error(f"Error in new_memories_with_actions: {e}") + new_memories_with_actions = [] - new_memories_with_actions = remove_code_blocks(new_memories_with_actions) - new_memories_with_actions = json.loads(new_memories_with_actions) + try: + new_memories_with_actions = remove_code_blocks(new_memories_with_actions) + new_memories_with_actions = json.loads(new_memories_with_actions) + except Exception as e: + logging.error(f"Invalid JSON response: {e}") + new_memories_with_actions = [] returned_memories = [] try: