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/litellm.py
+++ b/mem0/llms/litellm.py
@@ -8,6 +8,7 @@ except ImportError:

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


 class LiteLLM(LLMBase):
@@ -39,7 +40,7 @@ class LiteLLM(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)),
                        }
                    )