From 09195182c27231e6c972c850c5cf6c3fd70a2e7e Mon Sep 17 00:00:00 2001
From: ArnoChen <arnochen101@gmail.com>
Date: Mon, 10 Feb 2025 01:01:48 +0800
Subject: [PATCH] enable MongoGraphStorage based on config

mongo graph
---
 lightrag/api/lightrag_server.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py
index 80c51442..8d13fab0 100644
--- a/lightrag/api/lightrag_server.py
+++ b/lightrag/api/lightrag_server.py
@@ -125,11 +125,14 @@ if qdrant_uri:
 # MongoDB config
 mongo_uri = config.get("mongodb", "uri", fallback=None)
 mongo_database = config.get("mongodb", "database", fallback="LightRAG")
+mongo_graph = config.getboolean("mongodb", "graph", fallback=False)
 if mongo_uri:
     os.environ["MONGO_URI"] = mongo_uri
     os.environ["MONGO_DATABASE"] = mongo_database
     rag_storage_config.KV_STORAGE = "MongoKVStorage"
     rag_storage_config.DOC_STATUS_STORAGE = "MongoKVStorage"
+    if mongo_graph:
+        rag_storage_config.GRAPH_STORAGE = "MongoGraphStorage"
 
 
 def get_default_host(binding_type: str) -> str: