wafer-cli 0.2.8__tar.gz → 0.2.10__tar.gz

{wafer_cli-0.2.8 → wafer_cli-0.2.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-cli
-Version: 0.2.8
+Version: 0.2.10
 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.8 → wafer_cli-0.2.10}/README.md

@@ -43,10 +43,16 @@ wafer remote-run --upload-dir ./my_code -- python3 train.py
 
 Create and manage persistent GPU environments.
 
+**Available GPUs:**
+
+- `MI300X` - AMD Instinct MI300X (192GB HBM3, ROCm)
+- `B200` - NVIDIA Blackwell B200 (180GB HBM3e, CUDA) - default
+
 ```bash
 wafer workspaces list
-wafer workspaces create my-workspace
-wafer workspaces attach <workspace-id>    # Get SSH credentials
+wafer workspaces create my-workspace --gpu B200 --wait   # NVIDIA B200
+wafer workspaces create amd-dev --gpu MI300X             # AMD MI300X
+wafer workspaces ssh <workspace-id>
 wafer workspaces delete <workspace-id>
 ```
 
@@ -197,6 +203,39 @@ Now you can tab-complete:
 
 ---
 
+## AI Assistant Skills
+
+Install the Wafer CLI skill to make wafer commands discoverable by your AI coding assistant:
+
+```bash
+# Install for all supported tools (Claude Code, Codex CLI, Cursor)
+wafer skill install
+
+# Install for a specific tool
+wafer skill install -t cursor    # Cursor
+wafer skill install -t claude    # Claude Code
+wafer skill install -t codex     # Codex CLI
+
+# Check installation status
+wafer skill status
+
+# Uninstall
+wafer skill uninstall
+```
+
+### Installing from GitHub (Cursor)
+
+You can also install the skill directly from GitHub in Cursor:
+
+1. Open Cursor Settings (Cmd+Shift+J / Ctrl+Shift+J)
+2. Navigate to **Rules** → **Add Rule** → **Remote Rule (Github)**
+3. Enter: `https://github.com/wafer-ai/skills`
+4. Cursor will automatically discover skills in `.cursor/skills/`
+
+The skill provides comprehensive guidance for GPU kernel development, including documentation lookup, trace analysis, kernel evaluation, and optimization workflows.
+
+---
+
 ## Requirements
 
 - Python 3.10+

