weco 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

weco/__init__.py

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

weco/cli.py

@@ -185,6 +185,9 @@ def main() -> None:
                 # 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"])
 
+                # Write the next solution to the source file
+                write_to_path(fp=source_fp, content=eval_and_next_solution_response["code"])
+
                 # Get the optimization session status for
                 # the best solution, its score, and the history to plot the tree
                 status_response = get_optimization_session_status(console=console, session_id=session_id, include_history=True)

weco/panels.py

@@ -5,7 +5,7 @@ from rich.layout import Layout
 from rich.panel import Panel
 from rich.syntax import Syntax
 from typing import Dict, List, Optional, Union, Tuple
-from .utils import format_number, truncate_text
+from .utils import format_number
 
 
 class SummaryPanel:
@@ -67,7 +67,6 @@ class PlanPanel:
 
     def __init__(self):
         self.plan = ""
-        self.max_lines = 3  # Approximate number of lines that can fit in the panel
 
     def update(self, plan: str):
         """Update the plan text."""
@@ -79,15 +78,7 @@ class PlanPanel:
 
     def get_display(self) -> Panel:
         """Create a panel displaying the plan with truncation if needed."""
-        display_text = truncate_text(text=self.plan, max_lines=self.max_lines)
-        return Panel(
-            display_text,
-            title="[bold]📝 Thinking...",
-            border_style="cyan",
-            expand=True,
-            padding=(0, 1),
-            subtitle="[cyan]↓ truncated ↓[/]" if len(self.plan) > self.max_lines else None,
-        )
+        return Panel(self.plan, title="[bold]📝 Thinking...", border_style="cyan", expand=True, padding=(0, 1))
 
 
 class Node:
@@ -238,7 +229,6 @@ class EvaluationOutputPanel:
 
     def __init__(self):
         self.output = ""
-        self.max_lines = 25  # Approximate number of lines that can fit in the panel
 
     def update(self, output: str) -> None:
         """Update the evaluation output."""
@@ -250,13 +240,7 @@ class EvaluationOutputPanel:
 
     def get_display(self) -> Panel:
         """Create a panel displaying the evaluation output with truncation if needed."""
-        display_text = truncate_text(text=self.output, max_lines=self.max_lines)
-
-        title = "[bold]📋 Evaluation Output"
-        if len(self.output) == len(display_text):
-            title += " (truncated)[/]"
-
-        return Panel(display_text, title=title, border_style="red", expand=True, padding=(0, 1))
+        return Panel(self.output, title="[bold]📋 Evaluation Output", border_style="red", expand=True, padding=(0, 1))
 
 
 class SolutionPanels:
@@ -267,7 +251,6 @@ class SolutionPanels:
         self.current_node = None
         # Best solution
         self.best_node = None
-        self.max_lines = 30  # Approximate number of lines that can fit in each panel
 
     def update(self, current_node: Union[Node, None], best_node: Union[Node, None]):
         """Update the current and best solutions."""
@@ -282,44 +265,24 @@ class SolutionPanels:
         best_code = self.best_node.code if self.best_node is not None else ""
         best_score = self.best_node.metric if self.best_node is not None else None
 
-        # Determine if code is too long (approximate)
-        current_lines = current_code.count("\n") + 1
-        best_lines = best_code.count("\n") + 1
-
         # 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,
-                line_range=(0, self.max_lines),  # Only show first max_lines lines
-            ),
+            Syntax(str(current_code), "python", theme="monokai", line_numbers=True, word_wrap=False),
             title=current_title,
             border_style="yellow",
             expand=True,
             padding=(0, 1),
-            subtitle="[yellow]↓ truncated ↓[/]" if current_lines > self.max_lines else None,
         )
 
         # Best solution
         best_title = f"[bold]🏆 Best Solution ([green]Score: {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,
-                line_range=(0, self.max_lines),  # Only show first max_lines lines
-            ),
+            Syntax(str(best_code), "python", theme="monokai", line_numbers=True, word_wrap=False),
             title=best_title,
             border_style="green",
             expand=True,
             padding=(0, 1),
-            subtitle="[yellow]↓ truncated ↓[/]" if best_lines > self.max_lines else None,
         )
 
         return current_panel, best_panel

weco/utils.py

@@ -13,12 +13,11 @@ import pathlib
 def read_api_keys_from_env() -> Dict[str, Any]:
     """Read API keys from environment variables."""
     keys = {}
