gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix linting

Browse source
This commit is contained in:
yangdx 2025-09-14 05:02:02 +08:00
parent cddd81a86c
commit 7cc2b69bcf
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
lightrag/operate.py
View file

@ -887,14 +887,22 @@ async def _process_extraction_result(
record, [f"{tuple_delimiter}entity{tuple_delimiter}"]
)
for entity_record in entity_records:
if not entity_record.startswith(f"entity{tuple_delimiter}") and not entity_record.startswith(f"relationship{tuple_delimiter}"):
if not entity_record.startswith(
f"entity{tuple_delimiter}"
) and not entity_record.startswith(f"relationship{tuple_delimiter}"):
entity_record = f"entity{tuple_delimiter}{entity_record}"
entity_relation_records = split_string_by_multi_markers(
entity_record, [f"{tuple_delimiter}relationship{tuple_delimiter}"]
)
for entity_relation_record in entity_relation_records:
if not entity_relation_record.startswith(f"entity{tuple_delimiter}") and not entity_relation_record.startswith(f"relationship{tuple_delimiter}"):
entity_relation_record = f"relationship{tuple_delimiter}{entity_relation_record}"
if not entity_relation_record.startswith(
f"entity{tuple_delimiter}"
) and not entity_relation_record.startswith(
f"relationship{tuple_delimiter}"
):
entity_relation_record = (
f"relationship{tuple_delimiter}{entity_relation_record}"
)
fixed_records = fixed_records + [entity_relation_record]
if len(fixed_records) != len(records):