xiamuceer
143 lines
6.1 KiB
Python
143 lines
6.1 KiB
Python
"""设置相关的Pydantic模型"""
|
|
from pydantic import BaseModel, Field, ConfigDict
|
|
from typing import Optional, List
|
|
from datetime import datetime
|
|
|
|
|
|
class SettingsBase(BaseModel):
|
|
"""设置基础模型"""
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
api_provider: Optional[str] = Field(default="openai", description="API提供商")
|
|
api_key: Optional[str] = Field(default=None, description="API密钥")
|
|
api_base_url: Optional[str] = Field(default=None, description="自定义API地址")
|
|
llm_model: Optional[str] = Field(default="gpt-4", description="模型名称")
|
|
temperature: Optional[float] = Field(default=0.7, ge=0.0, le=2.0, description="温度参数")
|
|
max_tokens: Optional[int] = Field(default=2000, ge=1, description="最大token数")
|
|
system_prompt: Optional[str] = Field(default=None, description="系统级别提示词，每次AI调用都会使用")
|
|
cover_api_provider: Optional[str] = Field(default=None, description="封面图片API提供商")
|
|
cover_api_key: Optional[str] = Field(default=None, description="封面图片API密钥")
|
|
cover_api_base_url: Optional[str] = Field(default=None, description="封面图片自定义API地址")
|
|
cover_image_model: Optional[str] = Field(default=None, description="封面图片模型名称")
|
|
cover_enabled: Optional[bool] = Field(default=False, description="是否启用封面图片生成")
|
|
preferences: Optional[str] = Field(default=None, description="其他偏好设置(JSON)")
|
|
|
|
|
|
class SettingsCreate(SettingsBase):
|
|
"""创建设置请求模型"""
|
|
pass
|
|
|
|
|
|
class SettingsUpdate(SettingsBase):
|
|
"""更新设置请求模型"""
|
|
pass
|
|
|
|
|
|
class SettingsResponse(SettingsBase):
|
|
"""设置响应模型"""
|
|
model_config = ConfigDict(from_attributes=True, protected_namespaces=())
|
|
|
|
id: str
|
|
user_id: str
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
|
|
class SystemSMTPSettingsBase(BaseModel):
|
|
"""系统 SMTP 设置基础模型"""
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
smtp_provider: str = Field(default="qq", description="SMTP 提供商")
|
|
smtp_host: Optional[str] = Field(default=None, description="SMTP 主机")
|
|
smtp_port: int = Field(default=465, ge=1, le=65535, description="SMTP 端口")
|
|
smtp_username: Optional[str] = Field(default=None, description="SMTP 用户名")
|
|
smtp_password: Optional[str] = Field(default=None, description="SMTP 密码或授权码")
|
|
smtp_use_tls: bool = Field(default=False, description="是否启用 TLS")
|
|
smtp_use_ssl: bool = Field(default=True, description="是否启用 SSL")
|
|
smtp_from_email: Optional[str] = Field(default=None, description="发件人邮箱")
|
|
smtp_from_name: str = Field(default="MuMuAINovel", description="发件人名称")
|
|
email_auth_enabled: bool = Field(default=True, description="是否启用邮箱认证")
|
|
email_register_enabled: bool = Field(default=True, description="是否启用邮箱注册")
|
|
verification_code_ttl_minutes: int = Field(default=10, ge=1, le=120, description="验证码有效期（分钟）")
|
|
verification_resend_interval_seconds: int = Field(default=60, ge=10, le=3600, description="验证码重发间隔（秒）")
|
|
|
|
|
|
class SystemSMTPSettingsUpdate(SystemSMTPSettingsBase):
|
|
"""系统 SMTP 设置更新模型"""
|
|
pass
|
|
|
|
|
|
class SystemSMTPSettingsResponse(SystemSMTPSettingsBase):
|
|
"""系统 SMTP 设置响应模型"""
|
|
model_config = ConfigDict(from_attributes=True, protected_namespaces=())
|
|
|
|
id: str
|
|
user_id: str
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
|
|
class SMTPTestRequest(BaseModel):
|
|
"""SMTP 测试请求模型"""
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
to_email: str = Field(..., min_length=3, max_length=255, description="测试收件邮箱")
|
|
|
|
|
|
# ========== API配置预设相关模型 ==========
|
|
|
|
class APIKeyPresetConfig(BaseModel):
|
|
"""预设配置内容"""
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
api_provider: str = Field(..., description="API提供商")
|
|
api_key: str = Field(..., description="API密钥")
|
|
api_base_url: Optional[str] = Field(None, description="自定义API地址")
|
|
llm_model: str = Field(..., description="模型名称")
|
|
temperature: float = Field(default=0.7, ge=0.0, le=2.0, description="温度参数")
|
|
max_tokens: int = Field(default=2000, ge=1, description="最大token数")
|
|
system_prompt: Optional[str] = Field(default=None, description="系统级别提示词")
|
|
|
|
|
|
class APIKeyPreset(BaseModel):
|
|
"""API配置预设"""
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
id: str = Field(..., description="预设ID")
|
|
name: str = Field(..., min_length=1, max_length=50, description="预设名称")
|
|
description: Optional[str] = Field(None, max_length=200, description="预设描述")
|
|
is_active: bool = Field(default=False, description="是否激活")
|
|
created_at: datetime = Field(..., description="创建时间")
|
|
config: APIKeyPresetConfig = Field(..., description="配置内容")
|
|
|
|
|
|
class PresetCreateRequest(BaseModel):
|
|
"""创建预设请求"""
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
name: str = Field(..., min_length=1, max_length=50, description="预设名称")
|
|
description: Optional[str] = Field(None, max_length=200, description="预设描述")
|
|
config: APIKeyPresetConfig = Field(..., description="配置内容")
|
|
|
|
|
|
class PresetUpdateRequest(BaseModel):
|
|
"""更新预设请求"""
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
name: Optional[str] = Field(None, min_length=1, max_length=50, description="预设名称")
|
|
description: Optional[str] = Field(None, max_length=200, description="预设描述")
|
|
config: Optional[APIKeyPresetConfig] = Field(None, description="配置内容")
|
|
|
|
|
|
class PresetResponse(APIKeyPreset):
|
|
"""预设响应"""
|
|
pass
|
|
|
|
|
|
class PresetListResponse(BaseModel):
|
|
"""预设列表响应"""
|
|
model_config = ConfigDict(protected_namespaces=())
|
|
|
|
presets: List[PresetResponse] = Field(..., description="预设列表")
|
|
total: int = Field(..., description="总数")
|
|
active_preset_id: Optional[str] = Field(None, description="当前激活的预设ID") |