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

Fix entity/relationship record parsing in extraction result processing

Browse source
This commit is contained in:
yangdx 2025-09-14 05:35:01 +08:00
parent 4f5ad76c2c
commit b820d8d588
1 changed files with 9 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
lightrag/operate.py
View file

@ -883,23 +883,24 @@ async def _process_extraction_result(
# Fix LLM output format error which use tuple_delimiter to seperate record instead of "\n" # Fix LLM output format error which use tuple_delimiter to seperate record instead of "\n"
fixed_records = [] fixed_records = []
for record in records: for record in records:
record = record.strip()
if record is None:
continue
entity_records = split_string_by_multi_markers( entity_records = split_string_by_multi_markers(
record, [f"{tuple_delimiter}entity{tuple_delimiter}"] record, [f"{tuple_delimiter}entity{tuple_delimiter}"]
) )
for entity_record in entity_records: for entity_record in entity_records:
if not entity_record.startswith( if not entity_record.startswith("entity") and not entity_record.startswith(
f"entity{tuple_delimiter}" "relationship"
) and not entity_record.startswith(f"relationship{tuple_delimiter}"): ):
entity_record = f"entity{tuple_delimiter}{entity_record}" entity_record = f"entity<|{entity_record}"
entity_relation_records = split_string_by_multi_markers( entity_relation_records = split_string_by_multi_markers(
entity_record, [f"{tuple_delimiter}relationship{tuple_delimiter}"] entity_record, [f"{tuple_delimiter}relationship{tuple_delimiter}"]
) )
for entity_relation_record in entity_relation_records: for entity_relation_record in entity_relation_records:
if not entity_relation_record.startswith( if not entity_relation_record.startswith(
f"entity{tuple_delimiter}" "entity"
) and not entity_relation_record.startswith( ) and not entity_relation_record.startswith("relationship"):
f"relationship{tuple_delimiter}"
):
entity_relation_record = ( entity_relation_record = (
f"relationship{tuple_delimiter}{entity_relation_record}" f"relationship{tuple_delimiter}{entity_relation_record}"
) )