gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #378 from doosenn/main

Browse Source 
fix neo4jstorage bug
This commit is contained in:
zrguo
2024-12-04 11:11:19 +08:00
committed by GitHub
parent 96962cd243 801619084f
commit 6927b57520
3 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -321,7 +321,7 @@ WORKING_DIR = "./local_neo4jWorkDir"
rag = LightRAG( rag = LightRAG(
working_dir=WORKING_DIR, 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", #<-----------override KG default graph_storage="Neo4JStorage", #<-----------override KG default
log_level="DEBUG" #<-----------override log_level default log_level="DEBUG" #<-----------override log_level default
) )
``` ```

4
lightrag/lightrag.py
View File

@@ -186,7 +186,9 @@ class LightRAG:
embedding_func=self.embedding_func, embedding_func=self.embedding_func,
) )
self.chunk_entity_relation_graph = self.graph_storage_cls( 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 # add embedding func by walter over

2
test_neo4j.py
View File

@@ -17,7 +17,7 @@ if not os.path.exists(WORKING_DIR):
rag = LightRAG( rag = LightRAG(
working_dir=WORKING_DIR, 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", graph_storage="Neo4JStorage",
log_level="INFO", log_level="INFO",
# llm_model_func=gpt_4o_complete # Optionally, use a stronger model # llm_model_func=gpt_4o_complete # Optionally, use a stronger model
) )