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-07-05 12:07:37 +08:00
parent a2e59dd078
commit fe13475234
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lightrag/operate.py
View file

@ -167,7 +167,7 @@ async def _handle_single_entity_extraction(
# Normalize entity name # Normalize entity name
entity_name = normalize_extracted_info(entity_name, is_entity=True) entity_name = normalize_extracted_info(entity_name, is_entity=True)
# Check if entity name became empty after normalization # Check if entity name became empty after normalization
if not entity_name or not entity_name.strip(): if not entity_name or not entity_name.strip():
logger.warning( logger.warning(
@ -216,20 +216,20 @@ async def _handle_single_relationship_extraction(
# Normalize source and target entity names # Normalize source and target entity names
source = normalize_extracted_info(source, is_entity=True) source = normalize_extracted_info(source, is_entity=True)
target = normalize_extracted_info(target, is_entity=True) target = normalize_extracted_info(target, is_entity=True)
# Check if source or target became empty after normalization # Check if source or target became empty after normalization
if not source or not source.strip(): if not source or not source.strip():
logger.warning( logger.warning(
f"Relationship extraction error: source entity became empty after normalization. Original: '{record_attributes[1]}'" f"Relationship extraction error: source entity became empty after normalization. Original: '{record_attributes[1]}'"
) )
return None return None
if not target or not target.strip(): if not target or not target.strip():
logger.warning( logger.warning(
f"Relationship extraction error: target entity became empty after normalization. Original: '{record_attributes[2]}'" f"Relationship extraction error: target entity became empty after normalization. Original: '{record_attributes[2]}'"
) )
return None return None
if source == target: if source == target:
logger.debug( logger.debug(
f"Relationship source and target are the same in: {record_attributes}" f"Relationship source and target are the same in: {record_attributes}"