From 801619084fc268891416ebdbaad4b55542b5d534 Mon Sep 17 00:00:00 2001
From: zuoluo <doosen612@gmail.com>
Date: Tue, 3 Dec 2024 16:04:58 +0800
Subject: [PATCH] fix neo4jstorage bug

---
 README.md            | 2 +-
 lightrag/lightrag.py | 4 +++-
 test_neo4j.py        | 2 +-
 3 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 269e8249..02ecbe4e 100644
--- a/README.md
+++ b/README.md
@@ -321,7 +321,7 @@ WORKING_DIR = "./local_neo4jWorkDir"
 rag = LightRAG(
     working_dir=WORKING_DIR,
     llm_model_func=gpt_4o_mini_complete,  # Use gpt_4o_mini_complete LLM model
-    kg="Neo4JStorage", #<-----------override KG default
+    graph_storage="Neo4JStorage", #<-----------override KG default
     log_level="DEBUG"  #<-----------override log_level default
 )
 ```
diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index 6511e61b..2b2d4820 100644
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -186,7 +186,9 @@ class LightRAG:
             embedding_func=self.embedding_func,
         )
         self.chunk_entity_relation_graph = self.graph_storage_cls(
-            namespace="chunk_entity_relation", global_config=asdict(self)
+            namespace="chunk_entity_relation",
+            global_config=asdict(self),
+            embedding_func=self.embedding_func,
         )
         ####
         # add embedding func by walter over
diff --git a/test_neo4j.py b/test_neo4j.py
index 822cd7bc..0048fc17 100644
--- a/test_neo4j.py
+++ b/test_neo4j.py
@@ -17,7 +17,7 @@ if not os.path.exists(WORKING_DIR):
 rag = LightRAG(
     working_dir=WORKING_DIR,
     llm_model_func=gpt_4o_mini_complete,  # Use gpt_4o_mini_complete LLM model
-    kg="Neo4JStorage",
+    graph_storage="Neo4JStorage",
     log_level="INFO",
     # llm_model_func=gpt_4o_complete  # Optionally, use a stronger model
 )