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

plato/chronos/api/agents/list_agents.py

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

plato/chronos/api/agents/lookup_agent.py

@@ -0,0 +1,74 @@
+"""Lookup Agent"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import AgentLookupResponse
+
+
+def _build_request_args(
+    name: str,
+    version: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/agents/lookup"
+
+    params: dict[str, Any] = {}
+    if name is not None:
+        params["name"] = name
+    if version is not None:
+        params["version"] = version
+
+    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,
+    name: str,
+    version: str | None = None,
+    x_api_key: str | None = None,
+) -> AgentLookupResponse:
+    """Lookup an agent by name and optional version. Returns the artifact info."""
+
+    request_args = _build_request_args(
+        name=name,
+        version=version,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return AgentLookupResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    name: str,
+    version: str | None = None,
+    x_api_key: str | None = None,
+) -> AgentLookupResponse:
+    """Lookup an agent by name and optional version. Returns the artifact info."""
+
+    request_args = _build_request_args(
+        name=name,
+        version=version,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return AgentLookupResponse.model_validate(response.json())

plato/chronos/api/auth/__init__.py

@@ -0,0 +1,9 @@
+"""API endpoints."""
+
+from . import debug_auth_api_auth_debug_get, get_auth_status_api_auth_status_get, get_current_user_route_api_auth_me_get
+
+__all__ = [
+    "get_auth_status_api_auth_status_get",
+    "get_current_user_route_api_auth_me_get",
+    "debug_auth_api_auth_debug_get",
+]

plato/chronos/api/auth/debug_auth_api_auth_debug_get.py

@@ -0,0 +1,43 @@
+"""Debug Auth"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+
+
+def _build_request_args() -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/auth/debug"
+
+    return {
+        "method": "GET",
+        "url": url,
+    }
+
+
+def sync(
+    client: httpx.Client,
+) -> dict[str, Any]:
+    """Debug endpoint to see what auth info is being received."""
+
+    request_args = _build_request_args()
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return response.json()
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+) -> dict[str, Any]:
+    """Debug endpoint to see what auth info is being received."""
+
+    request_args = _build_request_args()
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return response.json()

plato/chronos/api/auth/get_auth_status_api_auth_status_get.py

