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

plato/_generated/api/v2/jobs/disk_snapshot.py

@@ -12,7 +12,7 @@ from plato._generated.models import CreateDiskSnapshotRequest, CreateDiskSnapsho
 
 def _build_request_args(
     job_id: str,
-    body: CreateDiskSnapshotRequest | Any,
+    body: CreateDiskSnapshotRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
@@ -36,7 +36,7 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     job_id: str,
-    body: CreateDiskSnapshotRequest | Any,
+    body: CreateDiskSnapshotRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> CreateDiskSnapshotResult:
@@ -66,7 +66,7 @@ def sync(
 async def asyncio(
     client: httpx.AsyncClient,
     job_id: str,
-    body: CreateDiskSnapshotRequest | Any,
+    body: CreateDiskSnapshotRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> CreateDiskSnapshotResult:

plato/_generated/api/v2/jobs/log_for_job.py

@@ -7,31 +7,12 @@ from typing import Any
 import httpx
 
 from plato._generated.errors import raise_for_status
-from plato._generated.models import (
-    AgentLogInput,
-    AppApiV2SchemasSessionBatchLogRequest,
-    LogResponse,
-    OODRequestLogInput,
-    PassthroughRequestLogInput,
-    ScreenRecordingStartedLogInput,
-    SessionBackupLogInput,
-    SimResetLogInput,
-    StateMutationLogInput,
-    SystemLogInput,
-)
+from plato._generated.models import LogResponse, SystemLogInput
 
 
 def _build_request_args(
     job_id: str,
-    body: SystemLogInput
-    | AgentLogInput
-    | StateMutationLogInput
-    | OODRequestLogInput
-    | PassthroughRequestLogInput
-    | ScreenRecordingStartedLogInput
-    | SimResetLogInput
-    | SessionBackupLogInput
-    | AppApiV2SchemasSessionBatchLogRequest,
+    body: SystemLogInput,
 ) -> dict[str, Any]:
     """Build request arguments."""
     url = f"/api/v2/jobs/{job_id}/log"
@@ -46,15 +27,7 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     job_id: str,
-    body: SystemLogInput
-    | AgentLogInput
-    | StateMutationLogInput
-    | OODRequestLogInput
-    | PassthroughRequestLogInput
-    | ScreenRecordingStartedLogInput
-    | SimResetLogInput
-    | SessionBackupLogInput
-    | AppApiV2SchemasSessionBatchLogRequest,
+    body: SystemLogInput,
 ) -> LogResponse:
     """Log a state mutation or batch of mutations for a specific job.
 
@@ -83,15 +56,7 @@ def sync(
 async def asyncio(
     client: httpx.AsyncClient,
     job_id: str,
-    body: SystemLogInput
-    | AgentLogInput
-    | StateMutationLogInput
-    | OODRequestLogInput
-    | PassthroughRequestLogInput
-    | ScreenRecordingStartedLogInput
-    | SimResetLogInput
-    | SessionBackupLogInput
-    | AppApiV2SchemasSessionBatchLogRequest,
+    body: SystemLogInput,
 ) -> LogResponse:
     """Log a state mutation or batch of mutations for a specific job.
 

plato/_generated/api/v2/jobs/make.py

@@ -7,11 +7,11 @@ from typing import Any
 import httpx
 
 from plato._generated.errors import raise_for_status
-from plato._generated.models import EnvFromArtifact, EnvFromResource, EnvFromSimulator, EnvInfo
+from plato._generated.models import EnvFromSimulator, EnvInfo
 
 
 def _build_request_args(
-    body: EnvFromSimulator | EnvFromArtifact | EnvFromResource,
+    body: EnvFromSimulator,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
@@ -34,7 +34,7 @@ def _build_request_args(
 
 def sync(
     client: httpx.Client,
-    body: EnvFromSimulator | EnvFromArtifact | EnvFromResource,
+    body: EnvFromSimulator,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> EnvInfo:
@@ -57,7 +57,7 @@ def sync(
 
 async def asyncio(
     client: httpx.AsyncClient,
-    body: EnvFromSimulator | EnvFromArtifact | EnvFromResource,
+    body: EnvFromSimulator,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> EnvInfo:

plato/_generated/api/v2/jobs/setup_sandbox.py

@@ -0,0 +1,97 @@
+"""Setup Sandbox"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import AppApiV2SchemasSessionSetupSandboxRequest, SetupSandboxResult
+
+
+def _build_request_args(
+    job_id: str,
+    body: AppApiV2SchemasSessionSetupSandboxRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/jobs/{job_id}/setup_sandbox"
+
+    headers: dict[str, str] = {}
+    if authorization is not None:
+        headers["authorization"] = authorization
+    if x_api_key is not None:
+        headers["X-API-Key"] = x_api_key
+
+    return {
+        "method": "POST",
+        "url": url,
+        "json": body.model_dump(mode="json", exclude_none=True),
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    job_id: str,
+    body: AppApiV2SchemasSessionSetupSandboxRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> SetupSandboxResult:
+    """Setup sandbox environment with Docker overlay on a specific job.
+
+    This configures the VM for Docker usage with overlay2 storage driver,
+    which is significantly faster than the default vfs driver. Should be called
+    after job creation and before pulling Docker images.
+
+    Args:
+        job_id: The job public ID.
+        request: Optional SetupSandboxRequest with timeout.
+
+    Returns:
+        SetupSandboxResult with success status and command output."""
+
+    request_args = _build_request_args(
+        job_id=job_id,
+        body=body,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return SetupSandboxResult.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    job_id: str,
+    body: AppApiV2SchemasSessionSetupSandboxRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> SetupSandboxResult:
+    """Setup sandbox environment with Docker overlay on a specific job.
+
+    This configures the VM for Docker usage with overlay2 storage driver,
+    which is significantly faster than the default vfs driver. Should be called
+    after job creation and before pulling Docker images.
+
+    Args:
+        job_id: The job public ID.
+        request: Optional SetupSandboxRequest with timeout.
+
+    Returns:
+        SetupSandboxResult with success status and command output."""
+
+    request_args = _build_request_args(
+        job_id=job_id,
+        body=body,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return SetupSandboxResult.model_validate(response.json())

plato/_generated/api/v2/jobs/snapshot.py

@@ -12,7 +12,7 @@ from plato._generated.models import AppApiV2SchemasSessionCreateSnapshotRequest,
 
 def _build_request_args(
     job_id: str,
-    body: AppApiV2SchemasSessionCreateSnapshotRequest | Any,
+    body: AppApiV2SchemasSessionCreateSnapshotRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
@@ -36,7 +36,7 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     job_id: str,
-    body: AppApiV2SchemasSessionCreateSnapshotRequest | Any,
+    body: AppApiV2SchemasSessionCreateSnapshotRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> CreateSnapshotResult:
@@ -62,7 +62,7 @@ def sync(
 async def asyncio(
     client: httpx.AsyncClient,
     job_id: str,
-    body: AppApiV2SchemasSessionCreateSnapshotRequest | Any,
+    body: AppApiV2SchemasSessionCreateSnapshotRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> CreateSnapshotResult:

plato/_generated/api/v2/jobs/snapshot_store.py

@@ -0,0 +1,91 @@
+"""Snapshot Store"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import AppApiV2SchemasSessionCreateSnapshotRequest, CreateSnapshotResult
+
+
+def _build_request_args(
+    job_id: str,
+    body: AppApiV2SchemasSessionCreateSnapshotRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/jobs/{job_id}/snapshot_store"
+
+    headers: dict[str, str] = {}
+    if authorization is not None:
+        headers["authorization"] = authorization
+    if x_api_key is not None:
+        headers["X-API-Key"] = x_api_key
+
+    return {
+        "method": "POST",
+        "url": url,
+        "json": body.model_dump(mode="json", exclude_none=True),
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    job_id: str,
+    body: AppApiV2SchemasSessionCreateSnapshotRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> CreateSnapshotResult:
+    """Create a snapshot-store snapshot of a specific job (disk + memory).
+
+    Uses the snapshot-store pipeline for chunk-based deduplication and
+    efficient storage. This is the preferred method for new base snapshots.
+
+    Optional request body allows overriding artifact metadata:
+    - override_service: Override simulator/service name
+    - override_version: Override version/git_hash
+    - override_dataset: Override dataset name"""
+
+    request_args = _build_request_args(
+        job_id=job_id,
+        body=body,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return CreateSnapshotResult.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    job_id: str,
+    body: AppApiV2SchemasSessionCreateSnapshotRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> CreateSnapshotResult:
+    """Create a snapshot-store snapshot of a specific job (disk + memory).
+
+    Uses the snapshot-store pipeline for chunk-based deduplication and
+    efficient storage. This is the preferred method for new base snapshots.
+
+    Optional request body allows overriding artifact metadata:
+    - override_service: Override simulator/service name
+    - override_version: Override version/git_hash
+    - override_dataset: Override dataset name"""
+
+    request_args = _build_request_args(
+        job_id=job_id,
+        body=body,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return CreateSnapshotResult.model_validate(response.json())

plato/_generated/api/v2/sessions/__init__.py

@@ -22,7 +22,9 @@ from . import (
     make,
     reset,
     set_date,
+    setup_sandbox,
     snapshot,
+    snapshot_store,
     state,
     upload_session_documents,
     wait_for_ready,
@@ -38,7 +40,9 @@ __all__ = [
     "evaluate",
     "execute",
     "set_date",
+    "setup_sandbox",
     "snapshot",
+    "snapshot_store",
     "checkpoint",
     "disk_snapshot",
     "state",

plato/_generated/api/v2/sessions/checkpoint.py

@@ -12,7 +12,7 @@ from plato._generated.models import CreateCheckpointRequest, CreateCheckpointRes
 
 def _build_request_args(
     session_id: str,
-    body: CreateCheckpointRequest | Any,
+    body: CreateCheckpointRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
@@ -36,7 +36,7 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     session_id: str,
-    body: CreateCheckpointRequest | Any,
+    body: CreateCheckpointRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> CreateCheckpointResponse:
@@ -68,7 +68,7 @@ def sync(
 async def asyncio(
     client: httpx.AsyncClient,
     session_id: str,
-    body: CreateCheckpointRequest | Any,
+    body: CreateCheckpointRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> CreateCheckpointResponse:

plato/_generated/api/v2/sessions/disk_snapshot.py

@@ -12,7 +12,7 @@ from plato._generated.models import CreateDiskSnapshotRequest, CreateDiskSnapsho
 
 def _build_request_args(
     session_id: str,
-    body: CreateDiskSnapshotRequest | Any,
+    body: CreateDiskSnapshotRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
@@ -36,7 +36,7 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     session_id: str,
-    body: CreateDiskSnapshotRequest | Any,
+    body: CreateDiskSnapshotRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> CreateDiskSnapshotResponse:
@@ -66,7 +66,7 @@ def sync(
 async def asyncio(
     client: httpx.AsyncClient,
     session_id: str,
-    body: CreateDiskSnapshotRequest | Any,
+    body: CreateDiskSnapshotRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> CreateDiskSnapshotResponse:

plato/_generated/api/v2/sessions/evaluate.py

@@ -12,7 +12,7 @@ from plato._generated.models import AppApiV2SchemasSessionEvaluateRequest, AppAp
 
 def _build_request_args(
     session_id: str,
-    body: AppApiV2SchemasSessionEvaluateRequest | Any,
+    body: AppApiV2SchemasSessionEvaluateRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
@@ -36,7 +36,7 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     session_id: str,
-    body: AppApiV2SchemasSessionEvaluateRequest | Any,
+    body: AppApiV2SchemasSessionEvaluateRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> AppApiV2SchemasSessionEvaluateResponse:
@@ -57,7 +57,7 @@ def sync(
 async def asyncio(
     client: httpx.AsyncClient,
     session_id: str,
-    body: AppApiV2SchemasSessionEvaluateRequest | Any,
+    body: AppApiV2SchemasSessionEvaluateRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> AppApiV2SchemasSessionEvaluateResponse:

plato/_generated/api/v2/sessions/log_job_mutation.py

@@ -7,32 +7,13 @@ from typing import Any
 import httpx
 
 from plato._generated.errors import raise_for_status
-from plato._generated.models import (
-    AgentLogInput,
-    AppApiV2SchemasSessionBatchLogRequest,
-    LogResponse,
-    OODRequestLogInput,
-    PassthroughRequestLogInput,
-    ScreenRecordingStartedLogInput,
-    SessionBackupLogInput,
-    SimResetLogInput,
-    StateMutationLogInput,
-    SystemLogInput,
-)
+from plato._generated.models import LogResponse, SystemLogInput
 
 
 def _build_request_args(
     session_id: str,
     job_id: str,
-    body: SystemLogInput
-    | AgentLogInput
-    | StateMutationLogInput
-    | OODRequestLogInput
-    | PassthroughRequestLogInput
-    | ScreenRecordingStartedLogInput
-    | SimResetLogInput
-    | SessionBackupLogInput
-    | AppApiV2SchemasSessionBatchLogRequest,
+    body: SystemLogInput,
 ) -> dict[str, Any]:
     """Build request arguments."""
     url = f"/api/v2/sessions/{session_id}/jobs/{job_id}/log"
@@ -48,15 +29,7 @@ def sync(
     client: httpx.Client,
     session_id: str,
     job_id: str,
-    body: SystemLogInput
-    | AgentLogInput
-    | StateMutationLogInput
-    | OODRequestLogInput
-    | PassthroughRequestLogInput
-    | ScreenRecordingStartedLogInput
-    | SimResetLogInput
-    | SessionBackupLogInput
-    | AppApiV2SchemasSessionBatchLogRequest,
+    body: SystemLogInput,
 ) -> LogResponse:
     """Log agent mutations/events for a specific job in a session.
 
@@ -80,15 +53,7 @@ async def asyncio(
     client: httpx.AsyncClient,
     session_id: str,
     job_id: str,
-    body: SystemLogInput
-    | AgentLogInput
-    | StateMutationLogInput
-    | OODRequestLogInput
-    | PassthroughRequestLogInput
-    | ScreenRecordingStartedLogInput
-    | SimResetLogInput
-    | SessionBackupLogInput
-    | AppApiV2SchemasSessionBatchLogRequest,
+    body: SystemLogInput,
 ) -> LogResponse:
     """Log agent mutations/events for a specific job in a session.
 

plato/_generated/api/v2/sessions/make.py

@@ -7,11 +7,11 @@ from typing import Any
 import httpx
 
 from plato._generated.errors import raise_for_status
-from plato._generated.models import CreateSessionFromEnvs, CreateSessionFromTask, CreateSessionResponse
+from plato._generated.models import CreateSessionFromEnvs, CreateSessionResponse
 
 
 def _build_request_args(
-    body: CreateSessionFromEnvs | CreateSessionFromTask,
+    body: CreateSessionFromEnvs,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
@@ -34,7 +34,7 @@ def _build_request_args(
 
 def sync(
     client: httpx.Client,
-    body: CreateSessionFromEnvs | CreateSessionFromTask,
+    body: CreateSessionFromEnvs,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> CreateSessionResponse:
@@ -57,7 +57,7 @@ def sync(
 
 async def asyncio(
     client: httpx.AsyncClient,
-    body: CreateSessionFromEnvs | CreateSessionFromTask,
+    body: CreateSessionFromEnvs,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> CreateSessionResponse:

plato/_generated/api/v2/sessions/setup_sandbox.py

@@ -0,0 +1,98 @@
+"""Setup Sandbox"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import (
+    AppApiV2SchemasSessionSetupSandboxRequest,
+    AppApiV2SchemasSessionSetupSandboxResponse,
+)
+
+
+def _build_request_args(
+    session_id: str,
+    body: AppApiV2SchemasSessionSetupSandboxRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/sessions/{session_id}/setup_sandbox"
+
+    headers: dict[str, str] = {}
+    if authorization is not None:
+        headers["authorization"] = authorization
+    if x_api_key is not None:
+        headers["X-API-Key"] = x_api_key
+
+    return {
+        "method": "POST",
+        "url": url,
+        "json": body.model_dump(mode="json", exclude_none=True),
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    session_id: str,
+    body: AppApiV2SchemasSessionSetupSandboxRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> AppApiV2SchemasSessionSetupSandboxResponse:
+    """Setup sandbox environment with Docker overlay on all jobs in the session.
+
+    This configures the VM for Docker usage with overlay2 storage driver,
+    which is significantly faster than the default vfs driver. Should be called
+    after session creation and before pulling Docker images.
+
+    The setup includes:
+    - Mounting /dev/vdb to /mnt/docker for Docker storage
+    - Configuring Docker with overlay2 storage driver
+    - Setting up ECR and Docker Hub authentication
+    - Creating a docker-user service for non-root Docker access"""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        body=body,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return AppApiV2SchemasSessionSetupSandboxResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    session_id: str,
+    body: AppApiV2SchemasSessionSetupSandboxRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> AppApiV2SchemasSessionSetupSandboxResponse:
+    """Setup sandbox environment with Docker overlay on all jobs in the session.
+
+    This configures the VM for Docker usage with overlay2 storage driver,
+    which is significantly faster than the default vfs driver. Should be called
+    after session creation and before pulling Docker images.
+
+    The setup includes:
+    - Mounting /dev/vdb to /mnt/docker for Docker storage
+    - Configuring Docker with overlay2 storage driver
+    - Setting up ECR and Docker Hub authentication
+    - Creating a docker-user service for non-root Docker access"""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        body=body,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return AppApiV2SchemasSessionSetupSandboxResponse.model_validate(response.json())

plato/_generated/api/v2/sessions/snapshot.py

@@ -15,7 +15,7 @@ from plato._generated.models import (
 
 def _build_request_args(
     session_id: str,
-    body: AppApiV2SchemasSessionCreateSnapshotRequest | Any,
+    body: AppApiV2SchemasSessionCreateSnapshotRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
@@ -39,7 +39,7 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     session_id: str,
-    body: AppApiV2SchemasSessionCreateSnapshotRequest | Any,
+    body: AppApiV2SchemasSessionCreateSnapshotRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> AppApiV2SchemasSessionCreateSnapshotResponse:
@@ -65,7 +65,7 @@ def sync(
 async def asyncio(
     client: httpx.AsyncClient,
     session_id: str,
-    body: AppApiV2SchemasSessionCreateSnapshotRequest | Any,
+    body: AppApiV2SchemasSessionCreateSnapshotRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> AppApiV2SchemasSessionCreateSnapshotResponse: