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

code clean

Browse Source
This commit is contained in:
Yannick Stephan
2025-02-14 23:33:59 +01:00
parent 66f555677a
commit dfa8681924
1 changed files with 18 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
lightrag/lightrag.py
View File

@@ -955,26 +955,26 @@ class LightRAG:
# Insert entities into vector storage if needed
data_for_vdb = {
compute_mdhash_id(dp["entity_name"], prefix="ent-"): {
"content": dp["entity_name"] + dp["description"],
"entity_name": dp["entity_name"],
}
for dp in all_entities_data
compute_mdhash_id(dp["entity_name"], prefix="ent-"): {
"content": dp["entity_name"] + dp["description"],
"entity_name": dp["entity_name"],
}
for dp in all_entities_data
}
await self.entities_vdb.upsert(data_for_vdb)
# Insert relationships into vector storage if needed
data_for_vdb = {
compute_mdhash_id(dp["src_id"] + dp["tgt_id"], prefix="rel-"): {
"src_id": dp["src_id"],
"tgt_id": dp["tgt_id"],
"content": dp["keywords"]
+ dp["src_id"]
+ dp["tgt_id"]
+ dp["description"],
}
for dp in all_relationships_data
compute_mdhash_id(dp["src_id"] + dp["tgt_id"], prefix="rel-"): {
"src_id": dp["src_id"],
"tgt_id": dp["tgt_id"],
"content": dp["keywords"]
+ dp["src_id"]
+ dp["tgt_id"]
+ dp["description"],
}
for dp in all_relationships_data
}
await self.relationships_vdb.upsert(data_for_vdb)
finally:
@@ -982,10 +982,7 @@ class LightRAG:
await self._insert_done()
def query(
self,
query: str,
param: QueryParam = QueryParam(),
prompt: str | None = None
self, query: str, param: QueryParam = QueryParam(), prompt: str | None = None
) -> str:
"""
Perform a sync query.