255 lines
11 KiB
Python
255 lines
11 KiB
Python
"""聊天 API schema。"""
|
|
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
|
|
class CreateSessionRequest(BaseModel):
|
|
model_config = ConfigDict(
|
|
json_schema_extra={
|
|
"examples": [
|
|
{
|
|
"title": "大模型联调会话",
|
|
"character_id": "character_ec83d55746c54e47ab6f98d683032626",
|
|
}
|
|
]
|
|
}
|
|
)
|
|
|
|
title: str | None = Field(default=None, max_length=200, description="会话标题。")
|
|
character_id: str | None = Field(default=None, description="人物 ID；为空时使用当前激活人物。")
|
|
|
|
|
|
class UpdateSessionRequest(BaseModel):
|
|
model_config = ConfigDict(
|
|
json_schema_extra={
|
|
"examples": [
|
|
{"title": "新的会话标题"},
|
|
{"status": "archived"},
|
|
]
|
|
}
|
|
)
|
|
|
|
title: str | None = Field(default=None, max_length=200, description="新标题；为空时保持不变。")
|
|
status: str | None = Field(
|
|
default=None,
|
|
description="会话状态切换：active / archived。deleted 请走 DELETE 端点。",
|
|
)
|
|
|
|
|
|
class ChatSessionResponse(BaseModel):
|
|
model_config = ConfigDict(
|
|
json_schema_extra={
|
|
"examples": [
|
|
{
|
|
"session_id": "session_2e7c3cbd4ef8472bab59718b7d012f80",
|
|
"character_id": "character_ec83d55746c54e47ab6f98d683032626",
|
|
"title": "大模型联调会话",
|
|
"status": "active",
|
|
"created_at": "2026-06-19T09:06:47.515000Z",
|
|
"updated_at": "2026-06-19T09:10:52.690000Z",
|
|
}
|
|
]
|
|
}
|
|
)
|
|
|
|
session_id: str = Field(description="会话 ID，前缀为 session_。")
|
|
character_id: str = Field(description="会话所属人物 ID。")
|
|
title: str | None = Field(description="会话标题。")
|
|
status: str = Field(description="会话状态：active / archived / deleted。")
|
|
created_at: str = Field(description="创建时间，UTC ISO 8601 字符串。")
|
|
updated_at: str = Field(description="更新时间，UTC ISO 8601 字符串。")
|
|
message_count: int = Field(default=0, description="会话内未删除消息数（用户 + 助手）。")
|
|
last_message_preview: str | None = Field(default=None, description="最近一条消息的内容摘要。")
|
|
|
|
|
|
class ChatSessionListResponse(BaseModel):
|
|
model_config = ConfigDict(
|
|
json_schema_extra={
|
|
"examples": [
|
|
{
|
|
"items": [
|
|
{
|
|
"session_id": "session_2e7c3cbd4ef8472bab59718b7d012f80",
|
|
"character_id": "character_ec83d55746c54e47ab6f98d683032626",
|
|
"title": "大模型联调会话",
|
|
"status": "active",
|
|
"created_at": "2026-06-19T09:06:47.515000Z",
|
|
"updated_at": "2026-06-19T09:10:52.690000Z",
|
|
}
|
|
],
|
|
"total": 1,
|
|
}
|
|
]
|
|
}
|
|
)
|
|
|
|
items: list[ChatSessionResponse] = Field(description="当前页会话列表。")
|
|
total: int = Field(description="符合条件的会话总数。")
|
|
|
|
|
|
class SendMessageRequest(BaseModel):
|
|
model_config = ConfigDict(
|
|
json_schema_extra={
|
|
"examples": [
|
|
{
|
|
"content": "你好，请简单介绍一下你自己。",
|
|
"modality": "text",
|
|
"model_hint": "deepseek-v4-flash",
|
|
},
|
|
{
|
|
"content": "请用三句话总结我们刚才聊到的内容。",
|
|
"modality": "text",
|
|
"model_hint": "default-chat",
|
|
},
|
|
]
|
|
}
|
|
)
|
|
|
|
content: str = Field(min_length=1, description="用户消息正文。")
|
|
modality: str = Field(default="text", description="消息模态；当前阶段仅支持 text。")
|
|
model_hint: str | None = Field(
|
|
default=None,
|
|
description="逻辑模型名或能力槽位，例如 default-chat、deepseek-v4-flash；为空时使用人物默认模型。",
|
|
)
|
|
thinking: bool = Field(
|
|
default=False, description="是否开启思考模式（reasoning）；模型不支持时忽略。"
|
|
)
|
|
reasoning_effort: str | None = Field(
|
|
default=None,
|
|
description="思考力度：low / medium / high。thinking=false 时忽略，使用模型默认。",
|
|
)
|
|
prior_versions: list[dict] | None = Field(
|
|
default=None,
|
|
description="重答时携带的历史版本列表，每项含 content 和 created_at。后端将其合并到新消息的 versions_json 中（前置插入），"
|
|
"同时软删除前一条 assistant 消息，避免刷新后出现重复消息。",
|
|
)
|
|
|
|
|
|
class SendMessageResponse(BaseModel):
|
|
model_config = ConfigDict(
|
|
json_schema_extra={
|
|
"examples": [
|
|
{
|
|
"message_id": "message_462927da6f0e4af780dd28fd58f73314",
|
|
"assistant_message_id": "message_2f9b7bf7d3b84231b67b28bb39fbfa7a",
|
|
"run_id": "run_a2c8ecc73c0e4cc7bc2fafbaeada74e0",
|
|
"session_id": "session_2e7c3cbd4ef8472bab59718b7d012f80",
|
|
"character_id": "character_ec83d55746c54e47ab6f98d683032626",
|
|
"status": "completed",
|
|
"stream_url": "/api/v1/runs/run_a2c8ecc73c0e4cc7bc2fafbaeada74e0/stream",
|
|
}
|
|
]
|
|
}
|
|
)
|
|
|
|
message_id: str = Field(description="本次写入的用户消息 ID。")
|
|
assistant_message_id: str | None = Field(description="模型回复消息 ID；失败时可能为空。")
|
|
run_id: str = Field(description="本次对话运行 ID，可用于查询 Run 或 SSE。")
|
|
session_id: str = Field(description="会话 ID。")
|
|
character_id: str = Field(description="人物 ID。")
|
|
status: str = Field(description="运行状态。成功完成时为 completed。")
|
|
stream_url: str = Field(description="SSE 事件流 URL（前置于此，前端可直接 new EventSource）。")
|
|
|
|
|
|
class ChatMessageResponse(BaseModel):
|
|
model_config = ConfigDict(
|
|
json_schema_extra={
|
|
"examples": [
|
|
{
|
|
"message_id": "message_2f9b7bf7d3b84231b67b28bb39fbfa7a",
|
|
"session_id": "session_2e7c3cbd4ef8472bab59718b7d012f80",
|
|
"character_id": "character_ec83d55746c54e47ab6f98d683032626",
|
|
"role": "assistant",
|
|
"content": "你好，我是魂镜中的默认人物，会结合记忆和当前会话与你对话。",
|
|
"modality": "text",
|
|
"run_id": "run_a2c8ecc73c0e4cc7bc2fafbaeada74e0",
|
|
"metadata": {"model_hint": "deepseek-v4-flash", "provider": "deepseek"},
|
|
"sensitivity": "low",
|
|
"created_at": "2026-06-19T09:10:52.676462Z",
|
|
"deleted_at": None,
|
|
}
|
|
]
|
|
}
|
|
)
|
|
|
|
message_id: str = Field(description="消息 ID，前缀为 message_。")
|
|
session_id: str = Field(description="所属会话 ID。")
|
|
character_id: str = Field(description="所属人物 ID。")
|
|
role: str = Field(description="消息角色：user / assistant / system / tool。")
|
|
content: str = Field(description="消息正文。")
|
|
modality: str = Field(description="消息模态：text / image / audio / file。")
|
|
run_id: str | None = Field(description="关联运行 ID。")
|
|
metadata: dict[str, Any] = Field(description="消息元数据，例如模型名、provider 等。")
|
|
sensitivity: str = Field(description="敏感等级：none / low / medium / high。")
|
|
versions: list[dict] = Field(
|
|
default=[],
|
|
description="版本历史列表，每项含 content 和 created_at。",
|
|
)
|
|
active_version_idx: int = Field(
|
|
default=0,
|
|
description="当前活跃版本在 versions 中的索引。",
|
|
)
|
|
created_at: str = Field(description="创建时间，UTC ISO 8601 字符串。")
|
|
deleted_at: str | None = Field(description="删除时间；未删除时为空。")
|
|
|
|
|
|
class ChatMessageListResponse(BaseModel):
|
|
model_config = ConfigDict(
|
|
json_schema_extra={
|
|
"examples": [
|
|
{
|
|
"items": [
|
|
{
|
|
"message_id": "message_462927da6f0e4af780dd28fd58f73314",
|
|
"session_id": "session_2e7c3cbd4ef8472bab59718b7d012f80",
|
|
"character_id": "character_ec83d55746c54e47ab6f98d683032626",
|
|
"role": "user",
|
|
"content": "你好，请简单介绍一下你自己。",
|
|
"modality": "text",
|
|
"run_id": "run_a2c8ecc73c0e4cc7bc2fafbaeada74e0",
|
|
"metadata": {},
|
|
"sensitivity": "low",
|
|
"created_at": "2026-06-19T09:10:52.660153Z",
|
|
"deleted_at": None,
|
|
},
|
|
{
|
|
"message_id": "message_2f9b7bf7d3b84231b67b28bb39fbfa7a",
|
|
"session_id": "session_2e7c3cbd4ef8472bab59718b7d012f80",
|
|
"character_id": "character_ec83d55746c54e47ab6f98d683032626",
|
|
"role": "assistant",
|
|
"content": "你好，我是魂镜中的默认人物，会结合记忆和当前会话与你对话。",
|
|
"modality": "text",
|
|
"run_id": "run_a2c8ecc73c0e4cc7bc2fafbaeada74e0",
|
|
"metadata": {"model_hint": "deepseek-v4-flash", "provider": "deepseek"},
|
|
"sensitivity": "low",
|
|
"created_at": "2026-06-19T09:10:52.676462Z",
|
|
"deleted_at": None,
|
|
},
|
|
],
|
|
"total": 2,
|
|
}
|
|
]
|
|
}
|
|
)
|
|
|
|
items: list[ChatMessageResponse] = Field(description="按创建时间升序排列的消息列表。")
|
|
total: int = Field(description="符合条件的消息总数。")
|
|
|
|
|
|
class SelectVersionRequest(BaseModel):
|
|
model_config = ConfigDict(
|
|
json_schema_extra={
|
|
"examples": [
|
|
{"version_idx": 0},
|
|
{"version_idx": 2},
|
|
]
|
|
}
|
|
)
|
|
|
|
version_idx: int = Field(
|
|
ge=0,
|
|
description="要激活的版本在消息 versions 数组中的索引（0-based）。",
|
|
)
|