wafer-core 0.1.20__py3-none-any.whl → 0.1.22__py3-none-any.whl

wafer_core/auth.py

@@ -41,6 +41,18 @@ PROVIDERS = {
         "display_name": "Modal",
         "key_url": "https://modal.com/settings",
     },
+    "anthropic": {
+        "env_var": "ANTHROPIC_API_KEY",
+        "alt_env_var": "WAFER_ANTHROPIC_API_KEY",  # Check this first
+        "display_name": "Anthropic",
+        "key_url": "https://console.anthropic.com/settings/keys",
+    },
+    "openai": {
+        "env_var": "OPENAI_API_KEY",
+        "alt_env_var": "WAFER_OPENAI_KEY",  # Check this first
+        "display_name": "OpenAI",
+        "key_url": "https://platform.openai.com/api-keys",
+    },
 }
 
 
@@ -78,11 +90,12 @@ def get_api_key(provider: str) -> str | None:
     """Get API key for a provider.
 
     Checks in order:
-    1. Environment variable (e.g., WAFER_RUNPOD_API_KEY)
-    2. ~/.wafer/auth.json
+    1. Alt environment variable if defined (e.g., WAFER_ANTHROPIC_API_KEY)
+    2. Primary environment variable (e.g., ANTHROPIC_API_KEY)
+    3. ~/.wafer/auth.json
 
     Args:
-        provider: Provider name (runpod, digitalocean, modal)
+        provider: Provider name (runpod, digitalocean, modal, anthropic, openai)
 
     Returns:
         API key string or None if not found
@@ -92,7 +105,13 @@ def get_api_key(provider: str) -> str | None:
 
     config = PROVIDERS[provider]
 
-    # Check environment variable first
+    # Check alt environment variable first (e.g., WAFER_ANTHROPIC_API_KEY)
+    if "alt_env_var" in config:
+        alt_key = os.environ.get(config["alt_env_var"], "").strip()
+        if alt_key:
+            return alt_key
+
+    # Check primary environment variable
     env_key = os.environ.get(config["env_var"], "").strip()
     if env_key:
         return env_key
@@ -154,7 +173,7 @@ def get_auth_status(provider: str) -> AuthStatus:
     """Get authentication status for a provider.
 
     Args:
-        provider: Provider name (runpod, digitalocean, modal)
+        provider: Provider name (runpod, digitalocean, modal, anthropic, openai)
 
     Returns:
         AuthStatus with details about the auth state
@@ -164,7 +183,20 @@ def get_auth_status(provider: str) -> AuthStatus:
 
     config = PROVIDERS[provider]
 
-    # Check environment variable first
+    # Check alt environment variable first (e.g., WAFER_ANTHROPIC_API_KEY)
+    if "alt_env_var" in config:
+        alt_key = os.environ.get(config["alt_env_var"], "").strip()
+        if alt_key:
+            return AuthStatus(
+                provider=provider,
+                display_name=config["display_name"],
+                is_authenticated=True,
+                source="env",
+                key_preview=_format_key_preview(alt_key),
+                key_url=config["key_url"],
+            )
+
+    # Check primary environment variable
     env_key = os.environ.get(config["env_var"], "").strip()
     if env_key:
         return AuthStatus(

wafer_core/environments/coding.py

@@ -34,6 +34,7 @@ from wafer_core.tools import (
     GLOB_TOOL,
     GREP_TOOL,
     READ_TOOL,
+    SKILL_TOOL,
     WRITE_TOOL,
     ApprovalCallback,
     exec_bash,
@@ -41,6 +42,7 @@ from wafer_core.tools import (
     exec_glob,
     exec_grep,
     exec_read,
+    exec_skill,
     exec_write,
 )
 
@@ -61,6 +63,7 @@ ALL_TOOLS = {
     "glob": GLOB_TOOL,
     "grep": GREP_TOOL,
     "bash": BASH_TOOL,
+    "skill": SKILL_TOOL,
     # TODO(wafer-tool): "wafer": WAFER_TOOL,
 }
 
@@ -208,6 +211,7 @@ class CodingEnvironment:
                 self.bash_approval_callback,
                 self._sandbox_policy,
             ),
+            "skill": lambda tc: exec_skill(tc),
             # TODO(wafer-tool): "wafer": lambda tc: exec_wafer(
             #     tc, self.working_dir, self.enabled_tools, self.allow_spawn, cancel_scope
             # ),

wafer_core/rollouts/dtypes.py

@@ -1562,6 +1562,10 @@ class EvalConfig:
     resume_dir: Path | None = None
     report_batch_size: int = 1  # Write report after each sample for best recovery
 
+    # Custom metadata (flows to report.json for dashboard filtering)
+    # e.g., {"waferbench_category": "gemm", "github_runner": "elliot"}
+    metadata: dict[str, Any] | None = None
+
 
 # ── Session Types ──────────────────────────────────────────────────────────────
 # Types for persisting agent sessions (trajectories, config, environment state).

wafer_core/rollouts/environments/localfs.py

@@ -331,9 +331,9 @@ def generate_diff(old_content: str, new_content: str, context_lines: int = 3) ->
 
 # Tool preset configurations
 TOOL_PRESETS = {
-    "full": ["read", "write", "edit", "bash", "web_fetch"],
+    "full": ["read", "write", "edit", "bash", "web_fetch", "skill"],
     "readonly": ["read"],
-    "no-write": ["read", "edit", "bash", "web_fetch"],
+    "no-write": ["read", "edit", "bash", "web_fetch", "skill"],
 }
 
 
@@ -630,6 +630,24 @@ class LocalFilesystemEnvironment:
                     required=["url", "prompt"],
                 ),
             ),
