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

plato/_generated/__init__.py

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

plato/_generated/api/v1/evals/get_scores_by_user.py

@@ -15,6 +15,7 @@ def _build_request_args(
     page_size: int | None = None,
     days: float | None = None,
     include_human_results: bool | None = False,
+    date: str | None = None,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
@@ -32,6 +33,8 @@ def _build_request_args(
         params["days"] = days
     if include_human_results is not None:
         params["include_human_results"] = include_human_results
+    if date is not None:
+        params["date"] = date
 
     headers: dict[str, str] = {}
     if authorization is not None:
@@ -54,6 +57,7 @@ def sync(
     page_size: int | None = None,
     days: float | None = None,
     include_human_results: bool | None = False,
+    date: str | None = None,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> Any:
@@ -65,6 +69,7 @@ def sync(
         page_size=page_size,
         days=days,
         include_human_results=include_human_results,
+        date=date,
         authorization=authorization,
         x_api_key=x_api_key,
     )
@@ -81,6 +86,7 @@ async def asyncio(
     page_size: int | None = None,
     days: float | None = None,
     include_human_results: bool | None = False,
+    date: str | None = None,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> Any:
@@ -92,6 +98,7 @@ async def asyncio(
         page_size=page_size,
         days=days,
         include_human_results=include_human_results,
+        date=date,
         authorization=authorization,
         x_api_key=x_api_key,
     )

plato/_generated/api/v1/testcases/get_testcases.py

@@ -25,6 +25,8 @@ def _build_request_args(
     is_sample: bool | None = None,
     rejected: bool | None = None,
     exclude_assigned_to_annotators: bool | None = None,
+    workorder: str | None = None,
+    infeasible: bool | None = None,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
@@ -62,6 +64,10 @@ def _build_request_args(
         params["rejected"] = rejected
     if exclude_assigned_to_annotators is not None:
         params["exclude_assigned_to_annotators"] = exclude_assigned_to_annotators
+    if workorder is not None:
+        params["workorder"] = workorder
+    if infeasible is not None:
+        params["infeasible"] = infeasible
 
     headers: dict[str, str] = {}
     if authorization is not None:
@@ -94,6 +100,8 @@ def sync(
     is_sample: bool | None = None,
     rejected: bool | None = None,
     exclude_assigned_to_annotators: bool | None = None,
+    workorder: str | None = None,
+    infeasible: bool | None = None,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> Any:
@@ -115,6 +123,8 @@ def sync(
         is_sample=is_sample,
         rejected=rejected,
         exclude_assigned_to_annotators=exclude_assigned_to_annotators,
+        workorder=workorder,
+        infeasible=infeasible,
         authorization=authorization,
         x_api_key=x_api_key,
     )
@@ -141,6 +151,8 @@ async def asyncio(
     is_sample: bool | None = None,
     rejected: bool | None = None,
     exclude_assigned_to_annotators: bool | None = None,
+    workorder: str | None = None,
+    infeasible: bool | None = None,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> Any:
@@ -162,6 +174,8 @@ async def asyncio(
         is_sample=is_sample,
         rejected=rejected,
         exclude_assigned_to_annotators=exclude_assigned_to_annotators,
+        workorder=workorder,
+        infeasible=infeasible,
         authorization=authorization,
         x_api_key=x_api_key,
     )

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

@@ -53,7 +53,7 @@ def sync(
     - 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"""
+    - Adding plato user to docker group for shared Docker access"""
 
     request_args = _build_request_args(
         session_id=session_id,
@@ -84,7 +84,7 @@ async def asyncio(
     - 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"""
+    - Adding plato user to docker group for shared Docker access"""
 
     request_args = _build_request_args(
         session_id=session_id,

plato/_generated/models/__init__.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
-#   filename:  tmp28thllk8.json
-#   timestamp: 2026-01-17T04:45:09+00:00
+#   filename:  tmppwt3ublb.json
+#   timestamp: 2026-01-23T22:08:30+00:00
 
 from __future__ import annotations
 
@@ -840,6 +840,14 @@ class CreateSnapshotResult(BaseModel):
     """
     ID of the created snapshot artifact
     """
+    success: Annotated[bool | None, Field(title="Success")] = True
+    """
+    Whether snapshot creation succeeded
+    """
+    error: Annotated[str | None, Field(title="Error")] = None
+    """
+    Error message if failed
+    """
 
 
 class CreateTestCaseFromLabelsRequest(BaseModel):
@@ -1714,6 +1722,20 @@ class NodeNatsConfig(BaseModel):
     subject_prefix: Annotated[str | None, Field(title="Subject Prefix")] = "firecracker"
 
 
+class NodeNebulaConfig(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    lighthouse_upstream: Annotated[str | None, Field(title="Lighthouse Upstream")] = "100.64.1.1:4242"
+    """
+    Nebula lighthouse upstream address (host:port) for mesh connectivity.
+    """
+    lighthouse_ip: Annotated[str | None, Field(title="Lighthouse Ip")] = "100.64.1.1"
+    """
+    Nebula lighthouse mesh IP address.
+    """
+
+
 class NodePolicyConfig(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -1808,6 +1830,10 @@ class NodeSnapshotStoreConfig(BaseModel):
     """
     Worker threads for snapshot-store compression.
     """
+    upload_workers: Annotated[int | None, Field(title="Upload Workers")] = 6
+    """
+    Number of concurrent snapshot upload jobs.
+    """
     upload_max_attempts: Annotated[int | None, Field(title="Upload Max Attempts")] = 3
     """
     Max upload attempts for snapshot-store uploads.
@@ -1846,6 +1872,12 @@ class NodeStreamConfig(BaseModel):
     resource_matcher_key_prefix: Annotated[str | None, Field(title="Resource Matcher Key Prefix")] = (
         "vm-resource-matcher"
     )
+    artifact_upload_notification_channel: Annotated[str | None, Field(title="Artifact Upload Notification Channel")] = (
+        "artifact-upload-notifications"
+    )
+    """
+    Pubsub channel for artifact upload status updates.
+    """
 
 
 class OODRequestLogInput(BaseModel):
@@ -4358,6 +4390,10 @@ class NodeConfig(BaseModel):
     """
     Build configuration including SSH keys for VMs
     """
+    nebula: NodeNebulaConfig | None = None
+    """
+    Nebula mesh VPN configuration for lighthouse connectivity
+    """
 
 
 class QueueStatusResponse(BaseModel):
@@ -4501,6 +4537,8 @@ class TestCaseCreateRequest(BaseModel):
     simulatorId: Annotated[int | None, Field(title="Simulatorid")] = None
     simulatorArtifactId: Annotated[str | None, Field(title="Simulatorartifactid")] = None
     simulatorArtifactIds: Annotated[list[str] | None, Field(title="Simulatorartifactids")] = None
+    workorder: Annotated[str | None, Field(title="Workorder")] = None
+    infeasible: Annotated[bool | None, Field(title="Infeasible")] = None
 
 
 class TestCaseUpdateRequest(BaseModel):
@@ -4525,6 +4563,8 @@ class TestCaseUpdateRequest(BaseModel):
     simulatorArtifactId: Annotated[str | None, Field(title="Simulatorartifactid")] = None
     simulatorArtifactIds: Annotated[list[str] | None, Field(title="Simulatorartifactids")] = None
     mutationIgnoreConfig: DatabaseIgnoreConfig | None = None
+    workorder: Annotated[str | None, Field(title="Workorder")] = None
+    infeasible: Annotated[bool | None, Field(title="Infeasible")] = None
 
 
 class WaitForReadyResponse(BaseModel):

plato/chronos/api/admin/__init__.py

@@ -0,0 +1,17 @@
+"""API endpoints."""
+
+from . import (
+    clear_database_api_admin_clear_db_post,
+    sync_agents_api_admin_sync_agents_post,
+    sync_all_api_admin_sync_all_post,
+    sync_runtimes_api_admin_sync_runtimes_post,
+    sync_worlds_api_admin_sync_worlds_post,
+)
+
+__all__ = [
+    "sync_agents_api_admin_sync_agents_post",
+    "sync_worlds_api_admin_sync_worlds_post",
+    "sync_runtimes_api_admin_sync_runtimes_post",
+    "sync_all_api_admin_sync_all_post",
+    "clear_database_api_admin_clear_db_post",
+]

plato/chronos/api/admin/clear_database_api_admin_clear_db_post.py

@@ -0,0 +1,57 @@
+"""Clear Database"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import ClearDbResult
+
+
+def _build_request_args(
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/admin/clear-db"
+
+    headers: dict[str, str] = {}
+    if x_api_key is not None:
+        headers["X-API-Key"] = x_api_key
+
+    return {
+        "method": "POST",
+        "url": url,
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    x_api_key: str | None = None,
+) -> ClearDbResult:
+    """Clear ALL data from the database. Only available in local environment."""
+
+    request_args = _build_request_args(
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return ClearDbResult.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    x_api_key: str | None = None,
+) -> ClearDbResult:
+    """Clear ALL data from the database. Only available in local environment."""
+
+    request_args = _build_request_args(
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return ClearDbResult.model_validate(response.json())

plato/chronos/api/admin/sync_agents_api_admin_sync_agents_post.py

@@ -0,0 +1,67 @@
+"""Sync Agents"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import SyncResult
+
+
+def _build_request_args(
+    source_url: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/admin/sync/agents"
+
+    params: dict[str, Any] = {}
+    if source_url is not None:
+        params["source_url"] = source_url
+
+    headers: dict[str, str] = {}
+    if x_api_key is not None:
+        headers["X-API-Key"] = x_api_key
+
+    return {
+        "method": "POST",
+        "url": url,
+        "params": params,
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    source_url: str,
+    x_api_key: str | None = None,
+) -> SyncResult:
+    """Sync agents from source Chronos to local database."""
+
+    request_args = _build_request_args(
+        source_url=source_url,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return SyncResult.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    source_url: str,
+    x_api_key: str | None = None,
+) -> SyncResult:
+    """Sync agents from source Chronos to local database."""
+
+    request_args = _build_request_args(
+        source_url=source_url,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return SyncResult.model_validate(response.json())

plato/chronos/api/admin/sync_all_api_admin_sync_all_post.py

@@ -0,0 +1,67 @@
+"""Sync All"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import SyncAllResult
+
+
+def _build_request_args(
+    source_url: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/admin/sync/all"
+
+    params: dict[str, Any] = {}
+    if source_url is not None:
+        params["source_url"] = source_url
+
+    headers: dict[str, str] = {}
+    if x_api_key is not None:
+        headers["X-API-Key"] = x_api_key
+
+    return {
+        "method": "POST",
+        "url": url,
+        "params": params,
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    source_url: str,
+    x_api_key: str | None = None,
+) -> SyncAllResult:
+    """Sync all data (agents, worlds, runtimes) from source."""
+
+    request_args = _build_request_args(
+        source_url=source_url,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return SyncAllResult.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    source_url: str,
+    x_api_key: str | None = None,
+) -> SyncAllResult:
+    """Sync all data (agents, worlds, runtimes) from source."""
+
+    request_args = _build_request_args(
+        source_url=source_url,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return SyncAllResult.model_validate(response.json())

plato/chronos/api/admin/sync_runtimes_api_admin_sync_runtimes_post.py

@@ -0,0 +1,67 @@
+"""Sync Runtimes"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import SyncResult
+
+
+def _build_request_args(
+    source_url: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/admin/sync/runtimes"
+
+    params: dict[str, Any] = {}
+    if source_url is not None:
+        params["source_url"] = source_url
+
+    headers: dict[str, str] = {}
+    if x_api_key is not None:
+        headers["X-API-Key"] = x_api_key
+
+    return {
+        "method": "POST",
+        "url": url,
+        "params": params,
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    source_url: str,
+    x_api_key: str | None = None,
+) -> SyncResult:
+    """Sync runtimes from source Chronos to local database."""
+
+    request_args = _build_request_args(
+        source_url=source_url,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return SyncResult.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    source_url: str,
+    x_api_key: str | None = None,
+) -> SyncResult:
+    """Sync runtimes from source Chronos to local database."""
+
+    request_args = _build_request_args(
+        source_url=source_url,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return SyncResult.model_validate(response.json())

plato/chronos/api/admin/sync_worlds_api_admin_sync_worlds_post.py

@@ -0,0 +1,67 @@
+"""Sync Worlds"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import SyncResult
+
+
+def _build_request_args(
+    source_url: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/admin/sync/worlds"
+
+    params: dict[str, Any] = {}
+    if source_url is not None:
+        params["source_url"] = source_url
+
+    headers: dict[str, str] = {}
+    if x_api_key is not None:
+        headers["X-API-Key"] = x_api_key
+
+    return {
+        "method": "POST",
+        "url": url,
+        "params": params,
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    source_url: str,
+    x_api_key: str | None = None,
+) -> SyncResult:
+    """Sync worlds from source Chronos to local database."""
+
+    request_args = _build_request_args(
+        source_url=source_url,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return SyncResult.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    source_url: str,
+    x_api_key: str | None = None,
+) -> SyncResult:
+    """Sync worlds from source Chronos to local database."""
+
+    request_args = _build_request_args(
+        source_url=source_url,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return SyncResult.model_validate(response.json())

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/chronos/api/checkpoints/__init__.py

@@ -0,0 +1,8 @@
+"""API endpoints."""
+
+from . import list_checkpoints, preview_checkpoint
+
+__all__ = [
+    "preview_checkpoint",
+    "list_checkpoints",
+]

plato/chronos/api/checkpoints/list_checkpoints.py

@@ -0,0 +1,52 @@
+"""List Checkpoints"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import CheckpointListResponse
+
+
+def _build_request_args(
+    session_id: str,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/checkpoints/session/{session_id}"
+
+    return {
+        "method": "GET",
+        "url": url,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    session_id: str,
+) -> CheckpointListResponse:
+    """List all checkpoints for a Chronos session."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return CheckpointListResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    session_id: str,
+) -> CheckpointListResponse:
+    """List all checkpoints for a Chronos session."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return CheckpointListResponse.model_validate(response.json())