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

2
lightrag/utils_graph.py
View File

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