Add Memgraph integration (#2537)

2025-04-22 12:57:24 +02:00
parent cd5c3035ab
commit ba2e479902
10 changed files with 940 additions and 44 deletions
--- a/mem0/memory/main.py
+++ b/mem0/memory/main.py
@@ -59,7 +59,10 @@ class Memory(MemoryBase):
        self.enable_graph = False

        if self.config.graph_store.config:
-            from mem0.memory.graph_memory import MemoryGraph
+            if self.config.graph_store.provider == "memgraph":
+                from mem0.memory.memgraph_memory import MemoryGraph
+            else:
+                from mem0.memory.graph_memory import MemoryGraph

            self.graph = MemoryGraph(self.config)
            self.enable_graph = True