gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
Show stats 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())