weco 0.3.6__py3-none-any.whl → 0.3.8__py3-none-any.whl

weco/ui.py

@@ -0,0 +1,315 @@
+"""
+Optimization loop UI components.
+
+This module contains the UI protocol and implementations for displaying
+optimization progress in the CLI.
+"""
+
+import time
+from dataclasses import dataclass, field
+from typing import List, Optional, Protocol
+
+from rich.console import Console, Group
+from rich.live import Live
+from rich.panel import Panel
+from rich.table import Table
+from rich.text import Text
+
+
+class OptimizationUI(Protocol):
+    """Protocol for optimization UI event handlers."""
+
+    def on_polling(self, step: int) -> None:
+        """Called when polling for execution tasks."""
+        ...
+
+    def on_task_claimed(self, task_id: str, plan: Optional[str]) -> None:
+        """Called when a task is successfully claimed."""
+        ...
+
+    def on_executing(self, step: int) -> None:
+        """Called when starting to execute code."""
+        ...
+
+    def on_output(self, output: str, max_preview: int = 200) -> None:
+        """Called with execution output."""
+        ...
+
+    def on_submitting(self) -> None:
+        """Called when submitting result to backend."""
+        ...
+
+    def on_metric(self, step: int, value: float) -> None:
+        """Called when a metric value is received."""
+        ...
+
+    def on_complete(self, total_steps: int) -> None:
+        """Called when optimization completes successfully."""
+        ...
+
+    def on_stop_requested(self) -> None:
+        """Called when a stop request is received from dashboard."""
+        ...
+
+    def on_interrupted(self) -> None:
+        """Called when interrupted by user (Ctrl+C)."""
+        ...
+
+    def on_warning(self, message: str) -> None:
+        """Called for non-fatal warnings."""
+        ...
+
+    def on_error(self, message: str) -> None:
+        """Called for errors."""
+        ...
+
+
+@dataclass
+class UIState:
+    """Reactive state for the live optimization UI."""
+
+    step: int = 0
+    total_steps: int = 0
+    status: str = "initializing"  # polling, executing, submitting, complete, stopped, error
+    plan_preview: str = ""
+    output_preview: str = ""
+    metrics: List[tuple] = field(default_factory=list)  # (step, value)
+    error: Optional[str] = None
+
+
+class LiveOptimizationUI:
+    """
+    Rich Live implementation of OptimizationUI with dynamic single-panel updates.
+
+    Displays a compact, updating panel showing:
+    - Run info (ID, name, dashboard link)
+    - Current step and status with visual indicator
+    - Plan preview
+    - Output preview
+    - Metric history as sparkline
+    """
+
+    SPARKLINE_CHARS = "▁▂▃▄▅▆▇█"
+    SPINNER_FRAMES = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
+    # Statuses that show the spinner animation
+    ACTIVE_STATUSES = {"initializing", "polling", "executing", "submitting"}
+    STATUS_INDICATORS = {
+        "initializing": ("⏳", "dim"),
+        "polling": ("🔄", "cyan"),
+        "executing": ("⚡", "yellow"),
+        "submitting": ("🧠", "blue"),
+        "complete": ("✅", "green"),
+        "stopped": ("⏹", "yellow"),
+        "interrupted": ("⚠", "yellow"),
+        "error": ("❌", "red"),
+    }
+
+    def __init__(
+        self,
+        console: Console,
+        run_id: str,
+        run_name: str,
+        total_steps: int,
+        dashboard_url: str,
+        model: str = "",
+        metric_name: str = "",
+    ):
+        self.console = console
+        self.run_id = run_id
+        self.run_name = run_name
+        self.dashboard_url = dashboard_url
+        self.model = model
+        self.metric_name = metric_name
+        self.state = UIState(total_steps=total_steps)
+        self._live: Optional[Live] = None
+
+    def _sparkline(self, values: List[float], max_width: int) -> str:
+        """
+        Create a mini sparkline chart from metric values.
+
+        Automatically slides to show most recent values when they exceed max_width.
+        Shows "···" prefix when older values are hidden.
+        """
+        if not values:
+            return ""
+
+        # Reserve space for "···" prefix if we need to truncate
+        if len(values) > max_width:
+            prefix = "··"
+            available = max_width - len(prefix)
+            vals = values[-available:]  # Take most recent values that fit
+            sparkline_prefix = f"[dim]{prefix}[/]"
+        else:
+            vals = values
+            sparkline_prefix = ""
+
+        min_v, max_v = min(vals), max(vals)
+        if max_v == min_v:
+            return sparkline_prefix + self.SPARKLINE_CHARS[4] * len(vals)
+
+        chars = self.SPARKLINE_CHARS
+        sparkline = "".join(chars[int((v - min_v) / (max_v - min_v) * 7)] for v in vals)
+        return sparkline_prefix + sparkline
+
+    def _render(self) -> Group:
+        """Render the current UI state as a Rich Panel with top margin."""
+        emoji, style = self.STATUS_INDICATORS.get(self.state.status, ("⏳", "dim"))
+
+        # Build content grid - expands to full terminal width
+        grid = Table.grid(padding=(0, 1), expand=True)
+        grid.add_column(style="dim", width=10)
+        grid.add_column(overflow="ellipsis", no_wrap=True, ratio=1)
+
+        # Run info (always shown)
+        run_display = f"[bold]{self.run_name}[/] [dim]({self.run_id})[/]"
+        grid.add_row("Run", run_display)
+        grid.add_row("Dashboard", f"[link={self.dashboard_url}]{self.dashboard_url}[/link]")
+        if self.model:
+            grid.add_row("Model", f"[cyan]{self.model}[/]")
+        if self.metric_name:
+            grid.add_row("Metric", f"[magenta]{self.metric_name}[/]")
+        grid.add_row("", "")
+
+        # Progress (always shown)
+        progress_bar = self._render_progress_bar()
+        grid.add_row("Progress", progress_bar)
+
+        # Status (always shown) - with spinner for active states
+        status_text = Text()
+        status_text.append(f"{emoji} ", style=style)
+        status_text.append(self.state.status.replace("_", " ").title(), style=f"bold {style}")
+        if self.state.status in self.ACTIVE_STATUSES:
+            # Time-based frame calculation: ~10 fps spinner animation
+            frame = int(time.time() * 10) % len(self.SPINNER_FRAMES)
+            spinner = self.SPINNER_FRAMES[frame]
+            status_text.append(f" {spinner}", style=f"bold {style}")
+        grid.add_row("Status", status_text)
+
+        # Plan (always shown, placeholder when empty)
+        if self.state.plan_preview:
+            grid.add_row("Plan", f"[dim italic]{self.state.plan_preview}[/]")
+        else:
+            grid.add_row("Plan", "[dim]—[/]")
+
+        # Output (always shown, placeholder when empty)
+        if self.state.output_preview:
+            output_text = self.state.output_preview.replace("\n", " ")
+            grid.add_row("Output", f"[dim]{output_text}[/]")
+        else:
+            grid.add_row("Output", "[dim]—[/]")
+
+        # Metrics section (always shown, 3 rows: current, best, chart)
+        if self.state.metrics:
+            values = [m[1] for m in self.state.metrics]
+            latest = self.state.metrics[-1][1]
+            best = max(values)
+
+            # Current and best on separate lines
+            grid.add_row("Current", f"[bold cyan]{latest:.6g}[/]")
+            grid.add_row("Best", f"[bold green]{best:.6g}[/]")
+
+            # Chart line - calculate available width for sparkline
+            # Console width minus: label(10) + padding(4) + panel borders(4) + panel padding(4)
+            chart_width = max(self.console.width - 22, 20)
+            sparkline = self._sparkline(values, chart_width)
+            grid.add_row("History", f"[green]{sparkline}[/]")
+        else:
+            grid.add_row("Current", "[dim]—[/]")
+            grid.add_row("Best", "[dim]—[/]")
+            grid.add_row("History", "[dim]—[/]")
+
+        # Error row (always present, empty when no error)
+        if self.state.error:
+            grid.add_row("Error", f"[bold red]{self.state.error}[/]")
+        else:
+            grid.add_row("", "")  # Empty row to maintain height
+
+        panel = Panel(grid, title="[bold blue]⚡ Weco Optimization[/]", border_style="blue", padding=(1, 2), expand=True)
+        # Wrap panel with top margin for spacing
+        return Group(Text(""), panel)
+
+    def _render_progress_bar(self) -> Text:
+        """Render a simple ASCII progress bar."""
+        total = self.state.total_steps
+        current = min(self.state.step, total)  # Clamp to total to avoid >100%
+        width = 40
+
+        if total <= 0:
+            return Text(f"Step {self.state.step}", style="bold")
+
+        filled = min(int((current / total) * width), width)  # Clamp filled bars
+        bar = "█" * filled + "░" * (width - filled)
+        pct = min((current / total) * 100, 100)  # Clamp percentage
+        return Text(f"[{bar}] {current}/{total} ({pct:.0f}%)", style="bold")
+
+    def __rich__(self) -> Group:
+        """Called by Rich on each refresh cycle - enables auto-animated spinner."""
+        return self._render()
+
+    def _update(self) -> None:
+        """Trigger an immediate live update (for state changes)."""
+        if self._live:
+            self._live.refresh()
+
+    # --- Context manager for Live display ---
+    def __enter__(self) -> "LiveOptimizationUI":
+        # Pass self so Rich calls __rich__() on every auto-refresh (enables spinner animation)
+        # Use vertical_overflow="visible" to prevent clipping issues on exit
+        self._live = Live(self, console=self.console, refresh_per_second=10, transient=False, vertical_overflow="visible")
+        self._live.__enter__()
+        return self
+
+    def __exit__(self, *args) -> None:
+        if self._live:
+            self._live.__exit__(*args)
+            self._live = None
+
+    # --- OptimizationUI Protocol Implementation ---
+    def on_polling(self, step: int) -> None:
+        self.state.step = step
+        self.state.status = "polling"
+        self.state.output_preview = ""
+        self._update()
+
+    def on_task_claimed(self, task_id: str, plan: Optional[str]) -> None:
+        self.state.plan_preview = plan or ""
+        self._update()
+
+    def on_executing(self, step: int) -> None:
+        self.state.step = step
+        self.state.status = "executing"
+        self._update()
+
+    def on_output(self, output: str, max_preview: int = 200) -> None:
+        self.state.output_preview = output[:max_preview]
+        self._update()
+
+    def on_submitting(self) -> None:
+        self.state.status = "submitting"
+        self._update()
+
+    def on_metric(self, step: int, value: float) -> None:
+        self.state.metrics.append((step, value))
+        self._update()
+
+    def on_complete(self, total_steps: int) -> None:
+        self.state.step = total_steps
+        self.state.status = "complete"
+        self._update()
+
+    def on_stop_requested(self) -> None:
+        self.state.status = "stopped"
+        self._update()
+
+    def on_interrupted(self) -> None:
+        self.state.status = "interrupted"
+        self._update()
+
+    def on_warning(self, message: str) -> None:
+        # Warnings are less critical; we could add a warnings list but keeping it simple
+        pass
+
+    def on_error(self, message: str) -> None:
+        self.state.error = message
+        self.state.status = "error"
+        self._update()

