wafer-cli 0.2.16__py3-none-any.whl → 0.2.18__py3-none-any.whl

wafer/auth.py

@@ -191,6 +191,10 @@ def get_valid_token() -> str | None:
         if e.response.status_code != 401:
             # Not an auth error, re-raise
             raise
+    except httpx.RequestError:
+        # Network error (timeout, connection refused, DNS failure, etc.)
+        # Cannot verify token - return None to trigger re-login prompt
+        return None
 
     # Token expired, try refresh
     if not creds.refresh_token:
@@ -203,6 +207,9 @@ def get_valid_token() -> str | None:
     except httpx.HTTPStatusError:
         # Refresh failed, need to re-login
         return None
+    except httpx.RequestError:
+        # Network error during refresh - return None to trigger re-login prompt
+        return None
 
 
 def _find_free_port() -> int:

wafer/evaluate.py

@@ -354,7 +354,8 @@ def _build_docker_pip_install_cmd(target: BaremetalTarget | VMTarget) -> str:
     )
 
     # Install uv (fast, reliable) - use pip3 for compatibility
-    commands.append("pip3 install uv")
+    # Use --break-system-packages for Python 3.12+ with PEP 668 externally managed environments
+    commands.append("pip3 install --break-system-packages uv")
 
     # Install torch with custom index if specified (like Modal's two-phase install)
     # Use --system --break-system-packages to install to container's Python

wafer/workspaces.py

@@ -13,7 +13,7 @@ import httpx
 from .api_client import get_api_url
 from .auth import get_auth_headers
 
-VALID_STATUSES = {"creating", "running"}
+VALID_STATUSES = {"creating", "running", "error"}
 
 
 def _get_client() -> tuple[str, dict[str, str]]:
@@ -211,17 +211,39 @@ def list_workspaces(json_output: bool = False) -> str:
     lines = ["Workspaces:", ""]
     for ws in workspaces:
         status = ws.get("status", "unknown")
-        status_icon = {"running": "●", "creating": "◐"}.get(status, "?")
+        status_icon = {"running": "●", "creating": "◐", "error": "✗"}.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')}")
-        if ws.get("ssh_host") and ws.get("ssh_port") and ws.get("ssh_user"):
+
+        if status == "error":
+            lines.append(
+                f"    Status: Provisioning failed. Delete and recreate: wafer workspaces delete {ws['name']}"
+            )
+        elif ws.get("ssh_host") and ws.get("ssh_port") and ws.get("ssh_user"):
+            ssh_line = f"    SSH: ssh -p {ws['ssh_port']} {ws['ssh_user']}@{ws['ssh_host']}"
+            if status == "creating":
+                ssh_line += " (finalizing...)"
+            lines.append(ssh_line)
+        elif status == "running":
             lines.append(
-                f"    SSH: ssh -p {ws['ssh_port']} {ws['ssh_user']}@{ws['ssh_host']}"
+                f"    Status: Running but SSH not ready. Try: wafer workspaces delete {ws['name']} && wafer workspaces create {ws['name']} --wait"
             )
         else:
-            lines.append("    SSH: Not ready (run: wafer workspaces ssh <name>)")
+            lines.append("    SSH: Not ready (workspace is still creating)")
         lines.append("")
 
+    # Add SSH tip for users with running workspaces
+    has_running_with_ssh = any(
+        ws.get("status") == "running" and ws.get("ssh_host")
+        for ws in workspaces
+    )
+    if has_running_with_ssh:
+        lines.append("Tip: SSH directly for interactive work. 'exec' is for quick commands only.")
+    
+    has_error = any(ws.get("status") == "error" for ws in workspaces)
+    if has_error:
+        lines.append("Note: Error workspaces are auto-cleaned after 12 hours.")
+
     return "\n".join(lines)
 
 
@@ -441,6 +463,12 @@ def sync_files(
         f"Workspace {workspace_id} has invalid status '{workspace_status}'. "
         f"Valid statuses: {VALID_STATUSES}"
     )
