参数配置统一
This commit is contained in:
@@ -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,
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user