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

Fix file_path missing in acreate_entity and acreate_relation

Browse Source
This commit is contained in:
yangdx
2025-04-26 22:15:54 +08:00
parent 987c1e4265
commit 90cacdbcc3
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lightrag/kg/postgres_impl.py
View File

@@ -578,8 +578,7 @@ class PGVectorStorage(BaseVectorStorage):
"content": item["content"],
"content_vector": json.dumps(item["__vector__"].tolist()),
"chunk_ids": chunk_ids,
"file_path": item["file_path"],
# TODO: add document_id
"file_path": item.get("file_path", None),
}
return upsert_sql, data
@@ -599,8 +598,7 @@ class PGVectorStorage(BaseVectorStorage):
"content": item["content"],
"content_vector": json.dumps(item["__vector__"].tolist()),
"chunk_ids": chunk_ids,
"file_path": item["file_path"],
# TODO: add document_id
"file_path": item.get("file_path", None),
}
return upsert_sql, data

2
lightrag/utils_graph.py
View File

@@ -502,6 +502,7 @@ async def acreate_entity(
"source_id": source_id,
"description": description,
"entity_type": entity_type,
"file_path": entity_data.get("file_path", "manual_creation"),
}
}
@@ -607,6 +608,7 @@ async def acreate_relation(
"description": description,
"keywords": keywords,
"weight": weight,
"file_path": relation_data.get("file_path", "manual_creation"),
}
}