weco 0.2.28__py3-none-any.whl → 0.3.1__py3-none-any.whl

weco/panels.py

@@ -6,7 +6,6 @@ from rich.panel import Panel
 from rich.syntax import Syntax
 from rich import box
 from typing import Dict, List, Optional, Union, Tuple
-from .utils import format_number
 from pathlib import Path
 from .__init__ import __dashboard_url__
 
@@ -26,8 +25,6 @@ class SummaryPanel:
     ):
         self.maximize = maximize
         self.metric_name = metric_name
-        self.total_input_tokens = 0
-        self.total_output_tokens = 0
         self.total_steps = total_steps
         self.model = model
         self.runs_dir = runs_dir
@@ -62,13 +59,6 @@ class SummaryPanel:
         """Set the current step."""
         self.progress.update(self.task_id, completed=step)
 
-    def update_token_counts(self, usage: Dict[str, int]):
-        """Update token counts from usage data."""
-        if not isinstance(usage, dict) or "input_tokens" not in usage or "output_tokens" not in usage:
-            raise ValueError("Invalid token usage data received.")
-        self.total_input_tokens += usage["input_tokens"]
-        self.total_output_tokens += usage["output_tokens"]
-
     def update_thinking(self, thinking: str):
         """Update the thinking content."""
         self.thinking_content = thinking
@@ -85,6 +75,10 @@ class SummaryPanel:
         summary_table.add_column(justify="right")
         summary_table.add_row("")
 
+        # Run id
+        summary_table.add_row(f" Run ID: [bold cyan]{self.run_id}[/]")
+        summary_table.add_row("")
+
         # Dashboard url
         summary_table.add_row(f" Dashboard: [underline blue]{self.dashboard_url}[/]")
         summary_table.add_row("")
@@ -94,14 +88,8 @@ class SummaryPanel:
             summary_table.add_row(f"[bold cyan] Result:[/] {final_message}", "")
             summary_table.add_row("")
 
-        # Token info
-        token_info = (
-            f"[bold cyan] {self.model}:[/] "
-            f"↑[yellow]{format_number(self.total_input_tokens)}[/] "
-            f"↓[yellow]{format_number(self.total_output_tokens)}[/] = "
-            f"[green]{format_number(self.total_input_tokens + self.total_output_tokens)} Tokens[/]"
-        )
-        summary_table.add_row(token_info)
+        # Model info
+        summary_table.add_row(f" Model: [bold cyan]{self.model}[/]")
         summary_table.add_row("")
 
         # Progress bar

weco/utils.py

@@ -1,6 +1,5 @@
 from typing import Any, Dict, List, Tuple, Union
 import json
-import os
 import time
 import subprocess
 from rich.layout import Layout
@@ -10,45 +9,13 @@ import pathlib
 import requests
 from packaging.version import parse as parse_version
 
-from .constants import TRUNCATION_THRESHOLD, TRUNCATION_KEEP_LENGTH
+from .constants import TRUNCATION_THRESHOLD, TRUNCATION_KEEP_LENGTH, DEFAULT_MODEL, SUPPORTED_FILE_EXTENSIONS
 
 
 # Env/arg helper functions
-def read_api_keys_from_env() -> Dict[str, Any]:
-    """Read API keys from environment variables."""
-    keys = {}
-    keys_to_check = ["OPENAI_API_KEY", "ANTHROPIC_API_KEY", "GEMINI_API_KEY"]
-    for key in keys_to_check:
-        value = os.getenv(key)
-        if value is not None and len(value) > 0:
-            keys[key] = value
-    return keys
-
-
-def determine_default_model(llm_api_keys: Dict[str, Any]) -> str:
-    """Determine the default model based on available API keys.
-
-    Uses priority: OpenAI > Anthropic > Gemini
-
-    Args:
-        llm_api_keys: Dictionary of available LLM API keys
-
-    Returns:
-        str: The default model name to use
-
-    Raises:
-        ValueError: If no LLM API keys are found
-    """
-    if "OPENAI_API_KEY" in llm_api_keys:
-        return "o4-mini"
-    elif "ANTHROPIC_API_KEY" in llm_api_keys:
-        return "claude-sonnet-4-0"
-    elif "GEMINI_API_KEY" in llm_api_keys:
-        return "gemini-2.5-pro"
-    else:
-        raise ValueError(
-            "No LLM API keys found in environment variables. Please set one of the following: OPENAI_API_KEY, ANTHROPIC_API_KEY, or GEMINI_API_KEY based on your model of choice."
-        )
+def determine_model_for_onboarding() -> str:
+    """Determine which model to use for onboarding chatbot. Defaults to o4-mini."""
+    return DEFAULT_MODEL
 
 
 def read_additional_instructions(additional_instructions: str | None) -> str | None:
