jin commited on
Commit
f9fed52
·
1 Parent(s): 99f5836

Update lightrag_oracle_demo.py

Browse files

add embedding_func for graph database, it's deleted in commit 5661d76860436f7bf5aef2e50d9ee4a59660146c
This will fix the issue

Files changed (1) hide show
  1. examples/lightrag_oracle_demo.py +2 -0
examples/lightrag_oracle_demo.py CHANGED
@@ -104,6 +104,8 @@ async def main():
104
  rag.graph_storage_cls.db = oracle_db
105
  rag.key_string_value_json_storage_cls.db = oracle_db
106
  rag.vector_db_storage_cls.db = oracle_db
 
 
107
 
108
  # Extract and Insert into LightRAG storage
109
  with open("./dickens/demo.txt", "r", encoding="utf-8") as f:
 
104
  rag.graph_storage_cls.db = oracle_db
105
  rag.key_string_value_json_storage_cls.db = oracle_db
106
  rag.vector_db_storage_cls.db = oracle_db
107
+ # add embedding_func for graph database, it's deleted in commit 5661d76860436f7bf5aef2e50d9ee4a59660146c
108
+ rag.chunk_entity_relation_graph.embedding_func = rag.embedding_func
109
 
110
  # Extract and Insert into LightRAG storage
111
  with open("./dickens/demo.txt", "r", encoding="utf-8") as f: