weco 0.2.16__tar.gz → 0.2.17__tar.gz

{weco-0.2.16 → weco-0.2.17}/.github/workflows/release.yml

@@ -43,8 +43,6 @@ jobs:
             OLD_VERSION=""
           fi
 
-          OLD_VERSION="0.2.15"
-
           echo "Previous version: $OLD_VERSION"
           echo "Current  version: $NEW_VERSION"
 

{weco-0.2.16 → weco-0.2.17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weco
-Version: 0.2.16
+Version: 0.2.17
 Summary: Documentation for `weco`, a CLI for using Weco AI's code optimizer.
 Author-email: Weco AI Team <contact@weco.ai>
 License: MIT
@@ -20,12 +20,21 @@ Requires-Dist: build; extra == "dev"
 Requires-Dist: setuptools_scm; extra == "dev"
 Dynamic: license-file
 
-# Weco: The AI Research Engineer
+<div align="center">
 
-[![Python](https://img.shields.io/badge/Python-3.12.0-blue)](https://www.python.org)
+# Weco: The AI Code Optimizer
+
+[![Python](https://img.shields.io/badge/Python-3.8.0+-blue)](https://www.python.org)
+[![docs](https://img.shields.io/website?url=https://docs.weco.ai/&label=docs)](https://docs.weco.ai/)
 [![PyPI version](https://badge.fury.io/py/weco.svg)](https://badge.fury.io/py/weco)
 [![AIDE](https://img.shields.io/badge/AI--Driven_Exploration-arXiv-orange?style=flat-square&logo=arxiv)](https://arxiv.org/abs/2502.13138)
 
+<code>pip install weco</code>
+
+</div>
+
+---
+
 Weco systematically optimizes your code, guided directly by your evaluation metrics.
 
 Example applications include:
@@ -101,7 +110,7 @@ This command starts the optimization process.
 
 This basic example shows how to optimize a simple PyTorch function for speedup.
 
-For more advanced examples, including [Triton](/examples/triton/README.md), [CUDA kernel optimization](/examples/cuda/README.md)**, and **[ML model optimization](/examples/spaceship-titanic/README.md)**, please see the `README.md` files within the corresponding subdirectories under the [`examples/`](./examples/) folder.
+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](https://github.com/WecoAI/weco-cli/tree/main/examples/prompt), please see the `README.md` files within the corresponding subdirectories under the [`examples/`](./examples/) folder.
 
 ```bash
 # Navigate to the example directory
@@ -136,7 +145,6 @@ weco run --source optimize.py \
 | `--model`                   | Model identifier for the LLM to use (e.g., `gpt-4o`, `claude-3.5-sonnet`). Recommended models to try include `o3-mini`, `claude-3-haiku`, and `gemini-2.5-pro-exp-03-25`. | Yes      |
 | `--additional-instructions` | (Optional) Natural language description of specific instructions OR path to a file containing detailed instructions to guide the LLM.                                     | No       |
 | `--log-dir`                 | (Optional) Path to the directory to log intermediate steps and final optimization result. Defaults to `.runs/`.                                                           | No       |
-| `--preserve-source`         | (Optional) If set, do not overwrite the original `--source` file. Modifications and the best solution will still be saved in the `--log-dir`.                             | No       |
 
 ---
 

{weco-0.2.16 → weco-0.2.17}/README.md

@@ -1,9 +1,18 @@
-# Weco: The AI Research Engineer
+<div align="center">
 
-[![Python](https://img.shields.io/badge/Python-3.12.0-blue)](https://www.python.org)
+# Weco: The AI Code Optimizer
+
+[![Python](https://img.shields.io/badge/Python-3.8.0+-blue)](https://www.python.org)
+[![docs](https://img.shields.io/website?url=https://docs.weco.ai/&label=docs)](https://docs.weco.ai/)
 [![PyPI version](https://badge.fury.io/py/weco.svg)](https://badge.fury.io/py/weco)
 [![AIDE](https://img.shields.io/badge/AI--Driven_Exploration-arXiv-orange?style=flat-square&logo=arxiv)](https://arxiv.org/abs/2502.13138)
 
+<code>pip install weco</code>
+
+</div>
+
+---
+
 Weco systematically optimizes your code, guided directly by your evaluation metrics.
 
 Example applications include:
@@ -79,7 +88,7 @@ This command starts the optimization process.
 
 This basic example shows how to optimize a simple PyTorch function for speedup.
 
-For more advanced examples, including [Triton](/examples/triton/README.md), [CUDA kernel optimization](/examples/cuda/README.md)**, and **[ML model optimization](/examples/spaceship-titanic/README.md)**, please see the `README.md` files within the corresponding subdirectories under the [`examples/`](./examples/) folder.
+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](https://github.com/WecoAI/weco-cli/tree/main/examples/prompt), please see the `README.md` files within the corresponding subdirectories under the [`examples/`](./examples/) folder.
 
 ```bash
 # Navigate to the example directory
@@ -114,7 +123,6 @@ weco run --source optimize.py \
 | `--model`                   | Model identifier for the LLM to use (e.g., `gpt-4o`, `claude-3.5-sonnet`). Recommended models to try include `o3-mini`, `claude-3-haiku`, and `gemini-2.5-pro-exp-03-25`. | Yes      |
 | `--additional-instructions` | (Optional) Natural language description of specific instructions OR path to a file containing detailed instructions to guide the LLM.                                     | No       |
 | `--log-dir`                 | (Optional) Path to the directory to log intermediate steps and final optimization result. Defaults to `.runs/`.                                                           | No       |
-| `--preserve-source`         | (Optional) If set, do not overwrite the original `--source` file. Modifications and the best solution will still be saved in the `--log-dir`.                             | No       |
 
 ---
 

{weco-0.2.16 → weco-0.2.17}/examples/prompt/README.md

@@ -1,4 +1,3 @@
-# weco-cli/examples/prompt/README.md
 # AIME Prompt Engineering Example with Weco
 
 This example shows how **Weco** can iteratively improve a prompt for solving American Invitational Mathematics Examination (AIME) problems. The experiment runs locally, requires only two short Python files, and aims to improve the accuracy metric.
@@ -97,4 +96,4 @@ Weco then mutates the config, tries again, and gradually pushes the accuracy hig
 * `eval_aime.py` slices the **Maxwell‑Jia/AIME_2024** dataset to twenty problems for fast feedback. You can change the slice in one line.
 * The script sends model calls in parallel via `ThreadPoolExecutor`, so network latency is hidden.
 * Every five completed items, the script logs progress and elapsed time.
-* The final line `accuracy: value` is the only part Weco needs for guidance.
+* The final line `accuracy: value` is the only part Weco needs for guidance.

{weco-0.2.16 → weco-0.2.17}/pyproject.toml

@@ -10,7 +10,7 @@ authors = [
 ]
 description = "Documentation for `weco`, a CLI for using Weco AI's code optimizer."
 readme = "README.md"
-version = "0.2.16"
+version = "0.2.17"
 license = {text = "MIT"}
 requires-python = ">=3.8"
 dependencies = ["requests", "rich"]

{weco-0.2.16 → weco-0.2.17}/weco/__init__.py

@@ -1,7 +1,7 @@
 import os
 
 # DO NOT EDIT
-__pkg_version__ = "0.2.16"
+__pkg_version__ = "0.2.17"
 __api_version__ = "v1"
 
 __base_url__ = f"https://api.weco.ai/{__api_version__}"

{weco-0.2.16 → weco-0.2.17}/weco/cli.py

@@ -194,11 +194,6 @@ def main() -> None:
         type=str,
         help="Description of additional instruction or path to a file containing additional instructions",
     )
-    run_parser.add_argument(
-        "--preserve-source",
-        action="store_true",
-        help="If set, do not overwrite the original source file; only save modified versions in the runs directory",
-    )
 
     # --- Logout Command ---
     _ = subparsers.add_parser("logout", help="Log out from Weco and clear saved API key.")
@@ -310,9 +305,8 @@ def main() -> None:
                 # Write the initial code string to the logs
                 write_to_path(fp=runs_dir / f"step_0{source_fp.suffix}", content=session_response["code"])
 
-                # Write the initial code string to the source file path (if not preserving)
-                if not args.preserve_source:
-                    write_to_path(fp=source_fp, content=session_response["code"])
+                # Write the initial code string to the source file path
+                write_to_path(fp=source_fp, content=session_response["code"])
 
                 # Update the panels with the initial solution
                 summary_panel.set_session_id(session_id=session_id)  # Add session id now that we have it
@@ -398,8 +392,7 @@ def main() -> None:
                     )
 
                     # Write the next solution to the source file
-                    if not args.preserve_source:
-                        write_to_path(fp=source_fp, content=eval_and_next_solution_response["code"])
+                    write_to_path(fp=source_fp, content=eval_and_next_solution_response["code"])
 
                     # Get the optimization session status for
                     # the best solution, its score, and the history to plot the tree
@@ -560,8 +553,7 @@ def main() -> None:
                 write_to_path(fp=runs_dir / f"best{source_fp.suffix}", content=best_solution_content)
 
                 # write the best solution to the source file
-                if not args.preserve_source:
-                    write_to_path(fp=source_fp, content=best_solution_content)
+                write_to_path(fp=source_fp, content=best_solution_content)
 
             console.print(end_optimization_layout)
 

{weco-0.2.16 → weco-0.2.17}/weco.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weco
-Version: 0.2.16
+Version: 0.2.17
 Summary: Documentation for `weco`, a CLI for using Weco AI's code optimizer.
 Author-email: Weco AI Team <contact@weco.ai>
 License: MIT
@@ -20,12 +20,21 @@ Requires-Dist: build; extra == "dev"
 Requires-Dist: setuptools_scm; extra == "dev"
 Dynamic: license-file
 
-# Weco: The AI Research Engineer
+<div align="center">
 
-[![Python](https://img.shields.io/badge/Python-3.12.0-blue)](https://www.python.org)
+# Weco: The AI Code Optimizer
+
+[![Python](https://img.shields.io/badge/Python-3.8.0+-blue)](https://www.python.org)
+[![docs](https://img.shields.io/website?url=https://docs.weco.ai/&label=docs)](https://docs.weco.ai/)
 [![PyPI version](https://badge.fury.io/py/weco.svg)](https://badge.fury.io/py/weco)
 [![AIDE](https://img.shields.io/badge/AI--Driven_Exploration-arXiv-orange?style=flat-square&logo=arxiv)](https://arxiv.org/abs/2502.13138)
 
+<code>pip install weco</code>
+
+</div>
+
+---
+
 Weco systematically optimizes your code, guided directly by your evaluation metrics.
 
 Example applications include:
@@ -101,7 +110,7 @@ This command starts the optimization process.
 
 This basic example shows how to optimize a simple PyTorch function for speedup.
 
-For more advanced examples, including [Triton](/examples/triton/README.md), [CUDA kernel optimization](/examples/cuda/README.md)**, and **[ML model optimization](/examples/spaceship-titanic/README.md)**, please see the `README.md` files within the corresponding subdirectories under the [`examples/`](./examples/) folder.
+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](https://github.com/WecoAI/weco-cli/tree/main/examples/prompt), please see the `README.md` files within the corresponding subdirectories under the [`examples/`](./examples/) folder.
 
 ```bash
 # Navigate to the example directory
@@ -136,7 +145,6 @@ weco run --source optimize.py \
 | `--model`                   | Model identifier for the LLM to use (e.g., `gpt-4o`, `claude-3.5-sonnet`). Recommended models to try include `o3-mini`, `claude-3-haiku`, and `gemini-2.5-pro-exp-03-25`. | Yes      |
 | `--additional-instructions` | (Optional) Natural language description of specific instructions OR path to a file containing detailed instructions to guide the LLM.                                     | No       |
 | `--log-dir`                 | (Optional) Path to the directory to log intermediate steps and final optimization result. Defaults to `.runs/`.                                                           | No       |
-| `--preserve-source`         | (Optional) If set, do not overwrite the original `--source` file. Modifications and the best solution will still be saved in the `--log-dir`.                             | No       |
 
 ---