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 f576a28e0d
commit e7a7ff62b2
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], record_attributes: list[str],
chunk_key: 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 return None
# add this record as a node in the G # add this record as a node in the G
entity_name = clean_str(record_attributes[1].upper()) entity_name = clean_str(record_attributes[1].upper())
@ -97,7 +97,7 @@ async def _handle_single_relationship_extraction(
record_attributes: list[str], record_attributes: list[str],
chunk_key: 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 return None
# add this record as edge # add this record as edge
source = clean_str(record_attributes[1].upper()) source = clean_str(record_attributes[1].upper())