107 lines
4.1 KiB
Python
107 lines
4.1 KiB
Python
from typing import Any, Literal
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field, HttpUrl
|
|
|
|
|
|
class StrictPayload(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
|
|
class ReleasePayload(StrictPayload):
|
|
release_no: str = Field(min_length=1, max_length=64)
|
|
title: str = Field(min_length=1, max_length=128)
|
|
change_summary: str = Field(min_length=1, max_length=1000)
|
|
|
|
|
|
class ItemPayload(StrictPayload):
|
|
namespace: Literal["agent_tools", "memory", "relationship", "runtime", "portfolio_analysis"]
|
|
item_key: str = Field(min_length=1, max_length=128)
|
|
value_json: dict[str, Any]
|
|
schema_version: Literal["1"] = "1"
|
|
|
|
|
|
class EndpointPayload(StrictPayload):
|
|
endpoint_code: str = Field(min_length=1, max_length=64)
|
|
provider: str = Field(min_length=1, max_length=32)
|
|
model_name: str = Field(min_length=1, max_length=128)
|
|
base_url: HttpUrl
|
|
secret_ref: str = Field(pattern=r"^env:[A-Z][A-Z0-9_]{0,100}$")
|
|
capabilities: list[str] = Field(min_length=1, max_length=20)
|
|
allowed_data_levels: list[str] = Field(min_length=1, max_length=20)
|
|
context_window: int = Field(gt=0)
|
|
timeout_ms: int = Field(default=15000, ge=100, le=120000)
|
|
|
|
|
|
class FallbackPayload(StrictPayload):
|
|
endpoint_id: int = Field(gt=0)
|
|
fallback_order: int = Field(ge=1, le=2)
|
|
retryable_error_codes: list[str] = Field(default_factory=list)
|
|
|
|
|
|
class RoutingPayload(StrictPayload):
|
|
rule_code: str = Field(min_length=1, max_length=64)
|
|
agent_type: str = Field(min_length=2, max_length=32)
|
|
task_type: str = Field(min_length=1, max_length=48)
|
|
model_policy: str = Field(min_length=1, max_length=32)
|
|
primary_endpoint_id: int = Field(gt=0)
|
|
fallbacks: list[FallbackPayload] = Field(default_factory=list, max_length=2)
|
|
max_attempts: int = Field(default=2, ge=1, le=3)
|
|
latency_budget_ms: int = Field(default=15000, ge=100, le=120000)
|
|
priority: int = Field(default=100, ge=0)
|
|
|
|
|
|
class PromptPayload(StrictPayload):
|
|
release_id: int = Field(gt=0)
|
|
prompt_code: str = Field(min_length=1, max_length=64)
|
|
task_type: str = Field(min_length=1, max_length=48)
|
|
agent_type: str | None = Field(default=None, max_length=32)
|
|
version: int = Field(ge=1)
|
|
system_prompt: str = Field(min_length=1, max_length=32000)
|
|
user_prompt_template: str = Field(min_length=1, max_length=32000)
|
|
input_schema: dict[str, Any] | None = None
|
|
output_schema: dict[str, Any] | None = None
|
|
|
|
|
|
class IntentPayload(StrictPayload):
|
|
agent_type: str = Field(min_length=2, max_length=32)
|
|
intent_code: str = Field(min_length=1, max_length=64)
|
|
intent_name: str = Field(min_length=1, max_length=128)
|
|
description: str | None = Field(default=None, max_length=500)
|
|
examples: list[str]
|
|
classifier_instruction: str | None = None
|
|
confidence_threshold: str = Field(default="0.6000", pattern=r"^(0(\.\d{1,4})?|1(\.0{1,4})?)$")
|
|
max_clarification_rounds: int = Field(default=2, ge=0, le=10)
|
|
transfer_on_failure: bool = True
|
|
allowed_tools: list[str] = Field(default_factory=list)
|
|
priority: int = Field(default=100, ge=0)
|
|
version: int = Field(default=1, ge=1)
|
|
|
|
|
|
class ReplyPayload(StrictPayload):
|
|
template_code: str = Field(min_length=1, max_length=64)
|
|
scene: str = Field(min_length=1, max_length=32)
|
|
title: str = Field(min_length=1, max_length=128)
|
|
content_text: str = Field(min_length=1, max_length=32000)
|
|
variables: list[str] = Field(default_factory=list)
|
|
locale: str = Field(default="zh-CN", max_length=16)
|
|
version: int = Field(default=1, ge=1)
|
|
|
|
|
|
class NegativePayload(StrictPayload):
|
|
rule_code: str = Field(min_length=1, max_length=64)
|
|
word_pattern: str = Field(min_length=1, max_length=256)
|
|
match_type: Literal["contains", "exact"] = "contains"
|
|
category: str = Field(min_length=1, max_length=32)
|
|
severity: Literal["block", "replace", "warn"] = "block"
|
|
applicable_agents: list[str] | None = None
|
|
safe_reply_template_code: str | None = Field(default=None, max_length=64)
|
|
|
|
|
|
class ReviewPayload(StrictPayload):
|
|
decision: Literal["approved", "rejected"]
|
|
comment: str = Field(default="", max_length=1000)
|
|
|
|
|
|
class EmptyPayload(StrictPayload):
|
|
pass
|