diff --git a/api/init_db.py b/api/init_db.py index 5c1ace4..1d3c7a3 100644 --- a/api/init_db.py +++ b/api/init_db.py @@ -32,6 +32,12 @@ SEED_ASR_IDS = { "telespeech_asr": short_id("asr"), } +SEED_ASSISTANT_IDS = { + "default": short_id("ast"), + "customer_service": short_id("ast"), + "english_tutor": short_id("ast"), +} + def ensure_db_dir(): """确保 SQLite 数据目录存在。""" @@ -193,7 +199,7 @@ def init_default_assistants(): with db_session() as db: assistants = [ Assistant( - id="default", + id=SEED_ASSISTANT_IDS["default"], user_id=1, name="AI 助手", call_count=0, @@ -211,7 +217,7 @@ def init_default_assistants(): asr_model_id=SEED_ASR_IDS["sensevoice_small"], ), Assistant( - id="customer_service", + id=SEED_ASSISTANT_IDS["customer_service"], user_id=1, name="客服助手", call_count=0, @@ -227,7 +233,7 @@ def init_default_assistants(): config_mode="platform", ), Assistant( - id="english_tutor", + id=SEED_ASSISTANT_IDS["english_tutor"], user_id=1, name="英语导师", call_count=0,