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

wafer/global_config.py

@@ -234,7 +234,20 @@ def get_supabase_anon_key() -> str:
 
     The anon key is public and used for client-side auth operations
     like token refresh.
+
+    If SUPABASE_URL is set via env var, infer the matching anon key
+    from the built-in environments. Otherwise, use the config file's environment.
     """
+    supabase_url = get_supabase_url()
+    
+    # If SUPABASE_URL was set via env var, find matching environment
+    if os.environ.get("SUPABASE_URL"):
+        # Check built-in environments to find matching Supabase URL
+        for env_name, env in BUILTIN_ENVIRONMENTS.items():
+            if env.supabase_url == supabase_url:
+                return env.supabase_anon_key
+    
+    # Otherwise, use config file's environment
     return load_global_config().get_api_environment().supabase_anon_key
 
 

wafer/kernel_scope.py

@@ -95,7 +95,7 @@ def analyze_command(
             if not api_url or not auth_headers:
                 raise RuntimeError(
                     "API authentication required for .co file analysis. "
-                    "Run 'wafer login' first."
+                    "Run 'wafer auth login' first."
                 )
             result = analyze_code_object(target_path, api_url, auth_headers)
         # ISA files - use kernel_index parameter

wafer/ncu_analyze.py

@@ -520,7 +520,7 @@ def _analyze_remote_api(
 
     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

wafer/nsys_analyze.py

@@ -844,7 +844,7 @@ def _analyze_remote_api(
 
     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

wafer/skills/wafer-guide/SKILL.md

@@ -16,7 +16,7 @@ Before using Wafer CLI commands, install the tool:
 uv tool install wafer-cli
 
 # Authenticate (one-time setup)
-wafer login
+wafer auth login
 
 ```
 
@@ -71,15 +71,31 @@ Test correctness and measure speedup against a reference:
 wafer evaluate make-template ./my-kernel
 # Creates: kernel.py, reference.py, test_cases.json
 
-# Run evaluation on a configured target
-wafer evaluate \
+# test_cases.json format:
+# [{"name": "small", "n": 1024, "seed": 42}, {"name": "large", "n": 1048576, "seed": 42}]
+# Each dict is passed as **kwargs to generate_input() in reference.py
+
+# Run correctness check (GPUMode functional format)
+wafer evaluate gpumode \
   --impl ./my-kernel/kernel.py \
   --reference ./my-kernel/reference.py \
   --test-cases ./my-kernel/test_cases.json \
   --target <target-name>
 
-# With profiling
-wafer evaluate ... --profile
+# Run correctness + benchmark (measures speedup vs reference)
+wafer evaluate gpumode \
+  --impl ./my-kernel/kernel.py \
+  --reference ./my-kernel/reference.py \
+  --test-cases ./my-kernel/test_cases.json \
+  --target <target-name> --benchmark
+
+# Run with defensive timing (detects evaluation hacking)
+wafer evaluate gpumode ... --benchmark --defensive
+
+# KernelBench format (ModelNew class)
+wafer evaluate kernelbench \
+  --impl my_kernel.py --reference problem.py \
+  --target <target-name> --stages all
 ```
 
 ### 4. AI-Assisted Optimization
@@ -126,4 +142,4 @@ wafer config targets init runpod       # RunPod cloud GPUs
 wafer config targets init digitalocean # DigitalOcean AMD GPUs
 ```
 
-Then use: `wafer evaluate --target <name> ...`
+Then use: `wafer evaluate gpumode --target <name> ...`

wafer/specs_cli.py

@@ -0,0 +1,157 @@
+"""CLI commands for wafer specs — TargetSpec TOML management.
+
+These are the local config commands (no API calls).
+Registered as: wafer specs list|show|add|remove|default|init
+"""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+import typer
+
+specs_app = typer.Typer(
+    help="""Manage GPU target specs (provisioning blueprints).
+
+Specs define how to access or provision GPUs. They are TOML files in ~/.wafer/specs/.
+
+  wafer specs list                        # List all specs
+  wafer specs show runpod-mi300x          # Show one spec
+  wafer specs add /path/to/spec.toml      # Add from file
+  wafer specs remove old-target           # Remove a spec
+  wafer specs default runpod-mi300x       # Set default
+
+To create a new spec interactively:
+  wafer config targets init ssh           # (legacy, still works)
+  wafer config targets init runpod
+"""
+)
+
+
+@specs_app.command("list")
+def specs_list() -> None:
+    """List all configured specs.
+
+    Example:
+        wafer specs list
+    """
+    from wafer_core.targets.spec_store import list_spec_names, load_spec
+
+    from .targets import get_default_target
+
+    names = list_spec_names()
+    default = get_default_target()
+
+    if not names:
+        typer.echo("No specs configured.")
+        typer.echo("Add one with: wafer specs add <path/to/spec.toml>")
+        typer.echo("Or interactively: wafer config targets init ssh")
+        return
+
+    typer.echo("Configured specs:")
+    for name in names:
+        marker = " (default)" if name == default else ""
+        try:
+            spec = load_spec(name)
+            type_name = type(spec).__name__.replace("Target", "")
+            typer.echo(f"  {name}{marker}  [{type_name}]  gpu={spec.gpu_type}")
+        except Exception as e:
+            typer.echo(f"  {name}{marker}  [error: {e}]")
+
+
+@specs_app.command("show")
+def specs_show(
+    name: str = typer.Argument(..., help="Spec name"),
+) -> None:
+    """Show details for a spec.
+
+    Example:
+        wafer specs show runpod-mi300x
+    """
+    from wafer_core.targets.spec_store import load_spec
+
+    from .targets import get_target_info
+
+    try:
+        spec = load_spec(name)
+    except FileNotFoundError:
+        typer.echo(f"Spec not found: {name}", err=True)
+        raise typer.Exit(1) from None
+
+    typer.echo(f"Spec: {name}")
+    for key, value in get_target_info(spec).items():
+        typer.echo(f"  {key}: {value}")
+
+
+@specs_app.command("add")
+def specs_add(
+    file_path: Path = typer.Argument(..., help="Path to TOML spec file"),
+) -> None:
+    """Add a spec from a TOML file.
+
+    Example:
+        wafer specs add ./my-target.toml
+    """
+    import tomllib
+
+    from wafer_core.targets.spec_store import parse_spec, save_spec
+
+    if not file_path.exists():
+        typer.echo(f"File not found: {file_path}", err=True)
+        raise typer.Exit(1) from None
+
+    try:
+        with open(file_path, "rb") as f:
+            data = tomllib.load(f)
+        spec = parse_spec(data)
+        save_spec(spec)
+        typer.echo(f"Added spec: {spec.name}")
+    except Exception as e:
+        typer.echo(f"Error: {e}", err=True)
+        raise typer.Exit(1) from None
+
+
+@specs_app.command("remove")
+def specs_remove(
+    name: str = typer.Argument(..., help="Spec name to remove"),
+    force: bool = typer.Option(False, "--force", "-f", help="Skip confirmation"),
+) -> None:
+    """Remove a spec.
+
+    Example:
+        wafer specs remove old-target
+    """
+    from wafer_core.targets.spec_store import remove_spec
+
+    if not force:
+        confirm = typer.confirm(f"Remove spec '{name}'?")
+        if not confirm:
+            return
+
+    try:
+        remove_spec(name)
+        typer.echo(f"Removed spec: {name}")
+    except FileNotFoundError:
+        typer.echo(f"Spec not found: {name}", err=True)
+        raise typer.Exit(1) from None
+
+
+@specs_app.command("default")
+def specs_default(
+    name: str = typer.Argument(..., help="Spec name to set as default"),
+) -> None:
+    """Set the default spec.
+
+    Example:
+        wafer specs default runpod-mi300x
+    """
+    from wafer_core.targets.spec_store import list_spec_names
+
+    from .targets import set_default_target
+
+    if name not in list_spec_names():
+        typer.echo(f"Spec not found: {name}", err=True)
+        raise typer.Exit(1) from None
+
+    set_default_target(name)
+    typer.echo(f"Default spec set to: {name}")

wafer/ssh_keys.py

@@ -1,6 +1,6 @@
 """SSH Keys CLI - Manage SSH public keys for workspace access.
 
