mini-swe-agent 1.10.0__tar.gz → 1.11.1__tar.gz

{mini_swe_agent-1.10.0/src/mini_swe_agent.egg-info → mini_swe_agent-1.11.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini-swe-agent
-Version: 1.10.0
+Version: 1.11.1
 Summary: Nano SWE Agent - A simple AI software engineering agent
 Author-email: Kilian Lieret <kilian.lieret@posteo.de>, "Carlos E. Jimenez" <carlosej@princeton.edu>
 License: MIT License

{mini_swe_agent-1.10.0 → mini_swe_agent-1.11.1/src/mini_swe_agent.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini-swe-agent
-Version: 1.10.0
+Version: 1.11.1
 Summary: Nano SWE Agent - A simple AI software engineering agent
 Author-email: Kilian Lieret <kilian.lieret@posteo.de>, "Carlos E. Jimenez" <carlosej@princeton.edu>
 License: MIT License

{mini_swe_agent-1.10.0 → mini_swe_agent-1.11.1}/src/mini_swe_agent.egg-info/SOURCES.txt

@@ -34,6 +34,7 @@ src/minisweagent/environments/extra/swerex_docker.py
 src/minisweagent/models/__init__.py
 src/minisweagent/models/anthropic.py
 src/minisweagent/models/litellm_model.py
+src/minisweagent/models/openrouter_model.py
 src/minisweagent/models/test_models.py
 src/minisweagent/models/extra/__init__.py
 src/minisweagent/models/extra/roulette.py

{mini_swe_agent-1.10.0 → mini_swe_agent-1.11.1}/src/minisweagent/__init__.py

@@ -8,7 +8,7 @@ This file provides:
   unless you want the static type checking.
 """
 
-__version__ = "1.10.0"
+__version__ = "1.11.1"
 
 import os
 from pathlib import Path

{mini_swe_agent-1.10.0 → mini_swe_agent-1.11.1}/src/minisweagent/agents/default.py

@@ -5,7 +5,7 @@ import subprocess
 from collections.abc import Callable
 from dataclasses import asdict, dataclass
 
-from jinja2 import Template
+from jinja2 import StrictUndefined, Template
 
 from minisweagent import Environment, Model
 
@@ -63,7 +63,9 @@ class DefaultAgent:
 
     def render_template(self, template: str, **kwargs) -> str:
         template_vars = asdict(self.config) | self.env.get_template_vars() | self.model.get_template_vars()
-        return Template(template).render(**kwargs, **template_vars, **self.extra_template_vars)
+        return Template(template, undefined=StrictUndefined).render(
+            **kwargs, **template_vars, **self.extra_template_vars
+        )
 
     def add_message(self, role: str, content: str, **kwargs):
         self.messages.append({"role": role, "content": content, **kwargs})

{mini_swe_agent-1.10.0 → mini_swe_agent-1.11.1}/src/minisweagent/environments/docker.py

@@ -29,6 +29,8 @@ class DockerEnvironmentConfig:
     """
     container_timeout: str = "2h"
     """Max duration to keep container running. Uses the same format as the sleep command."""
+    pull_timeout: int = 120
+    """Timeout in seconds for pulling images."""
 
 
 class DockerEnvironment:
@@ -65,7 +67,7 @@ class DockerEnvironment:
             cmd,
             capture_output=True,
             text=True,
-            timeout=120,  # docker pull might take a while
+            timeout=self.config.pull_timeout,  # docker pull might take a while
             check=True,
         )
         self.logger.info(f"Started container {container_name} with ID {result.stdout.strip()}")

{mini_swe_agent-1.10.0 → mini_swe_agent-1.11.1}/src/minisweagent/models/__init__.py

@@ -74,6 +74,7 @@ def get_model_name(input_model_name: str | None = None, config: dict | None = No
 _MODEL_CLASS_MAPPING = {
     "anthropic": "minisweagent.models.anthropic.AnthropicModel",
     "litellm": "minisweagent.models.litellm_model.LitellmModel",
+    "openrouter": "minisweagent.models.openrouter_model.OpenRouterModel",
     "deterministic": "minisweagent.models.test_models.DeterministicModel",
 }
 

{mini_swe_agent-1.10.0 → mini_swe_agent-1.11.1}/src/minisweagent/models/litellm_model.py

@@ -66,8 +66,8 @@ class LitellmModel:
         except Exception as e:
             logger.critical(
                 f"Error calculating cost for model {self.config.model_name}: {e}. "
-                "Please check the 'Updating the model registry' section in the documentation. "
-                "http://bit.ly/4p31bi4 Still stuck? Please open a github issue for help!"
+                "Please check the 'Updating the model registry' section in the documentation at "
+                "https://klieret.short.gy/litellm-model-registry Still stuck? Please open a github issue for help!"
             )
             raise
         self.n_calls += 1
@@ -75,6 +75,9 @@ class LitellmModel:
         GLOBAL_MODEL_STATS.add(cost)
         return {
             "content": response.choices[0].message.content or "",  # type: ignore
+            "extra": {
+                "response": response.model_dump(),
+            },
         }
 
     def get_template_vars(self) -> dict[str, Any]:

mini_swe_agent-1.11.1/src/minisweagent/models/openrouter_model.py

@@ -0,0 +1,118 @@
+import json
+import logging
+import os
+from dataclasses import asdict, dataclass, field
+from typing import Any
+
+import requests
+from tenacity import (
+    before_sleep_log,
+    retry,
+    retry_if_not_exception_type,
+    stop_after_attempt,
+    wait_exponential,
+)
+
+from minisweagent.models import GLOBAL_MODEL_STATS
+
+logger = logging.getLogger("openrouter_model")
+
+
+@dataclass
+class OpenRouterModelConfig:
+    model_name: str
+    model_kwargs: dict[str, Any] = field(default_factory=dict)
+
+
+class OpenRouterAPIError(Exception):
+    """Custom exception for OpenRouter API errors."""
+
+    pass
+
+
+class OpenRouterAuthenticationError(Exception):
+    """Custom exception for OpenRouter authentication errors."""
+
+    pass
+
+
+class OpenRouterRateLimitError(Exception):
+    """Custom exception for OpenRouter rate limit errors."""
+
+    pass
+
+
+class OpenRouterModel:
+    def __init__(self, **kwargs):
+        self.config = OpenRouterModelConfig(**kwargs)
+        self.cost = 0.0
+        self.n_calls = 0
+        self._api_url = "https://openrouter.ai/api/v1/chat/completions"
+        self._api_key = os.getenv("OPENROUTER_API_KEY", "")
+
+    @retry(
+        stop=stop_after_attempt(10),
+        wait=wait_exponential(multiplier=1, min=4, max=60),
+        before_sleep=before_sleep_log(logger, logging.WARNING),
+        retry=retry_if_not_exception_type(
+            (
+                OpenRouterAuthenticationError,
+                KeyboardInterrupt,
+            )
+        ),
+    )
+    def _query(self, messages: list[dict[str, str]], **kwargs):
+        headers = {
+            "Authorization": f"Bearer {self._api_key}",
+            "Content-Type": "application/json",
+        }
+
+        payload = {
+            "model": self.config.model_name,
+            "messages": messages,
+            "usage": {"include": True},
+            **(self.config.model_kwargs | kwargs),
+        }
+
+        try:
+            response = requests.post(self._api_url, headers=headers, data=json.dumps(payload), timeout=60)
+            response.raise_for_status()
+            return response.json()
+        except requests.exceptions.HTTPError as e:
+            if response.status_code == 401:
+                error_msg = "Authentication failed. You can permanently set your API key with `mini-extra config set OPENROUTER_API_KEY YOUR_KEY`."
+                raise OpenRouterAuthenticationError(error_msg) from e
+            elif response.status_code == 429:
+                raise OpenRouterRateLimitError("Rate limit exceeded") from e
+            else:
+                raise OpenRouterAPIError(f"HTTP {response.status_code}: {response.text}") from e
+        except requests.exceptions.RequestException as e:
+            raise OpenRouterAPIError(f"Request failed: {e}") from e
+
+    def query(self, messages: list[dict[str, str]], **kwargs) -> dict:
+        response = self._query(messages, **kwargs)
+
+        # Extract cost from usage information
+        usage = response.get("usage", {})
+        cost = usage.get("cost", 0.0)
+
+        # If total_cost is not available, raise an error
+        if cost == 0.0:
+            raise OpenRouterAPIError(
+                f"No cost information available from OpenRouter API for model {self.config.model_name}. "
+                "Cost tracking is required but not provided by the API response."
+            )
+
+        self.n_calls += 1
+        self.cost += cost
+        GLOBAL_MODEL_STATS.add(cost)
+
+        return {
+            "content": response["choices"][0]["message"]["content"] or "",
+            "extra": {
+                "response": response,  # already is json
+            },
+        }
+
+    def get_template_vars(self) -> dict[str, Any]:
+        return asdict(self.config) | {"n_model_calls": self.n_calls, "model_cost": self.cost}

{mini_swe_agent-1.10.0 → mini_swe_agent-1.11.1}/src/minisweagent/run/extra/config.py

@@ -33,8 +33,11 @@ This setup will ask you for your model and an API key.
 
 Here's a few popular models and the required API keys:
 
-[bold green]claude-sonnet-4-20250514[/bold green] ([bold green]ANTHROPIC_API_KEY[/bold green])
+[bold green]anthropic/claude-sonnet-4-20250514[/bold green] ([bold green]ANTHROPIC_API_KEY[/bold green])
 [bold green]openai/gpt-5[/bold green] or [bold green]openai/gpt-5-mini[/bold green] ([bold green]OPENAI_API_KEY[/bold green])
+[bold green]gemini/gemini-2.5-pro[/bold green] ([bold green]GEMINI_API_KEY[/bold green])
+
+[bold]Note: Please always include the provider in the model name.[/bold]
 
 [bold yellow]You can leave any setting blank to skip it.[/bold yellow]
 

{mini_swe_agent-1.10.0 → mini_swe_agent-1.11.1}/src/minisweagent/run/extra/swebench.py

@@ -15,6 +15,7 @@ from pathlib import Path
 import typer
 import yaml
 from datasets import load_dataset
+from jinja2 import StrictUndefined, Template
 from rich.live import Live
 
 from minisweagent import Environment
@@ -72,17 +73,25 @@ def get_swebench_docker_image_name(instance: dict) -> str:
         # Docker doesn't allow double underscore, so we replace them with a magic token
         iid = instance["instance_id"]
         id_docker_compatible = iid.replace("__", "_1776_")
-        image_name = f"swebench/sweb.eval.x86_64.{id_docker_compatible}:latest".lower()
+        image_name = f"docker.io/swebench/sweb.eval.x86_64.{id_docker_compatible}:latest".lower()
     return image_name
 
 
 def get_sb_environment(config: dict, instance: dict) -> Environment:
-    image_name = get_swebench_docker_image_name(instance)
     env_config = config.setdefault("environment", {})
-    if env_config.get("environment_class") == "singularity":
-        image_name = "docker://" + image_name
-    env_config["image"] = image_name
-    return get_environment(env_config, default_type="docker")
+    env_config["environment_class"] = env_config.get("environment_class", "docker")
+    image_name = get_swebench_docker_image_name(instance)
+    if env_config["environment_class"] == "docker":
+        env_config["image"] = image_name
+    elif env_config["environment_class"] == "singularity":
+        env_config["image"] = "docker://" + image_name
+    env = get_environment(env_config)
+    if startup_command := config.get("run", {}).get("env_startup_command"):
+        startup_command = Template(startup_command, undefined=StrictUndefined).render(**instance)
+        out = env.execute(startup_command)
+        if out["returncode"] != 0:
+            raise RuntimeError(f"Error executing startup command: {out}")
+    return env
 
 
 def update_preds_file(output_path: Path, instance_id: str, model_name: str, result: str):

{mini_swe_agent-1.10.0 → mini_swe_agent-1.11.1}/src/minisweagent/run/github_issue.py

@@ -37,13 +37,16 @@ def fetch_github_issue(issue_url: str) -> str:
     return f"GitHub Issue: {title}\n\n{body}"
 
 
+# fmt: off
 @app.command()
 def main(
     issue_url: str = typer.Option(prompt="Enter GitHub issue URL", help="GitHub issue URL"),
     config: Path = typer.Option(DEFAULT_CONFIG, "-c", "--config", help="Path to config file"),
     model: str | None = typer.Option(None, "-m", "--model", help="Model to use"),
+    model_class: str | None = typer.Option(None, "--model-class", help="Model class to use (e.g., 'anthropic' or 'minisweagent.models.anthropic.AnthropicModel')", rich_help_panel="Advanced"),
     yolo: bool = typer.Option(False, "-y", "--yolo", help="Run without confirmation"),
 ) -> InteractiveAgent:
+    # fmt: on
     """Run mini-SWE-agent on a GitHub issue"""
     configure_if_first_time()
 
@@ -51,6 +54,8 @@ def main(
     _agent_config = _config.setdefault("agent", {})
     if yolo:
         _agent_config["mode"] = "yolo"
+    if model_class is not None:
+        _config.setdefault("model", {})["model_class"] = model_class
 
     task = fetch_github_issue(issue_url)
 

{mini_swe_agent-1.10.0 → mini_swe_agent-1.11.1}/src/minisweagent/run/mini.py

@@ -48,12 +48,13 @@ More information about the usage: [bold green]https://mini-swe-agent.com/latest/
 def main(
     visual: bool = typer.Option(False, "-v", "--visual", help="Toggle (pager-style) UI (Textual) depending on the MSWEA_VISUAL_MODE_DEFAULT environment setting",),
     model_name: str | None = typer.Option( None, "-m", "--model", help="Model to use",),
+    model_class: str | None = typer.Option(None, "--model-class", help="Model class to use (e.g., 'anthropic' or 'minisweagent.models.anthropic.AnthropicModel')", rich_help_panel="Advanced"),
     task: str | None = typer.Option(None, "-t", "--task", help="Task/problem statement", show_default=False),
     yolo: bool = typer.Option(False, "-y", "--yolo", help="Run without confirmation"),
     cost_limit: float | None = typer.Option(None, "-l", "--cost-limit", help="Cost limit. Set to 0 to disable."),
     config_spec: Path = typer.Option(DEFAULT_CONFIG, "-c", "--config", help="Path to config file"),
     output: Path | None = typer.Option(DEFAULT_OUTPUT, "-o", "--output", help="Output trajectory file"),
-    exit_immediately: bool = typer.Option( False, "--exit-immediately", help="Exit immediately when the agent wants to finish instead of prompting."),
+    exit_immediately: bool = typer.Option( False, "--exit-immediately", help="Exit immediately when the agent wants to finish instead of prompting.", rich_help_panel="Advanced"),
 ) -> Any:
     # fmt: on
     configure_if_first_time()
@@ -72,11 +73,14 @@ def main(
         )
         console.print("[bold green]Got that, thanks![/bold green]")
 
-    config["agent"]["mode"] = "confirm" if not yolo else "yolo"
+    if yolo:
+        config.setdefault("agent", {})["mode"] = "yolo"
     if cost_limit:
-        config["agent"]["cost_limit"] = cost_limit
+        config.setdefault("agent", {})["cost_limit"] = cost_limit
     if exit_immediately:
-        config["agent"]["confirm_exit"] = False
+        config.setdefault("agent", {})["confirm_exit"] = False
+    if model_class is not None:
+        config.setdefault("model", {})["model_class"] = model_class
     model = get_model(model_name, config.get("model", {}))
     env = LocalEnvironment(**config.get("env", {}))