增加exe打包配置

This commit is contained in:
lasean.zhou
2026-04-15 13:07:39 +08:00
parent 2d0b1ab36f
commit 150250817b
6 changed files with 119 additions and 15 deletions
+9 -7
View File
@@ -20,7 +20,7 @@ class SchemaIndexer:
功能:
1. 为表结构构建向量索引
2. 基于问题的表检索
3. Chroma 使用内存模式(EphemeralClient),进程退出后不保留;``persist_dir`` 仅作配置/统计引用
3. Chroma 使用持久化(PersistentClient),向量与元数据写入 ``persist_dir``,进程重启后可复用
"""
def __init__(
@@ -34,15 +34,17 @@ class SchemaIndexer:
Args:
embedder: Embedding模型实例
persist_dir: 历史配置中的向量库路径(仅展示与统计,不落盘)
persist_dir: Chroma 持久化根目录(磁盘路径)
collection_name: 集合名称
"""
self.embedder = embedder
self.persist_dir = Path(persist_dir)
self.persist_dir.mkdir(parents=True, exist_ok=True)
self.collection_name = collection_name
# 内存 Chroma:不落盘,每次进程需重新 build_index
self.client = chromadb.EphemeralClient(
# 持久化 Chroma:数据落盘至 persist_dir
self.client = chromadb.PersistentClient(
path=str(self.persist_dir),
settings=ChromaSettings(anonymized_telemetry=False),
)
@@ -53,8 +55,8 @@ class SchemaIndexer:
)
logger.info(
f"[OK] 初始化SchemaIndexer(Chroma内存): collection={self.collection_name}, "
f"配置路径引用={self.persist_dir}"
f"[OK] 初始化SchemaIndexer(Chroma持久化): collection={self.collection_name}, "
f"path={self.persist_dir}"
)
def build_index(
@@ -250,5 +252,5 @@ class SchemaIndexer:
"total_columns": total_columns,
"avg_columns": total_columns / count if count > 0 else 0,
"persist_dir": str(self.persist_dir),
"chroma_mode": "memory",
"chroma_mode": "persistent",
}