Fix: Memgraph Graph Generation Issue (#3109)

2025-07-05 22:36:13 +05:30
parent c0a930a7d3
commit 4491e7f9f4
2 changed files with 5 additions and 5 deletions
--- a/mem0/llms/openai.py
+++ b/mem0/llms/openai.py
@@ -79,7 +79,7 @@ class OpenAILLM(LLMBase):
        tool_choice: str = "auto",
    ):
        """
-        Generate a response based on the given messages using OpenAI.
+        Generate a JSON response based on the given messages using OpenAI.

        Args:
            messages (list): List of message dicts containing 'role' and 'content'.
@@ -88,7 +88,7 @@ class OpenAILLM(LLMBase):
            tool_choice (str, optional): Tool choice method. Defaults to "auto".

        Returns:
-            str: The generated response.
+            json: The generated response.
        """
        params = {
            "model": self.config.model,