xiamuceer
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
"""设置相关的Pydantic模型"""
|
|
from pydantic import BaseModel, Field, ConfigDict
|
|
from typing import Optional
|
|
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地址")
|
|
model_name: 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, le=32000, description="最大token数")
|
|
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 |