weco/validation.py

@@ -0,0 +1,112 @@
+"""Input validation for the Weco CLI.
+
+Provides early validation of user inputs with helpful, actionable error messages.
+Validation happens before expensive operations (auth, API calls) to fail fast.
+"""
+
+import pathlib
+from difflib import get_close_matches
+
+from rich.console import Console
+
+
+class ValidationError(Exception):
+    """Raised when user input validation fails."""
+
+    def __init__(self, message: str, suggestion: str | None = None):
+        self.message = message
+        self.suggestion = suggestion
+        super().__init__(message)
+
+
+def validate_source_file(source: str) -> None:
+    """
+    Validate that the source file exists and is readable.
+
+    Args:
+        source: Path to the source file.
+
+    Raises:
+        ValidationError: If the file doesn't exist, isn't readable, or isn't a valid text file.
+    """
+    path = pathlib.Path(source)
+
+    if not path.exists():
+        suggestion = _find_similar_files(path)
+        raise ValidationError(f"Source file '{source}' not found.", suggestion=suggestion)
+
+    if path.is_dir():
+        raise ValidationError(
+            f"'{source}' is a directory, not a file.", suggestion="Please specify a file path, e.g., 'src/model.py'"
+        )
+
+    # Try reading the file to catch permission and encoding issues early
+    try:
+        path.read_text(encoding="utf-8")
+    except PermissionError:
+        raise ValidationError(f"Cannot read '{source}' — permission denied.")
+    except UnicodeDecodeError:
+        raise ValidationError(
+            f"'{source}' doesn't appear to be a valid text file.",
+            suggestion="Weco optimizes source code files (e.g., .py, .cu, .rs)",
+        )
+    except OSError as e:
+        raise ValidationError(f"Cannot read '{source}': {e}")
+
+
+def validate_log_directory(log_dir: str) -> None:
+    """
+    Validate that the log directory is writable.
+
+    Args:
+        log_dir: Path to the log directory.
+
+    Raises:
+        ValidationError: If the directory can't be created or isn't writable.
+    """
+    path = pathlib.Path(log_dir)
+
+    try:
+        # Attempt to create the directory (no-op if exists)
+        path.mkdir(parents=True, exist_ok=True)
+    except PermissionError:
+        raise ValidationError(f"Cannot create log directory '{log_dir}' — permission denied.")
+    except OSError as e:
+        raise ValidationError(f"Cannot create log directory '{log_dir}': {e}")
+
+    # Check if writable by attempting to create a temp file
+    test_file = path / ".weco_write_test"
+    try:
+        test_file.touch()
+        test_file.unlink()
+    except PermissionError:
+        raise ValidationError(f"Log directory '{log_dir}' is not writable.")
+    except OSError:
+        pass  # Directory exists and is likely fine
+
+
+def _find_similar_files(path: pathlib.Path) -> str | None:
+    """Find similar filenames in the same directory to suggest as alternatives."""
+    parent = path.parent if path.parent.exists() else pathlib.Path(".")
+
+    try:
+        # Get files with the same extension, or all files if no extension
+        if path.suffix:
+            candidates = [f.name for f in parent.iterdir() if f.is_file() and f.suffix == path.suffix]
+        else:
+            candidates = [f.name for f in parent.iterdir() if f.is_file()]
+
+        matches = get_close_matches(path.name, candidates, n=3, cutoff=0.4)
+        if matches:
+            return f"Did you mean: {', '.join(matches)}?"
+    except OSError:
+        pass
+
+    return None
+
+
+def print_validation_error(error: ValidationError, console: Console) -> None:
+    """Print a validation error in a user-friendly format."""
+    console.print(f"[bold red]Error:[/] {error.message}")
+    if error.suggestion:
+        console.print(f"[dim]{error.suggestion}[/]")

