plato-sdk-v2 2.6.2__py3-none-any.whl → 2.7.1__py3-none-any.whl

plato/_generated/__init__.py

@@ -1,4 +1,4 @@
-"""Plato API SDK - v0.42.0"""
+"""Plato API SDK - v0.37.2"""
 
 from . import api, errors, models
 from .client import AsyncClient, Client

plato/_generated/api/v2/__init__.py

@@ -1,6 +1,6 @@
 """API version module."""
 
-from . import agents, artifacts, chronos, chronos_packages, cluster, jobs, pypi, releases, sessions, user
+from . import agents, artifacts, chronos, chronos_packages, cluster, jobs, networks, pypi, releases, sessions, user
 
 __all__ = [
     "agents",
@@ -9,6 +9,7 @@ __all__ = [
     "chronos_packages",
     "cluster",
     "jobs",
+    "networks",
     "pypi",
     "releases",
     "sessions",

plato/_generated/api/v2/networks/__init__.py

@@ -0,0 +1,23 @@
+"""API endpoints."""
+
+from . import (
+    add_member,
+    create_network,
+    delete_network,
+    get_network,
+    list_members,
+    list_networks,
+    remove_member,
+    update_member,
+)
+
+__all__ = [
+    "list_networks",
+    "create_network",
+    "get_network",
+    "delete_network",
+    "list_members",
+    "add_member",
+    "update_member",
+    "remove_member",
+]

plato/_generated/api/v2/networks/add_member.py

@@ -0,0 +1,75 @@
+"""Add Member"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import AddMemberRequest, NetworkMemberResponse
+
+
+def _build_request_args(
+    session_id: str,
+    body: AddMemberRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/networks/{session_id}/members"
+
+    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: AddMemberRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> NetworkMemberResponse:
+    """Add a member (job/VM) to a network."""
+
+    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 NetworkMemberResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    session_id: str,
+    body: AddMemberRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> NetworkMemberResponse:
+    """Add a member (job/VM) to a network."""
+
+    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 NetworkMemberResponse.model_validate(response.json())

plato/_generated/api/v2/networks/create_network.py

@@ -0,0 +1,70 @@
+"""Create Network"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import CreateNetworkRequest, CreateNetworkResponse
+
+
+def _build_request_args(
+    body: CreateNetworkRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/v2/networks/"
+
+    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,
+    body: CreateNetworkRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> CreateNetworkResponse:
+    """Create a new isolated network for a session."""
+
+    request_args = _build_request_args(
+        body=body,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return CreateNetworkResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    body: CreateNetworkRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> CreateNetworkResponse:
+    """Create a new isolated network for a session."""
+
+    request_args = _build_request_args(
+        body=body,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return CreateNetworkResponse.model_validate(response.json())

plato/_generated/api/v2/networks/delete_network.py

@@ -0,0 +1,68 @@
+"""Delete Network"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+
+
+def _build_request_args(
+    session_id: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/networks/{session_id}"
+
+    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,
+    session_id: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Delete/terminate a network."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        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,
+    session_id: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Delete/terminate a network."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        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/networks/get_network.py

@@ -0,0 +1,69 @@
+"""Get Network"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import NetworkResponse
+
+
+def _build_request_args(
+    session_id: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/networks/{session_id}"
+
+    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,
+    session_id: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> NetworkResponse:
+    """Get network details by session ID."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return NetworkResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    session_id: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> NetworkResponse:
+    """Get network details by session ID."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return NetworkResponse.model_validate(response.json())

plato/_generated/api/v2/networks/list_members.py

@@ -0,0 +1,69 @@
+"""List Members"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import NetworkMemberResponse
+
+
+def _build_request_args(
+    session_id: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/networks/{session_id}/members"
+
+    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,
+    session_id: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> list[NetworkMemberResponse]:
+    """List all members in a network."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        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,
+    session_id: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> list[NetworkMemberResponse]:
+    """List all members in a network."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        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/networks/list_networks.py

@@ -0,0 +1,74 @@
+"""List Networks"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import NetworkResponse
+
+
+def _build_request_args(
+    session_id: str | None = None,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = "/api/v2/networks/"
+
+    params: dict[str, Any] = {}
+    if session_id is not None:
+        params["session_id"] = session_id
+
+    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,
+        "params": params,
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    session_id: str | None = None,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> list[NetworkResponse]:
+    """List all networks for the organization."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        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,
+    session_id: str | None = None,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> list[NetworkResponse]:
+    """List all networks for the organization."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        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/networks/remove_member.py

@@ -0,0 +1,73 @@
+"""Remove Member"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+
+
+def _build_request_args(
+    session_id: str,
+    job_id: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/networks/{session_id}/members/{job_id}"
+
+    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,
+    session_id: str,
+    job_id: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Remove a member from a network."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        job_id=job_id,
+        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,
+    session_id: str,
+    job_id: str,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Remove a member from a network."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        job_id=job_id,
+        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/networks/update_member.py

@@ -0,0 +1,80 @@
+"""Update Member"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import NetworkMemberResponse, UpdateMemberRequest
+
+
+def _build_request_args(
+    session_id: str,
+    job_id: str,
+    body: UpdateMemberRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/networks/{session_id}/members/{job_id}"
+
+    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": "PUT",
+        "url": url,
+        "json": body.model_dump(mode="json", exclude_none=True),
+        "headers": headers,
+    }
+
+
+def sync(
+    client: httpx.Client,
+    session_id: str,
+    job_id: str,
+    body: UpdateMemberRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> NetworkMemberResponse:
+    """Update a network member (e.g., after VM joins with WireGuard info)."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        job_id=job_id,
+        body=body,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = client.request(**request_args)
+    raise_for_status(response)
+    return NetworkMemberResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    session_id: str,
+    job_id: str,
+    body: UpdateMemberRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> NetworkMemberResponse:
+    """Update a network member (e.g., after VM joins with WireGuard info)."""
+
+    request_args = _build_request_args(
+        session_id=session_id,
+        job_id=job_id,
+        body=body,
+        authorization=authorization,
+        x_api_key=x_api_key,
+    )
+
+    response = await client.request(**request_args)
+    raise_for_status(response)
+    return NetworkMemberResponse.model_validate(response.json())

plato/_generated/api/v2/sessions/__init__.py

@@ -1,8 +1,10 @@
 """API endpoints."""
 
 from . import (
+    add_ssh_key,
     checkpoint,
     close,
+    connect_network,
     disk_snapshot,
     evaluate,
     execute,
@@ -34,6 +36,8 @@ __all__ = [
     "make",
     "reset",
     "heartbeat",
+    "connect_network",
+    "add_ssh_key",
     "close",
     "upload_session_documents",
     "log_job_mutation",