参数配置统一

This commit is contained in:
2026-04-21 19:06:34 +08:00
parent e2eaac9498
commit 37e86f3bb1
10 changed files with 120 additions and 166 deletions

View File

@@ -37,11 +37,10 @@ logger = logging.getLogger(__name__)
@dataclass
class DocstoreConfig:
"""文档存储配置(用于父块存储)。"""
connection_string: Optional[str] = None
pool_config: Optional[Dict[str, Any]] = None
max_concurrency: Optional[int] = None
pool_config: Dict[str, Any] | None = None
max_concurrency: int | None = None
# 若要从外部注入已创建好的 docstore可直接设置此字段
instance: Optional[BaseStore] = None
instance: BaseStore | None = None
@dataclass
class IndexBuilderConfig:
@@ -147,7 +146,6 @@ class IndexBuilder:
# 使用工厂函数创建检索器,避免重复代码
self.retriever = create_parent_retriever(
collection_name=cfg.collection_name,
embeddings=self.embeddings,
parent_splitter=self.parent_splitter,
child_splitter=self.child_splitter,
docstore=self.docstore,
@@ -164,7 +162,6 @@ class IndexBuilder:
# 使用 create_docstore 创建 PostgreSQL 存储
docstore, conn_info = create_docstore(
connection_string=cfg.connection_string,
pool_config=cfg.pool_config,
max_concurrency=cfg.max_concurrency,
)