wafer-cli 0.2.14__py3-none-any.whl → 0.2.30__py3-none-any.whl

wafer/GUIDE.md

@@ -7,7 +7,7 @@ GPU development primitives for LLM agents.
 Run code on cloud GPUs instantly with workspaces:
 
 ```bash
-wafer login                              # One-time auth
+wafer auth login                         # One-time auth
 wafer workspaces create dev --gpu B200   # Create workspace (NVIDIA B200)
 wafer workspaces exec dev -- python -c "import torch; print(torch.cuda.get_device_name(0))"
 wafer workspaces sync dev ./my-project   # Sync files

wafer/agent_defaults.py

@@ -0,0 +1,42 @@
+"""Shared agent defaults for kernel optimization tasks.
+
+Single source of truth for bash allowlists and enabled tools used by both:
+- CLI templates (apps/wafer-cli/wafer/templates/optimize_kernelbench.py)
+- Eval configs (research/evals/optimize_kernelbench_eval/.../base_config.py)
+
+Import from here instead of defining your own copy.
+"""
+
+from __future__ import annotations
+
+# Tools available to the agent (coding environment tools)
+ENABLED_TOOLS: list[str] = ["read", "write", "edit", "glob", "grep", "bash"]
+
+# Bash commands allowed for kernel optimization agents.
+# Uses prefix matching — "wafer evaluate" also allows "wafer evaluate kernelbench".
+KERNELBENCH_BASH_ALLOWLIST: list[str] = [
+    # Kernel evaluation
+    "wafer evaluate",
+    # Profiling — AMD
+    "wafer amd rocprof-compute",
+    "wafer amd rocprof-sdk",
+    "wafer amd rocprof-systems",
+    # Profiling — NVIDIA
+    "wafer nvidia ncu",
+    "wafer nvidia nsys",
+    # Analysis
+    "wafer compiler-analyze",
+    # Sub-agents
+    "wafer agent -t ask-docs",
+    # General utilities
+    "python",
+    "python3",
+    "timeout",
+    "ls",
+    "cat",
+    "head",
+    "tail",
+    "wc",
+    "pwd",
+    "which",
+]

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/billing.py

@@ -1,6 +1,6 @@
 """Billing CLI - Manage credits and subscription.
 
-This module provides the implementation for the `wafer billing` subcommand.
+This module provides the implementation for the `wafer config billing` subcommand.
 """
 
 import json
@@ -126,7 +126,7 @@ def format_usage_text(usage: dict) -> str:
         lines.extend([
             "",
             "Upgrade to Pro for hardware counters and credit topups:",
-            "  wafer billing portal",
+            "  wafer config billing portal",
         ])
 
     return "\n".join(lines)
@@ -153,7 +153,7 @@ def get_usage(json_output: bool = False) -> str:
             usage = response.json()
     except httpx.HTTPStatusError as e:
         if e.response.status_code == 401:
-            raise RuntimeError("Not authenticated. Run: wafer login") from e
+            raise RuntimeError("Not authenticated. Run: wafer auth login") from e
         raise RuntimeError(f"API error: {e.response.status_code} - {e.response.text}") from e
     except httpx.RequestError as e:
         raise RuntimeError(f"Could not reach API: {e}") from e
@@ -188,7 +188,7 @@ def create_topup(amount_cents: int) -> dict:
             return response.json()
     except httpx.HTTPStatusError as e:
         if e.response.status_code == 401:
-            raise RuntimeError("Not authenticated. Run: wafer login") from e
+            raise RuntimeError("Not authenticated. Run: wafer auth login") from e
         if e.response.status_code == 400:
             # Invalid amount
             try:
@@ -200,7 +200,7 @@ def create_topup(amount_cents: int) -> dict:
             # Start tier or other restriction
             raise RuntimeError(
                 "Topup not available for your subscription tier.\n"
-                "Upgrade your subscription first: wafer billing portal"
+                "Upgrade your subscription first: wafer config billing portal"
             ) from e
         if e.response.status_code == 503:
             raise RuntimeError("Billing service temporarily unavailable. Please try again later.") from e
@@ -227,7 +227,7 @@ def get_portal_url() -> dict:
             return response.json()
     except httpx.HTTPStatusError as e:
         if e.response.status_code == 401:
-            raise RuntimeError("Not authenticated. Run: wafer login") from e
+            raise RuntimeError("Not authenticated. Run: wafer auth login") from e
         raise RuntimeError(f"API error: {e.response.status_code} - {e.response.text}") from e
     except httpx.RequestError as e:
         raise RuntimeError(f"Could not reach API: {e}") from e