-    openai_api_key = os.getenv("OPENAI_API_KEY")
-    anthropic_api_key = os.getenv("ANTHROPIC_API_KEY")
-    if openai_api_key is not None and len(openai_api_key) > 0:
-        keys["OPENAI_API_KEY"] = openai_api_key
-    if anthropic_api_key is not None and len(anthropic_api_key) > 0:
-        keys["ANTHROPIC_API_KEY"] = anthropic_api_key
+    keys_to_check = ["OPENAI_API_KEY", "ANTHROPIC_API_KEY", "GEMINI_API_KEY"]
+    for key in keys_to_check:
+        value = os.getenv(key)
+        if value is not None and len(value) > 0:
+            keys[key] = value
     return keys
 
 
@@ -106,14 +105,3 @@ def run_evaluation(eval_command: str) -> str:
             output += "\n"
         output += result.stdout
     return output
-
-
-def truncate_text(text: str, max_lines: int) -> str:
-    """Truncate text to a maximum number of lines."""
-    lines = text.split("\n")
-    if len(lines) > max_lines:
-        display_lines = lines[:max_lines]
-        display_text = "\n".join(display_lines)
-    else:
-        display_text = text
-    return display_text

{weco-0.2.0.dist-info → weco-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weco
-Version: 0.2.0
+Version: 0.2.2
 Summary: Documentation for `weco`, a CLI for using Weco AI's code optimizer.
 Author-email: Weco AI Team <dhruv@weco.ai>
 License: MIT
@@ -49,6 +49,7 @@ The `weco` CLI leverages advanced optimization techniques and language model str
    
    - **OpenAI:** `export OPENAI_API_KEY="your_key_here"`
    - **Anthropic:** `export ANTHROPIC_API_KEY="your_key_here"`
+   - **Google DeepMind:** `export GEMINI_API_KEY="your_key_here"`
 
 ---
 
@@ -99,6 +100,7 @@ The CLI supports the following model providers:
 
 - **OpenAI:** Set your API key using `OPENAI_API_KEY`.
 - **Anthropic:** Set your API key using `ANTHROPIC_API_KEY`.
+- **Google DeepMind:** Set your API key using `GEMINI_API_KEY`.
 
 ---
 

weco-0.2.2.dist-info/RECORD

@@ -0,0 +1,11 @@
+weco/__init__.py,sha256=piNl2k8gLYD32s0kA8QqBDe7vANYnzN_QN9bWtfRsks,124
+weco/api.py,sha256=JHAm60sO51Y2tzLnyAAFv9OLm75o_RNWeoK-gYF4VHE,3342
+weco/cli.py,sha256=OneQ8a7ySgtzrNGo0l53Ij_I_p0RTVQ8GV2n66wIJv0,16050
+weco/panels.py,sha256=iUZWtZfohaWeI6-x6pf2kKN1c8IL7U_AhpBBeFhtFhk,11642
+weco/utils.py,sha256=g1Lchd3_L-eGRNkqr2IyRZ5FJfZTrFf9aTmmpMtdhko,3761
+weco-0.2.2.dist-info/licenses/LICENSE,sha256=p_GQqJBvuZgkLNboYKyH-5dhpTDlKs2wq2TVM55WrWE,1065
+weco-0.2.2.dist-info/METADATA,sha256=jy-GYv-J4qga01Ic2HtzkG1P3MDlgNjq_aas4EukZ0o,5270
+weco-0.2.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+weco-0.2.2.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
+weco-0.2.2.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
+weco-0.2.2.dist-info/RECORD,,

weco-0.2.0.dist-info/RECORD

@@ -1,11 +0,0 @@
-weco/__init__.py,sha256=J-pJsj20yEXYYCDJyhKJBFWQUzb4bhww0ynqxNBA5Z4,124
-weco/api.py,sha256=JHAm60sO51Y2tzLnyAAFv9OLm75o_RNWeoK-gYF4VHE,3342
-weco/cli.py,sha256=q1UuQ8xe2R5dpLhxPGJEW9mOFozH0oRlTbyyVD70XlU,15895
-weco/panels.py,sha256=Cu249CAlpczrseBjqvERlo_wlg4RLbO0ylx1JPZGjRA,13058
-weco/utils.py,sha256=yCVlQyCP0P7t-864xuw55qEUVmyXXGin-a1Y5TDFHE4,4202
-weco-0.2.0.dist-info/licenses/LICENSE,sha256=p_GQqJBvuZgkLNboYKyH-5dhpTDlKs2wq2TVM55WrWE,1065
-weco-0.2.0.dist-info/METADATA,sha256=piZk0o15yLjq6qpKKFQqT4vRcGR3unNC2LzZOIdJAro,5140
-weco-0.2.0.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-weco-0.2.0.dist-info/entry_points.txt,sha256=ixJ2uClALbCpBvnIR6BXMNck8SHAab8eVkM9pIUowcs,39
-weco-0.2.0.dist-info/top_level.txt,sha256=F0N7v6e2zBSlsorFv-arAq2yDxQbzX3KVO8GxYhPUeE,5
-weco-0.2.0.dist-info/RECORD,,