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

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

@@ -0,0 +1,81 @@
+"""Add Ssh Key"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import AddSSHKeyRequest, AddSSHKeyResponse
+
+
+def _build_request_args(
+    session_id: str,
+    body: AddSSHKeyRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/sessions/{session_id}/add_ssh_key"
+
+    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: AddSSHKeyRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> AddSSHKeyResponse:
+    """Add an SSH public key to all VMs in a session.
+
+    This allows the user to SSH into VMs using their own key pair.
+    The key is added to the specified user's authorized_keys file."""
+
+    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 AddSSHKeyResponse.model_validate(response.json())
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    session_id: str,
+    body: AddSSHKeyRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> AddSSHKeyResponse:
+    """Add an SSH public key to all VMs in a session.
+
+    This allows the user to SSH into VMs using their own key pair.
+    The key is added to the specified user's authorized_keys file."""
+
+    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 AddSSHKeyResponse.model_validate(response.json())

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

@@ -0,0 +1,89 @@
+"""Connect Network"""
+
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+
+from plato._generated.errors import raise_for_status
+from plato._generated.models import ConnectNetworkRequest
+
+
+def _build_request_args(
+    session_id: str,
+    body: ConnectNetworkRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Build request arguments."""
+    url = f"/api/v2/sessions/{session_id}/connect_network"
+
+    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: ConnectNetworkRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Connect all jobs in a session to a WireGuard network.
+
+    Must be called after all jobs are ready (have worker assignments).
+    Pre-generates WireGuard keys and configures full mesh networking.
+
+    Args:
+        host_only: If True, force WireGuard to run on worker instead of in VM.
+                   Useful for VMs without WireGuard tools or for testing."""
+
+    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 response.json()
+
+
+async def asyncio(
+    client: httpx.AsyncClient,
+    session_id: str,
+    body: ConnectNetworkRequest,
+    authorization: str | None = None,
+    x_api_key: str | None = None,
+) -> dict[str, Any]:
+    """Connect all jobs in a session to a WireGuard network.
+
+    Must be called after all jobs are ready (have worker assignments).
+    Pre-generates WireGuard keys and configures full mesh networking.
+
+    Args:
+        host_only: If True, force WireGuard to run on worker instead of in VM.
+                   Useful for VMs without WireGuard tools or for testing."""
+
+    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 response.json()

plato/_generated/models/__init__.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
-#   filename:  tmpwzer8yl7.json
-#   timestamp: 2026-01-26T01:31:41+00:00
+#   filename:  tmph5p8qjzo.json
+#   timestamp: 2026-01-28T09:03:29+00:00
 
 from __future__ import annotations
 
@@ -18,6 +18,14 @@ class ActiveSessionResponse(BaseModel):
     status: Annotated[str | None, Field(title="Status")] = None
 
 
+class AddMemberRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    job_id: Annotated[str, Field(title="Job Id")]
+    alias: Annotated[str | None, Field(title="Alias")] = None
+
+
 class ReviewType(Enum):
     env = "env"
     data = "data"
@@ -40,6 +48,38 @@ class AddReviewRequest(BaseModel):
     comments: Annotated[str | None, Field(title="Comments")] = None
 
 
+class AddSSHKeyRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    public_key: Annotated[str, Field(title="Public Key")]
+    """
+    SSH public key in OpenSSH format
+    """
+    username: Annotated[str | None, Field(title="Username")] = "root"
+    """
+    Username to add the key for (default: root)
+    """
+
+
+class AddSSHKeyResult(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    success: Annotated[bool, Field(title="Success")]
+    """
+    Whether the key was added successfully
+    """
+    error: Annotated[str | None, Field(title="Error")] = None
+    """
+    Error message if the operation failed
+    """
+    output: Annotated[str | None, Field(title="Output")] = None
+    """
+    Command output (stdout)
+    """
+
+
 class AgentInfo(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -590,6 +630,13 @@ class CloseSessionResponse(BaseModel):
     """
 
 
+class ConnectNetworkRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    host_only: Annotated[bool | None, Field(title="Host Only")] = False
+
+
 class ConnectRoutingInfoResult(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -794,6 +841,25 @@ class CreateEnvRequest(BaseModel):
     timeout: Annotated[int | None, Field(title="Timeout")] = 1800
 
 
+class CreateNetworkRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    session_id: Annotated[str, Field(title="Session Id")]
+    name: Annotated[str | None, Field(title="Name")] = None
+
+
+class CreateNetworkResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    session_id: Annotated[str, Field(title="Session Id")]
+    name: Annotated[str | None, Field(title="Name")] = None
+    subnet: Annotated[str, Field(title="Subnet")]
+    dns_suffix: Annotated[str, Field(title="Dns Suffix")]
+    status: Annotated[str, Field(title="Status")]
+
+
 class CreateOrgApiKeyRequest(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -1536,6 +1602,10 @@ class JobInfo(BaseModel):
     """
     Job status
     """
+    status_reason: Annotated[str | None, Field(title="Status Reason")] = None
+    """
+    Job status reason
+    """
 
 
 class JobInfoResponse(BaseModel):
@@ -1686,6 +1756,31 @@ class NavigateStep(BaseModel):
     """
 
 
+class NetworkMemberResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    job_id: Annotated[str, Field(title="Job Id")]
+    alias: Annotated[str | None, Field(title="Alias")] = None
+    dns_name: Annotated[str | None, Field(title="Dns Name")] = None
+    wireguard_ip: Annotated[str | None, Field(title="Wireguard Ip")] = None
+    status: Annotated[str, Field(title="Status")]
+    joined_at: Annotated[str | None, Field(title="Joined At")] = None
+
+
+class NetworkResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    session_id: Annotated[str, Field(title="Session Id")]
+    name: Annotated[str | None, Field(title="Name")] = None
+    subnet: Annotated[str, Field(title="Subnet")]
+    dns_suffix: Annotated[str, Field(title="Dns Suffix")]
+    status: Annotated[str, Field(title="Status")]
+    member_count: Annotated[int, Field(title="Member Count")]
+    created_at: Annotated[str | None, Field(title="Created At")] = None
+
+
 class NodeBuildConfig(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -1698,13 +1793,11 @@ class NodeBuildConfig(BaseModel):
     """
     SSH public key for VM access. Derived from private key if not set.
     """
-    ecr_enabled: Annotated[bool | None, Field(title="Ecr Enabled")] = True
-    """
-    Whether to use ECR for base images. Disable for local builds.
-    """
-    default_image: Annotated[str | None, Field(title="Default Image")] = "plato-ubuntu-22.04"
+    default_image: Annotated[str | None, Field(title="Default Image")] = (
+        "383806609161.dkr.ecr.us-west-1.amazonaws.com/vm/rootfs/plato-ubuntu-22.04:1.1.0"
+    )
     """
-    Default base image name for VM builds (e.g., 'plato-ubuntu-22.04', 'dind').
+    Default VM image URL with tag (e.g., 'ecr-url/repo:tag').
     """
 
 
@@ -1726,20 +1819,6 @@ class NodeNatsConfig(BaseModel):
     subject_prefix: Annotated[str | None, Field(title="Subject Prefix")] = "firecracker"
 
 
-class NodeNebulaConfig(BaseModel):
-    model_config = ConfigDict(
-        extra="allow",
-    )
-    lighthouse_upstream: Annotated[str | None, Field(title="Lighthouse Upstream")] = "100.64.1.1:4242"
-    """
-    Nebula lighthouse upstream address (host:port) for mesh connectivity.
-    """
-    lighthouse_ip: Annotated[str | None, Field(title="Lighthouse Ip")] = "100.64.1.1"
-    """
-    Nebula lighthouse mesh IP address.
-    """
-
-
 class NodePolicyConfig(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -1798,6 +1877,16 @@ class NodeS3Config(BaseModel):
     """
 
 
+class NodeSessionNetworkConfig(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    enabled: Annotated[bool | None, Field(title="Enabled")] = False
+    """
+    Enable session networking (WireGuard-based VM-to-VM communication).
+    """
+
+
 class NodeSnapshotStoreConfig(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -1846,6 +1935,10 @@ class NodeSnapshotStoreConfig(BaseModel):
     """
     Local storage root for snapshot-store data.
     """
+    dedup_rootfs: Annotated[bool | None, Field(title="Dedup Rootfs")] = False
+    """
+    Enable rootfs-level dedup via snapshot-store FUSE mount instead of file copy.
+    """
 
 
 class NodeStatusResponse(BaseModel):
@@ -3087,6 +3180,16 @@ class UpdateConcurrencyLimitResponse(BaseModel):
     plan_type: Annotated[str, Field(title="Plan Type")]
 
 
+class UpdateMemberRequest(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    wireguard_ip: Annotated[str | None, Field(title="Wireguard Ip")] = None
+    wireguard_public_key: Annotated[str | None, Field(title="Wireguard Public Key")] = None
+    worker_endpoint: Annotated[str | None, Field(title="Worker Endpoint")] = None
+    status: Annotated[str | None, Field(title="Status")] = None
+
+
 class UpdateReleaseRequest(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -4047,6 +4150,24 @@ class ModelsScoringConfigJSONSchemaDraft7(BaseModel):
     """
 
 
+class AddSSHKeyResponse(BaseModel):
+    model_config = ConfigDict(
+        extra="allow",
+    )
+    success: Annotated[bool, Field(title="Success")]
+    """
+    Whether all keys were added successfully
+    """
+    session_id: Annotated[str, Field(title="Session Id")]
+    """
+    Session ID
+    """
+    results: Annotated[dict[str, AddSSHKeyResult], Field(title="Results")]
+    """
+    Results keyed by job_id
+    """
+
+
 class AnnotatorAssignmentListItem(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -4395,9 +4516,9 @@ class NodeConfig(BaseModel):
     """
     Build configuration including SSH keys for VMs
     """
-    nebula: NodeNebulaConfig | None = None
+    session_network: NodeSessionNetworkConfig | None = None
     """
-    Nebula mesh VPN configuration for lighthouse connectivity
+    Session network configuration for WireGuard-based VM-to-VM networking
     """
 
 

plato/v1/cli/agent.py

@@ -616,21 +616,17 @@ def agent_run(
     model: str = typer.Option(None, "--model", "-m", help="Model name (e.g., 'anthropic/claude-sonnet-4')"),
     dataset: str = typer.Option(None, "--dataset", "-d", help="Dataset to run on"),
 ):
-    """
-    Run an agent using Harbor's runner infrastructure.
-
-    This command wraps `harbor run` to execute agents. For full Harbor options,
-    use `harbor run` directly or pass additional arguments after --.
+    """Run an agent using Harbor's runner infrastructure.
 
-    Harbor agents: claude-code, openhands, codex, aider, gemini-cli, goose,
-    swe-agent, mini-swe-agent, cline-cli, cursor-cli, opencode, qwen-coder
+    Wraps `harbor run` to execute agents on a dataset. Supports Harbor built-in agents
+    (claude-code, openhands, codex, aider, etc.) and Plato custom agents (computer-use).
 
-    Plato agents: computer-use (requires pip install plato-agent-computer-use)
+    Options:
+        -a, --agent: Agent name to run. See 'plato agent list' for available agents.
+        -m, --model: Model name for the agent (e.g., 'anthropic/claude-sonnet-4')
+        -d, --dataset: Dataset to run on (e.g., 'swe-bench-lite', 'terminal-bench')
 
-    Examples:
-        plato agent run -a claude-code -m anthropic/claude-sonnet-4 -d swe-bench-lite
-        plato agent run -a openhands -m openai/gpt-4o -d terminal-bench
-        plato agent run -- -a claude-code -m anthropic/claude-sonnet-4 -d swe-bench-lite --limit 10
+    Additional arguments can be passed to Harbor after '--' separator.
     """
     # Check if harbor is installed
     if not shutil.which("harbor"):
@@ -683,8 +679,10 @@ def agent_run(
 
 @agent_app.command(name="list")
 def agent_list():
-    """
-    List all available agents (Harbor built-in + Plato custom).
+    """List all available agents.
+
+    Shows Harbor built-in agents (claude-code, openhands, etc.) and Plato custom agents
+    (computer-use) that can be used with 'plato agent run' and 'plato agent publish'.
     """
     console.print("[bold]Harbor Agents:[/bold]\n")
 
@@ -728,14 +726,13 @@ def agent_list():
 def agent_schema(
     agent_name: str = typer.Argument(..., help="Agent name to get schema for"),
 ):
-    """
-    Get the configuration schema for a Harbor agent.
+    """Get the configuration schema for a Harbor agent.
 
-    Shows the JSON schema defining the configuration options for the specified agent.
+    Shows the JSON schema defining configuration options for the specified agent.
+    The schema describes what fields are available when configuring the agent for runs.
 
-    Example:
-        plato agent schema claude-code
-        plato agent schema openhands
+    Arguments:
+        agent_name: Name of the agent (e.g., 'claude-code', 'openhands')
     """
     try:
         from plato.agents import AGENT_SCHEMAS, get_agent_schema
@@ -763,23 +760,19 @@ def agent_publish(
     all_agents: bool = typer.Option(False, "--all", "-a", help="Publish all agents in directory"),
     dry_run: bool = typer.Option(False, "--dry-run", help="Build without pushing to ECR"),
 ):
-    """
-    Build and publish an agent Docker image to ECR.
-
-    Can publish either:
-    - A custom agent from a directory (with Dockerfile + pyproject.toml)
-    - A Harbor built-in agent by name
-    - All agents in a directory with --all
-
-    Version is automatic:
-    - Custom agents: from pyproject.toml
-    - Harbor agents: from installed Harbor package version
-
-    Examples:
-        plato agent publish ./agents/my-agent      # Custom agent
-        plato agent publish claude-code            # Harbor built-in
-        plato agent publish --all ./agents         # All agents in directory
-        plato agent publish claude-code --dry-run  # Dry run
+    """Build and publish an agent Docker image to ECR.
+
+    Builds a Docker image for the agent and pushes it to the Plato ECR registry.
+    Version is determined automatically from pyproject.toml (custom agents) or the
+    installed Harbor package version (Harbor agents).
+
+    Arguments:
+        target: Path to custom agent directory with Dockerfile + pyproject.toml,
+            OR name of a Harbor built-in agent (e.g., 'claude-code')
+
+    Options:
+        -a, --all: Publish all agents found in the target directory
+        --dry-run: Build the Docker image without pushing to ECR
     """
 
     # Handle --all flag with directory
@@ -935,11 +928,10 @@ def _push_single_agent(pkg_path: Path, dry_run: bool) -> None:
 
 @agent_app.command(name="images")
 def agent_images():
-    """
-    List all published agent images for your organization.
+    """List all published agent images for your organization.
 
-    Example:
-        plato agent images
+    Queries the Plato API to show all agent Docker images that have been published
+    to your organization's ECR registry. Requires PLATO_API_KEY.
     """
     import httpx
 
@@ -977,11 +969,12 @@ def agent_images():
 def agent_versions(
     agent_name: str = typer.Argument(..., help="Agent name"),
 ):
-    """
-    List all published versions of an agent.
+    """List all published versions of an agent.
 
-    Example:
-        plato agent versions claude-code
+    Shows all available versions of the specified agent in your organization's registry.
+
+    Arguments:
+        agent_name: Name of the agent to list versions for
     """
     import httpx
 
@@ -1021,15 +1014,15 @@ def agent_versions(
 def agent_deploy(
     path: str = typer.Argument(".", help="Path to the agent package directory (default: current directory)"),
 ):
-    """
-    Deploy a Chronos agent package to AWS CodeArtifact.
+    """Deploy a Chronos agent package to AWS CodeArtifact.
+
+    Builds the Python package, discovers @ai agents from the codebase, and uploads
+    to CodeArtifact via the Plato API for use in Chronos jobs.
 
-    Builds the package, discovers @ai agents, and uploads to CodeArtifact
-    via the Plato API. Requires PLATO_API_KEY environment variable.
+    Arguments:
+        path: Path to the agent package directory with pyproject.toml (default: current directory)
 
-    Example:
-        plato agent deploy
-        plato agent deploy ./my-agent-package
+    Requires PLATO_API_KEY environment variable.
     """
     try:
         import tomli