Merge remote-tracking branch 'origin/main' into graph-viewer-webui

2025-02-11 22:52:01 +08:00
parent 3cacc088fe 18acb4a2b1
commit 1af72b4ce9
4 changed files with 38 additions and 7 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,