Add: Json Parsing to solve Hallucination Errors (#3013)

2025-06-23 21:50:16 +05:30
parent eb24b92227
commit 2bb0653e67
15 changed files with 44 additions and 17 deletions
--- a/mem0/llms/deepseek.py
+++ b/mem0/llms/deepseek.py
@@ -6,6 +6,7 @@ from openai import OpenAI

 from mem0.configs.llms.base import BaseLlmConfig
 from mem0.llms.base import LLMBase
+from mem0.memory.utils import extract_json


 class DeepSeekLLM(LLMBase):
@@ -41,7 +42,7 @@ class DeepSeekLLM(LLMBase):
                    processed_response["tool_calls"].append(
                        {
                            "name": tool_call.function.name,
-                            "arguments": json.loads(tool_call.function.arguments),
+                            "arguments": json.loads(extract_json(tool_call.function.arguments)),
                        }
                    )