From 4491e7f9f493da5784c565b5b3a52aa03a5cc91c Mon Sep 17 00:00:00 2001 From: Akshat Jain <125379408+akshat1423@users.noreply.github.com> Date: Sat, 5 Jul 2025 22:36:13 +0530 Subject: [PATCH] Fix: Memgraph Graph Generation Issue (#3109) --- mem0/llms/openai.py | 4 ++-- mem0/memory/memgraph_memory.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/mem0/llms/openai.py b/mem0/llms/openai.py index 8f3cf807..d94a0f3a 100644 --- 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, diff --git a/mem0/memory/memgraph_memory.py b/mem0/memory/memgraph_memory.py index beb8af2b..97af282c 100644 --- a/mem0/memory/memgraph_memory.py +++ b/mem0/memory/memgraph_memory.py @@ -3,7 +3,7 @@ import logging from mem0.memory.utils import format_entities try: - from langchain_memgraph import Memgraph + from langchain_memgraph.graphs.memgraph import Memgraph except ImportError: raise ImportError("langchain_memgraph is not installed. Please install it using pip install langchain-memgraph") @@ -76,8 +76,8 @@ class MemoryGraph: # TODO: Batch queries with APOC plugin # TODO: Add more filter support - deleted_entities = self._delete_entities(to_be_deleted, filters["user_id"]) - added_entities = self._add_entities(to_be_added, filters["user_id"], entity_type_map) + deleted_entities = self._delete_entities(to_be_deleted, filters) + added_entities = self._add_entities(to_be_added, filters, entity_type_map) return {"deleted_entities": deleted_entities, "added_entities": added_entities}