Merge pull request #741 from Brenon28/fix-update-doc-status-postgresql

fix(postgres): update document status with partial update instead of …
This commit is contained in:
zrguo
2025-02-11 22:14:52 +08:00
committed by GitHub
3 changed files with 34 additions and 5 deletions

View File

@@ -226,3 +226,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)