plato-sdk-v2 2.5.0__py3-none-any.whl → 2.6.0__py3-none-any.whl

plato/chronos/models/__init__.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
-#   filename:  tmp7svjeeij.json
-#   timestamp: 2026-01-17T02:28:53+00:00
+#   filename:  tmpxmdo22xa.json
+#   timestamp: 2026-01-23T22:08:21+00:00
 
 from __future__ import annotations
 
@@ -9,22 +9,6 @@ from typing import Annotated, Any
 from pydantic import AwareDatetime, BaseModel, ConfigDict, Field
 
 
-class AgentConfig(BaseModel):
-    """Agent config - supports multiple formats.
-
-    New format: agent + version (version optional, defaults to latest)
-    Legacy format: agent_id (public_id)
-    """
-
-    model_config = ConfigDict(
-        extra="allow",
-    )
-    agent: Annotated[str | None, Field(title="Agent")] = None
-    version: Annotated[str | None, Field(title="Version")] = None
-    agent_id: Annotated[str | None, Field(title="Agent Id")] = None  # backwards compat
-    config: Annotated[dict[str, Any] | None, Field(title="Config")] = {}
-
-
 class AgentCreate(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -33,8 +17,6 @@ class AgentCreate(BaseModel):
     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
 
 
@@ -42,57 +24,17 @@ 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
-    """
+    version: Annotated[str, Field(title="Version")]
+    description: Annotated[str | None, Field(title="Description")] = None
+    image_uri: Annotated[str, Field(title="Image Uri")]
 
 
-class AgentLogsResponse(BaseModel):
+class AgentListResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
     )
-    success: Annotated[bool, Field(title="Success")]
-    count: Annotated[int, Field(title="Count")]
-    """
-    Number of logs added
-    """
+    agents: Annotated[list[AgentInfo], Field(title="Agents")]
 
 
 class AgentLookupResponse(BaseModel):
@@ -114,8 +56,6 @@ class AgentResponse(BaseModel):
     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
@@ -132,28 +72,6 @@ class AgentSchemaResponse(BaseModel):
     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",
@@ -170,42 +88,113 @@ class AgentVersionsResponse(BaseModel):
     versions: Annotated[list[AgentVersionInfo], Field(title="Versions")]
 
 
-class ArtifactsUploadRequest(BaseModel):
+class AuthStatusResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
     )
-    session_id: Annotated[str, Field(title="Session Id")]
+    authenticated: Annotated[bool, Field(title="Authenticated")]
+    user: Annotated[dict[str, Any] | None, Field(title="User")] = None
+    env: Annotated[str, Field(title="Env")]
+
+
+class CheckpointListItem(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    public_id: Annotated[str, Field(title="Public Id")]
+    step_number: Annotated[int, Field(title="Step Number")]
+    env_count: Annotated[int, Field(title="Env Count")]
+    has_state: Annotated[bool, Field(title="Has State")]
+    created_at: Annotated[str, Field(title="Created At")]
+
+
+class CheckpointListResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    checkpoints: Annotated[list[CheckpointListItem], Field(title="Checkpoints")]
+
+
+class ClearDbResult(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    sessions_deleted: Annotated[int, Field(title="Sessions Deleted")]
+    runtimes_deleted: Annotated[int, Field(title="Runtimes Deleted")]
+    agents_deleted: Annotated[int, Field(title="Agents Deleted")]
+    worlds_deleted: Annotated[int, Field(title="Worlds Deleted")]
+
+
+class CloseSessionResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    success: Annotated[bool, Field(title="Success")]
+    message: Annotated[str, Field(title="Message")]
+
+
+class CompleteSessionRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    status: Annotated[str, Field(title="Status")]
     """
-    Chronos session ID
+    Final status: 'completed' or 'failed'
     """
-    trajectory: Annotated[dict[str, Any] | None, Field(title="Trajectory")] = None
+    exit_code: Annotated[int | None, Field(title="Exit Code")] = None
     """
-    AITF trajectory JSON
+    Exit code from world runner
     """
-    logs_base64: Annotated[str | None, Field(title="Logs Base64")] = None
+    error_message: Annotated[str | None, Field(title="Error Message")] = None
     """
-    Base64 encoded zip file
+    Error message if failed
     """
 
 
-class ArtifactsUploadResponse(BaseModel):
+class CreateSessionRequest(BaseModel):
     model_config = ConfigDict(
         extra="allow",
     )
-    success: Annotated[bool, Field(title="Success")]
-    logs_url: Annotated[str | None, Field(title="Logs Url")] = None
+    world_name: Annotated[str, Field(title="World Name")]
+    """
+    Name of the world to run
+    """
+    world_config: Annotated[dict[str, Any] | None, Field(title="World Config")] = None
+    """
+    World configuration
     """
-    S3 URL of uploaded logs
+    plato_session_id: Annotated[str | None, Field(title="Plato Session Id")] = None
+    """
+    Plato session ID if already created
     """
 
 
-class AuthStatusResponse(BaseModel):
+class CreateSessionResponse(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")]
+    public_id: Annotated[str, Field(title="Public Id")]
+    otel_url: Annotated[str, Field(title="Otel Url")]
+    upload_url: Annotated[str, Field(title="Upload Url")]
+
+
+class EnvInfo(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    alias: Annotated[str, Field(title="Alias")]
+    job_id: Annotated[str, Field(title="Job Id")]
+    artifact_id: Annotated[str, Field(title="Artifact Id")]
+
+
+class EnvInfoResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    alias: Annotated[str, Field(title="Alias")]
+    job_id: Annotated[str, Field(title="Job Id")]
+    artifact_id: Annotated[str | None, Field(title="Artifact Id")] = None
+    simulator: Annotated[str | None, Field(title="Simulator")] = None
 
 
 class EnvSecretCreate(BaseModel):
@@ -276,36 +265,6 @@ class JobTemplateUpdate(BaseModel):
     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",
@@ -324,6 +283,15 @@ class LaunchJobResponse(BaseModel):
     """
 
 
+class LiveLogsResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    public_id: Annotated[str, Field(title="Public Id")]
+    status: Annotated[str, Field(title="Status")]
+    logs: Annotated[str, Field(title="Logs")]
+
+
 class LogsDownloadResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -332,21 +300,62 @@ class LogsDownloadResponse(BaseModel):
     expires_in: Annotated[int | None, Field(title="Expires In")] = 3600
 
 
-class LogsUploadRequest(BaseModel):
+class OTelSpan(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    trace_id: Annotated[str, Field(title="Trace Id")]
+    span_id: Annotated[str, Field(title="Span Id")]
+    parent_span_id: Annotated[str | None, Field(title="Parent Span Id")] = None
+    name: Annotated[str, Field(title="Name")]
+    start_time_unix_nano: Annotated[int, Field(title="Start Time Unix Nano")]
+    end_time_unix_nano: Annotated[int | None, Field(title="End Time Unix Nano")] = None
+    attributes: Annotated[dict[str, Any] | None, Field(title="Attributes")] = {}
+    status: Annotated[dict[str, Any] | None, Field(title="Status")] = None
+
+
+class OTelSpanListResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    events: Annotated[list[OTelSpan], Field(title="Events")]
+
+
+class OTelTraceResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
     )
     session_id: Annotated[str, Field(title="Session Id")]
+    spans: Annotated[list[OTelSpan], Field(title="Spans")]
+
+
+class PreviewResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    success: Annotated[bool, Field(title="Success")]
+    plato_session_id: Annotated[str | None, Field(title="Plato Session Id")] = None
+    """
+    Plato session ID
     """
-    Chronos session ID
+    serialized_session: Annotated[dict[str, Any] | None, Field(title="Serialized Session")] = None
     """
-    logs_base64: Annotated[str, Field(title="Logs Base64")]
+    Serialized session for Session.load()
     """
-    Base64 encoded zip file
+    envs: Annotated[list[EnvInfo] | None, Field(title="Envs")] = None
+    """
+    Environment info
     """
 
 
-LogsUploadResponse = ArtifactsUploadResponse
+class RuntimeConfig(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    artifact_id: Annotated[str | None, Field(title="Artifact Id")] = None
+    """
+    Runtime artifact ID for cached environment
+    """
 
 
 class RuntimeCreate(BaseModel):
@@ -384,6 +393,7 @@ class RuntimeResponse(BaseModel):
     agent_images: Annotated[list[str], Field(title="Agent Images")]
     status: Annotated[str, Field(title="Status")]
     description: Annotated[str | None, Field(title="Description")] = None
+    plato_session_id: Annotated[str | None, Field(title="Plato Session Id")] = None
     created_at: Annotated[AwareDatetime, Field(title="Created At")]
 
 
@@ -396,6 +406,31 @@ class ServiceStatuses(BaseModel):
     plato: Annotated[str, Field(title="Plato")]
 
 
+class SessionEnvsResponse(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
+    envs: Annotated[list[EnvInfoResponse], Field(title="Envs")]
+
+
+class SessionLogsResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    logs: Annotated[list[OTelSpan], Field(title="Logs")]
+
+
+class SessionStatusResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    public_id: Annotated[str, Field(title="Public Id")]
+    status: Annotated[str, Field(title="Status")]
+    status_reason: Annotated[str | None, Field(title="Status Reason")] = None
+
+
 class StatusResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -404,38 +439,39 @@ class StatusResponse(BaseModel):
     services: ServiceStatuses
 
 
-class TrajectoryResponse(BaseModel):
+class SyncResult(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")]
+    synced: Annotated[int, Field(title="Synced")]
+    skipped: Annotated[int, Field(title="Skipped")]
+    errors: Annotated[list[str], Field(title="Errors")]
 
 
-class TrajectoryUploadRequest(BaseModel):
+class TagsListResponse(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")]
+    tags: Annotated[list[str], Field(title="Tags")]
+
+
+class UpdateTagsRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    tags: Annotated[list[str], Field(title="Tags")]
     """
-    AITF trajectory JSON
+    List of tags for the session (use '.' for hierarchy)
     """
 
 
-class TrajectoryUploadResponse(BaseModel):
+class UserInfo(BaseModel):
     model_config = ConfigDict(
         extra="allow",
     )
-    success: Annotated[bool, Field(title="Success")]
+    id: Annotated[str, Field(title="Id")]
+    email: Annotated[str | None, Field(title="Email")] = None
+    name: Annotated[str | None, Field(title="Name")] = None
 
 
 class ValidationError(BaseModel):
@@ -456,6 +492,20 @@ class VersionResponse(BaseModel):
     deployment: Annotated[str | None, Field(title="Deployment")] = None
 
 
+class WorldConfig(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    package: Annotated[str, Field(title="Package")]
+    """
+    World package name with optional version (e.g., plato-world-structured-execution:0.1.17)
+    """
+    config: Annotated[dict[str, Any] | None, Field(title="Config")] = None
+    """
+    World configuration - passed directly to world runner
+    """
+
+
 class WorldCreate(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -472,8 +522,10 @@ class WorldInfo(BaseModel):
         extra="allow",
     )
     name: Annotated[str, Field(title="Name")]
-    package_name: Annotated[str, Field(title="Package Name")]
-    version: Annotated[str | None, Field(title="Version")] = None
+    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 WorldResponse(BaseModel):
@@ -510,32 +562,11 @@ class WorldVersionsResponse(BaseModel):
     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):
+class ChronosModelsAgentAgentListResponse(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
+    agents: Annotated[list[AgentResponse], Field(title="Agents")]
 
 
 class ChronosModelsAgentAgentSchemaResponse(BaseModel):
@@ -545,8 +576,6 @@ class ChronosModelsAgentAgentSchemaResponse(BaseModel):
     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):
@@ -565,11 +594,14 @@ class ChronosModelsAgentAgentVersionsResponse(BaseModel):
     versions: Annotated[list[ChronosModelsAgentAgentVersionInfo], Field(title="Versions")]
 
 
-class AgentListResponse(BaseModel):
+class ChronosModelsSessionWorldInfo(BaseModel):
     model_config = ConfigDict(
         extra="allow",
     )
-    agents: Annotated[list[AgentResponse], Field(title="Agents")]
+    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
 
 
 class EnvSecretListResponse(BaseModel):
@@ -593,18 +625,29 @@ class JobTemplateListResponse(BaseModel):
     templates: Annotated[list[JobTemplateResponse], Field(title="Templates")]
 
 
-class RuntimeListResponse(BaseModel):
+class LaunchJobRequest(BaseModel):
     model_config = ConfigDict(
         extra="allow",
     )
-    runtimes: Annotated[list[RuntimeResponse], Field(title="Runtimes")]
+    world: WorldConfig
+    """
+    World package and configuration
+    """
+    runtime: RuntimeConfig | None = None
+    """
+    Runtime configuration
+    """
+    tags: Annotated[list[str] | None, Field(title="Tags")] = None
+    """
+    Tags for organizing/filtering sessions (use '.' for hierarchy, '_' instead of '-')
+    """
 
 
-class SessionLogsResponse(BaseModel):
+class RuntimeListResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
     )
-    logs: Annotated[list[TrajectoryResponse], Field(title="Logs")]
+    runtimes: Annotated[list[RuntimeResponse], Field(title="Runtimes")]
 
 
 class SessionResponse(BaseModel):
@@ -614,31 +657,33 @@ class SessionResponse(BaseModel):
     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")]
+    status_reason: Annotated[str | None, Field(title="Status Reason")] = None
     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: ChronosModelsSessionWorldInfo | 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
+    tags: Annotated[list[str] | None, Field(title="Tags")] = []
+    created_by: UserInfo | None = None
 
 
-class TrajectoryListResponse(BaseModel):
+class SyncAllResult(BaseModel):
     model_config = ConfigDict(
         extra="allow",
     )
-    trajectories: Annotated[list[TrajectoryResponse], Field(title="Trajectories")]
+    agents: SyncResult
+    worlds: SyncResult
+    runtimes: SyncResult
 
 
 class WorldCatalogResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
     )
-    worlds: Annotated[list[ChronosApiRegistryWorldInfo], Field(title="Worlds")]
+    worlds: Annotated[list[WorldInfo], Field(title="Worlds")]
 
 
 class WorldListResponse(BaseModel):