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

weco/api.py

@@ -178,7 +178,6 @@ def evaluate_feedback_then_suggest_next_solution(
     run_id: str,
     step: int,
     execution_output: str,
-    additional_instructions: str = None,
     auth_headers: dict = {},
     timeout: Union[int, Tuple[int, int]] = (10, 3650),
 ) -> Dict[str, Any]:
@@ -189,7 +188,7 @@ def evaluate_feedback_then_suggest_next_solution(
 
         response = requests.post(
             f"{__base_url__}/runs/{run_id}/suggest",
-            json={"execution_output": truncated_output, "additional_instructions": additional_instructions, "metadata": {}},
+            json={"execution_output": truncated_output, "metadata": {}},
             headers=auth_headers,
             timeout=timeout,
         )

weco/cli.py

@@ -49,7 +49,7 @@ def configure_run_parser(run_parser: argparse.ArgumentParser) -> None:
         "--model",
         type=str,
         default=None,
-        help="Model to use for optimization. Defaults to `o4-mini` when `OPENAI_API_KEY` is set, `claude-sonnet-4-0` when `ANTHROPIC_API_KEY` is set, and `gemini-2.5-pro` when `GEMINI_API_KEY` is set. When multiple keys are set, the priority is `OPENAI_API_KEY` > `ANTHROPIC_API_KEY` > `GEMINI_API_KEY`.",
+        help="Model to use for optimization. Defaults to `o4-mini`. See full list at https://docs.weco.ai/cli/supported-models",
     )
     run_parser.add_argument(
         "-l", "--log-dir", type=str, default=".runs", help="Directory to store logs and results. Defaults to `.runs`."
@@ -164,7 +164,7 @@ def main() -> None:
         "--model",
         type=str,
         default=None,
-        help="Model to use for optimization. Defaults to `o4-mini` when `OPENAI_API_KEY` is set, `claude-sonnet-4-0` when `ANTHROPIC_API_KEY` is set, and `gemini-2.5-pro` when `GEMINI_API_KEY` is set. When multiple keys are set, the priority is `OPENAI_API_KEY` > `ANTHROPIC_API_KEY` > `GEMINI_API_KEY`.",
+        help="Model to use for optimization. Defaults to `o4-mini`. See full list at docs.weco.ai/cli/supported-models",
     )
 
     subparsers = parser.add_subparsers(

weco/optimizer.py

@@ -337,8 +337,6 @@ def execute_optimization(
 
             # Starting from step 1 to steps (inclusive) because the baseline solution is step 0, so we want to optimize for steps worth of steps
             for step in range(1, steps + 1):
-                # Re-read instructions from the original source (file path or string) BEFORE each suggest call
-                current_additional_instructions = read_additional_instructions(additional_instructions=additional_instructions)
                 if run_id:
                     try:
                         current_status_response = get_optimization_run_status(
@@ -356,12 +354,7 @@ def execute_optimization(
 
                 # Send feedback and get next suggestion
                 eval_and_next_solution_response = evaluate_feedback_then_suggest_next_solution(
-                    console=console,
-                    step=step,
-                    run_id=run_id,
-                    execution_output=term_out,
-                    additional_instructions=current_additional_instructions,
-                    auth_headers=auth_headers,
+                    console=console, step=step, run_id=run_id, execution_output=term_out, auth_headers=auth_headers
                 )
                 # Save next solution (.runs/<run-id>/step_<step>.<extension>)
                 write_to_path(fp=runs_dir / f"step_{step}{source_fp.suffix}", content=eval_and_next_solution_response["code"])
@@ -415,16 +408,9 @@ def execute_optimization(
                 )
 
             if not user_stop_requested_flag:
-                # Re-read instructions from the original source (file path or string) BEFORE each suggest call
-                current_additional_instructions = read_additional_instructions(additional_instructions=additional_instructions)
                 # Evaluate the final solution thats been generated
                 eval_and_next_solution_response = evaluate_feedback_then_suggest_next_solution(
-                    console=console,
-                    step=steps,
-                    run_id=run_id,
-                    execution_output=term_out,
-                    additional_instructions=current_additional_instructions,
-                    auth_headers=auth_headers,
+                    console=console, step=steps, run_id=run_id, execution_output=term_out, auth_headers=auth_headers
                 )
                 summary_panel.set_step(step=steps)
                 status_response = get_optimization_run_status(
@@ -632,7 +618,6 @@ def resume_optimization(run_id: str, console: Optional[Console] = None) -> bool:
         log_dir = resume_resp.get("log_dir", ".runs")
         save_logs = bool(resume_resp.get("save_logs", False))
         eval_timeout = resume_resp.get("eval_timeout")
-        additional_instructions = resume_resp.get("additional_instructions")
 
         # Write last solution code to source path
         source_fp = pathlib.Path(source_path)
@@ -739,7 +724,6 @@ def resume_optimization(run_id: str, console: Optional[Console] = None) -> bool:
                     step=step,
                     run_id=resume_resp["run_id"],
                     execution_output=term_out,
-                    additional_instructions=additional_instructions,
                     auth_headers=auth_headers,
                 )
 
@@ -795,7 +779,6 @@ def resume_optimization(run_id: str, console: Optional[Console] = None) -> bool:
                     step=total_steps,
                     run_id=resume_resp["run_id"],
                     execution_output=term_out,
-                    additional_instructions=additional_instructions,
                     auth_headers=auth_headers,
                 )
                 summary_panel.set_step(step=total_steps)

weco/utils.py

@@ -2,13 +2,13 @@ from typing import Any, Dict, List, Tuple, Union
 import json
 import time
 import subprocess
+import psutil
 from rich.layout import Layout
 from rich.live import Live
 from rich.panel import Panel
 import pathlib
 import requests
 from packaging.version import parse as parse_version
-
 from .constants import TRUNCATION_THRESHOLD, TRUNCATION_KEEP_LENGTH, DEFAULT_MODEL, SUPPORTED_FILE_EXTENSIONS
 
 
@@ -108,22 +108,52 @@ def truncate_output(output: str) -> str:
 
 def run_evaluation(eval_command: str, timeout: int | None = None) -> str:
     """Run the evaluation command on the code and return the output."""
+    process = subprocess.Popen(
+        eval_command, shell=True, stdin=subprocess.DEVNULL, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True
+    )
 
-    # Run the eval command as is
     try:
-        result = subprocess.run(eval_command, shell=True, capture_output=True, text=True, check=False, timeout=timeout)
-        # Combine stdout and stderr for complete output
-        output = result.stderr if result.stderr else ""
-        if result.stdout:
-            if len(output) > 0:
-                output += "\n"
-            output += result.stdout
-        return output  # Return full output, no truncation
+        # NOTE: Process tree cleanup only happens on timeout. Normal completion relies on the OS/shell to clean up child processes, which works for typical evaluation scripts.
+        output, _ = process.communicate(timeout=timeout)
+        return output
+
     except subprocess.TimeoutExpired:
+        # Kill process tree
+        try:
+            parent = psutil.Process(process.pid)
+            children = parent.children(recursive=True)
+
+            # Terminate gracefully
+            for child in children:
+                try:
+                    child.terminate()
+                except psutil.NoSuchProcess:
+                    pass
+            try:
+                parent.terminate()
+            except psutil.NoSuchProcess:
+                pass
+
+            # Wait, then force kill survivors
+            _, alive = psutil.wait_procs(children + [parent], timeout=1)
+            for proc in alive:
+                try:
+                    proc.kill()
+                except psutil.NoSuchProcess:
+                    pass
+
+        except psutil.NoSuchProcess:
+            pass
+
+        # Drain pipes
+        try:
+            process.communicate(timeout=1)
+        except (subprocess.TimeoutExpired, ValueError, OSError):
+            pass
+
         return f"Evaluation timed out after {'an unspecified duration' if timeout is None else f'{timeout} seconds'}."
 
 
-# Update Check Function
 def check_for_cli_updates():
     """Checks PyPI for a newer version of the weco package and notifies the user."""
     try:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weco
-Version: 0.3.1
+Version: 0.3.3
 Summary: Documentation for `weco`, a CLI for using Weco AI's code optimizer.
 Author-email: Weco AI Team <contact@weco.ai>
 License: 
@@ -219,6 +219,7 @@ Requires-Dist: packaging
 Requires-Dist: gitingest
 Requires-Dist: fastapi
 Requires-Dist: slowapi
+Requires-Dist: psutil
 Provides-Extra: dev
 Requires-Dist: ruff; extra == "dev"
 Requires-Dist: build; extra == "dev"
@@ -229,7 +230,7 @@ Dynamic: license-file
 
 <div align="center">
   <img src="assets/weco.svg" alt="Weco Logo" width="120" height="120" style="margin-bottom: 20px;">
-  <h1>Weco: The Platform for Self-Improving Code</h1>
+  <h1>Weco: The Code Optimization Agent</h1>
 </div>
 
 [![Python](https://img.shields.io/badge/Python-3.8.0+-blue)](https://www.python.org)
@@ -237,7 +238,7 @@ Dynamic: license-file
 [![docs](https://img.shields.io/website?url=https://docs.weco.ai/&label=docs)](https://docs.weco.ai/)
 [![PyPI Downloads](https://static.pepy.tech/badge/weco?color=4c1)](https://pepy.tech/projects/weco)
 [![arXiv on AIDE](https://img.shields.io/badge/arXiv-AIDE-b31b1b?logo=arxiv&logoColor=white)](https://arxiv.org/abs/2502.13138)
-[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg?labelColor=ffffff&color=F17E01)](https://colab.research.google.com/github/WecoAI/weco-cli/blob/main/examples/hello-kernel-world/colab_notebook_walkthrough.ipynb)
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg?labelColor=ffffff&color=F17E01)](https://colab.research.google.com/github/WecoAI/weco-cli/blob/main/examples/hello-world/colab_notebook_walkthrough.ipynb)
 
 `pip install weco`
 
@@ -261,77 +262,32 @@ Example applications include:
 
 The `weco` CLI leverages a tree search approach guided by LLMs to iteratively explore and refine your code. It automatically applies changes, runs your evaluation script, parses the results, and proposes further improvements based on the specified goal.
 
-![image](https://github.com/user-attachments/assets/a6ed63fa-9c40-498e-aa98-a873e5786509)
 
----
-
-## Setup
-
-1.  **Install the Package:**
-
-    ```bash
-    pip install weco
-    ```
-
-2.  **Authenticate (Required):**
-
-    `weco` now uses a **credit-based billing system** with centralized LLM access. You need to authenticate to use the service:
-
-    - **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)
-
----
-
-## Get Started
-
-### Quick Start (Recommended for New Users)
-
-The easiest way to get started with Weco is to use the **interactive copilot**. Simply navigate to your project directory and run:
-
-```bash
-weco
-```
-
-Or specify a project path:
+## Install the Package
 
 ```bash
-weco /path/to/your/project
+pip install weco
 ```
 
-This launches Weco's interactive copilot that will:
+## Getting Started
 
-1. **Analyze your codebase** using AI to understand your project structure and identify optimization opportunities
-2. **Suggest specific optimizations** tailored to your code (e.g., GPU kernel optimization, model improvements, prompt engineering)
-3. **Generate evaluation scripts** automatically or help you configure existing ones
-4. **Set up the complete optimization pipeline** with appropriate metrics and commands
-5. **Run the optimization** or provide you with the exact command to execute
-
-<div style="background-color: #fff3cd; border: 1px solid #ffeeba; padding: 15px; border-radius: 4px; margin-bottom: 15px;">
-  <strong>⚠️ Warning: Code Modification</strong><br>
-  <code>weco</code> directly modifies the file specified by <code>--source</code> during the optimization process. It is <strong>strongly recommended</strong> to use version control (like Git) to track changes and revert if needed. Alternatively, ensure you have a backup of your original file before running the command. Upon completion, the file will contain the best-performing version of the code found during the run.
-</div>
-
-### Manual Setup
+### Quickstart with an example project
 
 **Configure optimization parameters yourself** - If you need precise control over the optimization parameters, you can use the direct `weco run` command:
 
 **Example: Optimizing Simple PyTorch Operations**
 
 ```bash
-# Navigate to the example directory
-cd examples/hello-kernel-world
-
-# Install dependencies
-pip install torch
+git clone https://github.com/WecoAI/weco-cli.git
+cd weco-cli/examples/hello-world/
+pip install -r requirements.txt
 
-# Run Weco with manual configuration
-weco run --source optimize.py \
-     --eval-command "python evaluate.py --solution-path optimize.py --device cpu" \
+# Run Weco with configuration
+weco run --source module.py \
+     --eval-command "python evaluate.py --path module.py" \
      --metric speedup \
      --goal maximize \
-     --steps 15 \
+     --steps 10 \
      --additional-instructions "Fuse operations in the forward method while ensuring the max float deviation remains small. Maintain the same format of the code."
 ```
 
@@ -361,7 +317,7 @@ 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` | `-M o4-mini`         |
+| `-M, --model`                  | Model identifier for the LLM to use (e.g., `o4-mini`, `claude-sonnet-4-5`, `gpt-5`).                                                                                                        | `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`             |
@@ -369,37 +325,12 @@ For more advanced examples, including [Triton](/examples/triton/README.md), [CUD
 
 ---
 
-### Authentication & Dashboard
-
-The CLI requires a Weco account for authentication and billing.
-
-#### 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
-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:**
-- **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
-
----
-
 ## Command Reference
 
 ### Basic Usage Patterns
 
 | Command | Description | When to Use |
 |---------|-------------|-------------|
-| `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 |
@@ -409,19 +340,23 @@ Weco now requires authentication for all operations. This enables our credit-bas
 You can specify which LLM model to use with the `-M` or `--model` flag:
 
 ```bash
-# Use with onboarding
-weco --model gpt-4o
-
-# Use with direct execution
-weco run --model claude-3.5-sonnet --source optimize.py [other options...]
+weco run --model gpt-5 --source optimize.py [other options...]
 ```
 
-**Available models:**
-- `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)
+**Available models (30 total):**
+
+**OpenAI Models:**
+- GPT-5 Series: `gpt-5.1`, `gpt-5.1-codex`, `gpt-5.1-codex-mini`, `gpt-5-codex`, `gpt-5-pro`, `gpt-5`, `gpt-5-mini`, `gpt-5-nano`
+- O-Series Reasoning: `o3-pro`, `o3`, `o3-mini`, `o4-mini`, `o1-pro`, `o1`, `codex-mini-latest`
+- GPT-4 Series: `gpt-4.1`, `gpt-4.1-mini`, `gpt-4.1-nano`, `gpt-4o`, `gpt-4o-mini`
+
+**Anthropic Claude (via Vertex AI):**
+- `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`
 
-All models are available through Weco's centralized system. If no model is specified, Weco automatically selects the best model for your optimization task.
+All models are available through Weco. If no model is specified, Weco automatically selects the best model for your optimization task.
 
 ---
 

weco-0.3.3.dist-info/RECORD

@@ -0,0 +1,16 @@
+weco/__init__.py,sha256=ClO0uT6GKOA0iSptvP0xbtdycf0VpoPTq37jHtvlhtw,303
+weco/api.py,sha256=zKcI4riwruK6CjV_vcL8RlsJGRXO40iP0WxeETtzPIY,18430
+weco/auth.py,sha256=O31Hoj-Loi8DWJJG2LfeWgUMuNqAUeGDpd2ZGjA9Ah0,9997
+weco/chatbot.py,sha256=EIK2WaOul9gn_yHLThjsZV7RnE8t3XQPwgRkO5tybSU,38415
+weco/cli.py,sha256=5TusCKQ8o4CUdqxOSkvWyjMzt86O9sj21SJoeU0Ni5w,11098
+weco/constants.py,sha256=V6yFugTznKm5EC2_jr4I_whd7sqI80HiPggRn0az580,406
+weco/credits.py,sha256=C08x-TRcLg3ccfKqMGNRY7zBn7t3r7LZ119bxgfztaI,7629
+weco/optimizer.py,sha256=nOKFmwPdFLcQ7RF4ielsD7iRfPMxvGr07pS9ocbW9C8,41282
+weco/panels.py,sha256=fnGPtmvxpx21AuBCtCFu1f_BpSxybNr2lhjIIKIutrY,16133
+weco/utils.py,sha256=erDDrA_g3KSlel6YEAGALlV_k8ftT-VQnPT1BrmzK8k,7021
+weco-0.3.3.dist-info/licenses/LICENSE,sha256=9LUfoGHjLPtak2zps2kL2tm65HAZIICx_FbLaRuS4KU,11337
+weco-0.3.3.dist-info/METADATA,sha256=aAE9oMp_SKJiQSG189k7fsb5ovvidj2t1DTuqNfvc68,28700
+weco-0.3.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+weco-0.3.3.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
+weco-0.3.3.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
+weco-0.3.3.dist-info/RECORD,,

weco-0.3.1.dist-info/RECORD

@@ -1,16 +0,0 @@
-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,,