wafer-cli 0.2.11__py3-none-any.whl → 0.2.13__py3-none-any.whl

wafer/cli.py

@@ -54,6 +54,19 @@ _command_start_time: float | None = None
 _command_outcome: str = "failure"
 
 
+def _show_version() -> None:
+    """Show CLI version and environment, then exit."""
+    from .analytics import _get_cli_version
+    from .global_config import load_global_config
+    
+    version = _get_cli_version()
+    config = load_global_config()
+    environment = config.environment
+    
+    typer.echo(f"wafer-cli {version} ({environment})")
+    raise typer.Exit()
+
+
 def _get_command_path(ctx: typer.Context) -> tuple[str, str | None]:
     """Extract command and subcommand from Typer context.
 
@@ -84,9 +97,20 @@ def _mark_command_success() -> None:
     _command_outcome = "success"
 
 
-@app.callback()
-def main_callback(ctx: typer.Context) -> None:
+@app.callback(invoke_without_command=True)
+def main_callback(
+    ctx: typer.Context,
+    version: bool = typer.Option(
+        False,
+        "--version",
+        help="Show version and exit",
+    ),
+) -> None:
     """Initialize analytics and track command execution."""
+    if version:
+        _show_version()
+        return
+    
     global _command_start_time, _command_outcome
     _command_start_time = time.time()
     _command_outcome = "success"  # Default to success, mark failure on exceptions
@@ -4615,9 +4639,6 @@ def workspaces_ssh(
 ) -> None:
     """SSH into a workspace.
 
-    Uses workspace SSH credentials once the workspace is running.
-    If workspace is not specified, uses the default workspace.
-
     Examples:
         wafer workspaces ssh dev
         wafer workspaces ssh           # uses default workspace
