weco 0.2.13__py3-none-any.whl → 0.2.15__py3-none-any.whl

weco/__init__.py

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

weco/api.py

@@ -7,8 +7,7 @@ import sys
 
 def handle_api_error(e: requests.exceptions.HTTPError, console: rich.console.Console) -> None:
     """Extract and display error messages from API responses in a structured format."""
-    error_message = str(e)  # Default message
-    console.print(f"[bold red]Error:[/] {error_message}")
+    console.print(f"[bold red]{e.response.json()['detail']}[/]")
     sys.exit(1)
 
 

weco/cli.py

@@ -201,14 +201,13 @@ def main() -> None:
     )
 
     # --- Logout Command ---
-    logout_parser = subparsers.add_parser("logout", help="Log out from Weco and clear saved API key.")  # noqa F841
+    _ = subparsers.add_parser("logout", help="Log out from Weco and clear saved API key.")
 
     args = parser.parse_args()
 
     # --- Handle Logout Command ---
     if args.command == "logout":
         clear_api_key()
-        console.print("[green]Logged out successfully.[/]")  # Added feedback
         sys.exit(0)
 
     # --- Handle Run Command ---
@@ -257,20 +256,23 @@ def main() -> None:
             maximize = args.maximize == "true"
             steps = args.steps
             code_generator_config = {"model": args.model}
-            evaluator_config = {"model": args.model}
+            evaluator_config = {
+                "model": args.model,
+                "include_analysis": False,  # NOTE: False for now
+            }
             search_policy_config = {
                 "num_drafts": max(1, math.ceil(0.15 * steps)),
                 "debug_prob": 0.5,
                 "max_debug_depth": max(1, math.ceil(0.1 * steps)),
             }
+            # API request timeout
+            timeout = 800
             # Read additional instructions
             additional_instructions = read_additional_instructions(additional_instructions=args.additional_instructions)
             # Read source code path
             source_fp = pathlib.Path(args.source)
             # Read source code content
             source_code = read_from_path(fp=source_fp, is_json=False)
-            # API request timeout
-            timeout = 800
 
             # --- Panel Initialization ---
             summary_panel = SummaryPanel(
@@ -308,9 +310,8 @@ def main() -> None:
                 runs_dir = pathlib.Path(args.log_dir) / session_id
                 runs_dir.mkdir(parents=True, exist_ok=True)
 
-                # Save the original code (.runs/<session-id>/original.<extension>)
-                runs_copy_source_fp = runs_dir / f"original{source_fp.suffix}"  # Use correct suffix
-                write_to_path(fp=runs_copy_source_fp, content=source_code)
+                # 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:
@@ -378,7 +379,8 @@ def main() -> None:
                     transition_delay=0.1,
                 )
 
-                for step in range(1, steps):
+                # 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=args.additional_instructions
@@ -551,9 +553,7 @@ def main() -> None:
                     best_solution_score = None
 
                 if best_solution_code is None or best_solution_score is None:
-                    best_solution_content = (
-                        f"# Weco could not find a better solution\n\n{read_from_path(fp=runs_copy_source_fp, is_json=False)}"
-                    )
+                    best_solution_content = f"# Weco could not find a better solution\n\n{read_from_path(fp=runs_dir / f'step_0{source_fp.suffix}', is_json=False)}"
                 else:
                     # Format score for the comment
                     best_score_str = (

{weco-0.2.13.dist-info → weco-0.2.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weco
-Version: 0.2.13
+Version: 0.2.15
 Summary: Documentation for `weco`, a CLI for using Weco AI's code optimizer.
 Author-email: Weco AI Team <contact@weco.ai>
 License: MIT

weco-0.2.15.dist-info/RECORD

@@ -0,0 +1,12 @@
+weco/__init__.py,sha256=2P-Z80aRiNeL8hyXD4ExZxizUgLb_Tj_NOydwoqTZ_k,426
+weco/api.py,sha256=z2DCe0kQJaaBHo-Vml52GRc9nwQBnrKATXoY3UfPojw,3842
+weco/auth.py,sha256=IPfiLthcNRkPyM8pWHTyDLvikw83sigacpY1PmeA03Y,2343
+weco/cli.py,sha256=rcntl-hFQorCNSBX1yaNoL-4eBJFSkN5Vo35b6-W4UU,28993
+weco/panels.py,sha256=gB4rZbCvqzewUCBcILvyyU4fnOQLwFgHCGmtn-ZlgSo,13385
+weco/utils.py,sha256=hhIebUPnetFMfNSFfcsKVw1TSpeu_Zw3rBPPnxDie0U,3911
+weco-0.2.15.dist-info/licenses/LICENSE,sha256=p_GQqJBvuZgkLNboYKyH-5dhpTDlKs2wq2TVM55WrWE,1065
+weco-0.2.15.dist-info/METADATA,sha256=B97o-KOP1o2sUVHqSz1maZGy-iysBKBbvq16QIyVE68,10851
+weco-0.2.15.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+weco-0.2.15.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
+weco-0.2.15.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
+weco-0.2.15.dist-info/RECORD,,

{weco-0.2.13.dist-info → weco-0.2.15.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (79.0.0)
+Generator: setuptools (79.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

weco-0.2.13.dist-info/RECORD

@@ -1,12 +0,0 @@
-weco/__init__.py,sha256=DvJyE-y8upXub4ecCkEZlC5tD5-XonqMe6x8ZtQmnQ0,426
-weco/api.py,sha256=_zgQ4AHDBgqfU2569W63c10qKuIWA5y63XT0qqvUag8,3881
-weco/auth.py,sha256=IPfiLthcNRkPyM8pWHTyDLvikw83sigacpY1PmeA03Y,2343
-weco/cli.py,sha256=rvUqxdJov_R74lkr0Gx-802RNsBYxhLahTue54n2bqc,28967
-weco/panels.py,sha256=gB4rZbCvqzewUCBcILvyyU4fnOQLwFgHCGmtn-ZlgSo,13385
-weco/utils.py,sha256=hhIebUPnetFMfNSFfcsKVw1TSpeu_Zw3rBPPnxDie0U,3911
-weco-0.2.13.dist-info/licenses/LICENSE,sha256=p_GQqJBvuZgkLNboYKyH-5dhpTDlKs2wq2TVM55WrWE,1065
-weco-0.2.13.dist-info/METADATA,sha256=QD0MYkpAfdETUUtQA2PC1HTuvaMf2zeAOEWyqP_lxrk,10851
-weco-0.2.13.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-weco-0.2.13.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
-weco-0.2.13.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
-weco-0.2.13.dist-info/RECORD,,