plato-sdk-v2 2.5.0__py3-none-any.whl → 2.6.0__py3-none-any.whl

plato/chronos/api/sessions/create_session.py

@@ -0,0 +1,69 @@
+"""Create Session"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import CreateSessionRequest, CreateSessionResponse
+
+
+def _build_request_args(
+    body: CreateSessionRequest,
+    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": "POST",
+        "url": url,
+        "json": body.model_dump(mode="json", exclude_none=True),
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    body: CreateSessionRequest,
+    x_api_key: str | None = None,
+) -> CreateSessionResponse:
+    """Create a new session for dev/testing.
+
+    This creates a chronos session record that can receive OTel traces.
+    Used by plato-world-runner dev mode to get a real session in chronos."""
+
+    request_args = _build_request_args(
+        body=body,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return CreateSessionResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    body: CreateSessionRequest,
+    x_api_key: str | None = None,
+) -> CreateSessionResponse:
+    """Create a new session for dev/testing.
+
+    This creates a chronos session record that can receive OTel traces.
+    Used by plato-world-runner dev mode to get a real session in chronos."""
+
+    request_args = _build_request_args(
+        body=body,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return CreateSessionResponse.model_validate(response.json())

plato/chronos/api/sessions/get_session.py

@@ -12,11 +12,16 @@ from plato.chronos.models import SessionResponse
 
 def _build_request_args(
     public_id: str,
+    reveal_secrets: bool | None = False,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
     """Build request arguments."""
     url = f"/api/sessions/{public_id}"
 
+    params: dict[str, Any] = {}
+    if reveal_secrets is not None:
+        params["reveal_secrets"] = reveal_secrets
+
     headers: dict[str, str] = {}
     if x_api_key is not None:
         headers["X-API-Key"] = x_api_key
@@ -24,6 +29,7 @@ def _build_request_args(
     return {
         "method": "GET",
         "url": url,
+        "params": params,
         "headers": headers,
     }
 
@@ -31,12 +37,18 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     public_id: str,
+    reveal_secrets: bool | None = False,
     x_api_key: str | None = None,
 ) -> SessionResponse:
-    """Get a session by public ID."""
+    """Get a session by public ID.
+
+    Args:
+        public_id: Session public ID
+        reveal_secrets: If true, decrypt and return actual secret values"""
 
     request_args = _build_request_args(
         public_id=public_id,
+        reveal_secrets=reveal_secrets,
         x_api_key=x_api_key,
     )
 
@@ -48,12 +60,18 @@ def sync(
 async def asyncio(
     client: httpx.AsyncClient,
     public_id: str,
+    reveal_secrets: bool | None = False,
     x_api_key: str | None = None,
 ) -> SessionResponse:
-    """Get a session by public ID."""
+    """Get a session by public ID.
+
+    Args:
+        public_id: Session public ID
+        reveal_secrets: If true, decrypt and return actual secret values"""
 
     request_args = _build_request_args(
         public_id=public_id,
+        reveal_secrets=reveal_secrets,
         x_api_key=x_api_key,
     )
 

plato/chronos/api/sessions/get_session_bash_logs_download.py

@@ -0,0 +1,61 @@
+"""Get Session Bash Logs Download"""
+
+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/sessions/{public_id}/bash-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,
+) -> Any:
+    """Download the session's bash/execution logs directly."""
+
+    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,
+) -> Any:
+    """Download the session's bash/execution logs directly."""
+
+    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/sessions/get_session_envs.py

@@ -0,0 +1,62 @@
+"""Get Session Envs"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import SessionEnvsResponse
+
+
+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}/envs"
+
+    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,
+) -> SessionEnvsResponse:
+    """Get env info (job IDs, aliases) from Plato for a session."""
+
+    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 SessionEnvsResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> SessionEnvsResponse:
+    """Get env info (job IDs, aliases) from Plato for a session."""
+
+    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 SessionEnvsResponse.model_validate(response.json())

plato/chronos/api/sessions/get_session_live_logs.py

@@ -0,0 +1,62 @@
+"""Get Session Live Logs"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import LiveLogsResponse
+
+
+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}/live-logs"
+
+    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,
+) -> LiveLogsResponse:
+    """Get live execution logs from the VM via SSH."""
+
+    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 LiveLogsResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> LiveLogsResponse:
+    """Get live execution logs from the VM via SSH."""
+
+    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 LiveLogsResponse.model_validate(response.json())

plato/chronos/api/sessions/get_session_logs.py

