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

weco/api.py

@@ -1,29 +1,11 @@
 import sys
 from typing import Dict, Any, Optional, Union, Tuple, List
-
 import requests
-from requests.adapters import HTTPAdapter
-from urllib3.util.retry import Retry
 from rich.console import Console
 
 from weco import __pkg_version__, __base_url__
 
 
-# --- Session Configuration ---
-def _get_weco_session() -> requests.Session:
-    session = requests.Session()
-    retry_strategy = Retry(
-        total=3,
-        status_forcelist=[429, 500, 502, 503, 504],  # Retry on these server errors and rate limiting
-        allowed_methods=["HEAD", "GET", "PUT", "POST", "DELETE", "OPTIONS"],  # Case-insensitive
-        backoff_factor=1,  # e.g., sleep for 0s, 2s, 4s between retries (factor * (2 ** ({number of total retries} - 1)))
-    )
-    adapter = HTTPAdapter(max_retries=retry_strategy)
-    session.mount("http://", adapter)
-    session.mount("https://", adapter)
-    return session
-
-
 def handle_api_error(e: requests.exceptions.HTTPError, console: Console) -> None:
     """Extract and display error messages from API responses in a structured format."""
     try:
@@ -53,8 +35,7 @@ def start_optimization_run(
     """Start the optimization run."""
     with console.status("[bold green]Starting Optimization..."):
         try:
-            session = _get_weco_session()
-            response = session.post(
+            response = requests.post(
                 f"{__base_url__}/runs",
                 json={
                     "source_code": source_code,
@@ -76,8 +57,8 @@ def start_optimization_run(
         except requests.exceptions.HTTPError as e:
             handle_api_error(e, console)
             sys.exit(1)
-        except requests.exceptions.RequestException as e:
-            console.print(f"[bold red]Network Error starting run: {e}[/]")
+        except Exception as e:
+            console.print(f"[bold red]Error starting run: {e}[/]")
             sys.exit(1)
 
 
@@ -91,8 +72,7 @@ def evaluate_feedback_then_suggest_next_solution(
 ) -> Dict[str, Any]:
     """Evaluate the feedback and suggest the next solution."""
     try:
-        session = _get_weco_session()
-        response = session.post(
+        response = requests.post(
             f"{__base_url__}/runs/{run_id}/suggest",
             json={
                 "execution_output": execution_output,
@@ -108,8 +88,8 @@ def evaluate_feedback_then_suggest_next_solution(
         # Allow caller to handle suggest errors, maybe retry or terminate
         handle_api_error(e, Console())  # Use default console if none passed
         raise  # Re-raise the exception
-    except requests.exceptions.RequestException as e:
-        print(f"Network Error during suggest: {e}")  # Use print as console might not be available
+    except Exception as e:
+        print(f"Error: {e}")  # Use print as console might not be available
         raise  # Re-raise the exception
 
 
@@ -118,8 +98,7 @@ def get_optimization_run_status(
 ) -> Dict[str, Any]:
     """Get the current status of the optimization run."""
     try:
-        session = _get_weco_session()
-        response = session.get(
+        response = requests.get(
             f"{__base_url__}/runs/{run_id}", params={"include_history": include_history}, headers=auth_headers, timeout=timeout
         )
         response.raise_for_status()
@@ -127,16 +106,15 @@ def get_optimization_run_status(
     except requests.exceptions.HTTPError as e:
         handle_api_error(e, Console())  # Use default console
         raise  # Re-raise
-    except requests.exceptions.RequestException as e:
-        print(f"Network Error getting status: {e}")
+    except Exception as e:
+        print(f"Error getting run status: {e}")
         raise  # Re-raise
 
 
 def send_heartbeat(run_id: str, auth_headers: dict = {}, timeout: Union[int, Tuple[int, int]] = 10) -> bool:
     """Send a heartbeat signal to the backend."""
     try:
-        session = _get_weco_session()
-        response = session.put(f"{__base_url__}/runs/{run_id}/heartbeat", headers=auth_headers, timeout=timeout)
+        response = requests.put(f"{__base_url__}/runs/{run_id}/heartbeat", headers=auth_headers, timeout=timeout)
         response.raise_for_status()
         return True
     except requests.exceptions.HTTPError as e:
@@ -145,8 +123,8 @@ def send_heartbeat(run_id: str, auth_headers: dict = {}, timeout: Union[int, Tup
         else:
             print(f"Heartbeat failed for run {run_id}: HTTP {e.response.status_code}", file=sys.stderr)
         return False
-    except requests.exceptions.RequestException as e:
-        print(f"Heartbeat network error for run {run_id}: {e}", file=sys.stderr)
+    except Exception as e:
+        print(f"Error sending heartbeat for run {run_id}: {e}", file=sys.stderr)
         return False
 
 
@@ -160,8 +138,7 @@ def report_termination(
 ) -> bool:
     """Report the termination reason to the backend."""
     try:
-        session = _get_weco_session()
-        response = session.post(
+        response = requests.post(
             f"{__base_url__}/runs/{run_id}/terminate",
             json={"status_update": status_update, "termination_reason": reason, "termination_details": details},
             headers=auth_headers,
@@ -169,7 +146,7 @@ def report_termination(
         )
         response.raise_for_status()
         return True
-    except requests.exceptions.RequestException as e:
+    except Exception as e:
         print(f"Warning: Failed to report termination to backend for run {run_id}: {e}", file=sys.stderr)
         return False
 
@@ -211,8 +188,7 @@ def get_optimization_suggestions_from_codebase(
     """Analyze codebase and get optimization suggestions using the model-agnostic backend API."""
     try:
         model, api_key_dict = _determine_model_and_api_key()
-        session = _get_weco_session()
-        response = session.post(
+        response = requests.post(
             f"{__base_url__}/onboard/analyze-codebase",
             json={
                 "gitingest_summary": gitingest_summary,
@@ -231,11 +207,8 @@ def get_optimization_suggestions_from_codebase(
     except requests.exceptions.HTTPError as e:
         handle_api_error(e, console)
         return None
-    except requests.exceptions.RequestException as e:
-        console.print(f"[bold red]Network Error getting optimization suggestions: {e}[/]")
-        return None
     except Exception as e:
-        console.print(f"[bold red]Error calling backend API: {e}[/]")
+        console.print(f"[bold red]Error: {e}[/]")
         return None
 
 
@@ -250,8 +223,7 @@ def generate_evaluation_script_and_metrics(
     """Generate evaluation script and determine metrics using the model-agnostic backend API."""
     try:
         model, api_key_dict = _determine_model_and_api_key()
-        session = _get_weco_session()
-        response = session.post(
+        response = requests.post(
             f"{__base_url__}/onboard/generate-script",
             json={
                 "target_file": target_file,
@@ -266,15 +238,11 @@ def generate_evaluation_script_and_metrics(
         response.raise_for_status()
         result = response.json()
         return result.get("script_content"), result.get("metric_name"), result.get("goal"), result.get("reasoning")
-
     except requests.exceptions.HTTPError as e:
         handle_api_error(e, console)
         return None, None, None, None
-    except requests.exceptions.RequestException as e:
-        console.print(f"[bold red]Network Error generating evaluation script: {e}[/]")
-        return None, None, None, None
     except Exception as e:
-        console.print(f"[bold red]Error calling backend API: {e}[/]")
+        console.print(f"[bold red]Error: {e}[/]")
         return None, None, None, None
 
 
@@ -291,8 +259,7 @@ def analyze_evaluation_environment(
     """Analyze existing evaluation scripts and environment using the model-agnostic backend API."""
     try:
         model, api_key_dict = _determine_model_and_api_key()
-        session = _get_weco_session()
-        response = session.post(
+        response = requests.post(
             f"{__base_url__}/onboard/analyze-environment",
             json={
                 "target_file": target_file,
@@ -312,11 +279,8 @@ def analyze_evaluation_environment(
     except requests.exceptions.HTTPError as e:
         handle_api_error(e, console)
         return None
-    except requests.exceptions.RequestException as e:
-        console.print(f"[bold red]Network Error analyzing evaluation environment: {e}[/]")
-        return None
     except Exception as e:
-        console.print(f"[bold red]Error calling backend API: {e}[/]")
+        console.print(f"[bold red]Error: {e}[/]")
         return None
 
 
@@ -331,8 +295,7 @@ def analyze_script_execution_requirements(
     """Analyze script to determine proper execution command using the model-agnostic backend API."""
     try:
         model, api_key_dict = _determine_model_and_api_key()
-        session = _get_weco_session()
-        response = session.post(
+        response = requests.post(
             f"{__base_url__}/onboard/analyze-script",
             json={
                 "script_content": script_content,
@@ -351,9 +314,6 @@ def analyze_script_execution_requirements(
     except requests.exceptions.HTTPError as e:
         handle_api_error(e, console)
         return f"python {script_path}"
-    except requests.exceptions.RequestException as e:
-        console.print(f"[bold red]Network Error analyzing script execution: {e}[/]")
-        return f"python {script_path}"
     except Exception as e:
-        console.print(f"[bold red]Error calling backend API: {e}[/]")
+        console.print(f"[bold red]Error: {e}[/]")
         return f"python {script_path}"

{weco-0.2.22.dist-info → weco-0.2.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weco
-Version: 0.2.22
+Version: 0.2.23
 Summary: Documentation for `weco`, a CLI for using Weco AI's code optimizer.
 Author-email: Weco AI Team <contact@weco.ai>
 License: MIT
@@ -30,10 +30,11 @@ Dynamic: license-file
 </div>
 
 [![Python](https://img.shields.io/badge/Python-3.8.0+-blue)](https://www.python.org)
+[![PyPI version](https://img.shields.io/pypi/v/weco?label=PyPI%20version&color=f05138&labelColor=555555)](https://badge.fury.io/py/weco)
 [![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)
-[![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)
+[![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)
 
 `pip install weco`
 
@@ -73,9 +74,9 @@ The `weco` CLI leverages a tree search approach guided by LLMs to iteratively ex
 
     `weco` requires API keys for the LLMs it uses internally. You **must** provide these keys via environment variables:
 
-    - **OpenAI:** `export OPENAI_API_KEY="your_key_here"` (Create your API key [here](https://platform.openai.com/api-keys))
-    - **Anthropic:** `export ANTHROPIC_API_KEY="your_key_here"` (Create your 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 API key [here](https://aistudio.google.com/apikey) to use `weco` for free.)
+    - **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.)
 
 ---
 
@@ -252,20 +253,17 @@ Weco will parse this output to extract the numerical value (1.5 in this case) as
 We welcome your contributions! To get started:
 
 1.  **Fork & Clone the Repository:**
-
     ```bash
     git clone https://github.com/WecoAI/weco-cli.git
     cd weco-cli
     ```
 
 2.  **Install Dependencies:**
-
     ```bash
     pip install -e ".[dev]"
     ```
 
 3.  **Create a Feature Branch:**
-
     ```bash
     git checkout -b feature/your-feature-name
     ```

{weco-0.2.22.dist-info → weco-0.2.23.dist-info}/RECORD

@@ -1,14 +1,14 @@
 weco/__init__.py,sha256=ClO0uT6GKOA0iSptvP0xbtdycf0VpoPTq37jHtvlhtw,303
-weco/api.py,sha256=X2b9iKhpXkEqmQWjecHcQruj0bRIiTpMOGGJMLpkCug,13827
+weco/api.py,sha256=hqAyMHGQhJr8BZTUo0jznfO0ossGU3ehEZZehPu6Mu8,11838
 weco/auth.py,sha256=6bDQv07sx7uxA9CrN3HqUdCHV6nqXO41PGCicquvB00,9919
 weco/chatbot.py,sha256=H6d5yK9MB3pqpE7XVh_HAi1YAmxKy0v_xdozVSlKPCc,36959
 weco/cli.py,sha256=Jy7kQEsNKdV7Wds9Z0DIWBeLpEVyssIiOBiQ4zCl3Lw,7862
 weco/optimizer.py,sha256=z86-js_rvLMv3J8zCqvtc1xJC0EA0WqrN9_BlmX2RK4,23259
 weco/panels.py,sha256=Cnro4Q65n7GGh0FBXuB_OGSxRVobd4k5lOuBViTQaaM,15591
 weco/utils.py,sha256=5Pbhv_5wbTRv93Ws7aJfIOtcxeeqNrDRT3bV6YFOdgM,6032
-weco-0.2.22.dist-info/licenses/LICENSE,sha256=p_GQqJBvuZgkLNboYKyH-5dhpTDlKs2wq2TVM55WrWE,1065
-weco-0.2.22.dist-info/METADATA,sha256=EQ6vcDHjDyU5MbyNucVQU9LccJtz2i6wzghjMB25M6U,15251
-weco-0.2.22.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-weco-0.2.22.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
-weco-0.2.22.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
-weco-0.2.22.dist-info/RECORD,,
+weco-0.2.23.dist-info/licenses/LICENSE,sha256=p_GQqJBvuZgkLNboYKyH-5dhpTDlKs2wq2TVM55WrWE,1065
+weco-0.2.23.dist-info/METADATA,sha256=E_u4w-WaRzBVYrlEzraepi0y7E1Cc1zHOhBu4Cp01TI,15445
+weco-0.2.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+weco-0.2.23.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
+weco-0.2.23.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
+weco-0.2.23.dist-info/RECORD,,