43 lines
911 B
Python
43 lines
911 B
Python
from typing import Any
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
|
|
class AgentRunCreateRequest(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
agent_type: str
|
|
message: str = Field(min_length=1)
|
|
session_id: str
|
|
idempotency_key: str = Field(min_length=16, max_length=128)
|
|
|
|
|
|
class AgentRunAcceptedResponse(BaseModel):
|
|
run_id: str
|
|
trace_id: str
|
|
status: str
|
|
status_url: str
|
|
events_url: str
|
|
|
|
|
|
class AgentRunAcceptedEnvelope(BaseModel):
|
|
data: AgentRunAcceptedResponse
|
|
meta: dict[str, str]
|
|
|
|
|
|
class AgentRunStatusResponse(BaseModel):
|
|
run_id: str
|
|
trace_id: str
|
|
status: str
|
|
agent_type: str
|
|
session_id: str
|
|
result: dict[str, Any] | None = None
|
|
error_code: str | None = None
|
|
created_at: str
|
|
completed_at: str | None = None
|
|
|
|
|
|
class AgentRunStatusEnvelope(BaseModel):
|
|
data: AgentRunStatusResponse
|
|
meta: dict[str, str]
|