mini-swe-agent 1.13.0__tar.gz → 1.13.3__tar.gz

{mini_swe_agent-1.13.0/src/mini_swe_agent.egg-info → mini_swe_agent-1.13.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini-swe-agent
-Version: 1.13.0
+Version: 1.13.3
 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
@@ -65,6 +65,7 @@ Requires-Dist: mkdocstrings[python]>=0.18; extra == "dev"
 Requires-Dist: mike; extra == "dev"
 Requires-Dist: mkdocs-material; extra == "dev"
 Requires-Dist: mkdocs-glightbox; extra == "dev"
+Requires-Dist: mkdocs-redirects; extra == "dev"
 Dynamic: license-file
 
 <div align="center">

{mini_swe_agent-1.13.0 → mini_swe_agent-1.13.3}/pyproject.toml

@@ -64,6 +64,7 @@ dev = [
     "mike",
     "mkdocs-material",
     "mkdocs-glightbox",
+    "mkdocs-redirects",
 ]
 
 [project.urls]

{mini_swe_agent-1.13.0 → mini_swe_agent-1.13.3/src/mini_swe_agent.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini-swe-agent
-Version: 1.13.0
+Version: 1.13.3
 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
@@ -65,6 +65,7 @@ Requires-Dist: mkdocstrings[python]>=0.18; extra == "dev"
 Requires-Dist: mike; extra == "dev"
 Requires-Dist: mkdocs-material; extra == "dev"
 Requires-Dist: mkdocs-glightbox; extra == "dev"
+Requires-Dist: mkdocs-redirects; extra == "dev"
 Dynamic: license-file
 
 <div align="center">

{mini_swe_agent-1.13.0 → mini_swe_agent-1.13.3}/src/mini_swe_agent.egg-info/requires.txt

@@ -24,6 +24,7 @@ mkdocstrings[python]>=0.18
 mike
 mkdocs-material
 mkdocs-glightbox
+mkdocs-redirects
 
 [full]
 mini-swe-agent[dev]

{mini_swe_agent-1.13.0 → mini_swe_agent-1.13.3}/src/minisweagent/__init__.py

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

{mini_swe_agent-1.13.0 → mini_swe_agent-1.13.3}/src/minisweagent/config/default.yaml

@@ -40,7 +40,7 @@ agent:
        However, you can prefix any action with `MY_ENV_VAR=MY_VALUE cd /path/to/working/dir && ...` or write/load environment variables from files
 
     <system_information>
-    {{system}} {{release}} {{version}} {{machine}} {{processor}}
+    {{system}} {{release}} {{version}} {{machine}}
     </system_information>
 
     ## Formatting your response

{mini_swe_agent-1.13.0 → mini_swe_agent-1.13.3}/src/minisweagent/config/mini.yaml

@@ -40,7 +40,7 @@ agent:
        However, you can prefix any action with `MY_ENV_VAR=MY_VALUE cd /path/to/working/dir && ...` or write/load environment variables from files
 
     <system_information>
-    {{system}} {{release}} {{version}} {{machine}} {{processor}}
+    {{system}} {{release}} {{version}} {{machine}}
     </system_information>
 
     ## Formatting your response

{mini_swe_agent-1.13.0 → mini_swe_agent-1.13.3}/src/minisweagent/config/mini_no_temp.yaml

@@ -41,7 +41,7 @@ agent:
        However, you can prefix any action with `MY_ENV_VAR=MY_VALUE cd /path/to/working/dir && ...` or write/load environment variables from files
 
     <system_information>
-    {{system}} {{release}} {{version}} {{machine}} {{processor}}
+    {{system}} {{release}} {{version}} {{machine}}
     </system_information>
 
     ## Formatting your response

{mini_swe_agent-1.13.0 → mini_swe_agent-1.13.3}/src/minisweagent/models/portkey_model.py

@@ -30,7 +30,12 @@ class PortkeyModelConfig:
     model_kwargs: dict[str, Any] = field(default_factory=dict)
     litellm_model_registry: Path | str | None = os.getenv("LITELLM_MODEL_REGISTRY_PATH")
     """We currently use litellm to calculate costs. Here you can register additional models to litellm's model registry.
-    Note that this might change if we get better support for Portkey.
+    Note that this might change if we get better support for Portkey and change how we calculate costs.
+    """
+    litellm_model_name_override: str = ""
+    """We currently use litellm to calculate costs. Here you can override the model name to use for litellm in case it
+    doesn't match the Portkey model name.
+    Note that this might change if we get better support for Portkey and change how we calculate costs.
     """
 
 
@@ -81,11 +86,17 @@ class PortkeyModel:
 
     def query(self, messages: list[dict[str, str]], **kwargs) -> dict:
         response = self._query(messages, **kwargs)
+        response_for_cost_calc = response.model_copy()
+        if self.config.litellm_model_name_override:
+            if response_for_cost_calc.model:
+                response_for_cost_calc.model = self.config.litellm_model_name_override
         try:
-            cost = litellm.cost_calculator.completion_cost(response)
+            cost = litellm.cost_calculator.completion_cost(
+                response_for_cost_calc, model=self.config.litellm_model_name_override or None
+            )
         except Exception as e:
             logger.critical(
-                f"Error calculating cost for model {self.config.model_name}: {e}. "
+                f"Error calculating cost for model {self.config.model_name} based on {response_for_cost_calc.model_dump()}: {e}. "
                 "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!"
             )