refactor(backend): update data schemas

- schemas/orchestrator.py: update orchestrator schemas
- schemas/settings.py: update settings schemas
- schemas/user_agent.py: update user agent schemas
This commit is contained in:
caoxiaozhu
2026-05-12 06:35:17 +00:00
parent 665a744a43
commit a6a28ba865
3 changed files with 130 additions and 0 deletions

View File

@@ -1,5 +1,6 @@
from __future__ import annotations
from datetime import datetime
from typing import Any, Literal
from pydantic import BaseModel, Field
@@ -12,6 +13,7 @@ OrchestratorStatus = Literal["succeeded", "blocked", "failed"]
class OrchestratorRequest(BaseModel):
source: OrchestratorSource = Field(description="请求来源。")
user_id: str | None = Field(default=None, description="当前用户 ID任务触发可为空。")
conversation_id: str | None = Field(default=None, description="多轮对话会话 ID。")
message: str | None = Field(default=None, description="用户消息或任务描述。")
task_id: str | None = Field(default=None, description="任务资产 IDschedule 触发时优先使用。")
context_json: dict[str, Any] = Field(
@@ -34,6 +36,7 @@ class OrchestratorTraceSummary(BaseModel):
class OrchestratorResponse(BaseModel):
run_id: str = Field(description="本次运行的唯一 run_id。")
conversation_id: str | None = Field(default=None, description="当前会话 ID。")
selected_agent: OrchestratorAgent | None = Field(
default=None,
description="最终路由到的下游 Agent。",
@@ -44,3 +47,37 @@ class OrchestratorResponse(BaseModel):
result: dict[str, Any] = Field(default_factory=dict, description="对前端可直接展示的最小结果。")
requires_confirmation: bool = Field(default=False, description="是否需要用户或管理员确认。")
trace_summary: OrchestratorTraceSummary = Field(description="简化后的 Trace 摘要。")
class ConversationMessageRead(BaseModel):
id: str = Field(description="消息 ID。")
role: str = Field(description="消息角色。")
content: str = Field(description="消息正文。")
run_id: str | None = Field(default=None, description="关联运行 ID。")
message_json: dict[str, Any] = Field(default_factory=dict, description="扩展消息载荷。")
created_at: datetime | None = Field(default=None, description="消息创建时间。")
class ConversationRead(BaseModel):
conversation_id: str = Field(description="会话 ID。")
user_id: str | None = Field(default=None, description="所属用户 ID。")
source: str | None = Field(default=None, description="来源。")
entry_source: str | None = Field(default=None, description="入口来源。")
title: str | None = Field(default=None, description="会话标题。")
last_run_id: str | None = Field(default=None, description="最近一次运行 ID。")
last_scenario: str | None = Field(default=None, description="最近场景。")
last_intent: str | None = Field(default=None, description="最近意图。")
draft_claim_id: str | None = Field(default=None, description="关联草稿单 ID。")
state_json: dict[str, Any] = Field(default_factory=dict, description="会话状态。")
message_count: int = Field(default=0, ge=0, description="消息数量。")
updated_at: datetime | None = Field(default=None, description="更新时间。")
messages: list[ConversationMessageRead] = Field(default_factory=list, description="历史消息。")
class ConversationLookupResponse(BaseModel):
found: bool = Field(default=False, description="是否找到可恢复会话。")
conversation: ConversationRead | None = Field(default=None, description="会话详情。")
class ConversationDeleteResponse(BaseModel):
deleted_count: int = Field(default=0, ge=0, description="删除的会话数量。")