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-10 18:38:21 +08:00
parent db6bba80c9
commit 7fe47fac84
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
lightrag/operate.py
View file

@ -878,32 +878,32 @@ async def _process_extraction_result(
for record in records: for record in records:
# Remove outer brackets (support English and Chinese brackets with enhanced tolerance) # Remove outer brackets (support English and Chinese brackets with enhanced tolerance)
record = record.strip() record = record.strip()
# Define allowed leading and trailing characters # Define allowed leading and trailing characters
leading_trailing_chars = r'[`<>"\']*' leading_trailing_chars = r'[`<>"\']*'
# Handle leading characters before left bracket # Handle leading characters before left bracket
if record.startswith("(") or record.startswith("（"): if record.startswith("(") or record.startswith("（"):
record = record[1:] record = record[1:]
else: else:
# Check for leading characters + left bracket pattern # Check for leading characters + left bracket pattern
leading_bracket_pattern = r'^' + leading_trailing_chars + r'([（(])' leading_bracket_pattern = r"^" + leading_trailing_chars + r"([（(])"
match = re.search(leading_bracket_pattern, record) match = re.search(leading_bracket_pattern, record)
if match: if match:
# Extract content from the left bracket position # Extract content from the left bracket position
bracket_pos = match.start(1) bracket_pos = match.start(1)
record = record[bracket_pos + 1:] record = record[bracket_pos + 1 :]
else: else:
logger.warning( logger.warning(
f"{chunk_key}: Record starting bracket can not be found in extraction result" f"{chunk_key}: Record starting bracket can not be found in extraction result"
) )
# Handle trailing characters after right bracket # Handle trailing characters after right bracket
if record.endswith(")") or record.endswith("）"): if record.endswith(")") or record.endswith("）"):
record = record[:-1] record = record[:-1]
else: else:
# Check for right bracket + trailing characters pattern # Check for right bracket + trailing characters pattern
trailing_bracket_pattern = r'([)）])' + leading_trailing_chars + r'$' trailing_bracket_pattern = r"([)）])" + leading_trailing_chars + r"$"
match = re.search(trailing_bracket_pattern, record) match = re.search(trailing_bracket_pattern, record)
if match: if match:
# Extract content up to the right bracket position # Extract content up to the right bracket position
@ -1474,7 +1474,7 @@ async def _merge_edges_then_upsert(
pipeline_status["history_messages"].append(status_message) pipeline_status["history_messages"].append(status_message)
else: else:
logger.debug(status_message) logger.debug(status_message)
else: else:
logger.error(f"Edge {src_id} - {tgt_id} has no description") logger.error(f"Edge {src_id} - {tgt_id} has no description")
description = "(no description)" description = "(no description)"