fix neo4jstorage bug

2024-12-03 16:04:58 +08:00
parent b00f2d6027
commit 2786004972
3 changed files with 5 additions and 3 deletions
--- 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
 )