diff --git a/config.yaml b/config.yaml index 7f82f47f..7a9aebce 100644 --- a/config.yaml +++ b/config.yaml @@ -97,6 +97,10 @@ LLM: api_key: 你的DifyLLM api key GeminiLLM: type: gemini + # 谷歌Gemini API,需要先在Google Cloud控制台创建API密钥并获取api_key + # 若在中国境内使用,请遵守《生成式人工智能服务管理暂行办法》 + # token申请地址: https://aistudio.google.com/apikey + # 若部署地无法访问接口,需要开启科学上网 api_key: 你的gemini api key model_name: "gemini-1.5-pro" # gemini-1.5-pro 是免费的 TTS: @@ -120,6 +124,8 @@ TTS: cluster: volcano_tts CosyVoiceSiliconflow: type: siliconflow + # 硅基流动TTS + # token申请地址 https://cloud.siliconflow.cn/account/ak model: FunAudioLLM/CosyVoice2-0.5B voice: FunAudioLLM/CosyVoice2-0.5B:alex output_file: tmp/ diff --git a/core/utils/llm.py b/core/utils/llm.py index 79b48dbb..1cdfcefb 100644 --- a/core/utils/llm.py +++ b/core/utils/llm.py @@ -1,15 +1,11 @@ import os import sys -import json import logging -import openai -import requests import importlib from datetime import datetime from core.utils.util import is_segment from core.utils.util import get_string_no_punctuation_or_emoji from core.utils.util import read_config, get_project_dir -from abc import ABC, abstractmethod logger = logging.getLogger(__name__) diff --git a/requirements.txt b/requirements.txt index 2e69445e..9fa53a9c 100755 --- a/requirements.txt +++ b/requirements.txt @@ -10,4 +10,5 @@ torchaudio==2.2.2 openai==1.61.0 google-generativeai==0.8.4 edge_tts==7.0.0 -httpx==0.27.2 \ No newline at end of file +httpx==0.27.2 +google-generativeai==0.8.4 \ No newline at end of file