@@ -4626,52 +4647,37 @@ def workspaces_ssh(
 
     from .workspaces import get_workspace_raw, resolve_workspace
 
-    # Resolve workspace
+    # Resolve workspace name/ID
     try:
         resolved_workspace = resolve_workspace(workspace)
     except RuntimeError as e:
         typer.echo(f"Error: {e}", err=True)
         raise typer.Exit(1) from None
 
-    typer.echo(f"Connecting to workspace: {resolved_workspace}...", err=True)
-
-    # Get SSH credentials from workspace
+    # Get workspace SSH credentials
     try:
         ws = get_workspace_raw(resolved_workspace)
     except RuntimeError as e:
         typer.echo(f"Error: {e}", err=True)
         raise typer.Exit(1) from None
 
-    from .workspaces import VALID_STATUSES
-    
-    workspace_status = ws.get("status")
-    assert workspace_status in VALID_STATUSES, (
-        f"Workspace {resolved_workspace} has invalid status '{workspace_status}'. "
-        f"Valid statuses: {VALID_STATUSES}"
-    )
+    ssh_host = ws.get("ssh_host")
+    ssh_port = ws.get("ssh_port")
+    ssh_user = ws.get("ssh_user")
     
-    if workspace_status != "running":
-        typer.echo(f"Error: Workspace is {workspace_status}. Wait for it to be running.", err=True)
-        raise typer.Exit(1)
-    if not ws.get("ssh_host") or not ws.get("ssh_port") or not ws.get("ssh_user"):
-        typer.echo("Error: SSH credentials not available yet.", err=True)
+    if not ssh_host or not ssh_port or not ssh_user:
+        typer.echo("Error: Workspace not ready. Wait a few seconds and retry.", err=True)
         raise typer.Exit(1)
 
-    # Build SSH args - key_path is None for BYOK model (uses default SSH key)
-    ssh_args = ["ssh"]
-    ssh_args.extend([
-        "-p",
-        str(ws.get("ssh_port")),
-        "-o",
-        "StrictHostKeyChecking=no",
-        "-o",
-        "UserKnownHostsFile=/dev/null",
-        f"{ws.get('ssh_user')}@{ws.get('ssh_host')}",
+    # Connect via SSH
+    os.execvp("ssh", [
+        "ssh",
+        "-p", str(ssh_port),
+        "-o", "StrictHostKeyChecking=no",
+        "-o", "UserKnownHostsFile=/dev/null",
+        f"{ssh_user}@{ssh_host}",
     ])
 
-    # Replace current process with SSH
-    os.execvp("ssh", ssh_args)
-
 
 @workspaces_app.command("sync")
 def workspaces_sync(

wafer/workspaces.py

@@ -214,11 +214,12 @@ def list_workspaces(json_output: bool = False) -> str:
         status_icon = {"running": "●", "creating": "◐"}.get(status, "?")
         lines.append(f"  {status_icon} {ws['name']} ({ws['id']})")
         lines.append(f"    GPU: {ws.get('gpu_type', 'N/A')} | Image: {ws.get('image', 'N/A')}")
-        lines.append(f"    Status: {status} | Created: {ws.get('created_at', 'N/A')}")
-        if status == "running" and ws.get("ssh_host") and ws.get("ssh_port") and ws.get("ssh_user"):
+        if ws.get("ssh_host") and ws.get("ssh_port") and ws.get("ssh_user"):
             lines.append(
                 f"    SSH: ssh -p {ws['ssh_port']} {ws['ssh_user']}@{ws['ssh_host']}"
             )
+        else:
+            lines.append("    SSH: Not ready (run: wafer workspaces ssh <name>)")
         lines.append("")
 
     return "\n".join(lines)

{wafer_cli-0.2.11.dist-info → wafer_cli-0.2.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-cli
-Version: 0.2.11
+Version: 0.2.13
 Summary: CLI tool for running commands on remote GPUs and GPU kernel optimization agent
 Requires-Python: >=3.11
 Requires-Dist: typer>=0.12.0

{wafer_cli-0.2.11.dist-info → wafer_cli-0.2.13.dist-info}/RECORD

@@ -5,7 +5,7 @@ wafer/api_client.py,sha256=i_Az2b2llC3DSW8yOL-BKqa7LSKuxOr8hSN40s-oQXY,6313
 wafer/auth.py,sha256=acBVOz-3la6avztDGjtLRopdjNRIqbrV4tRMM1FAmHI,13682
 wafer/autotuner.py,sha256=6gH0Ho7T58EFerMQcHQxshWe3DF4qU7fb5xthAh5SPM,44364
 wafer/billing.py,sha256=jbLB2lI4_9f2KD8uEFDi_ixLlowe5hasC0TIZJyIXRg,7163
-wafer/cli.py,sha256=c7IEKZ-Ge-w4LQ1GVAqvuKkodBYtT6NnxNeGf6me9pc,252787
+wafer/cli.py,sha256=PaWvp7h130Pct5U6Kx_9_zyy20GDPaIQzri2VIDtv5Q,252625
 wafer/config.py,sha256=h5Eo9_yfWqWGoPNdVQikI9GoZVUeysunSYiixf1mKcw,3411
 wafer/corpus.py,sha256=x5aFhCsTSAtgzFG9AMFpqq92Ej63mXofL-vvvpjj1sM,12913
 wafer/evaluate.py,sha256=Lf_H6Afrdo4k9JOnI27wchFwEteqy73gKt5gLQgaXSE,172671
@@ -27,14 +27,14 @@ wafer/targets.py,sha256=9r-iRWoKSH5cQl1LcamaX-T7cNVOg99ngIm_hlRk-qU,26922
 wafer/targets_ops.py,sha256=FJQhlQ4MfOMN5ZNaVfqUvrkRwGjOXI22cNTIEVSKeSE,21488
 wafer/tracelens.py,sha256=g9ZIeFyNojZn4uTd3skPqIrRiL7aMJOz_-GOd3aiyy4,7998
 wafer/wevin_cli.py,sha256=4cZ05GFCGBq11ekVQH_AgaqnITVq6IUfwwHo6CeHFN4,22179
-wafer/workspaces.py,sha256=kN8-XFffodFflI9cuIllqd6VQGFnlV1h-Z28oBh4Lms,30100
+wafer/workspaces.py,sha256=w6VHhCV189BdVZNLZ6obL71qlB5N4iLWX5fvyqFB87c,30085
 wafer/skills/wafer-guide/SKILL.md,sha256=KWetJw2TVTbz11_nzqazqOJWWRlbHRFShs4sOoreiWo,3255
 wafer/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wafer/templates/ask_docs.py,sha256=Lxs-faz9v5m4Qa4NjF2X_lE8KwM9ES9MNJkxo7ep56o,2256
 wafer/templates/optimize_kernel.py,sha256=u6AL7Q3uttqlnBLzcoFdsiPq5lV2TV3bgqwCYYlK9gk,2357
 wafer/templates/trace_analyze.py,sha256=XE1VqzVkIUsZbXF8EzQdDYgg-AZEYAOFpr6B_vnRELc,2880
-wafer_cli-0.2.11.dist-info/METADATA,sha256=JK2VwDV7PgVjmRA5utdsWOLsT1xY1CRbp8WkqpFO4cA,560
-wafer_cli-0.2.11.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-wafer_cli-0.2.11.dist-info/entry_points.txt,sha256=WqB7hB__WhtPY8y1cO2sZiUz7fCq6Ik-usAigpeFvWE,41
-wafer_cli-0.2.11.dist-info/top_level.txt,sha256=2MK1IVMWfpLL8BZCQ3E9aG6L6L666gSA_teYlwan4fs,6
-wafer_cli-0.2.11.dist-info/RECORD,,
+wafer_cli-0.2.13.dist-info/METADATA,sha256=wRWRnySLYOqvUTJT6NPPVBY5KBLL0CGquHK98fQ_Y_g,560
+wafer_cli-0.2.13.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+wafer_cli-0.2.13.dist-info/entry_points.txt,sha256=WqB7hB__WhtPY8y1cO2sZiUz7fCq6Ik-usAigpeFvWE,41
+wafer_cli-0.2.13.dist-info/top_level.txt,sha256=2MK1IVMWfpLL8BZCQ3E9aG6L6L666gSA_teYlwan4fs,6
+wafer_cli-0.2.13.dist-info/RECORD,,

{wafer_cli-0.2.11.dist-info → wafer_cli-0.2.13.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.10.1)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any