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

fix(postgres): update document status with partial update instead of …
2025-02-11 22:14:52 +08:00
parent f902f552da 4723e9b535
commit a272c7b3f7
3 changed files with 34 additions and 5 deletions
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -632,7 +632,7 @@ class LightRAG:
                ]
                try:
                    await asyncio.gather(*tasks)
-                    await self.doc_status.upsert(
+                    await self.doc_status.update_doc_status(
                        {
                            doc_status_id: {
                                "status": DocStatus.PROCESSED,
@@ -649,7 +649,7 @@ class LightRAG:

                except Exception as e:
                    logger.error(f"Failed to process document {doc_id}: {str(e)}")
-                    await self.doc_status.upsert(
+                    await self.doc_status.update_doc_status(
                        {
                            doc_status_id: {
                                "status": DocStatus.FAILED,