weco 0.2.22__py3-none-any.whl → 0.2.24__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (15) hide show
  1. weco/api.py +84 -87
  2. weco/auth.py +7 -5
  3. weco/chatbot.py +34 -23
  4. weco/cli.py +10 -1
  5. weco/constants.py +7 -0
  6. weco/optimizer.py +26 -21
  7. weco/panels.py +105 -59
  8. weco/utils.py +47 -12
  9. {weco-0.2.22.dist-info → weco-0.2.24.dist-info}/METADATA +36 -25
  10. weco-0.2.24.dist-info/RECORD +15 -0
  11. weco-0.2.22.dist-info/RECORD +0 -14
  12. {weco-0.2.22.dist-info → weco-0.2.24.dist-info}/WHEEL +0 -0
  13. {weco-0.2.22.dist-info → weco-0.2.24.dist-info}/entry_points.txt +0 -0
  14. {weco-0.2.22.dist-info → weco-0.2.24.dist-info}/licenses/LICENSE +0 -0
  15. {weco-0.2.22.dist-info → weco-0.2.24.dist-info}/top_level.txt +0 -0
{weco-0.2.22.dist-info → weco-0.2.24.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: weco
3
- Version: 0.2.22
3
+ Version: 0.2.24
4
4
  Summary: Documentation for `weco`, a CLI for using Weco AI's code optimizer.
5
5
  Author-email: Weco AI Team <contact@weco.ai>
6
6
  License: MIT
@@ -16,6 +16,8 @@ Requires-Dist: requests
16
16
  Requires-Dist: rich
17
17
  Requires-Dist: packaging
18
18
  Requires-Dist: gitingest
19
+ Requires-Dist: fastapi
20
+ Requires-Dist: slowapi
19
21
  Provides-Extra: dev
20
22
  Requires-Dist: ruff; extra == "dev"
21
23
  Requires-Dist: build; extra == "dev"
@@ -30,10 +32,11 @@ Dynamic: license-file
30
32
  </div>
31
33
 
32
34
  [![Python](https://img.shields.io/badge/Python-3.8.0+-blue)](https://www.python.org)
35
+ [![PyPI version](https://img.shields.io/pypi/v/weco?label=PyPI%20version&color=f05138&labelColor=555555)](https://badge.fury.io/py/weco)
33
36
  [![docs](https://img.shields.io/website?url=https://docs.weco.ai/&label=docs)](https://docs.weco.ai/)
34
- [![PyPI version](https://badge.fury.io/py/weco.svg)](https://badge.fury.io/py/weco)
35
- [![AIDE](https://img.shields.io/badge/AI--Driven_Exploration-arXiv-orange?style=flat-square&logo=arxiv)](https://arxiv.org/abs/2502.13138)
36
- [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/WecoAI/weco-cli/blob/main/examples/hello-kernel-world/colab_notebook_walkthrough.ipynb)
37
+ [![PyPI Downloads](https://static.pepy.tech/badge/weco?color=4c1)](https://pepy.tech/projects/weco)
38
+ [![arXiv on AIDE](https://img.shields.io/badge/arXiv-AIDE-b31b1b?logo=arxiv&logoColor=white)](https://arxiv.org/abs/2502.13138)
39
+ [![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)
37
40
 
38
41
  `pip install weco`
39
42
 
@@ -73,9 +76,9 @@ The `weco` CLI leverages a tree search approach guided by LLMs to iteratively ex
73
76
 
74
77
  `weco` requires API keys for the LLMs it uses internally. You **must** provide these keys via environment variables:
75
78
 
76
- - **OpenAI:** `export OPENAI_API_KEY="your_key_here"` (Create your API key [here](https://platform.openai.com/api-keys))
77
- - **Anthropic:** `export ANTHROPIC_API_KEY="your_key_here"` (Create your API key [here](https://console.anthropic.com/settings/keys))
78
- - **Google:** `export GEMINI_API_KEY="your_key_here"` (Google AI Studio has a free API usage quota. Create your API key [here](https://aistudio.google.com/apikey) to use `weco` for free.)
79
+ - **OpenAI:** `export OPENAI_API_KEY="your_key_here"` (Create your OpenAI API key [here](https://platform.openai.com/api-keys))
80
+ - **Anthropic:** `export ANTHROPIC_API_KEY="your_key_here"` (Create your Anthropic API key [here](https://console.anthropic.com/settings/keys))
81
+ - **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.)
79
82
 
80
83
  ---
81
84
 
@@ -157,6 +160,7 @@ For more advanced examples, including [Triton](/examples/triton/README.md), [CUD
157
160
  | `-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` |
158
161
  | `-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"`|
159
162
  | `-l, --log-dir` | Path to the directory to log intermediate steps and final optimization result. | `.runs/` | `-l ./logs/` |
163
+ | `--eval-timeout` | Timeout in seconds for each step in evaluation. | No timeout (unlimited) | `--eval-timeout 3600` |
160
164
 
161
165
  ---
162
166
 
@@ -247,31 +251,38 @@ Final speedup value = 1.5
247
251
 
248
252
  Weco will parse this output to extract the numerical value (1.5 in this case) associated with the metric name ('speedup').
249
253
 
250
- ## Contributing
254
+ ## Supported Models
251
255
 
252
- We welcome your contributions! To get started:
256
+ Weco supports the following LLM models:
253
257
 
254
- 1. **Fork & Clone the Repository:**
258
+ ### OpenAI Models
259
+ - `o3`
260
+ - `o3-mini`
261
+ - `o4-mini`
262
+ - `o1-pro`
263
+ - `o1`
264
+ - `gpt-4.1`
265
+ - `gpt-4.1-mini`
266
+ - `gpt-4.1-nano`
267
+ - `gpt-4o`
268
+ - `gpt-4o-mini`
255
269
 
256
- ```bash
257
- git clone https://github.com/WecoAI/weco-cli.git
258
- cd weco-cli
259
- ```
270
+ ### Anthropic Models
271
+ - `claude-opus-4-0`
272
+ - `claude-sonnet-4-0`
273
+ - `claude-3-7-sonnet-latest`
260
274
 
261
- 2. **Install Dependencies:**
262
-
263
- ```bash
264
- pip install -e ".[dev]"
265
- ```
275
+ ### Gemini Models
276
+ - `gemini-2.5-pro`
277
+ - `gemini-2.5-flash`
278
+ - `gemini-2.5-flash-lite`
266
279
 
267
- 3. **Create a Feature Branch:**
280
+ 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.
268
281
 
269
- ```bash
270
- git checkout -b feature/your-feature-name
271
- ```
282
+ ---
272
283
 
273
- 4. **Make Changes:** Ensure your code adheres to our style guidelines and includes relevant tests.
284
+ ## Contributing
274
285
 
275
- 5. **Commit, Push & Open a PR**: Commit your changes, and open a pull request with a clear description of your enhancements.
286
+ We welcome contributions! Please see [contributing.md](contributing.md) for detailed guidelines on how to contribute to this project.
276
287
 
277
288
  ---
weco-0.2.24.dist-info/RECORD ADDED
@@ -0,0 +1,15 @@
1
+ weco/__init__.py,sha256=ClO0uT6GKOA0iSptvP0xbtdycf0VpoPTq37jHtvlhtw,303
2
+ weco/api.py,sha256=sdRgSvbjStGr5Bywkp4Pkuo9jgtGp95tNzsVXrXmH00,13380
3
+ weco/auth.py,sha256=KMSAsN1V5wx7KUsYL1cEOOiG29Pqf4Exb3EPW4mAWC0,10003
4
+ weco/chatbot.py,sha256=MSG3yybZ0fHTpwW3If9TpOWOAE4EqrcUWpe3q16AjZE,37673
5
+ weco/cli.py,sha256=75JdYpUf0qdJW5pjycZoHUKA2n2MYj33qYM9WQwLT2s,8184
6
+ weco/constants.py,sha256=vfQGDf9_kzlN9BzEFvMsd0EeXOsRyzpvSWyxOJgRauE,168
7
+ weco/optimizer.py,sha256=E5Jii0rTdI9pxkstaM7ipjF4viX5XnSy5w71gdps4Ws,23662
8
+ weco/panels.py,sha256=jwAV_uoa0ZI9vjyey-hSY3rx4pfNNkZvPzqt-iz-RXo,16808
9
+ weco/utils.py,sha256=HecbOqD5rBuVhUkLixVrTWBMJ-ZMAhK-889N-lCk3dQ,7335
10
+ weco-0.2.24.dist-info/licenses/LICENSE,sha256=p_GQqJBvuZgkLNboYKyH-5dhpTDlKs2wq2TVM55WrWE,1065
11
+ weco-0.2.24.dist-info/METADATA,sha256=yFq-NPCUsh59B4Pj9IlYjDo53Kzjya5Rk_9-QmheGLM,16069
12
+ weco-0.2.24.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
13
+ weco-0.2.24.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
14
+ weco-0.2.24.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
15
+ weco-0.2.24.dist-info/RECORD,,
weco-0.2.22.dist-info/RECORD DELETED
@@ -1,14 +0,0 @@
1
- weco/__init__.py,sha256=ClO0uT6GKOA0iSptvP0xbtdycf0VpoPTq37jHtvlhtw,303
2
- weco/api.py,sha256=X2b9iKhpXkEqmQWjecHcQruj0bRIiTpMOGGJMLpkCug,13827
3
- weco/auth.py,sha256=6bDQv07sx7uxA9CrN3HqUdCHV6nqXO41PGCicquvB00,9919
4
- weco/chatbot.py,sha256=H6d5yK9MB3pqpE7XVh_HAi1YAmxKy0v_xdozVSlKPCc,36959
5
- weco/cli.py,sha256=Jy7kQEsNKdV7Wds9Z0DIWBeLpEVyssIiOBiQ4zCl3Lw,7862
6
- weco/optimizer.py,sha256=z86-js_rvLMv3J8zCqvtc1xJC0EA0WqrN9_BlmX2RK4,23259
7
- weco/panels.py,sha256=Cnro4Q65n7GGh0FBXuB_OGSxRVobd4k5lOuBViTQaaM,15591
8
- weco/utils.py,sha256=5Pbhv_5wbTRv93Ws7aJfIOtcxeeqNrDRT3bV6YFOdgM,6032
9
- weco-0.2.22.dist-info/licenses/LICENSE,sha256=p_GQqJBvuZgkLNboYKyH-5dhpTDlKs2wq2TVM55WrWE,1065
10
- weco-0.2.22.dist-info/METADATA,sha256=EQ6vcDHjDyU5MbyNucVQU9LccJtz2i6wzghjMB25M6U,15251
11
- weco-0.2.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
12
- weco-0.2.22.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
13
- weco-0.2.22.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
14
- weco-0.2.22.dist-info/RECORD,,
{weco-0.2.22.dist-info → weco-0.2.24.dist-info}/WHEEL RENAMED
File without changes