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

Fix track_id missing for query in PostgreSQL

Browse source
This commit is contained in:
yangdx 2025-07-30 03:44:20 +08:00
parent 5ec7eedf37
commit cfb7117dd6
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
lightrag/kg/postgres_impl.py
View file

@ -1820,6 +1820,7 @@ class PGDocStatusStorage(DocStatusStorage):
chunks_list=chunks_list, chunks_list=chunks_list,
metadata=metadata, metadata=metadata,
error_msg=result[0].get("error_msg"), 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]]: async def get_by_ids(self, ids: list[str]) -> list[dict[str, Any]]:
@ -1869,6 +1870,7 @@ class PGDocStatusStorage(DocStatusStorage):
"chunks_list": chunks_list, "chunks_list": chunks_list,
"metadata": metadata, "metadata": metadata,
"error_msg": row.get("error_msg"), "error_msg": row.get("error_msg"),
"track_id": row.get("track_id"),
} }
) )
@ -1927,6 +1929,7 @@ class PGDocStatusStorage(DocStatusStorage):
chunks_list=chunks_list, chunks_list=chunks_list,
metadata=metadata, metadata=metadata,
error_msg=element.get("error_msg"), error_msg=element.get("error_msg"),
track_id=element.get("track_id"),
) )
return docs_by_status return docs_by_status