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

plato/_generated/__init__.py

@@ -1,4 +1,4 @@
-"""Plato API SDK - v0.37.2"""
+"""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/api/v2/sessions/connect_network.py

@@ -7,12 +7,10 @@ from typing import Any
 import httpx
 
 from plato._generated.errors import raise_for_status
-from plato._generated.models import ConnectNetworkRequest
 
 
 def _build_request_args(
     session_id: str,
-    body: ConnectNetworkRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
@@ -28,7 +26,6 @@ def _build_request_args(
     return {
         "method": "POST",
         "url": url,
-        "json": body.model_dump(mode="json", exclude_none=True),
         "headers": headers,
     }
 
@@ -36,22 +33,16 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     session_id: str,
-    body: ConnectNetworkRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
     """Connect all jobs in a session to a WireGuard network.
 
     Must be called after all jobs are ready (have worker assignments).
-    Pre-generates WireGuard keys and configures full mesh networking.
-
-    Args:
-        host_only: If True, force WireGuard to run on worker instead of in VM.
-                   Useful for VMs without WireGuard tools or for testing."""
+    Pre-generates WireGuard keys and configures full mesh networking."""
 
     request_args = _build_request_args(
         session_id=session_id,
-        body=body,
         authorization=authorization,
         x_api_key=x_api_key,
     )
@@ -64,22 +55,16 @@ def sync(
 async def asyncio(
     client: httpx.AsyncClient,
     session_id: str,
-    body: ConnectNetworkRequest,
     authorization: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
     """Connect all jobs in a session to a WireGuard network.
 
     Must be called after all jobs are ready (have worker assignments).
-    Pre-generates WireGuard keys and configures full mesh networking.
-
-    Args:
-        host_only: If True, force WireGuard to run on worker instead of in VM.
-                   Useful for VMs without WireGuard tools or for testing."""
+    Pre-generates WireGuard keys and configures full mesh networking."""
 
     request_args = _build_request_args(
         session_id=session_id,
-        body=body,
         authorization=authorization,
         x_api_key=x_api_key,
     )