370 lines
14 KiB
Python
370 lines
14 KiB
Python
#!/usr/bin/env python3
|
||
"""初始化数据库"""
|
||
import os
|
||
import sys
|
||
|
||
# 添加路径
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
from app.db import Base, engine, DATABASE_URL
|
||
from app.models import Voice, Assistant, KnowledgeBase, Workflow, LLMModel, ASRModel
|
||
|
||
|
||
def init_db():
|
||
"""创建所有表"""
|
||
# 确保 data 目录存在
|
||
data_dir = os.path.dirname(DATABASE_URL.replace("sqlite:///", ""))
|
||
os.makedirs(data_dir, exist_ok=True)
|
||
|
||
print("📦 创建数据库表...")
|
||
Base.metadata.drop_all(bind=engine) # 删除旧表
|
||
Base.metadata.create_all(bind=engine)
|
||
print("✅ 数据库表创建完成")
|
||
|
||
|
||
def init_default_data():
|
||
from sqlalchemy.orm import Session
|
||
from app.db import SessionLocal
|
||
from app.models import Voice
|
||
|
||
db = SessionLocal()
|
||
try:
|
||
# 检查是否已有数据
|
||
if db.query(Voice).count() == 0:
|
||
# SiliconFlow CosyVoice 2.0 预设声音 (8个)
|
||
# 参考: https://docs.siliconflow.cn/cn/api-reference/audio/create-speech
|
||
voices = [
|
||
# 男声 (Male Voices)
|
||
Voice(id="alex", name="Alex", vendor="SiliconFlow", gender="Male", language="en",
|
||
description="Steady male voice.", is_system=True),
|
||
Voice(id="david", name="David", vendor="SiliconFlow", gender="Male", language="en",
|
||
description="Cheerful male voice.", is_system=True),
|
||
# 女声 (Female Voices)
|
||
Voice(id="bella", name="Bella", vendor="SiliconFlow", gender="Female", language="en",
|
||
description="Passionate female voice.", is_system=True),
|
||
Voice(id="claire", name="Claire", vendor="SiliconFlow", gender="Female", language="en",
|
||
description="Gentle female voice.", is_system=True),
|
||
]
|
||
for v in voices:
|
||
db.add(v)
|
||
db.commit()
|
||
print("✅ 默认声音数据已初始化 (SiliconFlow CosyVoice 2.0)")
|
||
finally:
|
||
db.close()
|
||
|
||
|
||
def init_default_assistants():
|
||
"""初始化默认助手"""
|
||
from sqlalchemy.orm import Session
|
||
from app.db import SessionLocal
|
||
|
||
db = SessionLocal()
|
||
try:
|
||
if db.query(Assistant).count() == 0:
|
||
assistants = [
|
||
Assistant(
|
||
id="default",
|
||
user_id=1,
|
||
name="AI 助手",
|
||
call_count=0,
|
||
opener="你好！我是AI助手，有什么可以帮你的吗？",
|
||
prompt="你是一个友好的AI助手，请用简洁清晰的语言回答用户的问题。",
|
||
language="zh",
|
||
voice="anna",
|
||
speed=1.0,
|
||
hotwords=[],
|
||
tools=["search", "calculator"],
|
||
interruption_sensitivity=500,
|
||
config_mode="platform",
|
||
llm_model_id="deepseek-chat",
|
||
asr_model_id="paraformer-v2",
|
||
),
|
||
Assistant(
|
||
id="customer_service",
|
||
user_id=1,
|
||
name="客服助手",
|
||
call_count=0,
|
||
opener="您好，欢迎致电客服中心，请问有什么可以帮您？",
|
||
prompt="你是一个专业的客服人员，耐心解答客户问题，提供优质的服务体验。",
|
||
language="zh",
|
||
voice="bella",
|
||
speed=1.0,
|
||
hotwords=["客服", "投诉", "咨询"],
|
||
tools=["search"],
|
||
interruption_sensitivity=600,
|
||
config_mode="platform",
|
||
),
|
||
Assistant(
|
||
id="english_tutor",
|
||
user_id=1,
|
||
name="英语导师",
|
||
call_count=0,
|
||
opener="Hello! I'm your English learning companion. How can I help you today?",
|
||
prompt="You are a friendly English tutor. Help users practice English conversation and explain grammar points clearly.",
|
||
language="en",
|
||
voice="alex",
|
||
speed=1.0,
|
||
hotwords=["grammar", "vocabulary", "practice"],
|
||
tools=[],
|
||
interruption_sensitivity=400,
|
||
config_mode="platform",
|
||
),
|
||
]
|
||
for a in assistants:
|
||
db.add(a)
|
||
db.commit()
|
||
print("✅ 默认助手数据已初始化")
|
||
finally:
|
||
db.close()
|
||
|
||
|
||
def init_default_workflows():
|
||
"""初始化默认工作流"""
|
||
from sqlalchemy.orm import Session
|
||
from app.db import SessionLocal
|
||
from datetime import datetime
|
||
|
||
db = SessionLocal()
|
||
try:
|
||
if db.query(Workflow).count() == 0:
|
||
now = datetime.utcnow().strftime("%Y-%m-%d %H:%M:%S")
|
||
workflows = [
|
||
Workflow(
|
||
id="simple_conversation",
|
||
user_id=1,
|
||
name="简单对话",
|
||
node_count=2,
|
||
created_at=now,
|
||
updated_at=now,
|
||
global_prompt="处理简单的对话流程，用户问什么答什么。",
|
||
nodes=[
|
||
{"id": "1", "type": "start", "position": {"x": 100, "y": 100}, "data": {"label": "开始"}},
|
||
{"id": "2", "type": "ai_reply", "position": {"x": 300, "y": 100}, "data": {"label": "AI回复"}},
|
||
],
|
||
edges=[{"source": "1", "target": "2", "id": "e1-2"}],
|
||
),
|
||
Workflow(
|
||
id="voice_input_flow",
|
||
user_id=1,
|
||
name="语音输入流程",
|
||
node_count=4,
|
||
created_at=now,
|
||
updated_at=now,
|
||
global_prompt="处理语音输入的完整流程。",
|
||
nodes=[
|
||
{"id": "1", "type": "start", "position": {"x": 100, "y": 100}, "data": {"label": "开始"}},
|
||
{"id": "2", "type": "asr", "position": {"x": 250, "y": 100}, "data": {"label": "语音识别"}},
|
||
{"id": "3", "type": "llm", "position": {"x": 400, "y": 100}, "data": {"label": "LLM处理"}},
|
||
{"id": "4", "type": "tts", "position": {"x": 550, "y": 100}, "data": {"label": "语音合成"}},
|
||
],
|
||
edges=[
|
||
{"source": "1", "target": "2", "id": "e1-2"},
|
||
{"source": "2", "target": "3", "id": "e2-3"},
|
||
{"source": "3", "target": "4", "id": "e3-4"},
|
||
],
|
||
),
|
||
]
|
||
for w in workflows:
|
||
db.add(w)
|
||
db.commit()
|
||
print("✅ 默认工作流数据已初始化")
|
||
finally:
|
||
db.close()
|
||
|
||
|
||
def init_default_knowledge_bases():
|
||
"""初始化默认知识库"""
|
||
from sqlalchemy.orm import Session
|
||
from app.db import SessionLocal
|
||
|
||
db = SessionLocal()
|
||
try:
|
||
if db.query(KnowledgeBase).count() == 0:
|
||
kb = KnowledgeBase(
|
||
id="default_kb",
|
||
user_id=1,
|
||
name="默认知识库",
|
||
description="系统默认知识库，用于存储常见问题解答。",
|
||
embedding_model="text-embedding-3-small",
|
||
chunk_size=500,
|
||
chunk_overlap=50,
|
||
doc_count=0,
|
||
chunk_count=0,
|
||
status="active",
|
||
)
|
||
db.add(kb)
|
||
db.commit()
|
||
print("✅ 默认知识库已初始化")
|
||
finally:
|
||
db.close()
|
||
|
||
|
||
def init_default_llm_models():
|
||
"""初始化默认LLM模型"""
|
||
from sqlalchemy.orm import Session
|
||
from app.db import SessionLocal
|
||
|
||
db = SessionLocal()
|
||
try:
|
||
if db.query(LLMModel).count() == 0:
|
||
llm_models = [
|
||
LLMModel(
|
||
id="deepseek-chat",
|
||
user_id=1,
|
||
name="DeepSeek Chat",
|
||
vendor="SiliconFlow",
|
||
type="text",
|
||
base_url="https://api.deepseek.com",
|
||
api_key="YOUR_API_KEY", # 用户需替换
|
||
model_name="deepseek-chat",
|
||
temperature=0.7,
|
||
context_length=4096,
|
||
enabled=True,
|
||
),
|
||
LLMModel(
|
||
id="deepseek-reasoner",
|
||
user_id=1,
|
||
name="DeepSeek Reasoner",
|
||
vendor="SiliconFlow",
|
||
type="text",
|
||
base_url="https://api.deepseek.com",
|
||
api_key="YOUR_API_KEY",
|
||
model_name="deepseek-reasoner",
|
||
temperature=0.7,
|
||
context_length=4096,
|
||
enabled=True,
|
||
),
|
||
LLMModel(
|
||
id="gpt-4o",
|
||
user_id=1,
|
||
name="GPT-4o",
|
||
vendor="OpenAI",
|
||
type="text",
|
||
base_url="https://api.openai.com/v1",
|
||
api_key="YOUR_API_KEY",
|
||
model_name="gpt-4o",
|
||
temperature=0.7,
|
||
context_length=16384,
|
||
enabled=True,
|
||
),
|
||
LLMModel(
|
||
id="glm-4",
|
||
user_id=1,
|
||
name="GLM-4",
|
||
vendor="ZhipuAI",
|
||
type="text",
|
||
base_url="https://open.bigmodel.cn/api/paas/v4",
|
||
api_key="YOUR_API_KEY",
|
||
model_name="glm-4",
|
||
temperature=0.7,
|
||
context_length=8192,
|
||
enabled=True,
|
||
),
|
||
LLMModel(
|
||
id="text-embedding-3-small",
|
||
user_id=1,
|
||
name="Embedding 3 Small",
|
||
vendor="OpenAI",
|
||
type="embedding",
|
||
base_url="https://api.openai.com/v1",
|
||
api_key="YOUR_API_KEY",
|
||
model_name="text-embedding-3-small",
|
||
enabled=True,
|
||
),
|
||
]
|
||
for m in llm_models:
|
||
db.add(m)
|
||
db.commit()
|
||
print("✅ 默认LLM模型已初始化")
|
||
finally:
|
||
db.close()
|
||
|
||
|
||
def init_default_asr_models():
|
||
"""初始化默认ASR模型"""
|
||
from sqlalchemy.orm import Session
|
||
from app.db import SessionLocal
|
||
|
||
db = SessionLocal()
|
||
try:
|
||
if db.query(ASRModel).count() == 0:
|
||
asr_models = [
|
||
ASRModel(
|
||
id="paraformer-v2",
|
||
user_id=1,
|
||
name="Paraformer V2",
|
||
vendor="SiliconFlow",
|
||
language="zh",
|
||
base_url="https://api.siliconflow.cn/v1",
|
||
api_key="YOUR_API_KEY",
|
||
model_name="paraformer-v2",
|
||
hotwords=["人工智能", "机器学习"],
|
||
enable_punctuation=True,
|
||
enable_normalization=True,
|
||
enabled=True,
|
||
),
|
||
ASRModel(
|
||
id="paraformer-en",
|
||
user_id=1,
|
||
name="Paraformer English",
|
||
vendor="SiliconFlow",
|
||
language="en",
|
||
base_url="https://api.siliconflow.cn/v1",
|
||
api_key="YOUR_API_KEY",
|
||
model_name="paraformer-en",
|
||
hotwords=[],
|
||
enable_punctuation=True,
|
||
enable_normalization=True,
|
||
enabled=True,
|
||
),
|
||
ASRModel(
|
||
id="whisper-1",
|
||
user_id=1,
|
||
name="Whisper",
|
||
vendor="OpenAI",
|
||
language="Multi-lingual",
|
||
base_url="https://api.openai.com/v1",
|
||
api_key="YOUR_API_KEY",
|
||
model_name="whisper-1",
|
||
hotwords=[],
|
||
enable_punctuation=True,
|
||
enable_normalization=True,
|
||
enabled=True,
|
||
),
|
||
ASRModel(
|
||
id="sensevoice",
|
||
user_id=1,
|
||
name="SenseVoice",
|
||
vendor="SiliconFlow",
|
||
language="Multi-lingual",
|
||
base_url="https://api.siliconflow.cn/v1",
|
||
api_key="YOUR_API_KEY",
|
||
model_name="sensevoice",
|
||
hotwords=[],
|
||
enable_punctuation=True,
|
||
enable_normalization=True,
|
||
enabled=True,
|
||
),
|
||
]
|
||
for m in asr_models:
|
||
db.add(m)
|
||
db.commit()
|
||
print("✅ 默认ASR模型已初始化")
|
||
finally:
|
||
db.close()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# 确保 data 目录存在
|
||
data_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "data")
|
||
os.makedirs(data_dir, exist_ok=True)
|
||
|
||
init_db()
|
||
init_default_data()
|
||
init_default_assistants()
|
||
init_default_workflows()
|
||
init_default_knowledge_bases()
|
||
init_default_llm_models()
|
||
init_default_asr_models()
|
||
print("🎉 数据库初始化完成！")
|