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

Improve extraction error handling and field validation

Browse source 
• Add field count validation warnings
• Fix relationship field count (5→6)
• Change error logs to warnings
This commit is contained in:
yangdx 2025-08-31 17:33:42 +08:00
parent b747417961
commit 97c9600085
1 changed files with 13 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
lightrag/operate.py
View file

@ -315,6 +315,11 @@ async def _handle_single_entity_extraction(
file_path: str = "unknown_source",
):
if len(record_attributes) < 4 or '"entity"' not in record_attributes[0]:
if len(record_attributes) > 1 and '"entity"' in record_attributes[0]:
logger.warning(
f"Entity extraction failed in {chunk_key}: expecting 4 fields but got {len(record_attributes)}"
)
logger.warning(f"Entity extracted: {record_attributes[1]}")
return None
try:
@ -376,7 +381,12 @@ async def _handle_single_relationship_extraction(
chunk_key: str,
file_path: str = "unknown_source",
):
if len(record_attributes) < 5 or '"relationship"' not in record_attributes[0]:
if len(record_attributes) < 6 or '"relationship"' not in record_attributes[0]:
if len(record_attributes) > 1 and '"relationship"' in record_attributes[0]:
logger.warning(
f"Relationship extraction failed in {chunk_key}: expecting 6 fields but got {len(record_attributes)}"
)
logger.warning(f"Relationship extracted: {record_attributes[1]}")
return None
try:
@ -433,12 +443,12 @@ async def _handle_single_relationship_extraction(
)
except ValueError as e:
logger.error(
logger.warning(
f"Relationship extraction failed due to encoding issues in chunk {chunk_key}: {e}"
)
return None
except Exception as e:
logger.error(
logger.warning(
f"Relationship extraction failed with unexpected error in chunk {chunk_key}: {e}"
)
return None