-This module provides the implementation for the `wafer ssh-keys` subcommand.
+This module provides the implementation for the `wafer config ssh-keys` subcommand.
 Users register their SSH public keys here, which are then installed in all
 workspaces they attach to (BYOK - Bring Your Own Key model).
 """
@@ -94,7 +94,7 @@ def list_ssh_keys(json_output: bool = False) -> str:
             keys = 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
@@ -107,7 +107,7 @@ def list_ssh_keys(json_output: bool = False) -> str:
             "No SSH keys registered.\n"
             "\n"
             "Add your SSH key:\n"
-            "  wafer ssh-keys add\n"
+            "  wafer config ssh-keys add\n"
             "\n"
             "This will auto-detect your key from ~/.ssh/"
         )
@@ -149,7 +149,7 @@ def add_ssh_key(
                 "  ssh-keygen -t ed25519\n"
                 "\n"
                 "Or specify a path:\n"
-                "  wafer ssh-keys add /path/to/key.pub"
+                "  wafer config ssh-keys add /path/to/key.pub"
             )
         pubkey_path = detected[0]
 
@@ -202,7 +202,7 @@ def add_ssh_key(
             key_data = 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:
             # Parse error detail
             try:
@@ -248,7 +248,7 @@ def remove_ssh_key(key_id: str, json_output: bool = False) -> str:
             response.raise_for_status()
     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 == 404:
             raise RuntimeError(f"SSH key not found: {key_id}") from e
         raise RuntimeError(f"API error: {e.response.status_code} - {e.response.text}") from e