wafer-cli 0.2.51__tar.gz → 0.2.52__tar.gz

{wafer_cli-0.2.51 → wafer_cli-0.2.52}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-cli
-Version: 0.2.51
+Version: 0.2.52
 Summary: CLI for running GPU workloads, managing remote workspaces, and evaluating/optimizing kernels
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{wafer_cli-0.2.51 → wafer_cli-0.2.52}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "wafer-cli"
-version = "0.2.51"
+version = "0.2.52"
 description = "CLI for running GPU workloads, managing remote workspaces, and evaluating/optimizing kernels"
 readme = "README.md"
 requires-python = ">=3.11"

{wafer_cli-0.2.51 → wafer_cli-0.2.52}/wafer/cli.py

@@ -1405,6 +1405,11 @@ def agent(  # noqa: PLR0913
         "--env",
         help="Environment name (coding, remote-target, runpod-sandbox, modal-sandbox). Uses --args for config.",
     ),
+    workspace: str | None = typer.Option(
+        None,
+        "--workspace",
+        help="Workspace ID (requires --env workspace)",
+    ),
 ) -> None:
     """AI assistant for GPU kernel development.
 
@@ -1580,6 +1585,8 @@ def _make_agent_alias(name: str, doc: str) -> None:
             corpus=corpus,
             no_sandbox=no_sandbox,
             no_proxy=no_proxy,
+            env=None,
+            workspace=None,
         )
 
     alias_cmd.__doc__ = doc
@@ -4905,9 +4912,12 @@ def targets_probe(
         # ProbeError already has actionable context
         typer.echo(f"\nError: {e}", err=True)
         raise typer.Exit(1) from None
-    except Exception as e:
-        # Unexpected errors - include type for debugging
-        typer.echo(f"\nUnexpected error probing target: {type(e).__name__}: {e}", err=True)
+    except BaseException as e:
+        # Unwrap ExceptionGroup (from Trio nurseries) to show the actual cause
+        cause = e
+        while isinstance(cause, BaseExceptionGroup) and len(cause.exceptions) == 1:
+            cause = cause.exceptions[0]
+        typer.echo(f"\nUnexpected error probing target: {type(cause).__name__}: {cause}", err=True)
         raise typer.Exit(1) from None
 
     # Display results

{wafer_cli-0.2.51 → wafer_cli-0.2.52}/wafer/cli_instructions.py

@@ -100,7 +100,11 @@ def _format_command_help(cmd_path: str, cmd: click.BaseCommand) -> str:
     return "\n".join(lines)
 
 
-def build_cli_instructions(bash_allowlist: list[str]) -> str:
+def build_cli_instructions(
+    bash_allowlist: list[str],
+    *,
+    has_local_gpu: bool = False,
+) -> str:
     """Generate CLI instruction text from --help for allowed wafer commands.
 
     Walks the typer/click command tree and extracts help text for each
@@ -110,6 +114,10 @@ def build_cli_instructions(bash_allowlist: list[str]) -> str:
     Args:
         bash_allowlist: List of allowed bash command prefixes.
             Example: ["wafer evaluate", "wafer nvidia ncu", "python"]
+        has_local_gpu: When True, the header tells the agent it has a local GPU
+            (e.g. CLI agents running on a GPU sandbox). When False (default),
+            it says there is no local GPU and the agent must use wafer CLI
+            for remote execution.
 
     Returns:
         Markdown-formatted CLI instructions, or empty string if no wafer
@@ -144,7 +152,12 @@ def build_cli_instructions(bash_allowlist: list[str]) -> str:
     if not sections and not template_help:
         return ""
 
-    header = "## Wafer CLI Commands\n\n"
+    if has_local_gpu:
+        gpu_line = "You have a local GPU. Use `--target local` with wafer CLI commands.\n"
+    else:
+        gpu_line = "You do not have a local GPU. Use the wafer CLI to run on remote GPU hardware.\n"
+
+    header = "## Wafer CLI Commands\n\n" + gpu_line
 
     result = header + "\n\n".join(sections)
     if template_help:

{wafer_cli-0.2.51 → wafer_cli-0.2.52}/wafer/evaluate.py

@@ -5073,7 +5073,6 @@ async def run_evaluate_kernelbench_local(
 
         ref_path = run_path / "reference.py"
         ref_path.write_text(args.reference.read_text())
-
         inputs_path = None
         if args.inputs:
             inputs_path = run_path / "custom_inputs.py"
@@ -5081,7 +5080,6 @@ async def run_evaluate_kernelbench_local(
 
         eval_script_path = run_path / "kernelbench_eval.py"
         eval_script_path.write_text(KERNELBENCH_EVAL_SCRIPT)
-
         defense_module_path = None
         if args.defensive:
             defense_src = (
@@ -5140,7 +5138,6 @@ async def run_evaluate_kernelbench_local(
             env["CUDA_VISIBLE_DEVICES"] = str(gpu_id)
 
         print(f"Running: {python_exe} kernelbench_eval.py ...")
-
         try:
             result = subprocess.run(
                 cmd_parts,
@@ -5339,6 +5336,9 @@ async def run_evaluate_kernelbench(args: KernelBenchEvaluateArgs) -> EvaluateRes
                 ),
             )
         return await run_evaluate_kernelbench_docker(args, target)
+    elif isinstance(target, LocalTarget):
+        # Local GPU - no SSH, runs directly as subprocess
+        return await run_evaluate_kernelbench_local(args, target)
     else:
         return EvaluateResult(
             success=False,
@@ -5349,7 +5349,7 @@ async def run_evaluate_kernelbench(args: KernelBenchEvaluateArgs) -> EvaluateRes
             total_tests=0,
             error_message=(
                 f"Target type '{type(target).__name__}' not yet supported for KernelBench format. "
-                "Use a DigitalOcean, RunPod, Baremetal, or VM target."
+                "Use a Local, DigitalOcean, RunPod, Baremetal, or VM target."
             ),
         )
 

{wafer_cli-0.2.51 → wafer_cli-0.2.52}/wafer/targets.py

@@ -57,7 +57,7 @@ def _parse_target(data: dict[str, Any]) -> TargetConfig:
     target_type = data.get("type")
     if not target_type:
         raise ValueError(
-            "Target must have 'type' field (baremetal, vm, modal, workspace, runpod, or digitalocean)"
+            "Target must have 'type' field (baremetal, vm, modal, workspace, runpod, digitalocean, or local)"
         )
 
     # Remove type field before passing to dataclass

{wafer_cli-0.2.51 → wafer_cli-0.2.52}/wafer_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-cli
-Version: 0.2.51
+Version: 0.2.52
 Summary: CLI for running GPU workloads, managing remote workspaces, and evaluating/optimizing kernels
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown