33 lines
705 B
Python
33 lines
705 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 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
|