Fix: Memgraph Graph Generation Issue (#3109)
This commit is contained in:
@@ -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}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user