Update pyproject.toml to version 0.2.18. Refactor heartbeat logic in cli.py to streamline error handling by removing redundant try-except blocks, enhancing code clarity and maintainability.

Author: yuxiang-wu

pyproject.toml

@@ -5,20 +5,18 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "weco"
-authors = [
-    {name = "Weco AI Team", email = "contact@weco.ai"},
-]
+authors = [{ name = "Weco AI Team", email = "contact@weco.ai" }]
 description = "Documentation for `weco`, a CLI for using Weco AI's code optimizer."
 readme = "README.md"
-version = "0.2.17"
-license = {text = "MIT"}
+version = "0.2.18"
+license = { text = "MIT" }
 requires-python = ">=3.8"
 dependencies = ["requests", "rich"]
 keywords = ["AI", "Code Optimization", "Code Generation"]
 classifiers = [
     "Programming Language :: Python :: 3",
     "Operating System :: OS Independent",
-    "License :: OSI Approved :: MIT License"
+    "License :: OSI Approved :: MIT License",
 ]
 
 [project.scripts]

weco/cli.py

@@ -437,8 +437,8 @@ def main() -> None:
                     transition_delay=0.1,
                 )
 
-                # Send initial heartbeat immediately after starting
-                send_heartbeat(session_id, auth_headers)
+                # # Send initial heartbeat immediately after starting
+                # send_heartbeat(session_id, auth_headers)
 
                 # Run evaluation on the initial solution
                 term_out = run_evaluation(eval_command=args.eval_command)
@@ -460,23 +460,14 @@ def main() -> None:
                     )
 
                     # Send feedback and get next suggestion
-                    try:
-                        eval_and_next_solution_response = evaluate_feedback_then_suggest_next_solution(
-                            session_id=session_id,
-                            execution_output=term_out,
-                            additional_instructions=current_additional_instructions,  # Pass current instructions
-                            api_keys=llm_api_keys,  # Pass client LLM keys
-                            auth_headers=auth_headers,  # Pass Weco key if logged in
-                            timeout=timeout,
-                        )
-                    except (requests.exceptions.HTTPError, requests.exceptions.RequestException) as suggest_error:
-                        # If suggest fails, we consider it a CLI error and break the loop
-                        console.print(
-                            f"\n[bold red]Error communicating with API during step {step}. Stopping optimization.[/]"
-                        )
-                        # Error details should have been printed by handle_api_error or the exception handler in evaluate_feedback_then_suggest_next_solution
-                        # Prepare to report error in finally block
-                        raise RuntimeError(f"API error during suggest at step {step}") from suggest_error
+                    eval_and_next_solution_response = evaluate_feedback_then_suggest_next_solution(
+                        session_id=session_id,
+                        execution_output=term_out,
+                        additional_instructions=current_additional_instructions,  # Pass current instructions
+                        api_keys=llm_api_keys,  # Pass client LLM keys
+                        auth_headers=auth_headers,  # Pass Weco key if logged in
+                        timeout=timeout,
+                    )
 
                     # Save next solution (.runs/<session-id>/step_<step>.<extension>)
                     write_to_path(
@@ -569,20 +560,14 @@ def main() -> None:
                 )
 
                 # Final evaluation report
-                try:
-                    eval_and_next_solution_response = evaluate_feedback_then_suggest_next_solution(
-                        session_id=session_id,
-                        execution_output=term_out,
-                        additional_instructions=current_additional_instructions,
-                        api_keys=llm_api_keys,
-                        timeout=timeout,
-                        auth_headers=auth_headers,
-                    )
-                except (requests.exceptions.HTTPError, requests.exceptions.RequestException) as suggest_error:
-                    console.print(
-                        "\n[bold red]Error communicating with API during final evaluation report. Results might be incomplete.[/]"
-                    )
-                    raise RuntimeError("API error during final suggest call") from suggest_error
+                eval_and_next_solution_response = evaluate_feedback_then_suggest_next_solution(
+                    session_id=session_id,
+                    execution_output=term_out,
+                    additional_instructions=current_additional_instructions,
+                    api_keys=llm_api_keys,
+                    timeout=timeout,
+                    auth_headers=auth_headers,
+                )
 
                 # Update the progress bar
                 summary_panel.set_step(step=steps)