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

weco/api.py

@@ -14,8 +14,6 @@ def handle_api_error(e: requests.exceptions.HTTPError, console: Console) -> None
     except (ValueError, KeyError):  # Handle cases where response is not JSON or detail key is missing
         detail = f"HTTP {e.response.status_code} Error: {e.response.text}"
     console.print(f"[bold red]{detail}[/]")
-    # Avoid exiting here, let the caller decide if the error is fatal
-    # sys.exit(1)
 
 
 def start_optimization_run(
@@ -32,7 +30,7 @@ def start_optimization_run(
     api_keys: Dict[str, Any] = {},
     auth_headers: dict = {},
     timeout: Union[int, Tuple[int, int]] = DEFAULT_API_TIMEOUT,
-) -> Dict[str, Any]:
+) -> Optional[Dict[str, Any]]:
     """Start the optimization run."""
     with console.status("[bold green]Starting Optimization..."):
         try:
@@ -61,12 +59,12 @@ def start_optimization_run(
             if result.get("code") is None:
                 result["code"] = ""
             return result
-        except Exception as e:
+        except requests.exceptions.HTTPError as e:
             handle_api_error(e, console)
-            sys.exit(1)
+            return None
         except Exception as e:
             console.print(f"[bold red]Error starting run: {e}[/]")
-            sys.exit(1)
+            return None
 
 
 def evaluate_feedback_then_suggest_next_solution(
@@ -101,11 +99,11 @@ def evaluate_feedback_then_suggest_next_solution(
         return result
     except requests.exceptions.HTTPError as e:
         # 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
+        handle_api_error(e, console)
+        raise
     except Exception as e:
-        print(f"Error: {e}")  # Use print as console might not be available
-        raise  # Re-raise the exception
+        console.print(f"[bold red]Error: {e}[/]")
+        raise
 
 
 def get_optimization_run_status(
@@ -137,11 +135,11 @@ def get_optimization_run_status(
                     result["nodes"][i]["code"] = ""
         return result
     except requests.exceptions.HTTPError as e:
-        handle_api_error(e, console)  # Use default console
-        raise  # Re-raise
+        handle_api_error(e, console)
+        raise
     except Exception as e:
-        print(f"Error getting run status: {e}")
-        raise  # Re-raise
+        console.print(f"[bold red]Error getting run status: {e}[/]")
+        raise
 
 
 def send_heartbeat(run_id: str, auth_headers: dict = {}, timeout: Union[int, Tuple[int, int]] = (10, 10)) -> bool:
@@ -152,7 +150,7 @@ def send_heartbeat(run_id: str, auth_headers: dict = {}, timeout: Union[int, Tup
         return True
     except requests.exceptions.HTTPError as e:
         if e.response.status_code == 409:
-            print("Polling ignore: Run {run_id} is not running.", file=sys.stderr)
+            print(f"Polling ignore: Run {run_id} is not running.", file=sys.stderr)
         else:
             print(f"Polling failed for run {run_id}: HTTP {e.response.status_code}", file=sys.stderr)
         return False
@@ -221,7 +219,6 @@ def get_optimization_suggestions_from_codebase(
     """Analyze codebase and get optimization suggestions using the model-agnostic backend API."""
     model, api_key_dict = _determine_model_and_api_key()
     try:
-        model, api_key_dict = _determine_model_and_api_key()
         response = requests.post(
             f"{__base_url__}/onboard/analyze-codebase",
             json={
@@ -238,7 +235,7 @@ def get_optimization_suggestions_from_codebase(
         result = response.json()
         return [option for option in result.get("options", [])]
 
-    except Exception as e:
+    except requests.exceptions.HTTPError as e:
         handle_api_error(e, console)
         return None
     except Exception as e:
@@ -257,7 +254,6 @@ def generate_evaluation_script_and_metrics(
     """Generate evaluation script and determine metrics using the model-agnostic backend API."""
     model, api_key_dict = _determine_model_and_api_key()
     try:
-        model, api_key_dict = _determine_model_and_api_key()
         response = requests.post(
             f"{__base_url__}/onboard/generate-script",
             json={
@@ -294,7 +290,6 @@ def analyze_evaluation_environment(
     """Analyze existing evaluation scripts and environment using the model-agnostic backend API."""
     model, api_key_dict = _determine_model_and_api_key()
     try:
-        model, api_key_dict = _determine_model_and_api_key()
         response = requests.post(
             f"{__base_url__}/onboard/analyze-environment",
             json={
@@ -312,7 +307,7 @@ def analyze_evaluation_environment(
         response.raise_for_status()
         return response.json()
 
-    except Exception as e:
+    except requests.exceptions.HTTPError as e:
         handle_api_error(e, console)
         return None
     except Exception as e:
@@ -331,7 +326,6 @@ def analyze_script_execution_requirements(
     """Analyze script to determine proper execution command using the model-agnostic backend API."""
     model, api_key_dict = _determine_model_and_api_key()
     try:
-        model, api_key_dict = _determine_model_and_api_key()
         response = requests.post(
             f"{__base_url__}/onboard/analyze-script",
             json={
@@ -348,7 +342,7 @@ def analyze_script_execution_requirements(
         result = response.json()
         return result.get("command", f"python {script_path}")
 
-    except Exception as e:
+    except requests.exceptions.HTTPError as e:
         handle_api_error(e, console)
         return f"python {script_path}"
     except Exception as e:

weco/chatbot.py

@@ -2,6 +2,7 @@ import pathlib
 import shlex
 import argparse
 from typing import List, Optional, Dict, Any, Tuple
+import sys
 
 from rich.console import Console
 from rich.prompt import Prompt
@@ -682,9 +683,9 @@ class Chatbot:
 
         # Import and execute the actual optimization function
         # (Import here to avoid circular imports)
-        from .optimizer import execute_optimization as actual_execute_optimization
+        from .optimizer import execute_optimization as execute_optimization_run
 
-        success = actual_execute_optimization(
+        success = execute_optimization_run(
             source=target_file,
             eval_command=eval_config["eval_command"],
             metric=eval_config["metric_name"],
@@ -702,6 +703,9 @@ class Chatbot:
         else:
             self.console.print("\n[bold yellow]⚠️  Optimization ended early or encountered issues.[/]")
 
+        exit_code = 0 if success else 1
+        sys.exit(exit_code)
+
     def show_and_copy_command(self, command: str) -> None:
         """Show the command and copy it to clipboard."""
         import subprocess

weco/cli.py

@@ -71,7 +71,7 @@ def configure_run_parser(run_parser: argparse.ArgumentParser) -> None:
 
 def execute_run_command(args: argparse.Namespace) -> None:
     """Execute the 'weco run' command with all its logic."""
-    from .optimizer import execute_optimization  # Moved import inside
+    from .optimizer import execute_optimization
 
     success = execute_optimization(
         source=args.source,

weco/optimizer.py

@@ -183,6 +183,10 @@ def execute_optimization(
             auth_headers=auth_headers,
             timeout=api_timeout,
         )
+        # Indicate the endpoint failed to return a response and the optimization was unsuccessful
+        if run_response is None:
+            return False
+
         run_id = run_response["run_id"]
         run_name = run_response["run_name"]
         current_run_id_for_heartbeat = run_id

{weco-0.2.24.dist-info → weco-0.2.26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weco
-Version: 0.2.24
+Version: 0.2.26
 Summary: Documentation for `weco`, a CLI for using Weco AI's code optimizer.
 Author-email: Weco AI Team <contact@weco.ai>
 License: MIT
@@ -268,6 +268,7 @@ Weco supports the following LLM models:
 - `gpt-4o-mini`
 
 ### Anthropic Models
+- `claude-opus-4-1`
 - `claude-opus-4-0`
 - `claude-sonnet-4-0`
 - `claude-3-7-sonnet-latest`

weco-0.2.26.dist-info/RECORD

@@ -0,0 +1,15 @@
+weco/__init__.py,sha256=ClO0uT6GKOA0iSptvP0xbtdycf0VpoPTq37jHtvlhtw,303
+weco/api.py,sha256=sz97FI0cMm4ku6bMnXsY9Jqgu-lUNELpF6L6vqIaSDE,12997
+weco/auth.py,sha256=KMSAsN1V5wx7KUsYL1cEOOiG29Pqf4Exb3EPW4mAWC0,10003
+weco/chatbot.py,sha256=EkzKd5Q_IlcobBbY3gsbgN0jxbJMfP5eYtzxQaNQ3fg,37747
+weco/cli.py,sha256=jasJpAwibp2rfZflE1LebyQz1dh55Y32G6qDINw4B_U,8161
+weco/constants.py,sha256=vfQGDf9_kzlN9BzEFvMsd0EeXOsRyzpvSWyxOJgRauE,168
+weco/optimizer.py,sha256=iFI1h2HlKFYMiIxVoTFS5aulUIuNjMbsOQw3MGfXIwI,23819
+weco/panels.py,sha256=jwAV_uoa0ZI9vjyey-hSY3rx4pfNNkZvPzqt-iz-RXo,16808
+weco/utils.py,sha256=HecbOqD5rBuVhUkLixVrTWBMJ-ZMAhK-889N-lCk3dQ,7335
+weco-0.2.26.dist-info/licenses/LICENSE,sha256=p_GQqJBvuZgkLNboYKyH-5dhpTDlKs2wq2TVM55WrWE,1065
+weco-0.2.26.dist-info/METADATA,sha256=liTaJZxAI_4Xe21aZ-1YKUzHEvtn1vzyu0liTayEb98,16089
+weco-0.2.26.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+weco-0.2.26.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
+weco-0.2.26.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
+weco-0.2.26.dist-info/RECORD,,

weco-0.2.24.dist-info/RECORD

@@ -1,15 +0,0 @@
-weco/__init__.py,sha256=ClO0uT6GKOA0iSptvP0xbtdycf0VpoPTq37jHtvlhtw,303
-weco/api.py,sha256=sdRgSvbjStGr5Bywkp4Pkuo9jgtGp95tNzsVXrXmH00,13380
-weco/auth.py,sha256=KMSAsN1V5wx7KUsYL1cEOOiG29Pqf4Exb3EPW4mAWC0,10003
-weco/chatbot.py,sha256=MSG3yybZ0fHTpwW3If9TpOWOAE4EqrcUWpe3q16AjZE,37673
-weco/cli.py,sha256=75JdYpUf0qdJW5pjycZoHUKA2n2MYj33qYM9WQwLT2s,8184
-weco/constants.py,sha256=vfQGDf9_kzlN9BzEFvMsd0EeXOsRyzpvSWyxOJgRauE,168
-weco/optimizer.py,sha256=E5Jii0rTdI9pxkstaM7ipjF4viX5XnSy5w71gdps4Ws,23662
-weco/panels.py,sha256=jwAV_uoa0ZI9vjyey-hSY3rx4pfNNkZvPzqt-iz-RXo,16808
-weco/utils.py,sha256=HecbOqD5rBuVhUkLixVrTWBMJ-ZMAhK-889N-lCk3dQ,7335
-weco-0.2.24.dist-info/licenses/LICENSE,sha256=p_GQqJBvuZgkLNboYKyH-5dhpTDlKs2wq2TVM55WrWE,1065
-weco-0.2.24.dist-info/METADATA,sha256=yFq-NPCUsh59B4Pj9IlYjDo53Kzjya5Rk_9-QmheGLM,16069
-weco-0.2.24.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-weco-0.2.24.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
-weco-0.2.24.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
-weco-0.2.24.dist-info/RECORD,,