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

Merge pull request #971 from Volve-Technologies/abyl/adding_full_doc_id

Browse Source 
adding full_doc_id to insert
This commit is contained in:
zrguo
2025-03-03 14:44:50 +08:00
committed by GitHub
parent 2d6d0be128 bf0ddc6450
commit 0320aa4ccc
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lightrag/lightrag.py
View File

@@ -952,11 +952,11 @@ class LightRAG:
pipeline_status["latest_message"] = log_message pipeline_status["latest_message"] = log_message
pipeline_status["history_messages"].append(log_message) pipeline_status["history_messages"].append(log_message)
def insert_custom_kg(self, custom_kg: dict[str, Any]) -> None: def insert_custom_kg(self, custom_kg: dict[str, Any], full_doc_id: str = None) -> None:
loop = always_get_an_event_loop() loop = always_get_an_event_loop()
loop.run_until_complete(self.ainsert_custom_kg(custom_kg)) loop.run_until_complete(self.ainsert_custom_kg(custom_kg, full_doc_id))
async def ainsert_custom_kg(self, custom_kg: dict[str, Any]) -> None: async def ainsert_custom_kg(self, custom_kg: dict[str, Any], full_doc_id: str = None) -> None:
update_storage = False update_storage = False
try: try:
# Insert chunks into vector storage # Insert chunks into vector storage
@@ -982,7 +982,7 @@ class LightRAG:
"source_id": source_id, "source_id": source_id,
"tokens": tokens, "tokens": tokens,
"chunk_order_index": chunk_order_index, "chunk_order_index": chunk_order_index,
"full_doc_id": source_id, "full_doc_id": full_doc_id if full_doc_id is not None else source_id,
"status": DocStatus.PROCESSED, "status": DocStatus.PROCESSED,
} }
all_chunks_data[chunk_id] = chunk_entry all_chunks_data[chunk_id] = chunk_entry