Inject oracle db to LightRag storage class when needed

2025-02-11 03:54:54 +08:00
parent 8cfca5a141
commit a4cf7e66d3
2 changed files with 54 additions and 1 deletions
--- a/lightrag/kg/oracle_impl.py
+++ b/lightrag/kg/oracle_impl.py
@@ -361,7 +361,8 @@ class OracleVectorDBStorage(BaseVectorStorage):

@dataclass
 class OracleGraphStorage(BaseGraphStorage):
-    """基于Oracle的图存储模块"""
+    # should pass db object to self.db
+    db: OracleDB = None

    def __post_init__(self):
        """从graphml文件加载图"""