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

Update operate.py

Browse Source
This commit is contained in:
zrguo
2024-10-18 15:33:11 +08:00
parent 8e0877f4b0
commit f49de420cf
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lightrag/operate.py
View File

@@ -76,7 +76,7 @@ async def _handle_single_entity_extraction(
record_attributes: list[str],
chunk_key: str,
):
if record_attributes[0] != '"entity"' or len(record_attributes) < 4:
if len(record_attributes) < 4 or record_attributes[0] != '"entity"':
return None
# add this record as a node in the G
entity_name = clean_str(record_attributes[1].upper())
@@ -97,7 +97,7 @@ async def _handle_single_relationship_extraction(
record_attributes: list[str],
chunk_key: str,
):
if record_attributes[0] != '"relationship"' or len(record_attributes) < 5:
if len(record_attributes) < 5 or record_attributes[0] != '"relationship"':
return None
# add this record as edge
source = clean_str(record_attributes[1].upper())