gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Normalize keyword extration result

Browse Source
This commit is contained in:
yangdx
2025-05-08 16:05:52 +08:00
parent de40f1b5b3
commit d2d755db7b
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lightrag/operate.py
View File

@@ -218,7 +218,11 @@ async def _handle_single_relationship_extraction(
edge_description = clean_str(record_attributes[3]) edge_description = clean_str(record_attributes[3])
edge_description = normalize_extracted_info(edge_description) edge_description = normalize_extracted_info(edge_description)
edge_keywords = clean_str(record_attributes[4]).strip('"').strip("'") edge_keywords = normalize_extracted_info(
clean_str(record_attributes[4]), is_entity=True
)
edge_keywords = edge_keywords.replace("，", ",")
edge_source_id = chunk_key edge_source_id = chunk_key
weight = ( weight = (
float(record_attributes[-1].strip('"').strip("'")) float(record_attributes[-1].strip('"').strip("'"))