plato-sdk-v2 2.0.50__py3-none-any.whl → 2.2.4__py3-none-any.whl

plato/chronos/models/__init__.py

@@ -0,0 +1,647 @@
+# generated by datamodel-codegen:
+#   filename:  tmp7svjeeij.json
+#   timestamp: 2026-01-17T02:28:53+00:00
+
+from __future__ import annotations
+
+from typing import Annotated, Any
+
+from pydantic import AwareDatetime, BaseModel, ConfigDict, Field
+
+
+class AgentConfig(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    agent_id: Annotated[str, Field(title="Agent Id")]
+    config: Annotated[dict[str, Any] | None, Field(title="Config")] = {}
+
+
+class AgentCreate(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str, Field(title="Name")]
+    version: Annotated[str, Field(title="Version")]
+    image_uri: Annotated[str, Field(title="Image Uri")]
+    config_schema: Annotated[dict[str, Any] | None, Field(title="Config Schema")] = None
+    secrets_schema: Annotated[dict[str, Any] | None, Field(title="Secrets Schema")] = None
+    template_variables: Annotated[dict[str, str] | None, Field(title="Template Variables")] = None
+    description: Annotated[str | None, Field(title="Description")] = None
+
+
+class AgentInfo(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    slot_name: Annotated[str, Field(title="Slot Name")]
+    name: Annotated[str, Field(title="Name")]
+    public_id: Annotated[str, Field(title="Public Id")]
+    config: Annotated[dict[str, Any] | None, Field(title="Config")] = {}
+
+
+class AgentLogEntry(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    level: Annotated[str | None, Field(title="Level")] = "info"
+    """
+    Log level: debug, info, warning, error
+    """
+    message: Annotated[str, Field(title="Message")]
+    """
+    Log message
+    """
+    timestamp: Annotated[str | None, Field(title="Timestamp")] = None
+    """
+    ISO timestamp, defaults to now
+    """
+    extra: Annotated[dict[str, Any] | None, Field(title="Extra")] = None
+    """
+    Additional structured data
+    """
+
+
+class AgentLogsRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    session_id: Annotated[str, Field(title="Session Id")]
+    """
+    Chronos session ID
+    """
+    logs: Annotated[list[AgentLogEntry], Field(title="Logs")]
+    """
+    Log entries to add
+    """
+
+
+class AgentLogsResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    success: Annotated[bool, Field(title="Success")]
+    count: Annotated[int, Field(title="Count")]
+    """
+    Number of logs added
+    """
+
+
+class AgentLookupResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    public_id: Annotated[str, Field(title="Public Id")]
+    name: Annotated[str, Field(title="Name")]
+    version: Annotated[str, Field(title="Version")]
+    image_uri: Annotated[str, Field(title="Image Uri")]
+
+
+class AgentResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    public_id: Annotated[str, Field(title="Public Id")]
+    name: Annotated[str, Field(title="Name")]
+    version: Annotated[str, Field(title="Version")]
+    image_uri: Annotated[str, Field(title="Image Uri")]
+    config_schema: Annotated[dict[str, Any] | None, Field(title="Config Schema")] = None
+    secrets_schema: Annotated[dict[str, Any] | None, Field(title="Secrets Schema")] = None
+    template_variables: Annotated[dict[str, str] | None, Field(title="Template Variables")] = None
+    description: Annotated[str | None, Field(title="Description")] = None
+    created_at: Annotated[AwareDatetime, Field(title="Created At")]
+    archived_at: Annotated[AwareDatetime | None, Field(title="Archived At")] = None
+
+
+class AgentSchemaResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str, Field(title="Name")]
+    version: Annotated[str, Field(title="Version")]
+    config_schema: Annotated[dict[str, Any] | None, Field(title="Config Schema")] = None
+    secrets_schema: Annotated[dict[str, Any] | None, Field(title="Secrets Schema")] = None
+    template_variables: Annotated[dict[str, Any] | None, Field(title="Template Variables")] = None
+
+
+class AgentStatusRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    session_id: Annotated[str, Field(title="Session Id")]
+    """
+    Chronos session ID
+    """
+    status: Annotated[str, Field(title="Status")]
+    """
+    New status: running, completed, failed
+    """
+    message: Annotated[str | None, Field(title="Message")] = None
+    """
+    Status message
+    """
+    extra: Annotated[dict[str, Any] | None, Field(title="Extra")] = None
+    """
+    Additional data
+    """
+
+
+class AgentVersionInfo(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    version: Annotated[str, Field(title="Version")]
+    created_at: Annotated[str | None, Field(title="Created At")] = None
+
+
+class AgentVersionsResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str, Field(title="Name")]
+    versions: Annotated[list[AgentVersionInfo], Field(title="Versions")]
+
+
+class ArtifactsUploadRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    session_id: Annotated[str, Field(title="Session Id")]
+    """
+    Chronos session ID
+    """
+    trajectory: Annotated[dict[str, Any] | None, Field(title="Trajectory")] = None
+    """
+    AITF trajectory JSON
+    """
+    logs_base64: Annotated[str | None, Field(title="Logs Base64")] = None
+    """
+    Base64 encoded zip file
+    """
+
+
+class ArtifactsUploadResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    success: Annotated[bool, Field(title="Success")]
+    logs_url: Annotated[str | None, Field(title="Logs Url")] = None
+    """
+    S3 URL of uploaded logs
+    """
+
+
+class AuthStatusResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    authenticated: Annotated[bool, Field(title="Authenticated")]
+    user: Annotated[dict[str, Any] | None, Field(title="User")] = None
+    env: Annotated[str, Field(title="Env")]
+
+
+class EnvSecretCreate(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str, Field(title="Name")]
+    value: Annotated[str, Field(title="Value")]
+    description: Annotated[str | None, Field(title="Description")] = None
+
+
+class EnvSecretResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    id: Annotated[int, Field(title="Id")]
+    name: Annotated[str, Field(title="Name")]
+    value: Annotated[str, Field(title="Value")]
+    description: Annotated[str | None, Field(title="Description")] = None
+    created_at: Annotated[AwareDatetime, Field(title="Created At")]
+    updated_at: Annotated[AwareDatetime, Field(title="Updated At")]
+
+
+class EnvSecretUpdate(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    value: Annotated[str | None, Field(title="Value")] = None
+    description: Annotated[str | None, Field(title="Description")] = None
+
+
+class JobTemplateCreate(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str, Field(title="Name")]
+    description: Annotated[str | None, Field(title="Description")] = None
+    world_id: Annotated[str, Field(title="World Id")]
+    world_config: Annotated[dict[str, Any] | None, Field(title="World Config")] = None
+    agent_configs: Annotated[dict[str, Any] | None, Field(title="Agent Configs")] = {}
+    secrets: Annotated[dict[str, str] | None, Field(title="Secrets")] = {}
+
+
+class JobTemplateResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    public_id: Annotated[str, Field(title="Public Id")]
+    name: Annotated[str, Field(title="Name")]
+    description: Annotated[str | None, Field(title="Description")] = None
+    world_id: Annotated[str, Field(title="World Id")]
+    world_config: Annotated[dict[str, Any], Field(title="World Config")]
+    agent_configs: Annotated[dict[str, Any], Field(title="Agent Configs")]
+    secrets: Annotated[dict[str, str], Field(title="Secrets")]
+    created_at: Annotated[AwareDatetime, Field(title="Created At")]
+    updated_at: Annotated[AwareDatetime, Field(title="Updated At")]
+
+
+class JobTemplateUpdate(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str | None, Field(title="Name")] = None
+    description: Annotated[str | None, Field(title="Description")] = None
+    world_id: Annotated[str | None, Field(title="World Id")] = None
+    world_config: Annotated[dict[str, Any] | None, Field(title="World Config")] = None
+    agent_configs: Annotated[dict[str, Any] | None, Field(title="Agent Configs")] = None
+    secrets: Annotated[dict[str, str] | None, Field(title="Secrets")] = None
+
+
+class LaunchJobRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    world_package: Annotated[str, Field(title="World Package")]
+    """
+    World package name
+    """
+    world_version: Annotated[str | None, Field(title="World Version")] = None
+    """
+    World version (latest if not specified)
+    """
+    world_config: Annotated[dict[str, Any] | None, Field(title="World Config")] = None
+    """
+    World configuration
+    """
+    agent_configs: Annotated[dict[str, AgentConfig], Field(title="Agent Configs")]
+    """
+    Agent configs keyed by slot name
+    """
+    secret_ids: Annotated[list[int] | None, Field(title="Secret Ids")] = None
+    """
+    IDs of secrets from env_secrets table
+    """
+    runtime_artifact_id: Annotated[str | None, Field(title="Runtime Artifact Id")] = None
+    """
+    Runtime artifact ID for cached environment
+    """
+
+
+class LaunchJobResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    session_id: Annotated[str, Field(title="Session Id")]
+    """
+    Chronos session public ID
+    """
+    plato_session_id: Annotated[str, Field(title="Plato Session Id")]
+    """
+    Plato session ID
+    """
+    status: Annotated[str, Field(title="Status")]
+    """
+    Job status
+    """
+
+
+class LogsDownloadResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    download_url: Annotated[str | None, Field(title="Download Url")] = None
+    expires_in: Annotated[int | None, Field(title="Expires In")] = 3600
+
+
+class LogsUploadRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    session_id: Annotated[str, Field(title="Session Id")]
+    """
+    Chronos session ID
+    """
+    logs_base64: Annotated[str, Field(title="Logs Base64")]
+    """
+    Base64 encoded zip file
+    """
+
+
+LogsUploadResponse = ArtifactsUploadResponse
+
+
+class RuntimeCreate(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str, Field(title="Name")]
+    world_package: Annotated[str, Field(title="World Package")]
+    world_version: Annotated[str | None, Field(title="World Version")] = None
+    agent_images: Annotated[list[str] | None, Field(title="Agent Images")] = []
+    description: Annotated[str | None, Field(title="Description")] = None
+
+
+class RuntimeCreateResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    session_id: Annotated[str, Field(title="Session Id")]
+    name: Annotated[str, Field(title="Name")]
+    world_package: Annotated[str, Field(title="World Package")]
+    world_version: Annotated[str | None, Field(title="World Version")] = None
+    agent_images: Annotated[list[str], Field(title="Agent Images")]
+    status: Annotated[str, Field(title="Status")]
+    message: Annotated[str, Field(title="Message")]
+
+
+class RuntimeResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    artifact_id: Annotated[str, Field(title="Artifact Id")]
+    name: Annotated[str, Field(title="Name")]
+    world_package: Annotated[str, Field(title="World Package")]
+    world_version: Annotated[str | None, Field(title="World Version")] = None
+    agent_images: Annotated[list[str], Field(title="Agent Images")]
+    status: Annotated[str, Field(title="Status")]
+    description: Annotated[str | None, Field(title="Description")] = None
+    created_at: Annotated[AwareDatetime, Field(title="Created At")]
+
+
+class ServiceStatuses(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    database: Annotated[str, Field(title="Database")]
+    registry: Annotated[str, Field(title="Registry")]
+    plato: Annotated[str, Field(title="Plato")]
+
+
+class StatusResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    status: Annotated[str, Field(title="Status")]
+    services: ServiceStatuses
+
+
+class TrajectoryResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    public_id: Annotated[str, Field(title="Public Id")]
+    session_id: Annotated[int, Field(title="Session Id")]
+    agent_id: Annotated[int | None, Field(title="Agent Id")] = None
+    log_type: Annotated[str, Field(title="Log Type")]
+    content: Annotated[str, Field(title="Content")]
+    extra: Annotated[dict[str, Any] | None, Field(title="Extra")] = None
+    created_at: Annotated[AwareDatetime, Field(title="Created At")]
+
+
+class TrajectoryUploadRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    session_id: Annotated[str, Field(title="Session Id")]
+    """
+    Chronos session ID
+    """
+    trajectory: Annotated[dict[str, Any], Field(title="Trajectory")]
+    """
+    AITF trajectory JSON
+    """
+
+
+class TrajectoryUploadResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    success: Annotated[bool, Field(title="Success")]
+
+
+class ValidationError(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    loc: Annotated[list[str | int], Field(title="Location")]
+    msg: Annotated[str, Field(title="Message")]
+    type: Annotated[str, Field(title="Error Type")]
+
+
+class VersionResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    version: Annotated[str, Field(title="Version")]
+    sha: Annotated[str, Field(title="Sha")]
+    deployment: Annotated[str | None, Field(title="Deployment")] = None
+
+
+class WorldCreate(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str, Field(title="Name")]
+    package_name: Annotated[str, Field(title="Package Name")]
+    version: Annotated[str | None, Field(title="Version")] = None
+    config_schema: Annotated[dict[str, Any] | None, Field(title="Config Schema")] = None
+    description: Annotated[str | None, Field(title="Description")] = None
+
+
+class WorldInfo(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str, Field(title="Name")]
+    package_name: Annotated[str, Field(title="Package Name")]
+    version: Annotated[str | None, Field(title="Version")] = None
+
+
+class WorldResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    public_id: Annotated[str, Field(title="Public Id")]
+    name: Annotated[str, Field(title="Name")]
+    package_name: Annotated[str, Field(title="Package Name")]
+    version: Annotated[str | None, Field(title="Version")] = None
+    config_schema: Annotated[dict[str, Any] | None, Field(title="Config Schema")] = None
+    description: Annotated[str | None, Field(title="Description")] = None
+    created_at: Annotated[AwareDatetime, Field(title="Created At")]
+    archived_at: Annotated[AwareDatetime | None, Field(title="Archived At")] = None
+
+
+class WorldSchemaResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str | None, Field(title="Name")] = None
+    description: Annotated[str | None, Field(title="Description")] = None
+    version: Annotated[str | None, Field(title="Version")] = None
+    config_schema: Annotated[dict[str, Any] | None, Field(title="Config Schema")] = None
+    secrets_schema: Annotated[dict[str, Any] | None, Field(title="Secrets Schema")] = None
+    agent_slots: Annotated[list[dict[str, Any]] | None, Field(title="Agent Slots")] = None
+
+
+class WorldVersionsResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    package: Annotated[str, Field(title="Package")]
+    versions: Annotated[list[str], Field(title="Versions")]
+
+
+class ChronosApiRegistryAgentInfo(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str, Field(title="Name")]
+    version: Annotated[str, Field(title="Version")]
+    description: Annotated[str | None, Field(title="Description")] = None
+    image_uri: Annotated[str, Field(title="Image Uri")]
+
+
+class ChronosApiRegistryAgentListResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    agents: Annotated[list[ChronosApiRegistryAgentInfo], Field(title="Agents")]
+
+
+class ChronosApiRegistryWorldInfo(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str, Field(title="Name")]
+    package: Annotated[str, Field(title="Package")]
+    description: Annotated[str | None, Field(title="Description")] = None
+    versions: Annotated[list[str] | None, Field(title="Versions")] = []
+    latest_version: Annotated[str | None, Field(title="Latest Version")] = None
+
+
+class ChronosModelsAgentAgentSchemaResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str, Field(title="Name")]
+    version: Annotated[str, Field(title="Version")]
+    config_schema: Annotated[dict[str, Any] | None, Field(title="Config Schema")] = None
+    secrets_schema: Annotated[dict[str, Any] | None, Field(title="Secrets Schema")] = None
+    template_variables: Annotated[dict[str, str] | None, Field(title="Template Variables")] = None
+
+
+class ChronosModelsAgentAgentVersionInfo(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    version: Annotated[str, Field(title="Version")]
+    created_at: Annotated[AwareDatetime, Field(title="Created At")]
+
+
+class ChronosModelsAgentAgentVersionsResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    name: Annotated[str, Field(title="Name")]
+    versions: Annotated[list[ChronosModelsAgentAgentVersionInfo], Field(title="Versions")]
+
+
+class AgentListResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    agents: Annotated[list[AgentResponse], Field(title="Agents")]
+
+
+class EnvSecretListResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    secrets: Annotated[list[EnvSecretResponse], Field(title="Secrets")]
+
+
+class HTTPValidationError(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    detail: Annotated[list[ValidationError] | None, Field(title="Detail")] = None
+
+
+class JobTemplateListResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    templates: Annotated[list[JobTemplateResponse], Field(title="Templates")]
+
+
+class RuntimeListResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    runtimes: Annotated[list[RuntimeResponse], Field(title="Runtimes")]
+
+
+class SessionLogsResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    logs: Annotated[list[TrajectoryResponse], Field(title="Logs")]
+
+
+class SessionResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    public_id: Annotated[str, Field(title="Public Id")]
+    plato_session_id: Annotated[str | None, Field(title="Plato Session Id")] = None
+    status: Annotated[str, Field(title="Status")]
+    started_at: Annotated[AwareDatetime | None, Field(title="Started At")] = None
+    ended_at: Annotated[AwareDatetime | None, Field(title="Ended At")] = None
+    created_at: Annotated[AwareDatetime, Field(title="Created At")]
+    world: WorldInfo | None = None
+    world_config: Annotated[dict[str, Any] | None, Field(title="World Config")] = {}
+    agents: Annotated[list[AgentInfo] | None, Field(title="Agents")] = []
+    agent_configs: Annotated[dict[str, Any] | None, Field(title="Agent Configs")] = {}
+    secret_ids: Annotated[list[int] | None, Field(title="Secret Ids")] = []
+    trajectory: Annotated[dict[str, Any] | None, Field(title="Trajectory")] = None
+    logs_url: Annotated[str | None, Field(title="Logs Url")] = None
+    runtime_artifact_id: Annotated[str | None, Field(title="Runtime Artifact Id")] = None
+
+
+class TrajectoryListResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    trajectories: Annotated[list[TrajectoryResponse], Field(title="Trajectories")]
+
+
+class WorldCatalogResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    worlds: Annotated[list[ChronosApiRegistryWorldInfo], Field(title="Worlds")]
+
+
+class WorldListResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    worlds: Annotated[list[WorldResponse], Field(title="Worlds")]
+
+
+class SessionListResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    sessions: Annotated[list[SessionResponse], Field(title="Sessions")]