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

plato/chronos/api/status/get_status_api_status_get.py

@@ -0,0 +1,44 @@
+"""Get Status"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import StatusResponse
+
+
+def _build_request_args() -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/status"
+
+    return {
+        "method": "GET",
+        "url": url,
+    }
+
+
+def sync(
+    client: httpx.Client,
+) -> StatusResponse:
+    """Get status of all connected services."""
+
+    request_args = _build_request_args()
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return StatusResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+) -> StatusResponse:
+    """Get status of all connected services."""
+
+    request_args = _build_request_args()
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return StatusResponse.model_validate(response.json())

plato/chronos/api/status/get_version_info_api_version_get.py

@@ -0,0 +1,44 @@
+"""Get Version Info"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import VersionResponse
+
+
+def _build_request_args() -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/version"
+
+    return {
+        "method": "GET",
+        "url": url,
+    }
+
+
+def sync(
+    client: httpx.Client,
+) -> VersionResponse:
+    """Get version information."""
+
+    request_args = _build_request_args()
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return VersionResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+) -> VersionResponse:
+    """Get version information."""
+
+    request_args = _build_request_args()
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return VersionResponse.model_validate(response.json())

plato/chronos/api/templates/__init__.py

@@ -0,0 +1,11 @@
+"""API endpoints."""
+
+from . import create_template, delete_template, get_template, list_templates, update_template
+
+__all__ = [
+    "list_templates",
+    "create_template",
+    "get_template",
+    "update_template",
+    "delete_template",
+]

plato/chronos/api/templates/create_template.py

@@ -0,0 +1,63 @@
+"""Create Template"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import JobTemplateCreate, JobTemplateResponse
+
+
+def _build_request_args(
+    body: JobTemplateCreate,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/templates"
+
+    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: JobTemplateCreate,
+    x_api_key: str | None = None,
+) -> JobTemplateResponse:
+    """Create a new job template."""
+
+    request_args = _build_request_args(
+        body=body,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return JobTemplateResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    body: JobTemplateCreate,
+    x_api_key: str | None = None,
+) -> JobTemplateResponse:
+    """Create a new job template."""
+
+    request_args = _build_request_args(
+        body=body,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return JobTemplateResponse.model_validate(response.json())

plato/chronos/api/templates/delete_template.py

@@ -0,0 +1,61 @@
+"""Delete Template"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+
+
+def _build_request_args(
+    public_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/templates/{public_id}"
+
+    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,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Delete a job template."""
+
+    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 response.json()
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Delete a job template."""
+
+    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 response.json()

plato/chronos/api/templates/get_template.py

@@ -0,0 +1,62 @@
+"""Get Template"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import JobTemplateResponse
+
+
+def _build_request_args(
+    public_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/templates/{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,
+) -> JobTemplateResponse:
+    """Get a job template 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 JobTemplateResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> JobTemplateResponse:
+    """Get a job template 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 JobTemplateResponse.model_validate(response.json())

plato/chronos/api/templates/list_templates.py

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

plato/chronos/api/templates/update_template.py

@@ -0,0 +1,68 @@
+"""Update Template"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import JobTemplateResponse, JobTemplateUpdate
+
+
+def _build_request_args(
+    public_id: str,
+    body: JobTemplateUpdate,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/templates/{public_id}"
+
+    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,
+    public_id: str,
+    body: JobTemplateUpdate,
+    x_api_key: str | None = None,
+) -> JobTemplateResponse:
+    """Update a job template."""
+
+    request_args = _build_request_args(
+        public_id=public_id,
+        body=body,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return JobTemplateResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    public_id: str,
+    body: JobTemplateUpdate,
+    x_api_key: str | None = None,
+) -> JobTemplateResponse:
+    """Update a job template."""
+
+    request_args = _build_request_args(
+        public_id=public_id,
+        body=body,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return JobTemplateResponse.model_validate(response.json())

plato/chronos/api/trajectories/__init__.py

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

plato/chronos/api/trajectories/get_trajectory.py

@@ -0,0 +1,62 @@
+"""Get Trajectory"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import TrajectoryResponse
+
+
+def _build_request_args(
+    public_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/trajectories/{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,
+) -> TrajectoryResponse:
+    """Get a trajectory 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 TrajectoryResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> TrajectoryResponse:
+    """Get a trajectory 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 TrajectoryResponse.model_validate(response.json())

plato/chronos/api/trajectories/list_trajectories.py

@@ -0,0 +1,62 @@
+"""List Trajectories For Session"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import TrajectoryListResponse
+
+
+def _build_request_args(
+    session_public_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/trajectories/session/{session_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,
+    session_public_id: str,
+    x_api_key: str | None = None,
+) -> TrajectoryListResponse:
+    """List all trajectories for a session."""
+
+    request_args = _build_request_args(
+        session_public_id=session_public_id,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return TrajectoryListResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    session_public_id: str,
+    x_api_key: str | None = None,
+) -> TrajectoryListResponse:
+    """List all trajectories for a session."""
+
+    request_args = _build_request_args(
+        session_public_id=session_public_id,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return TrajectoryListResponse.model_validate(response.json())

plato/chronos/api/worlds/__init__.py

@@ -0,0 +1,10 @@
+"""API endpoints."""
+
+from . import create_world, delete_world, get_world, list_worlds
+
+__all__ = [
+    "list_worlds",
+    "create_world",
+    "get_world",
+    "delete_world",
+]

plato/chronos/api/worlds/create_world.py

@@ -0,0 +1,63 @@
+"""Create World"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import WorldCreate, WorldResponse
+
+
+def _build_request_args(
+    body: WorldCreate,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/worlds"
+
+    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: WorldCreate,
+    x_api_key: str | None = None,
+) -> WorldResponse:
+    """Create a new world."""
+
+    request_args = _build_request_args(
+        body=body,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return WorldResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    body: WorldCreate,
+    x_api_key: str | None = None,
+) -> WorldResponse:
+    """Create a new world."""
+
+    request_args = _build_request_args(
+        body=body,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return WorldResponse.model_validate(response.json())

plato/chronos/api/worlds/delete_world.py

@@ -0,0 +1,61 @@
+"""Delete World"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+
+
+def _build_request_args(
+    public_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/worlds/{public_id}"
+
+    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,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Archive a world."""
+
+    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 response.json()
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Archive a world."""
+
+    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 response.json()