+            # skill tool
+            Tool(
+                type="function",
+                function=ToolFunction(
+                    name="skill",
+                    description="Load a skill's full instructions. Skills provide domain-specific knowledge and workflows. Use this when you need detailed guidance for a task mentioned in your available skills.",
+                    parameters=ToolFunctionParameter(
+                        type="object",
+                        properties={
+                            "name": {
+                                "type": "string",
+                                "description": "Name of the skill to load (e.g., 'wafer-guide')",
+                            },
+                        },
+                    ),
+                    required=["name"],
+                ),
+            ),
         ]
 
     async def on_assistant_message(self, message: Message, state: AgentState) -> AgentState:
@@ -655,6 +673,8 @@ class LocalFilesystemEnvironment:
                 return await self._exec_bash(tool_call, current_state.session_id, cancel_scope)
             elif tool_call.name == "web_fetch":
                 return await self._exec_web_fetch(tool_call, current_state.session_id)
+            elif tool_call.name == "skill":
+                return await self._exec_skill(tool_call)
             else:
                 return ToolResult(
                     tool_call_id=tool_call.id,
@@ -1155,3 +1175,31 @@ class LocalFilesystemEnvironment:
             content=header + final_content,
             details={"output_file": output_file_path} if output_file_path else None,
         )
+
+    async def _exec_skill(self, tool_call: ToolCall) -> ToolResult:
+        """Load a skill's full instructions."""
+        from ..skills import load_skill
+
+        skill_name = tool_call.args["name"]
+        skill = load_skill(skill_name)
+
+        if skill is None:
+            # List available skills in error message
+            from ..skills import discover_skills
+
+            available = discover_skills()
+            available_names = [s.name for s in available]
+            return ToolResult(
+                tool_call_id=tool_call.id,
+                is_error=True,
+                content="",
+                error=f"Skill not found: {skill_name}. Available skills: {', '.join(available_names) or 'none'}",
+            )
+
+        # Return the full skill content
+        header = f"# Skill: {skill.name}\n\n"
+        return ToolResult(
+            tool_call_id=tool_call.id,
+            is_error=False,
+            content=header + skill.content,
+        )

wafer_core/rollouts/evaluation.py

@@ -642,6 +642,7 @@ class EvalReport:
     timestamp: str = field(default_factory=lambda: datetime.now().isoformat())
     git_info: dict[str, Any] = field(default_factory=_get_git_info)
     config_path: str | None = None  # Path to config file relative to repo root
+    metadata: dict[str, Any] | None = None  # Custom metadata (waferbench_category, github_runner, etc.)
 
     async def save(self, output_dir: Path) -> None:
         """Save evaluation results to directory."""
@@ -675,6 +676,8 @@ class EvalReport:
             "config_path": self.config_path,
             "sample_ids": [s.id for s in self.sample_results],
         }
+        if self.metadata:
+            summary["metadata"] = self.metadata
         # Sanitize API keys in the summary before saving
         summary = sanitize_api_keys(summary)
         report_file = output_dir / "report.json"
@@ -761,6 +764,9 @@ def _write_partial_report(
         "config_path": config.config_path,
     }
 
+    if config.metadata:
+        report["metadata"] = config.metadata
+
     if resume_from:
         report["resume_from"] = resume_from
 
@@ -1279,6 +1285,7 @@ async def evaluate(
             summary_metrics={},
             sample_results=[],
             config={"resumed_from": str(config.resume_dir)},
+            metadata=config.metadata,
         )
 
     if config.verbose:
@@ -1489,6 +1496,7 @@ async def evaluate(
             "evaluation_timestamp": datetime.now().isoformat(),
         },
         config_path=config.config_path,
+        metadata=config.metadata,
     )
 
     # Save if output directory specified
@@ -1546,7 +1554,7 @@ def compute_summary_metrics(results: list[Sample]) -> dict[str, float]:
             for m in r.score.metrics:
                 all_metric_names.add(m.name)
 
-    # Compute mean, min, max, std for each metric
+    # Compute mean, median, min, max, std for each metric
     for metric_name in all_metric_names:
         values = []
         for r in results:
@@ -1557,7 +1565,15 @@ def compute_summary_metrics(results: list[Sample]) -> dict[str, float]:
                         break
         if values:
             mean_val = sum(values) / len(values)
