Remove db migration code to init

This commit is contained in:
Xin Wang
2026-02-12 14:29:47 +08:00
parent d41db6418c
commit a17ef6f182
2 changed files with 32 additions and 27 deletions

View File

@@ -3,6 +3,7 @@
import argparse
import os
import sys
from sqlalchemy import inspect, text
# 添加路径
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
@@ -23,6 +24,35 @@ def init_db():
print("✅ 数据库表创建完成")
def migrate_db_schema():
"""对现有数据库执行非破坏性 schema 迁移。"""
inspector = inspect(engine)
table_names = set(inspector.get_table_names())
if "assistants" not in table_names:
print(" assistants 表不存在,跳过增量迁移")
return
columns = {col["name"] for col in inspector.get_columns("assistants")}
alter_statements = []
if "generated_opener_enabled" not in columns:
alter_statements.append(
"ALTER TABLE assistants ADD COLUMN generated_opener_enabled BOOLEAN DEFAULT 0"
)
if "bot_cannot_be_interrupted" not in columns:
alter_statements.append(
"ALTER TABLE assistants ADD COLUMN bot_cannot_be_interrupted BOOLEAN DEFAULT 0"
)
if not alter_statements:
print("✅ Schema 迁移检查完成(无需变更)")
return
with engine.begin() as conn:
for stmt in alter_statements:
conn.execute(text(stmt))
print(f"✅ Schema 迁移完成(应用 {len(alter_statements)} 条 ALTER")
def rebuild_vector_store(reset_doc_status: bool = True):
"""重建知识库向量集合(按 DB 中的 KB 列表重建 collection 壳)。"""
from app.db import SessionLocal
@@ -455,6 +485,8 @@ if __name__ == "__main__":
if args.rebuild_db:
init_db()
else:
migrate_db_schema()
if args.recreate_tool_db:
init_default_tools(recreate=True)