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

Fixed formatting

Browse Source
This commit is contained in:
Lukas Selch
2025-02-17 15:25:50 +01:00
parent 86f5a88db7
commit 537e10303d
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lightrag/lightrag.py
View File

@@ -860,17 +860,17 @@ class LightRAG:
all_chunks_data: dict[str, dict[str, str]] = {} all_chunks_data: dict[str, dict[str, str]] = {}
chunk_to_source_map: dict[str, str] = {} chunk_to_source_map: dict[str, str] = {}
for chunk_data in custom_kg.get("chunks", {}): for chunk_data in custom_kg.get("chunks", {}):
chunk_content = chunk_data["content"] chunk_content = chunk_data["content"].strip()
source_id = chunk_data["source_id"] source_id = chunk_data["source_id"]
chunk_id = compute_mdhash_id(chunk_content.strip(), prefix="chunk-") chunk_id = compute_mdhash_id(chunk_content, prefix="chunk-")
chunk_entry = { chunk_entry = {
"content": chunk_content.strip(), "content": chunk_content,
"source_id": source_id, "source_id": source_id,
"tokens": len(encode_string_by_tiktoken(chunk_content.strip())), "tokens": len(encode_string_by_tiktoken(chunk_content)),
"chunk_order_id": 0, "chunk_order_id": 0,
"full_doc_id": source_id, "full_doc_id": source_id,
"status": DocStatus.PROCESSED "status": DocStatus.PROCESSED,
} }
all_chunks_data[chunk_id] = chunk_entry all_chunks_data[chunk_id] = chunk_entry
chunk_to_source_map[source_id] = chunk_id chunk_to_source_map[source_id] = chunk_id