160 lines
6.9 KiB
Python
160 lines
6.9 KiB
Python
from datetime import datetime
|
|
from typing import Literal
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
from pydantic.alias_generators import to_camel
|
|
|
|
|
|
class AiChatRequestModel(BaseModel):
|
|
"""
|
|
AI对话请求模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
session_id: str | None = Field(default=None, description='会话ID')
|
|
model_id: int = Field(description='模型ID')
|
|
message: str = Field(description='用户消息')
|
|
is_reasoning: bool | None = Field(default=None, description='本次是否开启深度思考')
|
|
images: list[str] | None = Field(default=None, description='图片URL列表')
|
|
|
|
|
|
class AiChatConfigModel(BaseModel):
|
|
"""
|
|
AI对话配置模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
chat_config_id: int | None = Field(default=None, description='配置主键')
|
|
user_id: int | None = Field(default=None, description='用户ID')
|
|
temperature: float | None = Field(default=None, description='默认温度')
|
|
add_history_to_context: Literal['0', '1'] | None = Field(default=None, description='是否添加历史记录')
|
|
num_history_runs: int | None = Field(default=3, description='历史记录条数')
|
|
system_prompt: str | None = Field(default=None, description='系统提示词')
|
|
metrics_default_visible: Literal['0', '1'] | None = Field(default=None, description='默认显示指标')
|
|
vision_enabled: Literal['0', '1'] | None = Field(default=None, description='是否开启视觉')
|
|
image_max_size_mb: int | None = Field(default=None, description='图片最大大小(MB)')
|
|
create_time: datetime | None = Field(default=None, description='创建时间')
|
|
update_time: datetime | None = Field(default=None, description='更新时间')
|
|
|
|
|
|
class AiChatSessionBaseModel(BaseModel):
|
|
"""
|
|
AI对话会话基础模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
session_id: str = Field(description='会话ID')
|
|
session_title: str | None = Field(default=None, description='会话标题')
|
|
session_type: str | None = Field(default=None, description='会话类型')
|
|
user_id: str | None = Field(default=None, description='用户ID')
|
|
created_at: datetime | None = Field(default=None, description='创建时间')
|
|
updated_at: datetime | None = Field(default=None, description='更新时间')
|
|
|
|
|
|
class ModelInfoModel(BaseModel):
|
|
"""
|
|
对话会话数据模型-模型信息模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
id: str | None = Field(default=None, description='模型ID')
|
|
name: str | None = Field(default=None, description='模型名称')
|
|
provider: str | None = Field(default=None, description='模型提供者')
|
|
temperature: float | None = Field(default=None, description='模型温度')
|
|
|
|
|
|
class AgentDataModel(BaseModel):
|
|
"""
|
|
对话会话数据模型-智能体数据模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
model: ModelInfoModel | None = Field(default=None, description='模型信息')
|
|
agent_id: str | None = Field(default=None, description='智能体ID')
|
|
|
|
|
|
class SessionMetricsModel(BaseModel):
|
|
"""
|
|
对话会话数据模型-会话指标模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
cost: float | None = Field(default=None, description='成本（元）')
|
|
timer: float | None = Field(default=None, description='运行时长（秒）')
|
|
duration: float | None = Field(default=None, description='运行时长（秒）')
|
|
input_tokens: int | None = Field(default=None, description='输入Token数量')
|
|
total_tokens: int | None = Field(default=None, description='总Token数量')
|
|
output_tokens: int | None = Field(default=None, description='输出Token数量')
|
|
provider_metrics: dict | None = Field(default=None, description='提供者指标')
|
|
reasoning_tokens: int | None = Field(default=None, description='推理Token数量')
|
|
cache_read_tokens: int | None = Field(default=None, description='缓存读取Token数量')
|
|
additional_metrics: dict | None = Field(default=None, description='其他指标')
|
|
audio_input_tokens: int | None = Field(default=None, description='音频输入Token数量')
|
|
audio_total_tokens: int | None = Field(default=None, description='音频总Token数量')
|
|
cache_write_tokens: int | None = Field(default=None, description='缓存写入Token数量')
|
|
audio_output_tokens: int | None = Field(default=None, description='音频输出Token数量')
|
|
time_to_first_token: float | None = Field(default=None, description='到第一个Token的时间（秒）')
|
|
|
|
|
|
class SessionDataModel(BaseModel):
|
|
"""
|
|
对话会话数据模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
session_state: dict | None = Field(default=None, description='会话状态')
|
|
session_metrics: SessionMetricsModel | None = Field(default=None, description='会话指标')
|
|
|
|
|
|
class MessageMetrics(BaseModel):
|
|
"""
|
|
对话消息模型-消息指标模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
input_tokens: int | None = Field(default=None, description='输入Token数量')
|
|
total_tokens: int | None = Field(default=None, description='总Token数量')
|
|
output_tokens: int | None = Field(default=None, description='输出Token数量')
|
|
reasoning_tokens: int | None = Field(default=None, description='推理Token数量')
|
|
duration: float | None = Field(default=None, description='运行时长（秒）')
|
|
time_to_first_token: float | None = Field(default=None, description='到第一个Token的时间（秒）')
|
|
|
|
|
|
class ChatMessageModel(BaseModel):
|
|
"""
|
|
对话消息模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
id: str | None = Field(default=None, description='消息ID')
|
|
role: str | None = Field(default=None, description='角色')
|
|
content: str | None = Field(default=None, description='内容')
|
|
images: list[str] | None = Field(default=None, description='图片列表')
|
|
metrics: MessageMetrics | None = Field(default=None, description='Token使用统计')
|
|
created_at: datetime | None = Field(default=None, description='创建时间')
|
|
from_history: bool | None = Field(default=None, description='是否来自历史记录')
|
|
reasoning_content: str | None = Field(default=None, description='推理/思考内容')
|
|
stop_after_tool_call: bool | None = Field(default=None, description='是否在工具调用后停止')
|
|
|
|
|
|
class AiChatSessionModel(AiChatSessionBaseModel):
|
|
"""
|
|
AI对话会话模型
|
|
"""
|
|
|
|
model_config = ConfigDict(alias_generator=to_camel)
|
|
|
|
agent_id: str | None = Field(default=None, description='智能体ID')
|
|
session_data: SessionDataModel | None = Field(default=None, description='会话数据')
|
|
agent_data: AgentDataModel | None = Field(default=None, description='智能体数据')
|
|
messages: list[ChatMessageModel] | None = Field(default=None, description='消息列表')
|