{wafer_cli-0.2.8 → wafer_cli-0.2.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "wafer-cli"
-version = "0.2.8"
+version = "0.2.10"
 description = "CLI tool for running commands on remote GPUs and GPU kernel optimization agent"
 requires-python = ">=3.11"
 dependencies = [
@@ -78,6 +78,7 @@ ignore = [
 [tool.ruff.lint.per-file-ignores]
 "tests/**/*.py" = ["ANN201"]  # Don't require return type annotations in tests
 "wafer/evaluate.py" = ["PLR0915", "PLR1702", "E402"]  # complex deployment flows - TODO: refactor
+"wafer/output.py" = ["ANN401"]  # Output collector uses **kwargs for flexible event data
 
 [tool.ruff.lint.pylint]
 max-args = 7             # Max function arguments (Tiger Style: few parameters)

{wafer_cli-0.2.8 → wafer_cli-0.2.10}/tests/test_cli_coverage.py

@@ -61,7 +61,7 @@ def _cleanup_test_workspaces() -> None:
         return
     for ws in workspaces:
         if ws.get("name", "").startswith("test-"):
-            runner.invoke(app, ["workspaces", "delete", ws["id"]])
+            runner.invoke(app, ["workspaces", "delete", ws["id"], "-y"])
 
 
 @pytest.fixture(scope="session", autouse=True)
@@ -213,7 +213,7 @@ class TestWorkspacesCommands:
         finally:
             # Delete
             if ws_id:
-                result = runner.invoke(app, ["workspaces", "delete", ws_id])
+                result = runner.invoke(app, ["workspaces", "delete", ws_id, "-y"])
 
     def test_workspaces_create_pretty(self) -> None:
         """Create workspace with pretty output."""
@@ -228,53 +228,43 @@ class TestWorkspacesCommands:
                 return
 
             # Extract ID from output
-            # Output format: "Created workspace: <name> (<uuid>)"
+            # Output format: "Creating workspace: <name> (<uuid>)"
             match = re.search(r"\(([a-f0-9-]{36})\)", result.output)
             if match:
                 ws_id = match.group(1)
+            assert "Creating workspace:" in result.output
             assert ws_id is not None, f"Could not extract workspace ID from: {result.output}"
 
         finally:
             if ws_id:
-                runner.invoke(app, ["workspaces", "delete", ws_id])
+                runner.invoke(app, ["workspaces", "delete", ws_id, "-y"])
 
-    def test_workspaces_attach_json(self) -> None:
-        """Attach to workspace with JSON output."""
-        ws_name = f"test-attach-{os.getpid()}"
+    def test_workspaces_create_wait_json(self) -> None:
+        """Create workspace with --wait JSON output."""
+        ws_name = f"test-create-wait-{os.getpid()}"
         ws_id = None
 
         try:
-            # Create workspace first
-            result = runner.invoke(app, ["workspaces", "create", ws_name, "--json"])
+            result = runner.invoke(
+                app,
+                ["workspaces", "create", ws_name, "--wait", "--json"],
+            )
             if result.exit_code != 0:
                 if "auth" in result.output.lower() or "not authenticated" in result.output.lower():
                     pytest.skip("Not authenticated")
+                assert "error" in result.output.lower() or "no gpu" in result.output.lower()
                 return
 
             data = json.loads(result.stdout)
-            ws_id = data.get("id")
+            ws_id = data.get("workspace_id")
             assert ws_id is not None
-
-            # Attach with JSON (avoids file I/O side effects)
-            result = runner.invoke(app, ["workspaces", "attach", ws_id, "--json"])
-            if result.exit_code == 0:
-                attach_data = json.loads(result.stdout)
-                # Should have SSH connection info
-                assert "ssh_host" in attach_data
-                assert "ssh_port" in attach_data
-                assert "ssh_user" in attach_data
-                assert "private_key_pem" in attach_data
-            else:
-                # May fail with 503 if no GPU available - that's ok
-                assert (
-                    "503" in result.output
-                    or "No GPU" in result.output
-                    or "error" in result.output.lower()
-                )
+            assert "ssh_host" in data
+            assert "ssh_port" in data
+            assert "ssh_user" in data
 
         finally:
             if ws_id:
-                runner.invoke(app, ["workspaces", "delete", ws_id])
+                runner.invoke(app, ["workspaces", "delete", ws_id, "-y"])
 
     def test_workspaces_show_not_found(self) -> None:
         """Show workspace with invalid ID returns 404."""
@@ -288,10 +278,10 @@ class TestWorkspacesCommands:
                 or "auth" in result.output.lower()
             )
 
-    def test_workspaces_attach_not_found(self) -> None:
-        """Attach to workspace with invalid ID returns 404."""
+    def test_workspaces_ssh_not_found(self) -> None:
+        """SSH to workspace with invalid ID returns 404."""
         fake_id = "00000000-0000-0000-0000-000000000000"
-        result = runner.invoke(app, ["workspaces", "attach", fake_id])
+        result = runner.invoke(app, ["workspaces", "ssh", fake_id])
         if result.exit_code != 0:
             # Should be 404 or auth error
             assert (
@@ -300,6 +290,15 @@ class TestWorkspacesCommands:
                 or "auth" in result.output.lower()
             )
 
+    def test_workspaces_exec_option_order(self) -> None:
+        """Exec should reject options after workspace name."""
+        result = runner.invoke(
+            app,
+            ["workspaces", "exec", "dev", "--timeout", "10", "--", "echo", "hi"],
+        )
+        assert result.exit_code != 0
+        assert "options must come before the workspace name" in result.output.lower()
+
 
 class TestNsysAnalyzeCommand:
     """Test wafer nvidia nsys analyze command."""
@@ -333,7 +332,24 @@ class TestNsysAnalyzeCommand:
         result = runner.invoke(
             app, ["nvidia", "nsys", "analyze", str(real_nsys_file), "--remote", "--json"]
         )
-        assert result.exit_code == 0
+        if result.exit_code != 0:
+            combined = result.output.lower()
+            if "auth" in combined or "login" in combined or "401" in combined or "403" in combined:
+                pytest.skip("Not authenticated")
+            if (
+                "no gpu" in combined
+                or "no targets" in combined
+                or "unavailable" in combined
+                or "timed out" in combined
+                or "timeout" in combined
+                or "connect" in combined
+                or "service" in combined
+                or "billing" in combined
+                or "spend limit" in combined
+                or "402" in combined
+            ):
+                pytest.skip(f"Remote nsys unavailable: {result.output.strip()}")
+            assert result.exit_code == 0
         data = json.loads(result.stdout)
         assert "kernels" in data or "timeline" in data or "summary" in data
 
@@ -344,7 +360,24 @@ class TestNsysAnalyzeCommand:
             pytest.skip("No real .nsys-rep fixture available")
 
         result = runner.invoke(app, ["nvidia", "nsys", "analyze", str(real_nsys_file), "--remote"])
-        assert result.exit_code == 0
+        if result.exit_code != 0:
+            combined = result.output.lower()
+            if "auth" in combined or "login" in combined or "401" in combined or "403" in combined:
+                pytest.skip("Not authenticated")
+            if (
+                "no gpu" in combined
+                or "no targets" in combined
+                or "unavailable" in combined
+                or "timed out" in combined
+                or "timeout" in combined
+                or "connect" in combined
+                or "service" in combined
+                or "billing" in combined
+                or "spend limit" in combined
+                or "402" in combined
+            ):
+                pytest.skip(f"Remote nsys unavailable: {result.output.strip()}")
+            assert result.exit_code == 0
         assert "NSYS" in result.output or "Kernel" in result.output or "Profile" in result.output
 
     def test_nsys_analyze_local(self, real_nsys_file: Path | None) -> None:
@@ -627,3 +660,62 @@ class TestEvaluateCommand:
             or "error" in combined
             or result.exit_code in (0, 1)
         )
+
+
+class TestWorkspacesExecFlagPassthrough:
+    """Test that flags after -- are passed through to the command, not intercepted."""
+
+    def test_exec_passes_dash_i_flag(self) -> None:
+        """The -i flag should pass through, not be intercepted as --image."""
+        result = runner.invoke(app, [
+            "workspaces", "exec", "test-ws", "--", "echo", "-i", "test"
+        ])
+        # Should NOT fail with "no such option" or similar parse error
+        assert "no such option" not in result.output.lower()
+        assert "unrecognized" not in result.output.lower()
+        # May fail with workspace not found - that's fine, parsing succeeded
+
+    def test_exec_passes_dash_v_flag(self) -> None:
+        """The -v flag should pass through, not be intercepted as --verbose."""
+        result = runner.invoke(app, [
+            "workspaces", "exec", "test-ws", "--", "python", "-v", "-c", "print(1)"
+        ])
+        assert "no such option" not in result.output.lower()
+
+    def test_exec_passes_double_dash_help(self) -> None:
+        """--help after -- should pass to command, not show exec help."""
+        result = runner.invoke(app, [
+            "workspaces", "exec", "test-ws", "--", "python", "--help"
+        ])
+        # Should NOT show the exec command's help text
+        assert "Execute a command in workspace" not in result.output
+
+    def test_exec_passes_multiple_flags(self) -> None:
+        """Multiple flags like -i -v -n should all pass through."""
+        result = runner.invoke(app, [
+            "workspaces", "exec", "test-ws", "--", "grep", "-i", "-v", "-n", "pattern"
+        ])
+        assert "no such option" not in result.output.lower()
+
+    def test_exec_no_command_shows_error(self) -> None:
+        """Missing command after -- should show helpful error."""
+        result = runner.invoke(app, [
+            "workspaces", "exec", "test-ws", "--"
+        ])
+        assert result.exit_code != 0
+        assert "no command" in result.output.lower() or "error" in result.output.lower()
+
+    def test_exec_preserves_flag_order(self) -> None:
+        """Flags should be preserved in order they were given."""
+        # This tests the shlex.join behavior
+        result = runner.invoke(app, [
+            "workspaces", "exec", "test-ws", "--", "cmd", "-a", "1", "-b", "2"
+        ])
+        assert "no such option" not in result.output.lower()
+
+    def test_exec_with_equals_syntax(self) -> None:
+        """Flags with --flag=value syntax should pass through."""
+        result = runner.invoke(app, [
+            "workspaces", "exec", "test-ws", "--", "cmd", "--output=/tmp/out"
+        ])
+        assert "no such option" not in result.output.lower()

{wafer_cli-0.2.8 → wafer_cli-0.2.10}/tests/test_cli_parity_integration.py

@@ -77,7 +77,7 @@ class TestWorkspacesWorkflow:
         finally:
             # Cleanup
             if ws_id:
-                run_cli("workspaces", "delete", ws_id, check=False)
+                run_cli("workspaces", "delete", ws_id, "-y", check=False)
 
 
 @requires_api