+    if workspace_status == "error":
+        raise RuntimeError(
+            f"Workspace provisioning failed. Delete and recreate:\n"
+            f"  wafer workspaces delete {workspace_id}\n"
+            f"  wafer workspaces create {ws.get('name', workspace_id)} --wait"
+        )
     if workspace_status != "running":
         raise RuntimeError(
             f"Workspace is {workspace_status}. Wait for it to be running before syncing."
@@ -448,9 +476,14 @@ def sync_files(
     ssh_host = ws.get("ssh_host")
     ssh_port = ws.get("ssh_port")
     ssh_user = ws.get("ssh_user")
-    assert ssh_host, "Workspace missing ssh_host"
+    if not ssh_host or not ssh_port or not ssh_user:
+        # Workspace is running but SSH credentials are missing - unusual state
+        raise RuntimeError(
+            f"Workspace is running but SSH not ready.\n"
+            f"  Delete and recreate: wafer workspaces delete {workspace_id}\n"
+            f"  Then: wafer workspaces create {ws.get('name', workspace_id)} --wait"
+        )
     assert isinstance(ssh_port, int) and ssh_port > 0, "Workspace missing valid ssh_port"
-    assert ssh_user, "Workspace missing ssh_user"
 
     # Build rsync command
     # -a: archive mode (preserves permissions, etc.)
@@ -607,16 +640,34 @@ def get_workspace(workspace_id: str, json_output: bool = False) -> str:
         f"  Last Used: {workspace.get('last_used_at', 'N/A')}",
     ]
 
-    if workspace.get("ssh_host"):
+    if status == "error":
+        lines.extend([
+            "",
+            "Provisioning failed. Delete and recreate:",
+            f"  wafer workspaces delete {workspace['name']}",
+            f"  wafer workspaces create {workspace['name']} --wait",
+            "",
+            "Note: Error workspaces are auto-cleaned after 12 hours.",
+        ])
+    elif workspace.get("ssh_host"):
         lines.extend([
             "",
             "SSH Info:",
             f"  Host: {workspace['ssh_host']}",
             f"  Port: {workspace.get('ssh_port', 22)}",
             f"  User: {workspace.get('ssh_user', 'root')}",
+            "",
+            "Tip: SSH directly for interactive work. 'exec' is for quick commands only.",
         ])
     elif status == "creating":
         lines.extend(["", "SSH: available once workspace is running"])
+    elif status == "running":
+        # Running but no SSH credentials - unusual state
+        lines.extend([
+            "",
+            "Status: Running but SSH not ready.",
+            f"  Delete and recreate: wafer workspaces delete {workspace['name']}",
+        ])
 
     return "\n".join(lines)
 

{wafer_cli-0.2.16.dist-info → wafer_cli-0.2.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-cli
-Version: 0.2.16
+Version: 0.2.18
 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.16.dist-info → wafer_cli-0.2.18.dist-info}/RECORD

@@ -2,13 +2,13 @@ wafer/GUIDE.md,sha256=G6P4aFZslEXiHmVjtTB3_OIpGK5d1tSiqxtawASVUZg,3588
 wafer/__init__.py,sha256=kBM_ONCpU6UUMBOH8Tmg4A88sNFnbaD59o61cJs-uYM,90
 wafer/analytics.py,sha256=qLY6Z16usVHFD8TCv7XBuz7l47vXVdXk-qhOzA-hW_8,8179
 wafer/api_client.py,sha256=i_Az2b2llC3DSW8yOL-BKqa7LSKuxOr8hSN40s-oQXY,6313
-wafer/auth.py,sha256=nneKUjGwb5ggJEHRdF_GlFkT1ZozHP4kGyuXjhZjtgM,13677
+wafer/auth.py,sha256=dwss_se5P-FFc9IN38q4kh_dBrA6k-CguDBkivgcdj0,14003
 wafer/autotuner.py,sha256=41WYP41pTDvMijv2h42vm89bcHtDMJXObDlWmn6xpFU,44416
 wafer/billing.py,sha256=jbLB2lI4_9f2KD8uEFDi_ixLlowe5hasC0TIZJyIXRg,7163
 wafer/cli.py,sha256=lBBTQCcmKREqZDOQh27qSq8i6NedjHW5oh1JiuT9aho,254241
 wafer/config.py,sha256=h5Eo9_yfWqWGoPNdVQikI9GoZVUeysunSYiixf1mKcw,3411
 wafer/corpus.py,sha256=x5aFhCsTSAtgzFG9AMFpqq92Ej63mXofL-vvvpjj1sM,12913
-wafer/evaluate.py,sha256=bLTfL7jAGQlfqLL39hSGSB7bnBp5THTCY7nl6giVMkQ,176005
+wafer/evaluate.py,sha256=s1NszUBtxdWRonbi8YR3XWfCiCjNm14g2Pp1lu4kmtY,176125
 wafer/global_config.py,sha256=fhaR_RU3ufMksDmOohH1OLeQ0JT0SDW1hEip_zaP75k,11345
 wafer/gpu_run.py,sha256=TwqXy72T7f2I7e6n5WWod3xgxCPnDhU0BgLsB4CUoQY,9716
 wafer/inference.py,sha256=tZCO5i05FKY27ewis3CSBHFBeFbXY3xwj0DSjdoMY9s,4314
@@ -27,15 +27,15 @@ wafer/targets.py,sha256=9r-iRWoKSH5cQl1LcamaX-T7cNVOg99ngIm_hlRk-qU,26922
 wafer/targets_ops.py,sha256=jN1oIBx0mutxRNE9xpIc7SaBxPkVmOyus2eqn0kEKNI,21475
 wafer/tracelens.py,sha256=g9ZIeFyNojZn4uTd3skPqIrRiL7aMJOz_-GOd3aiyy4,7998
 wafer/wevin_cli.py,sha256=vF3GNH-qWXO4hAlXaDg98VZpS4uFexVUp94BHsJjjMU,22179
-wafer/workspaces.py,sha256=XZvN-13oq40fkpoJTB2UWTG9KkD-eO47ptXK0FY6360,30083
+wafer/workspaces.py,sha256=iUdioK7kA3z_gOTMNVDn9Q87c6qpkdXF4bOhJWkUPg8,32375
 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/optimize_kernelbench.py,sha256=aoOA13zWEl89r6QW03xF9NKxQ7j4mWe9rwua6-mlr4Y,4780
 wafer/templates/trace_analyze.py,sha256=XE1VqzVkIUsZbXF8EzQdDYgg-AZEYAOFpr6B_vnRELc,2880
-wafer_cli-0.2.16.dist-info/METADATA,sha256=jpGUUbYuCdCN1yyQV6YU4lJnrLrt_ItFGLtHV3XgQe0,560
-wafer_cli-0.2.16.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-wafer_cli-0.2.16.dist-info/entry_points.txt,sha256=WqB7hB__WhtPY8y1cO2sZiUz7fCq6Ik-usAigpeFvWE,41
-wafer_cli-0.2.16.dist-info/top_level.txt,sha256=2MK1IVMWfpLL8BZCQ3E9aG6L6L666gSA_teYlwan4fs,6
-wafer_cli-0.2.16.dist-info/RECORD,,
+wafer_cli-0.2.18.dist-info/METADATA,sha256=reFoNYKvPiYo1PO25LUKissdgm-UsaIem763Pme_H0E,560
+wafer_cli-0.2.18.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+wafer_cli-0.2.18.dist-info/entry_points.txt,sha256=WqB7hB__WhtPY8y1cO2sZiUz7fCq6Ik-usAigpeFvWE,41
+wafer_cli-0.2.18.dist-info/top_level.txt,sha256=2MK1IVMWfpLL8BZCQ3E9aG6L6L666gSA_teYlwan4fs,6
+wafer_cli-0.2.18.dist-info/RECORD,,