weco 0.2.0__tar.gz → 0.2.1__tar.gz

{weco-0.2.0 → weco-0.2.1}/.github/workflows/release.yml

@@ -79,7 +79,7 @@ jobs:
         path: dist/
 
     - name: Sign dists with Sigstore
-      uses: sigstore/gh-action-sigstore-python@v2.1.1
+      uses: sigstore/gh-action-sigstore-python@v3.0.0
       with:
         inputs: >-
           ./dist/*.tar.gz
@@ -90,7 +90,7 @@ jobs:
         GITHUB_TOKEN: ${{ github.token }}
       run: >-
         gh release create
-        'v0.2.0'
+        'v0.2.1'
         --repo '${{ github.repository }}'
         --notes ""
 
@@ -102,5 +102,5 @@ jobs:
       # sigstore-produced signatures and certificates.
       run: >-
         gh release upload
-        'v0.2.0' dist/**
+        'v0.2.1' dist/**
         --repo '${{ github.repository }}'

{weco-0.2.0 → weco-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weco
-Version: 0.2.0
+Version: 0.2.1
 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.0 → weco-0.2.1}/README.md

@@ -27,6 +27,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"`
 
 ---
 
@@ -77,6 +78,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.0 → weco-0.2.1}/pyproject.toml

@@ -10,7 +10,7 @@ authors = [
 ]
 description = "Documentation for `weco`, a CLI for using Weco AI's code optimizer."
 readme = "README.md"
-version = "0.2.0"
+version = "0.2.1"
 license = {text = "MIT"}
 requires-python = ">=3.12"
 dependencies = ["requests", "rich"]

{weco-0.2.0 → weco-0.2.1}/weco/__init__.py

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

{weco-0.2.0 → weco-0.2.1}/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-0.2.0 → weco-0.2.1}/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 → weco-0.2.1}/weco.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weco
-Version: 0.2.0
+Version: 0.2.1
 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`.
 
 ---