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

plato/chronos/api/runtimes/test_runtime.py

@@ -0,0 +1,67 @@
+"""Test Runtime"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+
+
+def _build_request_args(
+    artifact_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/runtimes/{artifact_id}/test"
+
+    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,
+    artifact_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Test a runtime by starting a session from it.
+
+    Creates a session from the runtime artifact, waits for it to be ready,
+    then returns the session info. The session stays open for manual testing."""
+
+    request_args = _build_request_args(
+        artifact_id=artifact_id,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return response.json()
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    artifact_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Test a runtime by starting a session from it.
+
+    Creates a session from the runtime artifact, waits for it to be ready,
+    then returns the session info. The session stays open for manual testing."""
+
+    request_args = _build_request_args(
+        artifact_id=artifact_id,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return response.json()

plato/chronos/api/secrets/__init__.py

@@ -0,0 +1,11 @@
+"""API endpoints."""
+
+from . import create_secret, delete_secret, get_secret, list_secrets, update_secret
+
+__all__ = [
+    "list_secrets",
+    "create_secret",
+    "get_secret",
+    "update_secret",
+    "delete_secret",
+]

plato/chronos/api/secrets/create_secret.py

@@ -0,0 +1,63 @@
+"""Create Secret"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import EnvSecretCreate, EnvSecretResponse
+
+
+def _build_request_args(
+    body: EnvSecretCreate,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/secrets"
+
+    headers: dict[str, str] = {}
+    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,
+    body: EnvSecretCreate,
+    x_api_key: str | None = None,
+) -> EnvSecretResponse:
+    """Create or update an environment secret."""
+
+    request_args = _build_request_args(
+        body=body,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return EnvSecretResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    body: EnvSecretCreate,
+    x_api_key: str | None = None,
+) -> EnvSecretResponse:
+    """Create or update an environment secret."""
+
+    request_args = _build_request_args(
+        body=body,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return EnvSecretResponse.model_validate(response.json())

plato/chronos/api/secrets/delete_secret.py

@@ -0,0 +1,61 @@
+"""Delete Secret"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+
+
+def _build_request_args(
+    name: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/secrets/{name}"
+
+    headers: dict[str, str] = {}
+    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,
+    name: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Delete an environment secret."""
+
+    request_args = _build_request_args(
+        name=name,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return response.json()
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    name: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Delete an environment secret."""
+
+    request_args = _build_request_args(
+        name=name,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return response.json()

plato/chronos/api/secrets/get_secret.py

@@ -0,0 +1,62 @@
+"""Get Secret"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import EnvSecretResponse
+
+
+def _build_request_args(
+    name: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/secrets/{name}"
+
+    headers: dict[str, str] = {}
+    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,
+    name: str,
+    x_api_key: str | None = None,
+) -> EnvSecretResponse:
+    """Get an environment secret by name."""
+
+    request_args = _build_request_args(
+        name=name,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return EnvSecretResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    name: str,
+    x_api_key: str | None = None,
+) -> EnvSecretResponse:
+    """Get an environment secret by name."""
+
+    request_args = _build_request_args(
+        name=name,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return EnvSecretResponse.model_validate(response.json())

plato/chronos/api/secrets/list_secrets.py

@@ -0,0 +1,57 @@
+"""List Secrets"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import EnvSecretListResponse
+
+
+def _build_request_args(
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/secrets"
+
+    headers: dict[str, str] = {}
+    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,
+    x_api_key: str | None = None,
+) -> EnvSecretListResponse:
+    """List all environment secrets for the org."""
+
+    request_args = _build_request_args(
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return EnvSecretListResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    x_api_key: str | None = None,
+) -> EnvSecretListResponse:
+    """List all environment secrets for the org."""
+
+    request_args = _build_request_args(
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return EnvSecretListResponse.model_validate(response.json())

plato/chronos/api/secrets/update_secret.py

@@ -0,0 +1,68 @@
+"""Update Secret"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import EnvSecretResponse, EnvSecretUpdate
+
+
+def _build_request_args(
+    name: str,
+    body: EnvSecretUpdate,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/secrets/{name}"
+
+    headers: dict[str, str] = {}
+    if x_api_key is not None:
+        headers["X-API-Key"] = x_api_key
+
+    return {
+        "method": "PUT",
+        "url": url,
+        "json": body.model_dump(mode="json", exclude_none=True),
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    name: str,
+    body: EnvSecretUpdate,
+    x_api_key: str | None = None,
+) -> EnvSecretResponse:
+    """Update an environment secret."""
+
+    request_args = _build_request_args(
+        name=name,
+        body=body,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return EnvSecretResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    name: str,
+    body: EnvSecretUpdate,
+    x_api_key: str | None = None,
+) -> EnvSecretResponse:
+    """Update an environment secret."""
+
+    request_args = _build_request_args(
+        name=name,
+        body=body,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return EnvSecretResponse.model_validate(response.json())

plato/chronos/api/sessions/__init__.py

@@ -0,0 +1,10 @@
+"""API endpoints."""
+
+from . import get_session, get_session_logs, get_session_logs_download, list_sessions
+
+__all__ = [
+    "list_sessions",
+    "get_session",
+    "get_session_logs",
+    "get_session_logs_download",
+]

plato/chronos/api/sessions/get_session.py

@@ -0,0 +1,62 @@
+"""Get Session"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import SessionResponse
+
+
+def _build_request_args(
+    public_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/sessions/{public_id}"
+
+    headers: dict[str, str] = {}
+    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,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> SessionResponse:
+    """Get a session by public ID."""
+
+    request_args = _build_request_args(
+        public_id=public_id,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return SessionResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> SessionResponse:
+    """Get a session by public ID."""
+
+    request_args = _build_request_args(
+        public_id=public_id,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return SessionResponse.model_validate(response.json())

plato/chronos/api/sessions/get_session_logs.py

@@ -0,0 +1,72 @@
+"""Get Session Logs"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import SessionLogsResponse
+
+
+def _build_request_args(
+    public_id: str,
+    limit: int | None = 100,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/sessions/{public_id}/logs"
+
+    params: dict[str, Any] = {}
+    if limit is not None:
+        params["limit"] = limit
+
+    headers: dict[str, str] = {}
+    if x_api_key is not None:
+        headers["X-API-Key"] = x_api_key
+
+    return {
+        "method": "GET",
+        "url": url,
+        "params": params,
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    public_id: str,
+    limit: int | None = 100,
+    x_api_key: str | None = None,
+) -> SessionLogsResponse:
+    """Get logs/events for a session."""
+
+    request_args = _build_request_args(
+        public_id=public_id,
+        limit=limit,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return SessionLogsResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    public_id: str,
+    limit: int | None = 100,
+    x_api_key: str | None = None,
+) -> SessionLogsResponse:
+    """Get logs/events for a session."""
+
+    request_args = _build_request_args(
+        public_id=public_id,
+        limit=limit,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return SessionLogsResponse.model_validate(response.json())

plato/chronos/api/sessions/get_session_logs_download.py

@@ -0,0 +1,62 @@
+"""Get Session Logs Download"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import LogsDownloadResponse
+
+
+def _build_request_args(
+    public_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/sessions/{public_id}/logs-download"
+
+    headers: dict[str, str] = {}
+    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,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> LogsDownloadResponse:
+    """Get a presigned download URL for the session's zipped logs."""
+
+    request_args = _build_request_args(
+        public_id=public_id,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return LogsDownloadResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> LogsDownloadResponse:
+    """Get a presigned download URL for the session's zipped logs."""
+
+    request_args = _build_request_args(
+        public_id=public_id,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return LogsDownloadResponse.model_validate(response.json())

plato/chronos/api/sessions/list_sessions.py

@@ -0,0 +1,57 @@
+"""List Sessions"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import SessionListResponse
+
+
+def _build_request_args(
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/sessions"
+
+    headers: dict[str, str] = {}
+    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,
+    x_api_key: str | None = None,
+) -> SessionListResponse:
+    """List all sessions for the org."""
+
+    request_args = _build_request_args(
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return SessionListResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    x_api_key: str | None = None,
+) -> SessionListResponse:
+    """List all sessions for the org."""
+
+    request_args = _build_request_args(
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return SessionListResponse.model_validate(response.json())

plato/chronos/api/status/__init__.py

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