plato-sdk-v2 2.7.7__tar.gz → 2.7.9__tar.gz

{plato_sdk_v2-2.7.7 → plato_sdk_v2-2.7.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plato-sdk-v2
-Version: 2.7.7
+Version: 2.7.9
 Summary: Python SDK for the Plato API
 Author-email: Plato <support@plato.so>
 License-Expression: MIT

plato_sdk_v2-2.7.9/VERSION

@@ -0,0 +1 @@
+2.7.9

{plato_sdk_v2-2.7.7 → plato_sdk_v2-2.7.9}/plato/_generated/__init__.py

@@ -1,4 +1,4 @@
-"""Plato API SDK - v0.47.0"""
+"""Plato API SDK - v0.48.2"""
 
 from . import api, errors, models
 from .client import AsyncClient, Client

{plato_sdk_v2-2.7.7 → plato_sdk_v2-2.7.9}/plato/_generated/api/v2/sessions/__init__.py

@@ -23,6 +23,7 @@ from . import (
     list_sessions,
     log_job_mutation,
     make,
+    remove_job,
     reset,
     set_date,
     setup_sandbox,
@@ -37,6 +38,7 @@ __all__ = [
     "make",
     "list_jobs",
     "add_job",
+    "remove_job",
     "reset",
     "heartbeat",
     "connect_network",

plato_sdk_v2-2.7.9/plato/_generated/api/v2/sessions/remove_job.py

@@ -0,0 +1,89 @@
+"""Remove Job"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import RemoveJobRequest, RemoveJobResponse
+
+
+def _build_request_args(
+    session_id: str,
+    body: RemoveJobRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/sessions/{session_id}/remove-job"
+
+    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: RemoveJobRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> RemoveJobResponse:
+    """Remove a job from a session.
+
+    This will:
+    1. Remove the job from the session's network (if connected)
+    2. Shut down the VM associated with the job
+    3. Cancel the job in the system
+
+    Use this to clean up individual jobs without closing the entire session."""
+
+    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 RemoveJobResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    session_id: str,
+    body: RemoveJobRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> RemoveJobResponse:
+    """Remove a job from a session.
+
+    This will:
+    1. Remove the job from the session's network (if connected)
+    2. Shut down the VM associated with the job
+    3. Cancel the job in the system
+
+    Use this to clean up individual jobs without closing the entire session."""
+
+    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 RemoveJobResponse.model_validate(response.json())

{plato_sdk_v2-2.7.7 → plato_sdk_v2-2.7.9}/plato/_generated/models/__init__.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
-#   filename:  tmpy5rfux2_.json
-#   timestamp: 2026-01-29T20:47:53+00:00
+#   filename:  tmp2hcn27e0.json
+#   timestamp: 2026-01-29T22:45:52+00:00
 
 from __future__ import annotations
 
@@ -1158,6 +1158,14 @@ class EnvFromArtifact(BaseModel):
     """
 
 
+class RootfsStorageBackend(Enum):
+    sparse_s3 = "sparse-s3"
+    blockdiff = "blockdiff"
+    blockdiff_checkpoint = "blockdiff_checkpoint"
+    blockdiff_disk = "blockdiff_disk"
+    snapshot_store = "snapshot-store"
+
+
 class EnvFromSimulator(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -1929,6 +1937,11 @@ class NodeSessionNetworkConfig(BaseModel):
     """
 
 
+class DefaultRootfsStorageBackend(Enum):
+    sparse_s3 = "sparse-s3"
+    snapshot_store = "snapshot-store"
+
+
 class NodeSnapshotStoreConfig(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -1977,9 +1990,12 @@ class NodeSnapshotStoreConfig(BaseModel):
     """
     Local storage root for snapshot-store data.
     """
-    dedup_rootfs: Annotated[bool | None, Field(title="Dedup Rootfs")] = False
+    default_rootfs_storage_backend: Annotated[
+        DefaultRootfsStorageBackend | None,
+        Field(title="Default Rootfs Storage Backend"),
+    ] = DefaultRootfsStorageBackend.sparse_s3
     """
-    Enable rootfs-level dedup via snapshot-store FUSE mount instead of file copy.
+    Default rootfs storage backend. 'sparse-s3' uses S3 archives, 'snapshot-store' uses FUSE-based dedup.
     """
 
 
@@ -2334,6 +2350,38 @@ class ReleaseStatus(Enum):
     failed = "failed"
 
 
+class RemoveJobRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    job_id: Annotated[str, Field(title="Job Id")]
+    """
+    Job ID to remove from the session
+    """
+
+
+class RemoveJobResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    session_id: Annotated[str, Field(title="Session Id")]
+    """
+    Session ID the job was removed from
+    """
+    job_id: Annotated[str, Field(title="Job Id")]
+    """
+    Job ID that was removed
+    """
+    success: Annotated[bool, Field(title="Success")]
+    """
+    Whether the removal was successful
+    """
+    message: Annotated[str | None, Field(title="Message")] = None
+    """
+    Status message or error details
+    """
+
+
 class RepositoryResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -4324,6 +4372,18 @@ class EnvFromResource(BaseModel):
     """
     Custom name for this environment
     """
+    docker_image_url: Annotated[str | None, Field(title="Docker Image Url")] = None
+    """
+    Custom Docker image URL (ECR). If not set, uses default from settings.
+    """
+    rootfs_storage_backend: Annotated[RootfsStorageBackend | None, Field(title="Rootfs Storage Backend")] = None
+    """
+    Storage backend for rootfs ('sparse-s3' or 'snapshot-store'). If not set, uses default.
+    """
+    upload_rootfs: Annotated[bool | None, Field(title="Upload Rootfs")] = True
+    """
+    Upload rootfs to S3/snapshot-store if not already cached. Set to False for one-off VMs.
+    """
 
 
 class ExecuteCommandResponse(BaseModel):

{plato_sdk_v2-2.7.7 → plato_sdk_v2-2.7.9}/plato/chronos/api/agents/get_agent_versions.py

@@ -7,7 +7,7 @@ from typing import Any
 import httpx
 
 from plato.chronos.errors import raise_for_status
-from plato.chronos.models import AgentVersionsResponse
+from plato.chronos.models import ChronosModelsAgentAgentVersionsResponse
 
 
 def _build_request_args(
@@ -32,7 +32,7 @@ def sync(
     client: httpx.Client,
     name: str,
     x_api_key: str | None = None,
-) -> AgentVersionsResponse:
+) -> ChronosModelsAgentAgentVersionsResponse:
     """Get all versions of an agent by name."""
 
     request_args = _build_request_args(
@@ -42,14 +42,14 @@ def sync(
 
     response = client.request(**request_args)
     raise_for_status(response)
-    return AgentVersionsResponse.model_validate(response.json())
+    return ChronosModelsAgentAgentVersionsResponse.model_validate(response.json())
 
 
 async def asyncio(
     client: httpx.AsyncClient,
     name: str,
     x_api_key: str | None = None,
-) -> AgentVersionsResponse:
+) -> ChronosModelsAgentAgentVersionsResponse:
     """Get all versions of an agent by name."""
 
     request_args = _build_request_args(
@@ -59,4 +59,4 @@ async def asyncio(
 
     response = await client.request(**request_args)
     raise_for_status(response)
-    return AgentVersionsResponse.model_validate(response.json())
+    return ChronosModelsAgentAgentVersionsResponse.model_validate(response.json())

{plato_sdk_v2-2.7.7 → plato_sdk_v2-2.7.9}/plato/chronos/api/agents/list_agents.py

@@ -7,7 +7,7 @@ from typing import Any
 import httpx
 
 from plato.chronos.errors import raise_for_status
-from plato.chronos.models import AgentListResponse
+from plato.chronos.models import ChronosModelsAgentAgentListResponse
 
 
 def _build_request_args(
@@ -30,7 +30,7 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     x_api_key: str | None = None,
-) -> AgentListResponse:
+) -> ChronosModelsAgentAgentListResponse:
     """List all agents for the org."""
 
     request_args = _build_request_args(
@@ -39,13 +39,13 @@ def sync(
 
     response = client.request(**request_args)
     raise_for_status(response)
-    return AgentListResponse.model_validate(response.json())
+    return ChronosModelsAgentAgentListResponse.model_validate(response.json())
 
 
 async def asyncio(
     client: httpx.AsyncClient,
     x_api_key: str | None = None,
-) -> AgentListResponse:
+) -> ChronosModelsAgentAgentListResponse:
     """List all agents for the org."""
 
     request_args = _build_request_args(
@@ -54,4 +54,4 @@ async def asyncio(
 
     response = await client.request(**request_args)
     raise_for_status(response)
-    return AgentListResponse.model_validate(response.json())
+    return ChronosModelsAgentAgentListResponse.model_validate(response.json())

{plato_sdk_v2-2.7.7 → plato_sdk_v2-2.7.9}/plato/chronos/api/registry/get_agent_versions_api_registry_agents__agent_name__versions_get.py

@@ -7,7 +7,7 @@ from typing import Any
 import httpx
 
 from plato.chronos.errors import raise_for_status
-from plato.chronos.models import ChronosApiRegistryAgentVersionsResponse
+from plato.chronos.models import AgentVersionsResponse
 
 
 def _build_request_args(
@@ -25,7 +25,7 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     agent_name: str,
-) -> ChronosApiRegistryAgentVersionsResponse:
+) -> AgentVersionsResponse:
     """Get versions for an agent from the registry."""
 
     request_args = _build_request_args(
@@ -34,13 +34,13 @@ def sync(
 
     response = client.request(**request_args)
     raise_for_status(response)
-    return ChronosApiRegistryAgentVersionsResponse.model_validate(response.json())
+    return AgentVersionsResponse.model_validate(response.json())
 
 
 async def asyncio(
     client: httpx.AsyncClient,
     agent_name: str,
-) -> ChronosApiRegistryAgentVersionsResponse:
+) -> AgentVersionsResponse:
     """Get versions for an agent from the registry."""
 
     request_args = _build_request_args(
@@ -49,4 +49,4 @@ async def asyncio(
 
     response = await client.request(**request_args)
     raise_for_status(response)
-    return ChronosApiRegistryAgentVersionsResponse.model_validate(response.json())
+    return AgentVersionsResponse.model_validate(response.json())

{plato_sdk_v2-2.7.7 → plato_sdk_v2-2.7.9}/plato/chronos/api/registry/list_registry_agents_api_registry_agents_get.py

@@ -7,7 +7,7 @@ from typing import Any
 import httpx
 
 from plato.chronos.errors import raise_for_status
-from plato.chronos.models import ChronosApiRegistryAgentListResponse
+from plato.chronos.models import AgentListResponse
 
 
 def _build_request_args() -> dict[str, Any]:
@@ -22,23 +22,23 @@ def _build_request_args() -> dict[str, Any]:
 
 def sync(
     client: httpx.Client,
-) -> ChronosApiRegistryAgentListResponse:
+) -> AgentListResponse:
     """List all agents from the registry with their ECR image URIs."""
 
     request_args = _build_request_args()
 
     response = client.request(**request_args)
     raise_for_status(response)
-    return ChronosApiRegistryAgentListResponse.model_validate(response.json())
+    return AgentListResponse.model_validate(response.json())
 
 
 async def asyncio(
     client: httpx.AsyncClient,
-) -> ChronosApiRegistryAgentListResponse:
+) -> AgentListResponse:
     """List all agents from the registry with their ECR image URIs."""
 
     request_args = _build_request_args()
 
     response = await client.request(**request_args)
     raise_for_status(response)
-    return ChronosApiRegistryAgentListResponse.model_validate(response.json())
+    return AgentListResponse.model_validate(response.json())

{plato_sdk_v2-2.7.7 → plato_sdk_v2-2.7.9}/plato/chronos/models/__init__.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
-#   filename:  tmp9iosw1i3.json
-#   timestamp: 2026-01-29T20:31:32+00:00
+#   filename:  tmpi9d4hw2u.json
+#   timestamp: 2026-01-29T22:45:54+00:00
 
 from __future__ import annotations
 
@@ -30,6 +30,13 @@ class AgentInfo(BaseModel):
     image_uri: Annotated[str, Field(title="Image Uri")]
 
 
+class AgentListResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    agents: Annotated[list[AgentInfo], Field(title="Agents")]
+
+
 class AgentLookupResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -71,14 +78,6 @@ class AgentVersionInfo(BaseModel):
     created_at: Annotated[AwareDatetime, Field(title="Created At")]
 
 
-class AgentVersionsResponse(BaseModel):
-    model_config = ConfigDict(
-        extra="allow",
-    )
-    name: Annotated[str, Field(title="Name")]
-    versions: Annotated[list[AgentVersionInfo], Field(title="Versions")]
-
-
 class AuthStatusResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -566,13 +565,6 @@ class WorldVersionsResponse(BaseModel):
     versions: Annotated[list[str], Field(title="Versions")]
 
 
-class ChronosApiRegistryAgentListResponse(BaseModel):
-    model_config = ConfigDict(
-        extra="allow",
-    )
-    agents: Annotated[list[AgentInfo], Field(title="Agents")]
-
-
 class ChronosApiRegistryAgentSchemaResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -592,12 +584,19 @@ class ChronosApiRegistryAgentVersionInfo(BaseModel):
     created_at: Annotated[str | None, Field(title="Created At")] = None
 
 
-class ChronosApiRegistryAgentVersionsResponse(BaseModel):
+class ChronosModelsAgentAgentListResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    agents: Annotated[list[AgentResponse], Field(title="Agents")]
+
+
+class ChronosModelsAgentAgentVersionsResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
     )
     name: Annotated[str, Field(title="Name")]
-    versions: Annotated[list[ChronosApiRegistryAgentVersionInfo], Field(title="Versions")]
+    versions: Annotated[list[AgentVersionInfo], Field(title="Versions")]
 
 
 class ChronosModelsSessionWorldInfo(BaseModel):
@@ -610,11 +609,12 @@ class ChronosModelsSessionWorldInfo(BaseModel):
     config_schema: Annotated[dict[str, Any] | None, Field(title="Config Schema")] = None
 
 
-class AgentListResponse(BaseModel):
+class AgentVersionsResponse(BaseModel):
     model_config = ConfigDict(
         extra="allow",
     )
-    agents: Annotated[list[AgentResponse], Field(title="Agents")]
+    name: Annotated[str, Field(title="Name")]
+    versions: Annotated[list[ChronosApiRegistryAgentVersionInfo], Field(title="Versions")]
 
 
 class CreatorsListResponse(BaseModel):

{plato_sdk_v2-2.7.7 → plato_sdk_v2-2.7.9}/plato/cli/pm.py

@@ -42,7 +42,6 @@ from plato.cli.utils import (
     require_sandbox_field,
     require_sandbox_state,
 )
-from plato.cli.verify import pm_verify_app
 from plato.v2.async_.client import AsyncPlato
 from plato.v2.async_.flow_executor import FlowExecutor
 from plato.v2.types import Env
@@ -66,7 +65,6 @@ submit_app = typer.Typer(help="Submit simulator artifacts for review")
 pm_app.add_typer(list_app, name="list")
 pm_app.add_typer(review_app, name="review")
 pm_app.add_typer(submit_app, name="submit")
-pm_app.add_typer(pm_verify_app, name="verify")
 
 
 # =============================================================================

{plato_sdk_v2-2.7.7 → plato_sdk_v2-2.7.9}/plato/v2/sync/sandbox.py

@@ -20,7 +20,6 @@ from urllib.parse import quote
 
 import httpx
 import yaml
-from playwright.async_api import async_playwright
 from pydantic import BaseModel
 from rich.console import Console
 
@@ -1128,6 +1127,8 @@ class SandboxClient:
 
         # Run the flow with Playwright
         async def _run_flow():
+            from playwright.async_api import async_playwright
+
             async with async_playwright() as p:
                 browser = await p.chromium.launch(headless=False)
                 try:

plato_sdk_v2-2.7.7/VERSION

@@ -1 +0,0 @@
-2.7.7