Clean up TTS module dependencies

- Remove references to deleted tts_config.py
- Simplify tts_routes.py to directly return TTSGeminiChatService
- Update __init__.py imports
- Prepare for multi-speaker TTS testing
This commit is contained in:
zzh
2025-07-15 15:44:55 +09:00
parent eeec45274b
commit af5b2fa2c9
2 changed files with 5 additions and 25 deletions

View File

@@ -4,17 +4,14 @@ Multi-speaker TTS functionality for conversation scenarios
""" """
from .tts_chat_service import TTSGeminiChatService from .tts_chat_service import TTSGeminiChatService
from .tts_config import TTSConfig, create_chat_service
from .tts_models import TTSGenerationConfig, MultiSpeakerVoiceConfig, SpeechConfig, TTSRequest from .tts_models import TTSGenerationConfig, MultiSpeakerVoiceConfig, SpeechConfig, TTSRequest
from .tts_response_handler import TTSResponseHandler from .tts_response_handler import TTSResponseHandler
from .tts_routes import get_tts_chat_service from .tts_routes import get_tts_chat_service
__all__ = [ __all__ = [
"TTSGeminiChatService", "TTSGeminiChatService",
"TTSConfig",
"create_chat_service",
"TTSGenerationConfig", "TTSGenerationConfig",
"MultiSpeakerVoiceConfig", "MultiSpeakerVoiceConfig",
"SpeechConfig", "SpeechConfig",
"TTSRequest", "TTSRequest",
"TTSResponseHandler", "TTSResponseHandler",

View File

@@ -1,17 +1,13 @@
""" """
TTS路由扩展 TTS路由扩展
可选的路由覆盖用于启用TTS功能 提供多人TTS增强服务
使用时可以替换原始路由的依赖注入
""" """
from fastapi import Depends from fastapi import Depends
from typing import Union
from app.config.config import settings from app.config.config import settings
from app.service.key.key_manager import KeyManager, get_key_manager_instance from app.service.key.key_manager import KeyManager, get_key_manager_instance
from app.service.chat.gemini_chat_service import GeminiChatService
from app.service.tts.multi_speaker.tts_chat_service import TTSGeminiChatService from app.service.tts.multi_speaker.tts_chat_service import TTSGeminiChatService
from app.service.tts.multi_speaker.tts_config import TTSConfig
async def get_key_manager(): async def get_key_manager():
@@ -19,23 +15,10 @@ async def get_key_manager():
return get_key_manager_instance() return get_key_manager_instance()
async def get_tts_chat_service(key_manager: KeyManager = Depends(get_key_manager)) -> Union[GeminiChatService, TTSGeminiChatService]: async def get_tts_chat_service(key_manager: KeyManager = Depends(get_key_manager)) -> TTSGeminiChatService:
""" """
获取聊天服务实例支持TTS 获取多人TTS增强聊天服务实例
根据配置返回原始服务或TTS增强服务
""" """
return TTSConfig.get_chat_service(settings.BASE_URL, key_manager) return TTSGeminiChatService(settings.BASE_URL, key_manager)
# 使用说明:
# 要启用TTS功能需要
# 1. 设置环境变量 ENABLE_TTS=true
# 2. 在路由中使用 get_tts_chat_service 替换 get_chat_service
#
# 例如在 gemini_routes.py 中:
# from app.service.tts.multi_speaker.tts_routes import get_tts_chat_service
#
# async def generate_content(
# chat_service = Depends(get_tts_chat_service) # 替换原来的依赖
# ):
# ...