From 637e7efa8375ec1f25aba3e7181500fa4133ec25 Mon Sep 17 00:00:00 2001
From: ArnoChen <arnochen101@gmail.com>
Date: Fri, 14 Feb 2025 02:49:00 +0800
Subject: [PATCH] add the missing MongoDocStatusStorage

---
 config.ini.example   | 1 -
 lightrag/lightrag.py | 7 ++++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/config.ini.example b/config.ini.example
index e6ceed0a..3041611e 100644
--- a/config.ini.example
+++ b/config.ini.example
@@ -6,7 +6,6 @@ password = your-password
 [mongodb]
 uri = mongodb+srv://name:password@your-cluster-address
 database = lightrag
-graph = false
 
 [redis]
 uri=redis://localhost:6379/1
diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index d53a252d..00c8fbbe 100644
--- 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"],
     },
 }