From cfb7117dd6a5f04805aa090b062825da8871cba3 Mon Sep 17 00:00:00 2001 From: yangdx Date: Wed, 30 Jul 2025 03:44:20 +0800 Subject: [PATCH] Fix track_id missing for query in PostgreSQL --- lightrag/kg/postgres_impl.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py index 0d2e7da3..1b5a7078 100644 --- a/lightrag/kg/postgres_impl.py +++ b/lightrag/kg/postgres_impl.py @@ -1820,6 +1820,7 @@ class PGDocStatusStorage(DocStatusStorage): chunks_list=chunks_list, metadata=metadata, error_msg=result[0].get("error_msg"), + track_id=result[0].get("track_id"), ) async def get_by_ids(self, ids: list[str]) -> list[dict[str, Any]]: @@ -1869,6 +1870,7 @@ class PGDocStatusStorage(DocStatusStorage): "chunks_list": chunks_list, "metadata": metadata, "error_msg": row.get("error_msg"), + "track_id": row.get("track_id"), } ) @@ -1927,6 +1929,7 @@ class PGDocStatusStorage(DocStatusStorage): chunks_list=chunks_list, metadata=metadata, error_msg=element.get("error_msg"), + track_id=element.get("track_id"), ) return docs_by_status