plato-sdk-v2 2.7.4__py3-none-any.whl → 2.7.5__py3-none-any.whl

plato/_generated/__init__.py

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

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

@@ -37,9 +37,9 @@ def sync(
     x_api_key: str | None = None,
 ) -> Any:
     """Get limited test case metadata for the scoring page.
-    Only returns: publicId, name, scoringTypes.
-    This endpoint is designed for annotators who should not see sensitive test case data
-    like prompts, hints, or expected outputs."""
+    Returns: publicId, name, scoringTypes, prompt.
+    This endpoint is designed for annotators - prompt is included to help them
+    understand what they're evaluating."""
 
     request_args = _build_request_args(
         public_id=public_id,
@@ -59,9 +59,9 @@ async def asyncio(
     x_api_key: str | None = None,
 ) -> Any:
     """Get limited test case metadata for the scoring page.
-    Only returns: publicId, name, scoringTypes.
-    This endpoint is designed for annotators who should not see sensitive test case data
-    like prompts, hints, or expected outputs."""
+    Returns: publicId, name, scoringTypes, prompt.
+    This endpoint is designed for annotators - prompt is included to help them
+    understand what they're evaluating."""
 
     request_args = _build_request_args(
         public_id=public_id,

plato/_generated/api/v2/__init__.py

@@ -1,8 +1,20 @@
 """API version module."""
 
-from . import agents, artifacts, chronos, chronos_packages, cluster, jobs, networks, pypi, releases, sessions, user
+from . import admin.cache
+from . import agents
+from . import artifacts
+from . import chronos
+from . import chronos_packages
+from . import cluster
+from . import jobs
+from . import networks
+from . import pypi
+from . import releases
+from . import sessions
+from . import user
 
 __all__ = [
+    "admin.cache",
     "agents",
     "artifacts",
     "chronos",
@@ -14,4 +26,4 @@ __all__ = [
     "releases",
     "sessions",
     "user",
-]
+]

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

@@ -0,0 +1,3 @@
+"""API version module."""
+
+__all__ = []

plato/_generated/api/v2/admin/cache/__init__.py

@@ -0,0 +1,11 @@
+"""API endpoints."""
+
+from . import clear_all, clear_single_cache, delete_key, get_stats, list_cache_names
+
+__all__ = [
+    "get_stats",
+    "clear_single_cache",
+    "clear_all",
+    "delete_key",
+    "list_cache_names",
+]

plato/_generated/api/v2/admin/cache/clear_all.py

@@ -0,0 +1,74 @@
+"""Clear All"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import CacheClearAllResponse
+
+
+def _build_request_args(
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/v2/admin/cache/clear-all"
+
+    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,
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> CacheClearAllResponse:
+    """Clear all registered caches.
+
+    This clears all caches on the current worker and publishes events
+    to clear them on all other workers in the cluster.
+
+    Requires admin privileges."""
+
+    request_args = _build_request_args(
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return CacheClearAllResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> CacheClearAllResponse:
+    """Clear all registered caches.
+
+    This clears all caches on the current worker and publishes events
+    to clear them on all other workers in the cluster.
+
+    Requires admin privileges."""
+
+    request_args = _build_request_args(
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return CacheClearAllResponse.model_validate(response.json())

plato/_generated/api/v2/admin/cache/clear_single_cache.py

@@ -0,0 +1,79 @@
+"""Clear Single Cache"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import CacheClearResponse, CacheName
+
+
+def _build_request_args(
+    cache_name: CacheName,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/admin/cache/clear/{cache_name}"
+
+    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,
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    cache_name: CacheName,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> CacheClearResponse:
+    """Clear a specific cache by name.
+
+    This clears the cache on the current worker and publishes an event
+    to clear the cache on all other workers in the cluster.
+
+    Requires admin privileges."""
+
+    request_args = _build_request_args(
+        cache_name=cache_name,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return CacheClearResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    cache_name: CacheName,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> CacheClearResponse:
+    """Clear a specific cache by name.
+
+    This clears the cache on the current worker and publishes an event
+    to clear the cache on all other workers in the cluster.
+
+    Requires admin privileges."""
+
+    request_args = _build_request_args(
+        cache_name=cache_name,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return CacheClearResponse.model_validate(response.json())

plato/_generated/api/v2/admin/cache/delete_key.py

@@ -0,0 +1,84 @@
+"""Delete Key"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import CacheKeyDeleteResponse, CacheName
+
+
+def _build_request_args(
+    cache_name: CacheName,
+    key: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/admin/cache/key/{cache_name}/{key}"
+
+    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": "DELETE",
+        "url": url,
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    cache_name: CacheName,
+    key: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> CacheKeyDeleteResponse:
+    """Delete a specific key from a cache.
+
+    Note: This only deletes from the current worker. For distributed
+    invalidation of a specific key, consider clearing the entire cache.
+
+    Requires admin privileges."""
+
+    request_args = _build_request_args(
+        cache_name=cache_name,
+        key=key,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return CacheKeyDeleteResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    cache_name: CacheName,
+    key: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> CacheKeyDeleteResponse:
+    """Delete a specific key from a cache.
+
+    Note: This only deletes from the current worker. For distributed
+    invalidation of a specific key, consider clearing the entire cache.
+
+    Requires admin privileges."""
+
+    request_args = _build_request_args(
+        cache_name=cache_name,
+        key=key,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return CacheKeyDeleteResponse.model_validate(response.json())

plato/_generated/api/v2/admin/cache/get_stats.py

@@ -0,0 +1,70 @@
+"""Get Stats"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import CacheStatsResponse
+
+
+def _build_request_args(
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/v2/admin/cache/stats"
+
+    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": "GET",
+        "url": url,
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> CacheStatsResponse:
+    """Get statistics for all registered caches.
+
+    Returns cache names with their current size and TTL configuration.
+    Requires admin privileges."""
+
+    request_args = _build_request_args(
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return CacheStatsResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> CacheStatsResponse:
+    """Get statistics for all registered caches.
+
+    Returns cache names with their current size and TTL configuration.
+    Requires admin privileges."""
+
+    request_args = _build_request_args(
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return CacheStatsResponse.model_validate(response.json())

plato/_generated/api/v2/admin/cache/list_cache_names.py

@@ -0,0 +1,67 @@
+"""List Cache Names"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+
+
+def _build_request_args(
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/v2/admin/cache/names"
+
+    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": "GET",
+        "url": url,
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """List all registered cache names and their TTL configuration.
+
+    Requires admin privileges."""
+
+    request_args = _build_request_args(
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return response.json()
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """List all registered cache names and their TTL configuration.
+
+    Requires admin privileges."""
+
+    request_args = _build_request_args(
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return response.json()

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

@@ -1,6 +1,7 @@
 """API endpoints."""
 
 from . import (
+    add_job,
     add_ssh_key,
     checkpoint,
     close,
@@ -34,6 +35,8 @@ from . import (
 
 __all__ = [
     "make",
+    "list_jobs",
+    "add_job",
     "reset",
     "heartbeat",
     "connect_network",
@@ -52,7 +55,6 @@ __all__ = [
     "state",
     "get_session_flows",
     "get_session_plato_config",
-    "list_jobs",
     "wait_for_ready",
     "get_public_url",
     "get_connect_url",

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

@@ -0,0 +1,97 @@
+"""Add Job"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import AddJobRequest, AddJobResponse
+
+
+def _build_request_args(
+    session_id: str,
+    body: AddJobRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/sessions/{session_id}/jobs"
+
+    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: AddJobRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> AddJobResponse:
+    """Add a new job to an existing session.
+
+    The new job will:
+    1. Become part of the session's job group
+    2. Be matched to an available VM via the resource matcher
+    3. Automatically join the session's WireGuard network if one exists
+
+    If the session has a network (from a previous connect_network call),
+    the new job will automatically be added as a member and connected
+    when the VM is allocated.
+
+    Use wait_for_ready on the job_id to wait for VM allocation and network setup."""
+
+    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 AddJobResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    session_id: str,
+    body: AddJobRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> AddJobResponse:
+    """Add a new job to an existing session.
+
+    The new job will:
+    1. Become part of the session's job group
+    2. Be matched to an available VM via the resource matcher
+    3. Automatically join the session's WireGuard network if one exists
+
+    If the session has a network (from a previous connect_network call),
+    the new job will automatically be added as a member and connected
+    when the VM is allocated.
+
+    Use wait_for_ready on the job_id to wait for VM allocation and network setup."""
+
+    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 AddJobResponse.model_validate(response.json())

plato/_generated/models/__init__.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
-#   filename:  tmp3gp2q7hj.json
-#   timestamp: 2026-01-29T10:43:35+00:00
+#   filename:  tmpio7emwyr.json
+#   timestamp: 2026-01-29T20:21:27+00:00
 
 from __future__ import annotations
 
@@ -446,6 +446,57 @@ class BulkArchiveTestCasesRequest(BaseModel):
     test_case_ids: Annotated[list[int], Field(title="Test Case Ids")]
 
 
+class CacheClearAllResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    caches_cleared: Annotated[dict[str, int], Field(title="Caches Cleared")]
+    total_cleared: Annotated[int, Field(title="Total Cleared")]
+
+
+class CacheClearResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    cache_name: Annotated[str, Field(title="Cache Name")]
+    entries_cleared: Annotated[int, Field(title="Entries Cleared")]
+
+
+class CacheKeyDeleteResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    cache_name: Annotated[str, Field(title="Cache Name")]
+    key: Annotated[str, Field(title="Key")]
+    deleted: Annotated[bool, Field(title="Deleted")]
+
+
+class CacheName(Enum):
+    artifact = "artifact"
+    artifact_tag = "artifact_tag"
+    simulator = "simulator"
+    handler = "handler"
+    service_type = "service_type"
+    organization = "organization"
+    api_key = "api_key"
+    access = "access"
+
+
+class CacheStats(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    size: Annotated[int, Field(title="Size")]
+    ttl_seconds: Annotated[int, Field(title="Ttl Seconds")]
+
+
+class CacheStatsResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    caches: Annotated[dict[str, CacheStats], Field(title="Caches")]
+
+
 class Capability(Enum):
     forms_and_inputs = "forms_and_inputs"
     search_and_extract = "search_and_extract"
@@ -4087,6 +4138,20 @@ class ModelsScoringConfigJSONSchemaDraft7(BaseModel):
     """
 
 
+class AddJobResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    session_id: Annotated[str, Field(title="Session Id")]
+    """
+    Session ID the job was added to
+    """
+    env: EnvInfo
+    """
+    Created environment info
+    """
+
+
 class AddReviewRequest(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -4750,6 +4815,27 @@ class AppSchemasBuildModelsSimConfigDataset(BaseModel):
     """
 
 
+class AddJobRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    env: Annotated[
+        EnvFromSimulator | EnvFromArtifact | EnvFromResource,
+        Field(discriminator="type", title="Env"),
+    ]
+    """
+    Environment configuration for the new job
+    """
+    timeout: Annotated[int | None, Field(title="Timeout")] = 1800
+    """
+    VM timeout in seconds (default: 30 minutes)
+    """
+    heartbeat_timeout: Annotated[int | None, Field(title="Heartbeat Timeout")] = None
+    """
+    Per-VM heartbeat timeout. None=use default (300s), 0=disabled.
+    """
+
+
 class AnnotatorAssignmentListItem(BaseModel):
     model_config = ConfigDict(
         extra="allow",