set kg by start param, defaults to networkx

2024-11-01 08:47:52 -04:00
parent c2ebb819fd
commit e44046f9e6
7 changed files with 48 additions and 13 deletions
--- a/testkg.py
+++ b/testkg.py
@@ -15,7 +15,8 @@ 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
+    llm_model_func=gpt_4o_mini_complete,  # Use gpt_4o_mini_complete LLM model
+    kg="Neo4JStorage"
    # llm_model_func=gpt_4o_complete  # Optionally, use a stronger model
 )