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

Merge pull request #1423 from tackhwa/main

Browse Source 
friendly implementation of entity extraction and relationship weight extract for Low-Capability LLMs
This commit is contained in:
Daniel.y
2025-04-22 19:11:04 +08:00
committed by GitHub
parent 0f1ad6d1b5 2e186ba488
commit a09f6eb170
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lightrag/operate.py
View File

@@ -152,7 +152,7 @@ async def _handle_single_entity_extraction(
chunk_key: str, chunk_key: str,
file_path: str = "unknown_source", file_path: str = "unknown_source",
): ):
if len(record_attributes) < 4 or record_attributes[0] != '"entity"': if len(record_attributes) < 4 or '"entity"' not in record_attributes[0]:
return None return None
# Clean and validate entity name # Clean and validate entity name
@@ -198,7 +198,7 @@ async def _handle_single_relationship_extraction(
chunk_key: str, chunk_key: str,
file_path: str = "unknown_source", file_path: str = "unknown_source",
): ):
if len(record_attributes) < 5 or record_attributes[0] != '"relationship"': if len(record_attributes) < 5 or '"relationship"' not in record_attributes[0]:
return None return None
# add this record as edge # add this record as edge
source = clean_str(record_attributes[1]) source = clean_str(record_attributes[1])
@@ -215,7 +215,7 @@ async def _handle_single_relationship_extraction(
edge_source_id = chunk_key edge_source_id = chunk_key
weight = ( weight = (
float(record_attributes[-1].strip('"').strip("'")) float(record_attributes[-1].strip('"').strip("'"))
if is_float_regex(record_attributes[-1]) if is_float_regex(record_attributes[-1].strip('"').strip("'"))
else 1.0 else 1.0
) )
return dict( return dict(