mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-05-20 15:50:10 +08:00
f38b5ae870
- 在.env.example中添加TTS模型、语音名称和语速的配置选项 - 更新README文件，增加TTS相关配置的说明 - 在配置类中添加TTS相关设置 - 新增TTS请求模型以支持文本转语音功能 - 更新智能路由中间件以支持音频请求 - 在路由中添加处理TTS请求的API接口 - 更新前端配置编辑器以支持TTS配置选项
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
from pydantic import BaseModel
|
|
from typing import Any, Dict, List, Optional, Union
|
|
|
|
from app.core.constants import DEFAULT_MODEL, DEFAULT_TEMPERATURE, DEFAULT_TOP_K, DEFAULT_TOP_P
|
|
|
|
|
|
class ChatRequest(BaseModel):
|
|
messages: List[dict]
|
|
model: str = DEFAULT_MODEL
|
|
temperature: Optional[float] = DEFAULT_TEMPERATURE
|
|
stream: Optional[bool] = False
|
|
max_tokens: Optional[int] = None
|
|
top_p: Optional[float] = DEFAULT_TOP_P
|
|
top_k: Optional[int] = DEFAULT_TOP_K
|
|
stop: Optional[Union[List[str],str]] = None
|
|
reasoning_effort: Optional[str] = None
|
|
tools: Optional[Union[List[Dict[str, Any]], Dict[str, Any]]] = []
|
|
tool_choice: Optional[str] = None
|
|
response_format: Optional[dict] = None
|
|
|
|
|
|
class EmbeddingRequest(BaseModel):
|
|
input: Union[str, List[str]]
|
|
model: str = "text-embedding-004"
|
|
encoding_format: Optional[str] = "float"
|
|
|
|
|
|
class ImageGenerationRequest(BaseModel):
|
|
model: str = "imagen-3.0-generate-002"
|
|
prompt: str = ""
|
|
n: int = 1
|
|
size: Optional[str] = "1024x1024"
|
|
quality: Optional[str] = None
|
|
style: Optional[str] = None
|
|
response_format: Optional[str] = "url"
|
|
|
|
|
|
class TTSRequest(BaseModel):
|
|
model: str = "gemini-2.5-flash-preview-tts"
|
|
input: str
|
|
voice: str = "Kore"
|
|
response_format: Optional[str] = "wav"
|