fix(postgres): update document status with partial update instead of full upsert

This commit is contained in:
Brenon
2025-02-10 15:05:44 +03:00
parent d0779209d9
commit 4723e9b535
3 changed files with 35 additions and 5 deletions

View File

@@ -202,3 +202,7 @@ class DocStatusStorage(BaseKVStorage):
async def get_pending_docs(self) -> dict[str, DocProcessingStatus]:
"""Get all pending documents"""
raise NotImplementedError
async def update_doc_status(self, data: dict[str, Any]) -> None:
"""Updates the status of a document. By default, it calls upsert."""
await self.upsert(data)