+            sorted_values = sorted(values)
+            n = len(sorted_values)
+            if n % 2 == 0:
+                median_val = (sorted_values[n // 2 - 1] + sorted_values[n // 2]) / 2
+            else:
+                median_val = sorted_values[n // 2]
+            
             summary[f"mean_{metric_name}"] = mean_val
+            summary[f"median_{metric_name}"] = median_val
             summary[f"min_{metric_name}"] = min(values)
             summary[f"max_{metric_name}"] = max(values)
             summary[f"std_{metric_name}"] = (

wafer_core/rollouts/prompt.py

@@ -232,7 +232,7 @@ Detailed docs: {docs}
 If asked about your capabilities, read these files."""
 
 
-def build_system_prompt(
+def build_system_prompt(  # noqa: PLR0913
     env_name: str,
     tools: list[Tool],
     cwd: Path | None = None,
@@ -240,6 +240,7 @@ def build_system_prompt(
     env_system_prompt: str | None = None,
     include_self_docs: bool = True,
     include_project_context: bool = True,
+    include_skills: bool = False,
 ) -> str:
     """Build complete system prompt with dynamic tool info.
 
@@ -251,6 +252,7 @@ def build_system_prompt(
         env_system_prompt: Environment-provided system prompt (from env.get_system_prompt())
         include_self_docs: Whether to include rollouts documentation paths
         include_project_context: Whether to load AGENTS.md/ROLLOUTS.md files
+        include_skills: Whether to discover and list available skills
     """
     # Assertions (Tiger Style: 2+ per function, split compound)
     assert env_name, "env_name required"
@@ -277,11 +279,19 @@ def build_system_prompt(
     if guidelines:
         sections.append("Guidelines:\n" + "\n".join(f"- {g}" for g in guidelines))
 
-    # 5. Self-documentation
+    # 5. Available skills (metadata only - agent loads full content via skill tool)
+    if include_skills:
+        from .skills import discover_skills, format_skill_metadata_for_prompt
+
+        skill_metadata = discover_skills()
+        if skill_metadata:
+            sections.append(format_skill_metadata_for_prompt(skill_metadata))
+
+    # 6. Self-documentation
     if include_self_docs:
         sections.append(build_self_doc_section())
 
-    # 6. Project context files (AGENTS.md, ROLLOUTS.md, etc.)
+    # 7. Project context files (AGENTS.md, ROLLOUTS.md, etc.)
     if include_project_context:
         context_files = load_project_context(working_dir)
         if context_files:
@@ -290,7 +300,7 @@ def build_system_prompt(
                 ctx_section += f"\n## {path}\n\n{content}\n"
             sections.append(ctx_section)
 
-    # 7. Runtime context
+    # 8. Runtime context
     now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     sections.append(f"Current time: {now}\nWorking directory: {working_dir}")
 

wafer_core/rollouts/skills.py

@@ -0,0 +1,176 @@
+"""Skill discovery and loading.
+
+Skills are documentation files that agents can load on demand.
+Format follows agentskills.io spec: SKILL.md with YAML frontmatter.
+
+Discovery order:
+1. ~/.wafer/skills/{name}/SKILL.md (user-installed)
+2. Bundled skills (wafer-cli package)
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from pathlib import Path
+
+from .paths import get_config_dir
+
+
+@dataclass(frozen=True)
+class SkillMetadata:
+    """Lightweight skill metadata for system prompt injection."""
+
+    name: str
+    description: str
+    path: Path  # Path to SKILL.md file
+
+
+@dataclass(frozen=True)
+class Skill:
+    """Full skill with content."""
+
+    name: str
+    description: str
+    content: str  # Full markdown content (without frontmatter)
+    path: Path
+
+
+def _parse_skill_file(path: Path) -> tuple[dict[str, str], str] | None:
+    """Parse SKILL.md file into (frontmatter, content).
+
+    Returns None if file doesn't exist or is malformed.
+    """
+    if not path.exists():
+        return None
+
+    try:
+        text = path.read_text()
+    except (OSError, PermissionError):
+        return None
+
+    # Parse YAML frontmatter (between --- markers)
+    if not text.startswith("---"):
+        return None
+
+    # Find closing ---
+    end_idx = text.find("---", 3)
+    if end_idx == -1:
+        return None
+
+    frontmatter_text = text[3:end_idx].strip()
+    content = text[end_idx + 3 :].strip()
+
+    # Parse YAML (simple key: value format, no dependencies)
+    frontmatter: dict[str, str] = {}
+    for raw_line in frontmatter_text.split("\n"):
+        stripped = raw_line.strip()
+        if not stripped or ":" not in stripped:
+            continue
+        key, _, value = stripped.partition(":")
+        frontmatter[key.strip()] = value.strip()
+
+    # Validate required fields
+    if "name" not in frontmatter or "description" not in frontmatter:
+        return None
+
+    return frontmatter, content
+
+
+def _get_bundled_skills_dir() -> Path | None:
+    """Get path to bundled skills in wafer-cli package."""
+    # Try to find wafer-cli's skills directory
+    try:
+        import wafer
+
+        wafer_cli_path = Path(wafer.__file__).parent
+        skills_dir = wafer_cli_path / "skills"
+        if skills_dir.exists():
+            return skills_dir
+    except ImportError:
+        pass
+
+    return None
+
+
+def discover_skills() -> list[SkillMetadata]:
+    """Discover all available skills.
+
+    Returns list of SkillMetadata (name + description only).
+    """
+    skills: dict[str, SkillMetadata] = {}
+
+    # 1. User-installed skills (~/.wafer/skills/)
+    user_skills_dir = get_config_dir() / "skills"
+    if user_skills_dir.exists():
+        for skill_dir in user_skills_dir.iterdir():
+            if not skill_dir.is_dir():
+                continue
+            skill_file = skill_dir / "SKILL.md"
+            parsed = _parse_skill_file(skill_file)
+            if parsed:
+                frontmatter, _ = parsed
+                skills[frontmatter["name"]] = SkillMetadata(
+                    name=frontmatter["name"],
+                    description=frontmatter["description"],
+                    path=skill_file,
+                )
+
+    # 2. Bundled skills (wafer-cli package)
+    bundled_dir = _get_bundled_skills_dir()
+    if bundled_dir:
+        for skill_dir in bundled_dir.iterdir():
+            if not skill_dir.is_dir():
+                continue
+            skill_file = skill_dir / "SKILL.md"
+            parsed = _parse_skill_file(skill_file)
+            if parsed:
+                frontmatter, _ = parsed
+                # User skills take precedence
+                if frontmatter["name"] not in skills:
+                    skills[frontmatter["name"]] = SkillMetadata(
+                        name=frontmatter["name"],
+                        description=frontmatter["description"],
+                        path=skill_file,
+                    )
+
+    return list(skills.values())
+
+
+def load_skill(name: str) -> Skill | None:
+    """Load a skill by name.
+
+    Returns full Skill with content, or None if not found.
+    """
+    # Find the skill
+    for metadata in discover_skills():
+        if metadata.name == name:
+            parsed = _parse_skill_file(metadata.path)
+            if parsed:
+                frontmatter, content = parsed
+                return Skill(
+                    name=frontmatter["name"],
+                    description=frontmatter["description"],
+                    content=content,
+                    path=metadata.path,
+                )
+    return None
+
+
+def format_skill_metadata_for_prompt(skills: list[SkillMetadata]) -> str:
+    """Format skill metadata for system prompt injection.
+
+    Returns a compact section listing available skills.
+    """
+    if not skills:
+        return ""
+
+    lines = ["## Available Skills", ""]
+    lines.append(
+        "You have access to the following skills. Use the `skill` tool to load full instructions when needed."
+    )
+    lines.append("")
+
+    for skill in skills:
+        lines.append(f"- **{skill.name}**: {skill.description}")
+
+    return "\n".join(lines)

wafer_core/rollouts/templates/base.py

@@ -222,6 +222,9 @@ class TemplateConfig:
     # Example: {"corpus": "./docs/", "format": "markdown"}
     defaults: dict[str, str] = field(default_factory=dict)
 
+    # Skill discovery - if True, discovers skills and adds skill tool
+    include_skills: bool = False
+
     def interpolate_prompt(self, args: dict[str, str] | None = None) -> str:
         """Interpolate template variables into the system prompt.
 

wafer_core/tools/__init__.py

@@ -49,6 +49,10 @@ from wafer_core.tools.rocprof_systems_tools import (
     exec_rocprof_systems_query,
     exec_rocprof_systems_sample,
 )
+from wafer_core.tools.skill_tool import (
+    SKILL_TOOL,
+    exec_skill,
+)
 from wafer_core.tools.tracelens_tools import (
     TRACELENS_COLLECTIVE_TOOL,
     TRACELENS_COMPARE_TOOL,
@@ -88,6 +92,9 @@ __all__ = [
     "BashPermissionResult",
     "check_bash_permissions",
     "exec_bash",
+    # Skill tool
+    "SKILL_TOOL",
+    "exec_skill",
     # Wafer tool
     "WAFER_TOOL",
     "WAFER_SUBCOMMANDS",

wafer_core/tools/skill_tool.py

@@ -0,0 +1,64 @@
+"""Skill tool.
+
+Loads skill content on demand from ~/.wafer/skills/ or bundled locations.
+"""
+
+from wafer_core.rollouts.dtypes import (
+    Tool,
+    ToolCall,
+    ToolFunction,
+    ToolFunctionParameter,
+    ToolResult,
+)
+
+# ── Tool Definition ──────────────────────────────────────────────────────────
+
+SKILL_TOOL = Tool(
+    type="function",
+    function=ToolFunction(
+        name="skill",
+        description="Load a skill's full instructions. Skills provide domain-specific knowledge and workflows. Use this when you need detailed guidance for a task mentioned in your available skills.",
+        parameters=ToolFunctionParameter(
+            type="object",
+            properties={
+                "name": {
+                    "type": "string",
+                    "description": "Name of the skill to load (e.g., 'wafer-guide')",
+                },
+            },
+        ),
+        required=["name"],
+    ),
+)
+
+
+# ── Pure Function Executor ───────────────────────────────────────────────────
+
+
+async def exec_skill(tool_call: ToolCall) -> ToolResult:
+    """Load a skill's full instructions.
+
+    Args:
+        tool_call: The tool call with skill name.
+    """
+    from wafer_core.rollouts.skills import discover_skills, load_skill
+
+    skill_name = tool_call.args["name"]
+    skill = load_skill(skill_name)
+
+    if skill is None:
+        available = discover_skills()
+        available_names = [s.name for s in available]
+        return ToolResult(
+            tool_call_id=tool_call.id,
+            is_error=True,
+            content="",
+            error=f"Skill not found: {skill_name}. Available skills: {', '.join(available_names) or 'none'}",
+        )
+
+    header = f"# Skill: {skill.name}\n\n"
+    return ToolResult(
+        tool_call_id=tool_call.id,
+        is_error=False,
+        content=header + skill.content,
+    )

wafer_core/utils/backend.py

@@ -33,6 +33,9 @@ def get_auth_token() -> str | None:
     Note:
         In local dev mode (localhost), no token is required.
         The API will use LOCAL_DEV_MODE to bypass auth.
+        
+        Callers (like wevin-extension) should pass WAFER_AUTH_TOKEN
+        as an environment variable when spawning Python processes.
     """
     return os.environ.get("WAFER_AUTH_TOKEN")
 

wafer_core/utils/kernel_utils/targets/config.py

@@ -61,7 +61,9 @@ class BaremetalTarget:
     ncu_available: bool = True  # Baremetal typically has NCU
 
     # Docker execution config (Modal-like). If docker_image is set, run in container.
-    docker_image: str | None = None  # Docker image to use (e.g., "nvcr.io/nvidia/cutlass:4.3-devel")
+    docker_image: str | None = (
+        None  # Docker image to use (e.g., "nvcr.io/nvidia/cutlass:4.3-devel")
+    )
     pip_packages: tuple[str, ...] = ()  # Packages to install via uv pip install
     torch_package: str | None = None  # Torch package spec (e.g., "torch>=2.8.0")
     torch_index_url: str | None = None  # Custom index for torch (e.g., PyTorch nightly)
@@ -69,7 +71,9 @@ class BaremetalTarget:
     def __post_init__(self) -> None:
         """Validate configuration."""
         assert len(self.gpu_ids) > 0, "Must specify at least one GPU ID"
-        assert ":" in self.ssh_target, f"ssh_target must include port (user@host:port), got: {self.ssh_target}"
+        assert ":" in self.ssh_target, (
+            f"ssh_target must include port (user@host:port), got: {self.ssh_target}"
+        )
         # If torch_index_url is set, torch_package must also be set
         if self.torch_index_url:
             assert self.torch_package, "torch_package must be set when torch_index_url is provided"
@@ -114,7 +118,9 @@ class VMTarget:
     ncu_available: bool = False  # VMs typically don't have NCU
 
     # Docker execution config (Modal-like). If docker_image is set, run in container.
-    docker_image: str | None = None  # Docker image to use (e.g., "nvcr.io/nvidia/pytorch:24.01-py3")
+    docker_image: str | None = (
+        None  # Docker image to use (e.g., "nvcr.io/nvidia/pytorch:24.01-py3")
+    )
     pip_packages: tuple[str, ...] = ()  # Packages to install via uv pip install
     torch_package: str | None = None  # Torch package spec (e.g., "torch>=2.8.0")
     torch_index_url: str | None = None  # Custom index for torch (e.g., PyTorch nightly)
@@ -122,7 +128,9 @@ class VMTarget:
     def __post_init__(self) -> None:
         """Validate configuration."""
         assert len(self.gpu_ids) > 0, "Must specify at least one GPU ID"
-        assert ":" in self.ssh_target, f"ssh_target must include port (user@host:port), got: {self.ssh_target}"
+        assert ":" in self.ssh_target, (
+            f"ssh_target must include port (user@host:port), got: {self.ssh_target}"
+        )
         # If torch_index_url is set, torch_package must also be set
         if self.torch_index_url:
             assert self.torch_package, "torch_package must be set when torch_index_url is provided"
@@ -325,12 +333,13 @@ class RunPodTarget:
 
         # Check for API key (env var or ~/.wafer/auth.json)
         api_key = get_api_key("runpod")
-        assert api_key, (
-            "RunPod API key not found.\n"
-            "Set WAFER_RUNPOD_API_KEY environment variable, or run:\n"
-            "  wafer auth login runpod\n"
-            "Get your API key from: https://runpod.io/console/user/settings"
-        )
+        if not api_key:
+            raise ValueError(
+                "RunPod API key not found.\n"
+                "Set WAFER_RUNPOD_API_KEY environment variable, or run:\n"
+                "  wafer auth login runpod\n"
+                "Get your API key from: https://runpod.io/console/user/settings"
+            )
 
 
 @dataclass(frozen=True)
@@ -370,7 +379,9 @@ class LocalTarget:
         """Validate configuration."""
         assert self.name, "name cannot be empty"
         assert len(self.gpu_ids) > 0, "Must specify at least one GPU ID"
-        assert self.vendor in ("nvidia", "amd"), f"vendor must be 'nvidia' or 'amd', got: {self.vendor}"
+        assert self.vendor in ("nvidia", "amd"), (
+            f"vendor must be 'nvidia' or 'amd', got: {self.vendor}"
+        )
 
 
 @dataclass(frozen=True)
@@ -443,16 +454,25 @@ class DigitalOceanTarget:
 
         # Check for API key (env var or ~/.wafer/auth.json)
         api_key = get_api_key("digitalocean")
-        assert api_key, (
-            "DigitalOcean API key not found.\n"
-            "Set WAFER_AMD_DIGITALOCEAN_API_KEY environment variable, or run:\n"
-            "  wafer auth login digitalocean\n"
-            "Get your API key from: https://cloud.digitalocean.com/account/api/tokens"
-        )
+        if not api_key:
+            raise ValueError(
+                "DigitalOcean API key not found.\n"
+                "Set WAFER_AMD_DIGITALOCEAN_API_KEY environment variable, or run:\n"
+                "  wafer auth login digitalocean\n"
+                "Get your API key from: https://cloud.digitalocean.com/account/api/tokens"
+            )
 
 
 # Union type for target configs
-TargetConfig = BaremetalTarget | VMTarget | ModalTarget | WorkspaceTarget | RunPodTarget | DigitalOceanTarget | LocalTarget
+TargetConfig = (
+    BaremetalTarget
+    | VMTarget
+    | ModalTarget
+    | WorkspaceTarget
+    | RunPodTarget
+    | DigitalOceanTarget
+    | LocalTarget
+)
 
 
 # Type guard functions for pattern matching
@@ -517,9 +537,9 @@ def target_to_deployment_config(target: TargetConfig, gpu_id: int) -> Deployment
     from wafer_core.utils.kernel_utils.deployment import DeploymentConfig
 
     # Type narrowing: Only SSH-based targets supported (not Modal)
-    assert not isinstance(
-        target, ModalTarget
-    ), f"target_to_deployment_config only supports SSH targets, got {type(target).__name__}"
+    assert not isinstance(target, ModalTarget), (
+        f"target_to_deployment_config only supports SSH targets, got {type(target).__name__}"
+    )
 
     return DeploymentConfig(
         ssh_target=target.ssh_target,

{wafer_core-0.1.20.dist-info → wafer_core-0.1.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wafer-core
-Version: 0.1.20
+Version: 0.1.22
 Summary: Core utilities and environments for Wafer GPU kernel optimization
 Requires-Python: >=3.10
 Requires-Dist: aiohttp>=3.9.0

{wafer_core-0.1.20.dist-info → wafer_core-0.1.22.dist-info}/RECORD

@@ -1,6 +1,6 @@
 wafer_core/__init__.py,sha256=syB2JjzvL91otODCKsGCgCPE6gVCINhTUv9j98k6tL0,3209
 wafer_core/async_ssh.py,sha256=ocw2Gh5p8ltKeoqG_q32DXOBfu5q-IE7jCnzMbQN9WI,28713
-wafer_core/auth.py,sha256=U1AeQE_Z5ev0ltXAEIh0STFIv46mie9G8Vy34oOyowE,5936
+wafer_core/auth.py,sha256=JpUkZ3bROIsgexayak5TLiGqUAR5kqGjekwqQRvIXH0,7235
 wafer_core/gpu.py,sha256=ENa92btjXsx6ldpoyKfRrAmfy-LHG2KpA5k7SWd6Q_s,28627
 wafer_core/gpu_detect.py,sha256=kpD8Q_G6GA9j-WnnnTNA3BBPulkGcWnZiogOmjKDao0,13650
 wafer_core/problem_config.py,sha256=8oqxL9-pvgzi8BtFxgDcqZW4e6DV2OCZOYkcPoyXrc8,10913
@@ -12,7 +12,7 @@ wafer_core/config/__init__.py,sha256=hKywfjA4YXd4lBeBFEcBoMwFoflPHJTiBnkTq7_JYOQ
 wafer_core/config/loader.py,sha256=k7JnILmO13TWUzIv9Lm8fvmj3UfYHZDgaFurjQ-GXpY,6623
 wafer_core/config/schema.py,sha256=2WhFlnG0VYYX4T-70BLeJK8Janvi4KEa8KKGZA7331w,3898
 wafer_core/environments/__init__.py,sha256=SIsResVtm22tr_d-oHPeeSxrkhFdmPOFico3DqtRqK8,238
-wafer_core/environments/coding.py,sha256=0TU1kggKH4fLaoeDOUKpWwBklDf5Q9dHTAV2b96GVnU,8313
+wafer_core/environments/coding.py,sha256=T-_JFU-n5OxPR8xAWp8qar4Y5xyC-TWTIBjRy4PDel8,8418
 wafer_core/environments/gpumode.py,sha256=8Da08nltvN_YloNyYI6-omN2D4n5C7aptKDCtUgT2bQ,17191
 wafer_core/lib/__init__.py,sha256=4-4p3mhwlquejWGglYXU8_nHdA0LoPaa_jGzcm13USA,1325
 wafer_core/lib/kernel_scope/__init__.py,sha256=WW2vu8jUlqOu-MCpgO40lIYacCA9N2u-uuECIs_JO2w,2817
@@ -336,9 +336,9 @@ wafer_core/rollouts/agents.py,sha256=Uv1kjYogUfdPl18YfkVxVqFTbmWfuJQrxem_iHTUgdw
 wafer_core/rollouts/cli.py,sha256=2NqgegKdlmxD0eJzGOMB5o_1Hb5t7O5JpP_32uvF2BE,80117
 wafer_core/rollouts/cli_agents.py,sha256=e4qqqYBzWLsbw8FsNnddGApWp_on9Cvzrfd1amiAyvI,20641
 wafer_core/rollouts/deploy.py,sha256=3t88fM_BMyAPkxIl8pS4r5ogHJvrlqWQDuIaltDZBRc,40924
-wafer_core/rollouts/dtypes.py,sha256=GXB3SaLzlSXs3Gxe6GLdU3_t4E1Pl7EmJamexRuaFDE,60962
+wafer_core/rollouts/dtypes.py,sha256=GUezPTzkd8E-nDlqdGE7idUthyZC-7jTrbpa4ye-v8k,61146
 wafer_core/rollouts/eval_helpers.py,sha256=OE7uQZRcbqQhpFqb4zOj8zafc9Gr6xZJpSrMvxXKVUw,1699
-wafer_core/rollouts/evaluation.py,sha256=Zu2hqkGWDjTRj9UxRVgNQ_FgEnFo6zKCw4ck2htvnpg,68483
+wafer_core/rollouts/evaluation.py,sha256=fk-pGZ5vpocVmw1iBbHtxMK0K6l8pYTLHCpDNvRY1Xo,69142
 wafer_core/rollouts/events.py,sha256=z85J8kq0LXPj5CiUk4RkiTQg--r9xiO7QeeJwkyUOto,7505
 wafer_core/rollouts/export.py,sha256=0CfdBB7Du4E3VekKEUcTwTEFS1bOMGZ9GbD5KU3CecQ,11583
 wafer_core/rollouts/feedback.py,sha256=mu17eQbAinXZWI3hMYLq-LyF4JAdH9SfNRWY-0S8jvQ,6769
@@ -350,11 +350,12 @@ wafer_core/rollouts/paths.py,sha256=9XtrA9ylhb5LttMFe2DE7X0IHeUMjuGUerII9OscYec,
 wafer_core/rollouts/pipeline.py,sha256=vlJTYE3ZX2XScpF9pmtv91K8Q0g8uLmcbI5jn6b5Hzg,15319
 wafer_core/rollouts/progress.py,sha256=szA9cvWT2xUxGVhF9BaAqJMmKDqMAUlxImxcOpcnqbY,29228
 wafer_core/rollouts/progress_display.py,sha256=it-IiI37k9whAuB6T_66GYgsZyidCq5x00URiOcxe2c,15769
-wafer_core/rollouts/prompt.py,sha256=KAFSS4HCtYtaPGGCFtC5VU7NrVmhEoQFNGWcA1HVW-Q,10566
+wafer_core/rollouts/prompt.py,sha256=EDmGb0rhWwke7tokIcO8dukc3q5c8x0n5Omi5CpAQmA,11022
 wafer_core/rollouts/providers.py,sha256=dcGJh1p30hstVbCDDtJ902lyafkg81DKjcOzb0uuKS0,1400
 wafer_core/rollouts/remote.py,sha256=cAYpRCONlsTeRxzLiegAUfjZWGtqBNwZTHehMhk5ldA,8816
 wafer_core/rollouts/scoring.py,sha256=qeIT8Z7pK51XRDmN2sGdg_hIPRabWqoQIYKsuytlvRo,8838
 wafer_core/rollouts/search.py,sha256=5BEDuw9FVbQhei3nvUXEVwBU5ouwgJE6ONhEqvU5Ldc,14696
+wafer_core/rollouts/skills.py,sha256=ATYoG02Cc6_VrtE415TnseBFJrKOMq27z-5YgBgPpZQ,5081
 wafer_core/rollouts/slice.py,sha256=darOZO53BuSPfvv_KjOSzulGVSWbL4OuoE3k6xXpBFg,20195
 wafer_core/rollouts/store.py,sha256=UDP9idDOEVs_0Pslx0K_Y8E1i-BeoqVSaxdQiaqtz1E,18051
 wafer_core/rollouts/transform_messages.py,sha256=yldzdLgugNYb5Zxju7myFBel1tmrHXx9M399ImqPLGI,20891
@@ -394,7 +395,7 @@ wafer_core/rollouts/environments/compose.py,sha256=DlJA_GdzByWjVvGeR4MrcQIB4ucV6
 wafer_core/rollouts/environments/cuda_grep.py,sha256=o4GPJcnKuB96KwE4UWkxXq5DdaKYMz-QizcZXleOKLs,22007
 wafer_core/rollouts/environments/git_worktree.py,sha256=f4S-OI-m6OEMyrEl3TfD4oYLXkkNgsiHuX6NOHCVfSQ,22397
 wafer_core/rollouts/environments/handoff.py,sha256=pvhcSDdltZ1zJ3Y_SAJlBmX6FhXjZLNYneXwjFqb9lE,16941
-wafer_core/rollouts/environments/localfs.py,sha256=fpk91XB9GNrtsHiFhSzvkfzRdEyqf8me_pR5PPxDmMg,43848
+wafer_core/rollouts/environments/localfs.py,sha256=xceepFc-eh9RqwTSJd_WiODVCs4Jm-d8rQm_MvutG0A,45766
 wafer_core/rollouts/environments/no_tools.py,sha256=FRsiMDzi8ma15xePTANteUxAOiOw0s90459XMvSq1_k,2970
 wafer_core/rollouts/environments/oracle.py,sha256=OtviDKed6DEIy67TdohqSGmvQT8XuuQLGnxJF9B7yrE,7563
 wafer_core/rollouts/environments/repl.py,sha256=DyvyqEbBBytliqZ2-uAJqm-F7gdROG5-9LDFVyCo6lo,36042
@@ -501,7 +502,7 @@ wafer_core/rollouts/providers/openai_completions.py,sha256=3vUA74qjrxG-aOjyngtnZ
 wafer_core/rollouts/providers/openai_responses.py,sha256=xlyeI9h7aZEbpFY_8_zQ6IIYbMeNgcsEVRi92PBEAWc,32470
 wafer_core/rollouts/providers/sglang.py,sha256=kahYlrFG008D4jRA-c6mylBsTe-qTryKMhsjUbidduU,35525
 wafer_core/rollouts/templates/__init__.py,sha256=8qANHtoWZe9zpAWufkXlo8tQ07_Lw-RhX-lm2i-0ORQ,976
-wafer_core/rollouts/templates/base.py,sha256=FP6zx43ba8rx2nTx8dK1qZu-8sbmude2lBlM0xjSyGY,6906
+wafer_core/rollouts/templates/base.py,sha256=aBojLjssj2YaPjgCZMvHYWxtGa83Y4uqt1Cxoy1J3v0,7010
 wafer_core/rollouts/templates/loader.py,sha256=eNRWP2zMTsygpewBhAO3Co0vLb4o4SwJo74Jp1DWS-0,4726
 wafer_core/rollouts/tests/test_slice.py,sha256=jUJFkTUzY6TNrimFH5kkfOavn3J5Gg-hsFWUOzs30lk,9148
 wafer_core/rollouts/tools/__init__.py,sha256=nLEJrlhT1anqInh6u4CeX8CCrjUJpAEMp0jx-G0V6gs,298
@@ -585,10 +586,11 @@ wafer_core/sessions/hooks.py,sha256=A-txm6ufnRGQCdtP3vwh7oEOdlLN9Tv0XsjORMihuAI,
 wafer_core/targets/__init__.py,sha256=sHndC7AAOaHXlrmDXFLB53a5Y8DBjuyqS6nwsO2nj-Y,1728
 wafer_core/targets/digitalocean.py,sha256=cvoYpYjtSyy5t2lQAPi7ERruuuibronah_ivOiduAHQ,16550
 wafer_core/targets/runpod.py,sha256=bYTLVRaASrewJLIcZRtPMfMU2en1McE6W5w-Edo_OPQ,15785
-wafer_core/tools/__init__.py,sha256=zdWVdQU8aP66HVoI1AhoUHJ9UKG95rT5MFufD6zV2Q0,3212
+wafer_core/tools/__init__.py,sha256=wBQD45GdSfkxcT6NHzIv0IMeXCc0enwwkpm3T_9j1X8,3341
 wafer_core/tools/bash_tool.py,sha256=daoKOVGSgL0x9X_3l8Apd6-wFH4VMXMGJwVemw2FIfc,16828
 wafer_core/tools/glob_tool.py,sha256=9X5PdOjQJj7kiVNqqCZC0-1LmnE6wHx3Zc9zfMjtXdc,3533
 wafer_core/tools/grep_tool.py,sha256=cStyDz-J47oDLLZCL83yOvYo8Ijv4qu3D372JKT_ptM,4580
+wafer_core/tools/skill_tool.py,sha256=JXsT5hBTUH5U4tmzHEywU7eHHt5xCEF79tL2tsuk4-c,2067
 wafer_core/tools/wafer_tool.py,sha256=-dgPTHbWXq3I3wFj0mP7-lj5iZqGRoFvFf9IEEo3plQ,6345
 wafer_core/tools/write_kernel_tool.py,sha256=dJjhr-WBhVNe06hcJQVmBZTbS8mid64KF1MwlE2s2R4,21547
 wafer_core/tools/autotuner/BENCHMARKING.md,sha256=RkJ2wFhbDFXuMbw0mOW4pRqntT0UirptXwIxyrA1_KM,3825
@@ -640,7 +642,7 @@ wafer_core/tools/tracelens_tools/tracelens_collective_tool.py,sha256=0E3FhfaA1N0
 wafer_core/tools/tracelens_tools/tracelens_compare_tool.py,sha256=99dUsB4wuYjxbh4X6Nsf2AtDMs94Uzy04tSemDOmKhg,4458
 wafer_core/tools/tracelens_tools/tracelens_report_tool.py,sha256=unuEx2zXaK42lA3qojS-WzFlBmIFrS75GHSgXUnDXGE,4720
 wafer_core/utils/__init__.py,sha256=oPHgkMkE7wS2lYKLlXrw4Ia5EHnpVcGHFfpWebIlVKs,354
-wafer_core/utils/backend.py,sha256=gNxv4jeOfZwrefTDhyp-UHzkWRyWRhmot76PRZkAwyk,8842
+wafer_core/utils/backend.py,sha256=zt5AX00OXSIstprvQ1_WNf_PQYphD2y53kOXWvg20RY,8986
 wafer_core/utils/code_validation.py,sha256=UqS4UVDxO-atdbn6i7JygX6IFPITvT56zZn1t-ZNuM8,4692
 wafer_core/utils/environment_serialization.py,sha256=cVDkapx0JC60CekazgirPEMAeGZhbLdX1WMIkFvId60,5047
 wafer_core/utils/event_streaming.py,sha256=Sg3-hI043Ofc2b29Z3DWrKgu4HkfJoIqhhbfGRJv70Q,2260
@@ -663,7 +665,7 @@ wafer_core/utils/kernel_utils/static_checker.py,sha256=GWC7RZdwL4WqMLK0nO-wnZd-V
 wafer_core/utils/kernel_utils/task.py,sha256=XcmKxKUWh5It6nX3zGqj77tWgA32uPfQMqNOqyD5T48,2682
 wafer_core/utils/kernel_utils/utils.py,sha256=uDZoJDxh07hJeLNlPdKN2vgB15pqIr1LbXf0YIBHU4E,43056
 wafer_core/utils/kernel_utils/targets/__init__.py,sha256=4NwRLsuJ__S4xKAfda4Ag82C5MQ3Qio-4xA5S-mQGlU,2067
-wafer_core/utils/kernel_utils/targets/config.py,sha256=eGpPOKlTxcS16Fs1-L9r9Cf_CDTG2-52bsG7LJllFw4,18900
+wafer_core/utils/kernel_utils/targets/config.py,sha256=3TvT2Hp3TV-cjsSiZ8NOmAz8epDoVii76p6DDAI2V64,19134
 wafer_core/utils/kernel_utils/targets/execution.py,sha256=bZuNXCo0sIdD6hFhetLPrtDC-zMSiIsAx_aml49VVL0,15033
 wafer_core/utils/kernel_utils/targets/selection.py,sha256=5I_RG_7cfhq7uaeR28meC2EeNNKssFsK-Tc3QFG6Ze0,3590
 wafer_core/utils/modal_execution/__init__.py,sha256=jkVqYOLzCT5K73N9Od0UIUsx-99A0m6bpDrxfyXxQZ8,945
@@ -671,6 +673,6 @@ wafer_core/utils/modal_execution/modal_app.py,sha256=ibBllC59R9bP9w4QweFocVGeSX5
 wafer_core/utils/modal_execution/modal_config.py,sha256=7cGX9TGqilQ3qxI3OFGXV5orjtyRU-PEDOJ4vP2oxno,4421
 wafer_core/utils/modal_execution/modal_execution.py,sha256=gChjnV6jqA3A7IRP3DfvV5cSfm_MN0X4f7JZufXgdZE,24594
 wafer_core/utils/modal_execution/test_modal.py,sha256=_jqou_hrLs1Daf1590Pnb0a_lXMMa2rczAPpW9HpoNQ,8153
-wafer_core-0.1.20.dist-info/METADATA,sha256=Dndu3eLqq_Gxvoa_tKupaqOmQ17FjVujI_KLTzk0MXQ,1420
-wafer_core-0.1.20.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-wafer_core-0.1.20.dist-info/RECORD,,
+wafer_core-0.1.22.dist-info/METADATA,sha256=wV6MLEufRIKPRacW_ErMoqgymAFNI2XgP4wobQoKjnM,1420
+wafer_core-0.1.22.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+wafer_core-0.1.22.dist-info/RECORD,,