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