参数配置统一

backend/rag_core/store/__init__.py

@@ -9,14 +9,13 @@
     
     >>> # 创建 PostgreSQL 存储
     >>> store, conn = create_docstore(
-    ...     connection_string="postgresql://user:pass@host:5432/db",
     ...     table_name="parent_docs"
     ... )
 """
 
 
 from .postgres import PostgresDocStore
-from .factory import create_docstore, get_docstore_uri, DEFAULT_DB_URI
+from .factory import create_docstore, get_docstore_uri
 
 __version__ = "2.0.0"
 
@@ -27,5 +26,4 @@ __all__ = [
     # 工厂函数
     "create_docstore",
     "get_docstore_uri",
-    "DEFAULT_DB_URI",
 ]

backend/rag_core/store/factory.py

@@ -5,17 +5,14 @@
 """
 
 import os
-from ..config import DB_URI, DOCSTORE_URI
+from ..config import DOCSTORE_URI
 import logging
-from typing import Optional, Tuple
+from typing import Tuple
 
 from langchain_core.stores import BaseStore
 from .postgres import PostgresDocStore
 
-logger = logging.getLogger(__name__)        
-
-# 默认连接字符串（从环境变量读取）
-DEFAULT_DB_URI = DB_URI
+logger = logging.getLogger(__name__)
 
 
 def get_docstore_uri() -> str:
@@ -24,48 +21,36 @@ def get_docstore_uri() -> str:
 
 
 def create_docstore(
-    store_type: str = "postgres",
-    connection_string: Optional[str] = None,
     table_name: str = "parent_documents",
-    pool_config: Optional[dict] = None,
-    max_concurrency: Optional[int] = None
-) -> Tuple[BaseStore, Optional[str]]:
+    pool_config: dict | None = None,
+    max_concurrency: int | None = None
+) -> Tuple[BaseStore, str]:
     """
     工厂函数，创建 PostgreSQL 文档存储。
-    
+
     Args:
-        store_type: 存储类型，目前仅支持 "postgres"（默认）
-        connection_string: PostgreSQL 连接字符串
         table_name: PostgreSQL 表名（默认：parent_documents）
         pool_config: 连接池配置
         max_concurrency: 最大并发操作数，如果为 None 则不限制
-    
+
     Returns:
         元组 (存储实例, 连接字符串)
-    
+
     Raises:
-        ValueError: 不支持的存储类型
         ImportError: 缺少必要的依赖
-    
+
     Example:
         >>> # 创建 PostgreSQL 存储
         >>> store, conn = create_docstore(
-        ...     connection_string="postgresql://user:pass@host:5432/db",
         ...     table_name="parent_docs",
         ...     max_concurrency=10
         ... )
     """
-    store_type = store_type.lower()
-    
-    if store_type == "postgres":
-        conn_str = connection_string or get_docstore_uri()
-        store = PostgresDocStore(
-            connection_string=conn_str,
-            table_name=table_name,
-            pool_config=pool_config,
-            max_concurrency=max_concurrency
-        )
-        return store, conn_str
-    
-    else:
-        raise ValueError(f"不支持的存储类型: {store_type}。目前仅支持: postgres")
+    conn_str = get_docstore_uri()
+    store = PostgresDocStore(
+        connection_string=conn_str,
+        table_name=table_name,
+        pool_config=pool_config,
+        max_concurrency=max_concurrency
+    )
+    return store, conn_str