@@ -12,7 +12,7 @@ from plato.chronos.models import SessionLogsResponse
 
 def _build_request_args(
     public_id: str,
-    limit: int | None = 100,
+    limit: int | None = 10000,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
     """Build request arguments."""
@@ -37,7 +37,7 @@ def _build_request_args(
 def sync(
     client: httpx.Client,
     public_id: str,
-    limit: int | None = 100,
+    limit: int | None = 10000,
     x_api_key: str | None = None,
 ) -> SessionLogsResponse:
     """Get logs/events for a session."""
@@ -56,7 +56,7 @@ def sync(
 async def asyncio(
     client: httpx.AsyncClient,
     public_id: str,
-    limit: int | None = 100,
+    limit: int | None = 10000,
     x_api_key: str | None = None,
 ) -> SessionLogsResponse:
     """Get logs/events for a session."""

plato/chronos/api/sessions/get_session_status.py

@@ -0,0 +1,62 @@
+"""Get Session Status"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import SessionStatusResponse
+
+
+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}/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,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> SessionStatusResponse:
+    """Get just the status of a session (lightweight endpoint)."""
+
+    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 SessionStatusResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    public_id: str,
+    x_api_key: str | None = None,
+) -> SessionStatusResponse:
+    """Get just the status of a session (lightweight endpoint)."""
+
+    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 SessionStatusResponse.model_validate(response.json())

plato/chronos/api/sessions/list_sessions.py

@@ -11,11 +11,16 @@ from plato.chronos.models import SessionListResponse
 
 
 def _build_request_args(
+    tag: str | None = None,
     x_api_key: str | None = None,
 ) -> dict[str, Any]:
     """Build request arguments."""
     url = "/api/sessions"
 
+    params: dict[str, Any] = {}
+    if tag is not None:
+        params["tag"] = tag
+
     headers: dict[str, str] = {}
     if x_api_key is not None:
         headers["X-API-Key"] = x_api_key
@@ -23,17 +28,24 @@ def _build_request_args(
     return {
         "method": "GET",
         "url": url,
+        "params": params,
         "headers": headers,
     }
 
 
 def sync(
     client: httpx.Client,
+    tag: str | None = None,
     x_api_key: str | None = None,
 ) -> SessionListResponse:
-    """List all sessions for the org."""
+    """List all sessions for the org, optionally filtered by tag.
+
+    Tag filtering uses ltree for hierarchical matching:
+    - 'project' matches 'project', 'project.foo', 'project.foo.bar'
+    - 'project.foo' matches 'project.foo', 'project.foo.bar'"""
 
     request_args = _build_request_args(
+        tag=tag,
         x_api_key=x_api_key,
     )
 
@@ -44,11 +56,17 @@ def sync(
 
 async def asyncio(
     client: httpx.AsyncClient,
+    tag: str | None = None,
     x_api_key: str | None = None,
 ) -> SessionListResponse:
-    """List all sessions for the org."""
+    """List all sessions for the org, optionally filtered by tag.
+
+    Tag filtering uses ltree for hierarchical matching:
+    - 'project' matches 'project', 'project.foo', 'project.foo.bar'
+    - 'project.foo' matches 'project.foo', 'project.foo.bar'"""
 
     request_args = _build_request_args(
+        tag=tag,
         x_api_key=x_api_key,
     )
 

plato/chronos/api/sessions/list_tags.py

@@ -0,0 +1,80 @@
+"""List Tags"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import TagsListResponse
+
+
+def _build_request_args(
+    q: str | None = None,
+    limit: int | None = 10,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/sessions/tags"
+
+    params: dict[str, Any] = {}
+    if q is not None:
+        params["q"] = q
+    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,
+    q: str | None = None,
+    limit: int | None = 10,
+    x_api_key: str | None = None,
+) -> TagsListResponse:
+    """Search unique tags for the org with ltree hierarchical matching.
+
+    Uses ltree for hierarchical tags with '.' separator.
+    Search matches tags containing the query as a substring or ancestor/descendant."""
+
+    request_args = _build_request_args(
+        q=q,
+        limit=limit,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return TagsListResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    q: str | None = None,
+    limit: int | None = 10,
+    x_api_key: str | None = None,
+) -> TagsListResponse:
+    """Search unique tags for the org with ltree hierarchical matching.
+
+    Uses ltree for hierarchical tags with '.' separator.
+    Search matches tags containing the query as a substring or ancestor/descendant."""
+
+    request_args = _build_request_args(
+        q=q,
+        limit=limit,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return TagsListResponse.model_validate(response.json())

plato/chronos/api/sessions/update_session_tags.py

@@ -0,0 +1,68 @@
+"""Update Session Tags"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato.chronos.errors import raise_for_status
+from plato.chronos.models import SessionResponse, UpdateTagsRequest
+
+
+def _build_request_args(
+    public_id: str,
+    body: UpdateTagsRequest,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/sessions/{public_id}/tags"
+
+    headers: dict[str, str] = {}
+    if x_api_key is not None:
+        headers["X-API-Key"] = x_api_key
+
+    return {
+        "method": "PATCH",
+        "url": url,
+        "json": body.model_dump(mode="json", exclude_none=True),
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    public_id: str,
+    body: UpdateTagsRequest,
+    x_api_key: str | None = None,
+) -> SessionResponse:
+    """Update tags for a session."""
+
+    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 SessionResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    public_id: str,
+    body: UpdateTagsRequest,
+    x_api_key: str | None = None,
+) -> SessionResponse:
+    """Update tags for a session."""
+
+    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 SessionResponse.model_validate(response.json())