{weco-0.3.6.dist-info → weco-0.3.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weco
-Version: 0.3.6
+Version: 0.3.8
 Summary: Documentation for `weco`, a CLI for using Weco AI's code optimizer.
 Author-email: Weco AI Team <contact@weco.ai>
 License: 
@@ -357,7 +357,7 @@ weco run --model gpt-5 --source optimize.py [other options...]
 - `claude-opus-4-5`, `claude-opus-4-1`, `claude-opus-4`, `claude-sonnet-4-5`, `claude-sonnet-4`, `claude-haiku-4-5`
 
 **Google Gemini:**
-- `gemini-3-pro-preview`, `gemini-2.5-pro`, `gemini-2.5-flash`, `gemini-2.5-flash-lite`
+- `gemini-3-pro-preview`, `gemini-3-flash-preview`, `gemini-2.5-pro`, `gemini-2.5-flash`, `gemini-2.5-flash-lite`
 
 All models are available through Weco. If no model is specified, Weco automatically selects the best model for your optimization task.
 

weco-0.3.8.dist-info/RECORD

@@ -0,0 +1,18 @@
+weco/__init__.py,sha256=ClO0uT6GKOA0iSptvP0xbtdycf0VpoPTq37jHtvlhtw,303
+weco/api.py,sha256=hy0D01x-AJ26DURtKEywQAS7nQgo38A-wAJfPKHGGqM,17395
+weco/auth.py,sha256=O31Hoj-Loi8DWJJG2LfeWgUMuNqAUeGDpd2ZGjA9Ah0,9997
+weco/browser.py,sha256=nsqQtLqbNOe9Zhu9Zogc8rMmBMyuDxuHzKZQL_w10Ps,923
+weco/cli.py,sha256=u5DSt2YGLOha-ldaW6qg3NO2z3rNfYw37FNtaIs-Kz4,12656
+weco/constants.py,sha256=rxL6yrpIzK8zvPTmPqOYl7LUMZ01vUJ9zUqfZD2n-0U,519
+weco/credits.py,sha256=C08x-TRcLg3ccfKqMGNRY7zBn7t3r7LZ119bxgfztaI,7629
+weco/optimizer.py,sha256=XygxTOTPaPx0dDVI9oWaMQHZ0-YAg1JFcMvGPcnEx-A,23990
+weco/panels.py,sha256=POHt0MdRKDykwUJYXcry92O41lpB9gxna55wFI9abWU,16272
+weco/ui.py,sha256=1shfWxeyfhjTzRZsuODuMx5XzeP9SngqpNpN4vIiDCI,11203
+weco/utils.py,sha256=v_rvgw-ktRoXrpPA2copngI8QDCB8UXmbiN-wAiYvEE,9450
+weco/validation.py,sha256=n5aDuF3BFgwVb4eZ9PuU48nogrseXYNI8S3ePqWZCoc,3736
+weco-0.3.8.dist-info/licenses/LICENSE,sha256=9LUfoGHjLPtak2zps2kL2tm65HAZIICx_FbLaRuS4KU,11337
+weco-0.3.8.dist-info/METADATA,sha256=a9A-tvBbsaojDEXmi1DYsoBAJALFQ6kPeIlWlf2Of0Y,29861
+weco-0.3.8.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+weco-0.3.8.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
+weco-0.3.8.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
+weco-0.3.8.dist-info/RECORD,,

{weco-0.3.6.dist-info → weco-0.3.8.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

weco-0.3.6.dist-info/RECORD

@@ -1,15 +0,0 @@
-weco/__init__.py,sha256=ClO0uT6GKOA0iSptvP0xbtdycf0VpoPTq37jHtvlhtw,303
-weco/api.py,sha256=xVVRk1pj9jpjTphaInkkAhjqhgFP2-6zHT_V-5Du1Fc,13629
-weco/auth.py,sha256=O31Hoj-Loi8DWJJG2LfeWgUMuNqAUeGDpd2ZGjA9Ah0,9997
-weco/cli.py,sha256=Mtkv3rE1rQLdeoVydn30EUi1ki3Cyu45Q3cONnQH4QY,11210
-weco/constants.py,sha256=Wt1VI76smsU0Gsg-HSB9e3V1ZjjsMYXJJg2G9Rb-nQc,519
-weco/credits.py,sha256=C08x-TRcLg3ccfKqMGNRY7zBn7t3r7LZ119bxgfztaI,7629
-weco/optimizer.py,sha256=2qYweESOAer26gjjhu4dg01XmuhtA2nMrJuijaxizzE,45492
-weco/panels.py,sha256=POHt0MdRKDykwUJYXcry92O41lpB9gxna55wFI9abWU,16272
-weco/utils.py,sha256=v_rvgw-ktRoXrpPA2copngI8QDCB8UXmbiN-wAiYvEE,9450
-weco-0.3.6.dist-info/licenses/LICENSE,sha256=9LUfoGHjLPtak2zps2kL2tm65HAZIICx_FbLaRuS4KU,11337
-weco-0.3.6.dist-info/METADATA,sha256=-BKVxw7j_Qe5ksblcmoVF_NVeDDIcWTg-CsUUaadAgs,29835
-weco-0.3.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-weco-0.3.6.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
-weco-0.3.6.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
-weco-0.3.6.dist-info/RECORD,,