From afb527281659a4ab3710234cab9208789147f7c1 Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Tue, 11 Feb 2025 17:07:53 +0800
Subject: [PATCH] Fix type and class name in doc status storage class
 assignment
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

• Corrected PGDocStatusStorage class name
• Fixed db assignment to class not instance
• Fixed incorrect Oracle reference
---
 lightrag/lightrag.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index a6ab4e26..b1670850 100644
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -554,8 +554,8 @@ class LightRAG:
                 self.vector_db_storage_cls.db = postgres_db
             if self.graph_storage == "PGGraphStorage":
                 self.graph_storage_cls.db = postgres_db
-            if self.doc_status_storage == "OracleGraphStorage":
-                self.doc_status_storage_cls = postgres_db
+            if self.doc_status_storage == "PGDocStatusStorage":
+                self.doc_status_storage_cls.db = postgres_db
 
         self.llm_response_cache = self.key_string_value_json_storage_cls(
             namespace=make_namespace(