@@ -60,6 +27,11 @@ def read_additional_instructions(additional_instructions: str | None) -> str | N
     potential_path = pathlib.Path(additional_instructions)
     try:
         if potential_path.exists() and potential_path.is_file():
+            # If it's a valid file path, check if we support the file extension
+            if potential_path.suffix.lower() not in SUPPORTED_FILE_EXTENSIONS:
+                raise ValueError(
+                    f"Unsupported file extension: {potential_path.suffix.lower()}. Supported extensions are: {', '.join(SUPPORTED_FILE_EXTENSIONS)}"
+                )
             return read_from_path(potential_path, is_json=False)  # type: ignore # read_from_path returns str when is_json=False
         else:
             # If it's not a valid file path, return the string itself
@@ -90,23 +62,6 @@ def write_to_path(fp: pathlib.Path, content: Union[str, Dict[str, Any]], is_json
 
 
 # Visualization helper functions
-def format_number(n: Union[int, float]) -> str:
-    """Format large numbers with K, M, B, T suffixes for better readability."""
-    if n >= 1e12:
-        return f"{n / 1e12:.1f}T"
-    elif n >= 1e9:
-        return f"{n / 1e9:.1f}B"
-    elif n >= 1e6:
-        return f"{n / 1e6:.1f}M"
-    elif n >= 1e3:
-        return f"{n / 1e3:.1f}K"
-    # Handle potential floats that don't need suffix but might need formatting
-    if isinstance(n, float):
-        # Format floats nicely, avoid excessive precision unless needed
-        return f"{n:.4g}"  # Use general format, up to 4 significant digits
-    return str(n)
-
-
 def smooth_update(
     live: Live, layout: Layout, sections_to_update: List[Tuple[str, Panel]], transition_delay: float = 0.05
 ) -> None:

{weco-0.2.28.dist-info → weco-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weco
-Version: 0.2.28
+Version: 0.3.1
 Summary: Documentation for `weco`, a CLI for using Weco AI's code optimizer.
 Author-email: Weco AI Team <contact@weco.ai>
 License: 
@@ -270,16 +270,17 @@ The `weco` CLI leverages a tree search approach guided by LLMs to iteratively ex
 1.  **Install the Package:**
 
     ```bash
-    pip install weco>=0.2.18
+    pip install weco
     ```
 
-2.  **Set Up LLM API Keys (Required):**
+2.  **Authenticate (Required):**
 
-    `weco` requires API keys for the LLMs it uses internally. You **must** provide these keys via environment variables:
+    `weco` now uses a **credit-based billing system** with centralized LLM access. You need to authenticate to use the service:
 
-    - **OpenAI:** `export OPENAI_API_KEY="your_key_here"` (Create your OpenAI API key [here](https://platform.openai.com/api-keys))
-    - **Anthropic:** `export ANTHROPIC_API_KEY="your_key_here"` (Create your Anthropic API key [here](https://console.anthropic.com/settings/keys))
-    - **Google:** `export GEMINI_API_KEY="your_key_here"` (Google AI Studio has a free API usage quota. Create your Gemini API key [here](https://aistudio.google.com/apikey) to use `weco` for free.)
+    - **Run the CLI**: `weco` will prompt you to authenticate via your web browser
+    - **Free Credits**: New users receive **free credits** upon signup
+    - **Centralized Keys**: All LLM provider API keys are managed by Weco (no BYOK required)
+    - **Credit Top-ups**: Purchase additional credits through the dashboard at [dashboard.weco.ai](https://dashboard.weco.ai)
 
 ---
 
@@ -338,6 +339,8 @@ weco run --source optimize.py \
 
 For more advanced examples, including [Triton](/examples/triton/README.md), [CUDA kernel optimization](/examples/cuda/README.md), [ML model optimization](/examples/spaceship-titanic/README.md), and [prompt engineering for math problems](examples/prompt/README.md), please see the `README.md` files within the corresponding subdirectories under the [`examples/`](examples/) folder.
 
+> Note: When recommend removing any backticks from your code if any are present. We currently don't support backticks but will support this in the future.
+
 ---
 
 ### Arguments for `weco run`
@@ -358,8 +361,8 @@ For more advanced examples, including [Triton](/examples/triton/README.md), [CUD
 | Argument                       | Description                                                                                                                                                                                                                | Default                                                                                                                                                | Example             |
 | :----------------------------- | :------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- | :------------------------------------------------------------------------------------------------------------------------------------------------------ | :------------------ |
 | `-n, --steps`                  | Number of optimization steps (LLM iterations) to run.                                                                                                                                                                      | 100                                                                                                                                                     | `-n 50`             |
-| `-M, --model`                  | Model identifier for the LLM to use (e.g., `o4-mini`, `claude-sonnet-4-0`).                                                                                                        | `o4-mini` when `OPENAI_API_KEY` is set; `claude-sonnet-4-0` when `ANTHROPIC_API_KEY` is set; `gemini-2.5-pro` when `GEMINI_API_KEY` is set. | `-M o4-mini`         |
-| `-i, --additional-instructions`| Natural language description of specific instructions **or** path to a file containing detailed instructions to guide the LLM.                                                                                             | `None`                                                                                                                                                  | `-i instructions.md` or `-i "Optimize the model for faster inference"`|
+| `-M, --model`                  | Model identifier for the LLM to use (e.g., `o4-mini`, `claude-sonnet-4-0`).                                                                                                        | `o4-mini` | `-M o4-mini`         |
+| `-i, --additional-instructions`| Natural language description of specific instructions **or** path to a file containing detailed instructions to guide the LLM. Supported file formats include - `.txt`, `.md`, and `.rst`.                                                                                             | `None`                                                                                                                                                  | `-i instructions.md` or `-i "Optimize the model for faster inference"`|
 | `-l, --log-dir`                | Path to the directory to log intermediate steps and final optimization result.                                                                                                                                             | `.runs/`                                                                                                                                               | `-l ./logs/`        |
 | `--eval-timeout`       | Timeout in seconds for each step in evaluation.                                                                                                                                                                             | No timeout (unlimited)                                                                                                                                                  | `--eval-timeout 3600`             |
 | `--save-logs`          | Save execution output from each optimization step to disk. Creates timestamped directories with raw output files and a JSONL index for tracking execution history.                                                        | `False`                                                                                                                                                 | `--save-logs`       |
@@ -368,24 +371,24 @@ For more advanced examples, including [Triton](/examples/triton/README.md), [CUD
 
 ### Authentication & Dashboard
 
-Weco offers both **anonymous** and **authenticated** usage:
-
-#### Anonymous Usage
-You can use Weco without creating an account by providing LLM API keys via environment variables. This is perfect for trying out Weco or for users who prefer not to create accounts.
+The CLI requires a Weco account for authentication and billing.
 
-#### Authenticated Usage (Recommended)
-To save your optimization runs and view them on the Weco dashboard, you can log in using Weco's secure device authentication flow:
+#### Credit-Based Authentication (Required)
+Weco now requires authentication for all operations. This enables our credit-based billing system and provides access to powerful optimizations:
 
-1. **During onboarding**: When you run `weco` for the first time, you'll be prompted to log in or skip
+1. **During onboarding**: When you run `weco` for the first time, you'll be prompted to log in
 2. **Manual login**: Use `weco logout` to clear credentials, then run `weco` again to re-authenticate
 3. **Device flow**: Weco will open your browser automatically and guide you through a secure OAuth-style authentication
 
 ![image (16)](https://github.com/user-attachments/assets/8a0a285b-4894-46fa-b6a2-4990017ca0c6)
 
-**Benefits of authenticated usage:**
-- **Run history**: View all your optimization runs on the Weco dashboard
-- **Progress tracking**: Monitor long-running optimizations remotely
-- **Enhanced support**: Get better assistance with your optimization challenges
+**Benefits:**
+- **No API Key Management**: All LLM provider keys are managed centrally
+- **Cost Transparency**: See exactly how many credits each optimization consumes
+- **Free Trial**: Free credits to get started with optimization projects
+- **Run History**: View all your optimization runs on the Weco dashboard
+- **Progress Tracking**: Monitor long-running optimizations remotely
+- **Budget Control**: Set spending limits and auto top-up preferences
 
 ---
 
@@ -398,6 +401,7 @@ To save your optimization runs and view them on the Weco dashboard, you can log
 | `weco` | Launch interactive onboarding | **Recommended for beginners** - Analyzes your codebase and guides you through setup |
 | `weco /path/to/project` | Launch onboarding for specific project | When working with a project in a different directory |
 | `weco run [options]` | Direct optimization execution | **For advanced users** - When you know exactly what to optimize and how |
+| `weco resume <run-id>` | Resume an interrupted run | Continue from the last completed step |
 | `weco logout` | Clear authentication credentials | To switch accounts or troubleshoot authentication issues |
 
 ### Model Selection
@@ -413,14 +417,37 @@ weco run --model claude-3.5-sonnet --source optimize.py [other options...]
 ```
 
 **Available models:**
-- `gpt-4o`, `o4-mini` (requires `OPENAI_API_KEY`)
-- `claude-3.5-sonnet`, `claude-sonnet-4-20250514` (requires `ANTHROPIC_API_KEY`)
-- `gemini-2.5-pro` (requires `GEMINI_API_KEY`)
+- `o4-mini`, `o3-mini`, `gpt-4o` (OpenAI models)
+- `claude-sonnet-4-0`, `claude-opus-4-0` (Anthropic models)  
+- `gemini-2.5-pro`, `gemini-2.5-flash` (Google models)
 
-If no model is specified, Weco automatically selects the best available model based on your API keys.
+All models are available through Weco's centralized system. If no model is specified, Weco automatically selects the best model for your optimization task.
 
 ---
 
+### Resuming Interrupted Runs
+
+If your optimization run is interrupted (network issues, restart, etc.), resume from the most recent node:
+
+```bash
+# Resume an interrupted run
+weco resume 0002e071-1b67-411f-a514-36947f0c4b31
+
+```
+
+Arguments for `weco resume`:
+
+| Argument | Description | Example |
+|----------|-------------|---------|
+| `run-id` | The UUID of the run to resume (shown at the start of each run) | `0002e071-1b67-411f-a514-36947f0c4b31` |
+
+Notes:
+- Works only for interrupted runs (status: `error`, `terminated`, etc.).
+- You’ll be prompted to confirm that your evaluation environment (source file + evaluation command) hasn’t changed.
+- The source file is restored to the most recent solution before continuing.
+- All progress and metrics from the original run are preserved.
+- Log directory, save-logs behavior, and evaluation timeout are reused from the original run.
+
 ### Performance & Expectations
 
 Weco, powered by the AIDE algorithm, optimizes code iteratively based on your evaluation results. Achieving significant improvements, especially on complex research-level tasks, often requires substantial exploration time.
@@ -493,37 +520,7 @@ Weco will parse this output to extract the numerical value (1.5 in this case) as
 
 ## Supported Models
 
-Weco supports the following LLM models:
-
-### OpenAI Models
-- `gpt-5` (recommended)
-- `gpt-5-mini`
-- `gpt-5-nano`
-- `o3-pro` (recommended)
-- `o3` (recommended)
-- `o4-mini` (recommended)
-- `o3-mini`
-- `o1-pro`
-- `o1`
-- `gpt-4.1`
-- `gpt-4.1-mini`
-- `gpt-4.1-nano`
-- `gpt-4o`
-- `gpt-4o-mini`
-- `codex-mini-latest`
-
-### Anthropic Models
-- `claude-opus-4-1`
-- `claude-opus-4-0`
-- `claude-sonnet-4-0`
-- `claude-3-7-sonnet-latest`
-
-### Gemini Models
-- `gemini-2.5-pro`
-- `gemini-2.5-flash`
-- `gemini-2.5-flash-lite`
-
-You can specify any of these models using the `-M` or `--model` flag. Ensure you have the corresponding API key set as an environment variable for the model provider you wish to use.
+A list of models we support can be found in our documentation [here](https://docs.weco.ai/cli/supported-models).
 
 ---
 

weco-0.3.1.dist-info/RECORD

@@ -0,0 +1,16 @@
+weco/__init__.py,sha256=ClO0uT6GKOA0iSptvP0xbtdycf0VpoPTq37jHtvlhtw,303
+weco/api.py,sha256=dUjzuOKKvayzZ_1B4j40eK9Ofk264jsc6vOR1afsszY,18523
+weco/auth.py,sha256=O31Hoj-Loi8DWJJG2LfeWgUMuNqAUeGDpd2ZGjA9Ah0,9997
+weco/chatbot.py,sha256=EIK2WaOul9gn_yHLThjsZV7RnE8t3XQPwgRkO5tybSU,38415
+weco/cli.py,sha256=579f6jf-ZWuFAmNXDisRY7zWr7vw2YZQuC_QX8-qxx0,11460
+weco/constants.py,sha256=V6yFugTznKm5EC2_jr4I_whd7sqI80HiPggRn0az580,406
+weco/credits.py,sha256=C08x-TRcLg3ccfKqMGNRY7zBn7t3r7LZ119bxgfztaI,7629
+weco/optimizer.py,sha256=mJU8_0bo_6dS2PEj1E3dQHvNH9V4e8NSLNE55tmvspw,42291
+weco/panels.py,sha256=fnGPtmvxpx21AuBCtCFu1f_BpSxybNr2lhjIIKIutrY,16133
+weco/utils.py,sha256=TT57S0YGMuMWPFNsn0tcexNHZd-kBEjDeiOLWxANiQU,6117
+weco-0.3.1.dist-info/licenses/LICENSE,sha256=9LUfoGHjLPtak2zps2kL2tm65HAZIICx_FbLaRuS4KU,11337
+weco-0.3.1.dist-info/METADATA,sha256=e5xozCmFPB7ih2ntFNYQAMXAU_O8Kw3NDSiRhaNEu4c,31856
+weco-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+weco-0.3.1.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
+weco-0.3.1.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
+weco-0.3.1.dist-info/RECORD,,

weco-0.2.28.dist-info/RECORD

@@ -1,15 +0,0 @@
-weco/__init__.py,sha256=ClO0uT6GKOA0iSptvP0xbtdycf0VpoPTq37jHtvlhtw,303
-weco/api.py,sha256=cdZEf-Zt0CxMOj_gka6rGHEK1MVwPAjG1YH16jgDEsg,13177
-weco/auth.py,sha256=KMSAsN1V5wx7KUsYL1cEOOiG29Pqf4Exb3EPW4mAWC0,10003
-weco/chatbot.py,sha256=EkzKd5Q_IlcobBbY3gsbgN0jxbJMfP5eYtzxQaNQ3fg,37747
-weco/cli.py,sha256=8hrlmHmaZiYQ7kotdpr4Ve-xAJZocDV6kcizPCmep0k,8380
-weco/constants.py,sha256=hVEIpNejUxZ6-1GrL7Qv97EqBoo5vqDdPdMfAjq-_24,345
-weco/optimizer.py,sha256=bXhNoa2qyC-CeqLHacy3xz2UKHuO_DVpC3z572NjFSU,26063
-weco/panels.py,sha256=jwAV_uoa0ZI9vjyey-hSY3rx4pfNNkZvPzqt-iz-RXo,16808
-weco/utils.py,sha256=P6efzBXg7m_Nnq6UUor9onCGxjE0CkTI2xYsymmCwZ4,7355
-weco-0.2.28.dist-info/licenses/LICENSE,sha256=9LUfoGHjLPtak2zps2kL2tm65HAZIICx_FbLaRuS4KU,11337
-weco-0.2.28.dist-info/METADATA,sha256=qnSbrHYphl5HLaqzAWghErM1I3gB-26czvbXfGZ_QzQ,31432
-weco-0.2.28.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-weco-0.2.28.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
-weco-0.2.28.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
-weco-0.2.28.dist-info/RECORD,,