Merge branch 'main' into docs

Update README

Author: DhruvSrikanth

.github/workflows/release.yml

@@ -90,7 +90,7 @@ jobs:
         GITHUB_TOKEN: ${{ github.token }}
       run: >-
         gh release create
-        'v0.2.7'
+        'v0.2.8'
         --repo '${{ github.repository }}'
         --notes ""
 
@@ -102,5 +102,5 @@ jobs:
       # sigstore-produced signatures and certificates.
       run: >-
         gh release upload
-        'v0.2.7' dist/**
+        'v0.2.8' dist/**
         --repo '${{ github.repository }}'

pyproject.toml

@@ -10,9 +10,9 @@ authors = [
 ]
 description = "Documentation for `weco`, a CLI for using Weco AI's code optimizer."
 readme = "README.md"
-version = "0.2.7"
+version = "0.2.8"
 license = {text = "MIT"}
-requires-python = ">=3.12"
+requires-python = ">=3.8"
 dependencies = ["requests", "rich"]
 keywords = ["AI", "Code Optimization", "Code Generation"]
 classifiers = [

weco/__init__.py

@@ -1,4 +1,4 @@
 # DO NOT EDIT
-__pkg_version__ = "0.2.7"
+__pkg_version__ = "0.2.8"
 __api_version__ = "v1"
 __base_url__ = f"https://api.aide.weco.ai/{__api_version__}"

weco/api.py

@@ -6,14 +6,9 @@
 
 
 def handle_api_error(e: requests.exceptions.HTTPError, console: rich.console.Console) -> None:
-    """Extract and display error messages from API responses."""
-    try:
-        error_data = e.response.json()
-        error_message = error_data.get("detail", str(e))
-        console.print(f"[bold red]Server Error:[/] {error_message}")
-    except Exception:
-        # If we can't parse the JSON, just show the original error
-        console.print(f"[bold red]Server Error:[/] {str(e)}")
+    """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}")
     sys.exit(1)
 
 

weco/cli.py

@@ -36,7 +36,7 @@ def main() -> None:
     parser = argparse.ArgumentParser(
         description="[bold cyan]Weco CLI[/]", formatter_class=argparse.RawDescriptionHelpFormatter
     )
-    parser.add_argument("--source", type=str, required=True, help="Path to the Python source code (e.g. optimize.py)")
+    parser.add_argument("--source", type=str, required=True, help="Path to the source code (e.g. optimize.py)")
     parser.add_argument(
         "--eval-command", type=str, required=True, help="Command to run for evaluation (e.g. 'python eval.py --arg1=val1')"
     )
@@ -88,7 +88,7 @@ def main() -> None:
             maximize=maximize, metric_name=metric_name, total_steps=steps, model=args.model, runs_dir=args.log_dir
         )
         plan_panel = PlanPanel()
-        solution_panels = SolutionPanels(metric_name=metric_name)
+        solution_panels = SolutionPanels(metric_name=metric_name, source_fp=source_fp)
         eval_output_panel = EvaluationOutputPanel()
         tree_panel = MetricTreePanel(maximize=maximize)
         layout = create_optimization_layout()
@@ -118,8 +118,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.py)
-            runs_copy_source_fp = runs_dir / "original.py"
+            # Save the original code (.runs/<session-id>/original.<extension>)
+            runs_copy_source_fp = runs_dir / f"original.{source_fp.suffix}"
             write_to_path(fp=runs_copy_source_fp, content=source_code)
 
             # Write the code string to the source file path
@@ -200,8 +200,8 @@ def main() -> None:
                     api_keys=api_keys,
                     timeout=timeout,
                 )
-                # Save next solution (.runs/<session-id>/step_<step>.py)
-                write_to_path(fp=runs_dir / f"step_{step}.py", content=eval_and_next_solution_response["code"])
+                # Save next solution (.runs/<session-id>/step_<step>.<extension>)
+                write_to_path(fp=runs_dir / f"step_{step}.{source_fp.suffix}", content=eval_and_next_solution_response["code"])
 
                 # Write the next solution to the source file
                 write_to_path(fp=source_fp, content=eval_and_next_solution_response["code"])
@@ -351,8 +351,8 @@ def main() -> None:
                 )
                 best_solution_content = f"# Best solution from Weco with a score of {best_score_str}\n\n{best_solution_code}"
 
-            # Save best solution to .runs/<session-id>/best.py
-            write_to_path(fp=runs_dir / "best.py", content=best_solution_content)
+            # Save best solution to .runs/<session-id>/best.<extension>
+            write_to_path(fp=runs_dir / f"best.{source_fp.suffix}", content=best_solution_content)
 
             # write the best solution to the source file
             write_to_path(fp=source_fp, content=best_solution_content)

weco/panels.py

@@ -6,6 +6,7 @@
 from rich.syntax import Syntax
 from typing import Dict, List, Optional, Union, Tuple
 from .utils import format_number
+import pathlib
 
 
 class SummaryPanel:
@@ -46,6 +47,8 @@ def get_display(self, final_message: Optional[str] = None) -> Panel:
         """Create a summary panel with the relevant information."""
         layout = Layout(name="summary")
         summary_table = Table(show_header=False, box=None, padding=(0, 1))
+
+        summary_table.add_row("")
         # Goal
         if final_message is not None:
             summary_table.add_row(f"[bold cyan]Result:[/] {final_message}")
@@ -256,13 +259,19 @@ def get_display(self) -> Panel:
 class SolutionPanels:
     """Displays the current and best solutions side by side."""
 
-    def __init__(self, metric_name: str):
+    def __init__(self, metric_name: str, source_fp: pathlib.Path):
         # Current solution
         self.current_node = None
         # Best solution
         self.best_node = None
         # Metric name
         self.metric_name = metric_name.capitalize()
+        # Determine the lexer for the source file
+        self.lexer = self._determine_lexer(source_fp)
+
+    def _determine_lexer(self, source_fp: pathlib.Path) -> str:
+        """Determine the lexer for the source file."""
+        return Syntax.from_path(source_fp).lexer
 
     def update(self, current_node: Union[Node, None], best_node: Union[Node, None]):
         """Update the current and best solutions."""
@@ -280,7 +289,7 @@ def get_display(self, current_step: int) -> Tuple[Panel, Panel]:
         # Current solution (without score)
         current_title = f"[bold]💡 Current Solution (Step {current_step})"
         current_panel = Panel(
-            Syntax(str(current_code), "python", theme="monokai", line_numbers=True, word_wrap=False),
+            Syntax(str(current_code), self.lexer, theme="monokai", line_numbers=True, word_wrap=False),
             title=current_title,
             border_style="yellow",
             expand=True,
@@ -290,7 +299,7 @@ def get_display(self, current_step: int) -> Tuple[Panel, Panel]:
         # Best solution
         best_title = f"[bold]🏆 Best Solution ([green]{self.metric_name}: {f'{best_score:.4f}' if best_score is not None else 'N/A'}[/])"
         best_panel = Panel(
-            Syntax(str(best_code), "python", theme="monokai", line_numbers=True, word_wrap=False),
+            Syntax(str(best_code), self.lexer, theme="monokai", line_numbers=True, word_wrap=False),
             title=best_title,
             border_style="green",
             expand=True,