Fix linting

This commit is contained in:
yangdx
2025-02-22 10:18:39 +08:00
parent 2fff9ae1ab
commit 411782797b

View File

@@ -591,7 +591,7 @@ class LightRAG:
if isinstance(input, str):
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))
# 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)}
else:
# Generate contents dict of MD5 hash IDs and documents
contents = {
compute_mdhash_id(doc, prefix="doc-"): doc
for doc in input
}
contents = {compute_mdhash_id(doc, prefix="doc-"): doc for doc in input}
# 2. Remove duplicate contents
unique_contents = {