47 lines
2.2 KiB
Python
47 lines
2.2 KiB
Python
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
from typing import Any, Literal
|
|||
|
|
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
|
|||
|
|
OrchestratorSource = Literal["user_message", "schedule", "system_event"]
|
|||
|
|
OrchestratorAgent = Literal["user_agent", "hermes"]
|
|||
|
|
OrchestratorStatus = Literal["succeeded", "blocked", "failed"]
|
|||
|
|
|
|||
|
|
|
|||
|
|
class OrchestratorRequest(BaseModel):
|
|||
|
|
source: OrchestratorSource = Field(description="请求来源。")
|
|||
|
|
user_id: str | None = Field(default=None, description="当前用户 ID，任务触发可为空。")
|
|||
|
|
message: str | None = Field(default=None, description="用户消息或任务描述。")
|
|||
|
|
task_id: str | None = Field(default=None, description="任务资产 ID，schedule 触发时优先使用。")
|
|||
|
|
context_json: dict[str, Any] = Field(
|
|||
|
|
default_factory=dict,
|
|||
|
|
description="用户上下文、测试开关或调用方附加信息。",
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class OrchestratorTraceSummary(BaseModel):
|
|||
|
|
scenario: str = Field(description="语义场景。")
|
|||
|
|
intent: str = Field(description="语义意图。")
|
|||
|
|
tool_count: int = Field(default=0, ge=0, description="工具调用总数。")
|
|||
|
|
failed_tool_count: int = Field(default=0, ge=0, description="失败工具调用数量。")
|
|||
|
|
selected_capability_codes: list[str] = Field(
|
|||
|
|
default_factory=list,
|
|||
|
|
description="本次路由命中的能力编码。",
|
|||
|
|
)
|
|||
|
|
degraded: bool = Field(default=False, description="是否发生降级。")
|
|||
|
|
|
|||
|
|
|
|||
|
|
class OrchestratorResponse(BaseModel):
|
|||
|
|
run_id: str = Field(description="本次运行的唯一 run_id。")
|
|||
|
|
selected_agent: OrchestratorAgent | None = Field(
|
|||
|
|
default=None,
|
|||
|
|
description="最终路由到的下游 Agent。",
|
|||
|
|
)
|
|||
|
|
route_reason: str = Field(description="路由原因摘要。")
|
|||
|
|
permission_level: str = Field(description="权限级别。")
|
|||
|
|
status: OrchestratorStatus = Field(description="最终运行状态。")
|
|||
|
|
result: dict[str, Any] = Field(default_factory=dict, description="对前端可直接展示的最小结果。")
|
|||
|
|
requires_confirmation: bool = Field(default=False, description="是否需要用户或管理员确认。")
|
|||
|
|
trace_summary: OrchestratorTraceSummary = Field(description="简化后的 Trace 摘要。")
|