mirror of
https://github.com/snailyp/gemini-balance.git
synced 2026-05-30 20:59:45 +08:00
b14bb93d8f
重构项目目录结构，提高代码组织性和可维护性 将schemas目录重命名为domain，更好地表达领域模型概念 将services目录细分为service/chat、service/image等子目录 将api目录重命名为router，更符合FastAPI惯例 创建utils目录存放通用工具函数 更新FastAPI应用程序生命周期管理 替换已弃用的on_event方法为推荐的lifespan事件处理器 添加应用程序关闭时的日志记录 代码质量改进 抽取常量到constants.py，减少硬编码值 添加helpers.py提供通用工具函数 优化配置管理，使用环境变量和默认值 完善文档字符串，提高代码可读性
33 lines
977 B
Python
33 lines
977 B
Python
from pydantic import BaseModel
|
|
from typing import List, Optional, Union
|
|
|
|
from app.core.constants import DEFAULT_MAX_TOKENS, 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
|
|
tools: Optional[List[dict]] = []
|
|
max_tokens: Optional[int] = DEFAULT_MAX_TOKENS
|
|
top_p: Optional[float] = DEFAULT_TOP_P
|
|
top_k: Optional[int] = DEFAULT_TOP_K
|
|
stop: Optional[List[str]] = []
|
|
|
|
|
|
class EmbeddingRequest(BaseModel):
|
|
input: Union[str, List[str]]
|
|
model: str = "text-embedding-004"
|
|
encoding_format: Optional[str] = "float"
|
|
|
|
|
|
class ImageGenerationRequest(BaseModel):
|
|
model: str = "DALL-E-3"
|
|
prompt: str = ""
|
|
n: int = 1
|
|
size: Optional[str] = "1024x1024"
|
|
quality: Optional[str] = ""
|
|
style: Optional[str] = ""
|
|
response_format: Optional[str] = "url"
|