@@ -0,0 +1,61 @@
+"""Get Auth Status"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import AuthStatusResponse
+
+
+def _build_request_args(
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/auth/status"
+
+    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,
+) -> AuthStatusResponse:
+    """Check authentication status.
+
+    Pass API key via X-API-Key header to authenticate."""
+
+    request_args = _build_request_args(
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return AuthStatusResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    x_api_key: str | None = None,
+) -> AuthStatusResponse:
+    """Check authentication status.
+
+    Pass API key via X-API-Key header to authenticate."""
+
+    request_args = _build_request_args(
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return AuthStatusResponse.model_validate(response.json())

plato/chronos/api/auth/get_current_user_route_api_auth_me_get.py

@@ -0,0 +1,60 @@
+"""Get Current User Route"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+
+
+def _build_request_args(
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/auth/me"
+
+    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,
+) -> dict[str, Any]:
+    """Get current authenticated user.
+
+    Requires X-API-Key header."""
+
+    request_args = _build_request_args(
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return response.json()
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Get current authenticated user.
+
+    Requires X-API-Key header."""
+
+    request_args = _build_request_args(
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return response.json()

plato/chronos/api/callback/__init__.py

@@ -0,0 +1,11 @@
+"""API endpoints."""
+
+from . import push_agent_logs, update_agent_status, upload_artifacts, upload_logs_zip, upload_trajectory
+
+__all__ = [
+    "push_agent_logs",
+    "update_agent_status",
+    "upload_trajectory",
+    "upload_logs_zip",
+    "upload_artifacts",
+]

plato/chronos/api/callback/push_agent_logs.py

@@ -0,0 +1,61 @@
+"""Push Agent Logs"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import AgentLogsRequest, AgentLogsResponse
+
+
+def _build_request_args(
+    body: AgentLogsRequest,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/callback/logs"
+
+    return {
+        "method": "POST",
+        "url": url,
+        "json": body.model_dump(mode="json", exclude_none=True),
+    }
+
+
+def sync(
+    client: httpx.Client,
+    body: AgentLogsRequest,
+) -> AgentLogsResponse:
+    """Receive logs from an agent running in a VM.
+
+    This endpoint acknowledges log receipt for real-time streaming.
+    Logs are NOT stored in the database - they're uploaded to S3 as a zip
+    at the end of the session via the /artifacts endpoint."""
+
+    request_args = _build_request_args(
+        body=body,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return AgentLogsResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    body: AgentLogsRequest,
+) -> AgentLogsResponse:
+    """Receive logs from an agent running in a VM.
+
+    This endpoint acknowledges log receipt for real-time streaming.
+    Logs are NOT stored in the database - they're uploaded to S3 as a zip
+    at the end of the session via the /artifacts endpoint."""
+
+    request_args = _build_request_args(
+        body=body,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return AgentLogsResponse.model_validate(response.json())

plato/chronos/api/callback/update_agent_status.py

@@ -0,0 +1,57 @@
+"""Update Agent Status"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import AgentLogsResponse, AgentStatusRequest
+
+
+def _build_request_args(
+    body: AgentStatusRequest,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/callback/status"
+
+    return {
+        "method": "POST",
+        "url": url,
+        "json": body.model_dump(mode="json", exclude_none=True),
+    }
+
+
+def sync(
+    client: httpx.Client,
+    body: AgentStatusRequest,
+) -> AgentLogsResponse:
+    """Update the status of a running session.
+
+    Called by agents to report completion or failure."""
+
+    request_args = _build_request_args(
+        body=body,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return AgentLogsResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    body: AgentStatusRequest,
+) -> AgentLogsResponse:
+    """Update the status of a running session.
+
+    Called by agents to report completion or failure."""
+
+    request_args = _build_request_args(
+        body=body,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return AgentLogsResponse.model_validate(response.json())

plato/chronos/api/callback/upload_artifacts.py

@@ -0,0 +1,59 @@
+"""Upload Artifacts Endpoint"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import ArtifactsUploadRequest, ArtifactsUploadResponse
+
+
+def _build_request_args(
+    body: ArtifactsUploadRequest,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/callback/artifacts"
+
+    return {
+        "method": "POST",
+        "url": url,
+        "json": body.model_dump(mode="json", exclude_none=True),
+    }
+
+
+def sync(
+    client: httpx.Client,
+    body: ArtifactsUploadRequest,
+) -> ArtifactsUploadResponse:
+    """Upload trajectory and/or logs for a session.
+
+    Convenience endpoint to upload both artifacts in one request.
+    Trajectory is stored in DB, logs are uploaded to S3."""
+
+    request_args = _build_request_args(
+        body=body,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return ArtifactsUploadResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    body: ArtifactsUploadRequest,
+) -> ArtifactsUploadResponse:
+    """Upload trajectory and/or logs for a session.
+
+    Convenience endpoint to upload both artifacts in one request.
+    Trajectory is stored in DB, logs are uploaded to S3."""
+
+    request_args = _build_request_args(
+        body=body,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return ArtifactsUploadResponse.model_validate(response.json())

plato/chronos/api/callback/upload_logs_zip.py

@@ -0,0 +1,57 @@
+"""Upload Logs Zip Endpoint"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import LogsUploadRequest, LogsUploadResponse
+
+
+def _build_request_args(
+    body: LogsUploadRequest,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/callback/logs-zip"
+
+    return {
+        "method": "POST",
+        "url": url,
+        "json": body.model_dump(mode="json", exclude_none=True),
+    }
+
+
+def sync(
+    client: httpx.Client,
+    body: LogsUploadRequest,
+) -> LogsUploadResponse:
+    """Upload zipped logs for a session.
+
+    Accepts base64-encoded zip file and uploads to S3."""
+
+    request_args = _build_request_args(
+        body=body,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return LogsUploadResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    body: LogsUploadRequest,
+) -> LogsUploadResponse:
+    """Upload zipped logs for a session.
+
+    Accepts base64-encoded zip file and uploads to S3."""
+
+    request_args = _build_request_args(
+        body=body,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return LogsUploadResponse.model_validate(response.json())

plato/chronos/api/callback/upload_trajectory.py

@@ -0,0 +1,57 @@
+"""Upload Trajectory Endpoint"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import TrajectoryUploadRequest, TrajectoryUploadResponse
+
+
+def _build_request_args(
+    body: TrajectoryUploadRequest,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/callback/trajectory"
+
+    return {
+        "method": "POST",
+        "url": url,
+        "json": body.model_dump(mode="json", exclude_none=True),
+    }
+
+
+def sync(
+    client: httpx.Client,
+    body: TrajectoryUploadRequest,
+) -> TrajectoryUploadResponse:
+    """Upload AITF trajectory for a session.
+
+    Accepts an AITF-formatted trajectory and stores it in the database."""
+
+    request_args = _build_request_args(
+        body=body,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return TrajectoryUploadResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    body: TrajectoryUploadRequest,
+) -> TrajectoryUploadResponse:
+    """Upload AITF trajectory for a session.
+
+    Accepts an AITF-formatted trajectory and stores it in the database."""
+
+    request_args = _build_request_args(
+        body=body,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return TrajectoryUploadResponse.model_validate(response.json())

plato/chronos/api/default/__init__.py

@@ -0,0 +1,7 @@
+"""API endpoints."""
+
+from . import health
+
+__all__ = [
+    "health",
+]

plato/chronos/api/default/health.py

@@ -0,0 +1,43 @@
+"""Health"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+
+
+def _build_request_args() -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/health"
+
+    return {
+        "method": "GET",
+        "url": url,
+    }
+
+
+def sync(
+    client: httpx.Client,
+) -> Any:
+    """Health check endpoint."""
+
+    request_args = _build_request_args()
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return response.json()
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+) -> Any:
+    """Health check endpoint."""
+
+    request_args = _build_request_args()
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return response.json()

plato/chronos/api/jobs/__init__.py

@@ -0,0 +1,7 @@
+"""API endpoints."""
+
+from . import launch_job
+
+__all__ = [
+    "launch_job",
+]