fix neo4jstorage bug

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
 )
 ```

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

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
 )