add the missing MongoDocStatusStorage

2025-02-14 02:49:00 +08:00
parent 9a91b68e62
commit 637e7efa83
2 changed files with 6 additions and 2 deletions
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -80,7 +80,12 @@ STORAGE_IMPLEMENTATIONS = {
        "required_methods": ["query", "upsert"],
    },
    "DOC_STATUS_STORAGE": {
-        "implementations": ["JsonDocStatusStorage", "PGDocStatusStorage"],
+        "implementations": [
+            "JsonDocStatusStorage",
+            "PGDocStatusStorage",
+            "PGDocStatusStorage",
+            "MongoDocStatusStorage",
+        ],
        "required_methods": ["get_pending_docs"],
    },
 }