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

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-02-22 10:18:39 +08:00
parent 2fd997ca4f
commit 351c8db849
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lightrag/lightrag.py
View File

@@ -591,7 +591,7 @@ class LightRAG:
if isinstance(input, str): if isinstance(input, str):
input = [input] input = [input]
# Clean input text and remove duplicates # Clean input text and remove duplicates
input = list(set(self.clean_text(doc) for doc in input)) input = list(set(self.clean_text(doc) for doc in input))
# 1. Validate ids if provided or generate MD5 hash IDs # 1. Validate ids if provided or generate MD5 hash IDs
@@ -608,10 +608,7 @@ class LightRAG:
contents = {id_: doc for id_, doc in zip(ids, input)} contents = {id_: doc for id_, doc in zip(ids, input)}
else: else:
# Generate contents dict of MD5 hash IDs and documents # Generate contents dict of MD5 hash IDs and documents
contents = { contents = {compute_mdhash_id(doc, prefix="doc-"): doc for doc in input}
compute_mdhash_id(doc, prefix="doc-"): doc
for doc in input
}
# 2. Remove duplicate contents # 2. Remove duplicate contents
unique_contents = { unique_contents = {