From c674625e88a2ff7a2426b7680a00bdc803535d7a Mon Sep 17 00:00:00 2001
From: Yuhang <lyh472617147@gmail.com>
Date: Thu, 5 Sep 2024 00:54:11 +0800
Subject: [PATCH] Fix bug about MemoryGraph can't find (#1810)

---
 mem0/memory/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mem0/memory/main.py b/mem0/memory/main.py
index 2e073cef..d5744b68 100644
--- a/mem0/memory/main.py
+++ b/mem0/memory/main.py
@@ -41,7 +41,7 @@ class Memory(MemoryBase):
         self.enable_graph = False
 
         if self.version == "v1.1" and self.config.graph_store.config:
-            from mem0.memory.main_graph import MemoryGraph
+            from mem0.memory.graph_memory import MemoryGraph
             self.graph = MemoryGraph(self.config)
             self.enable_graph = True