diff --git a/api/init_db.py b/api/init_db.py index fa8ff5d..d2e9b8e 100644 --- a/api/init_db.py +++ b/api/init_db.py @@ -303,32 +303,6 @@ def init_default_llm_models(): 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, @@ -362,56 +336,28 @@ def init_default_asr_models(): with db_session() as db: asr_models = [ ASRModel( - id="paraformer-v2", + id="FunAudioLLM/SenseVoiceSmall", user_id=1, - name="Paraformer V2", + name="FunAudioLLM/SenseVoiceSmall", vendor="SiliconFlow", - language="zh", + language="Multi-lingual", 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", + model_name="FunAudioLLM/SenseVoiceSmall", hotwords=[], enable_punctuation=True, enable_normalization=True, enabled=True, ), ASRModel( - id="whisper-1", + id="TeleAI/TeleSpeechASR", 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", + name="TeleAI/TeleSpeechASR", vendor="SiliconFlow", language="Multi-lingual", base_url="https://api.siliconflow.cn/v1", api_key="YOUR_API_KEY", - model_name="sensevoice", + model_name="TeleAI/TeleSpeechASR", hotwords=[], enable_punctuation=True, enable_normalization=True,