mini-swe-agent 1.13.0__py3-none-any.whl → 1.13.2__py3-none-any.whl

{mini_swe_agent-1.13.0.dist-info → mini_swe_agent-1.13.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini-swe-agent
-Version: 1.13.0
+Version: 1.13.2
 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.dist-info → mini_swe_agent-1.13.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
-mini_swe_agent-1.13.0.dist-info/licenses/LICENSE.md,sha256=D3luWPkdHAe7LBsdD4vzqDAXw6Xewb3G-uczss0uh1s,1094
-minisweagent/__init__.py,sha256=AV8y_FPibz99WsO-rGjKRkyf0gJhEx5RGP6IBamLGG0,2016
+mini_swe_agent-1.13.2.dist-info/licenses/LICENSE.md,sha256=D3luWPkdHAe7LBsdD4vzqDAXw6Xewb3G-uczss0uh1s,1094
+minisweagent/__init__.py,sha256=L9h6Hj8V819V2RbABXFOcXgPTwCdTacsJfSAG5X7tNE,2016
 minisweagent/__main__.py,sha256=FIyAOiw--c3FQ2g240FOM1FdL0lk_PxSpixu0pQ7WFo,194
 minisweagent/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 minisweagent/agents/__init__.py,sha256=cpjJLzg1IGxLM-tZpoMJV9S33ye13XtdBO0x7DU_Lrk,48
@@ -27,7 +27,7 @@ minisweagent/models/__init__.py,sha256=hMRTK5yfn2vymiXecZsSAGusxsTvrnQQVm8ih0y_P
 minisweagent/models/anthropic.py,sha256=D8nHvvbgzPjla0He8p0O9kaXASPWg1Sai0pHsAj_Yn8,855
 minisweagent/models/litellm_model.py,sha256=RSvNzK6ksOOcTEY-kM36KJql9-iSH29kq0cKYXj2mxE,3034
 minisweagent/models/openrouter_model.py,sha256=ugI3322pyUQ8wBOuMysPoKlp0oRAmR38a6GcdjE6qHI,3748
-minisweagent/models/portkey_model.py,sha256=hsaZJhQvzGSM88BJqTEZGvR7G_wczdKayhbtC5jex44,3814
+minisweagent/models/portkey_model.py,sha256=IcFNz33AYeYgj0N9PGWV7QCn1fT8MfMWZx_OEzLaPsc,4548
 minisweagent/models/test_models.py,sha256=ItCA6ddntzkYA7dzSuUEaLMV-AE8TBuXBFP8CzpiO3U,1351
 minisweagent/models/extra/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 minisweagent/models/extra/roulette.py,sha256=SqLj_wz9Vkbxou7i9Ef4Uzmg_eheDouNySkkV7pm2Ys,2093
@@ -50,8 +50,8 @@ minisweagent/run/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
 minisweagent/run/utils/save.py,sha256=2xd-UnUzI7Fr_AUZ5KEJ53Aa4kpuuGYxkLwyUcvqyMM,2503
 minisweagent/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 minisweagent/utils/log.py,sha256=ruDMNKMrVC9NPvCeHwO3QYz5jsVNUGQB2dRAEAPAWp8,996
-mini_swe_agent-1.13.0.dist-info/METADATA,sha256=Orz3vqB1mYJeHD0WXH4-IWS3GQkHKaFsd8rAxLKDItU,14102
-mini_swe_agent-1.13.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mini_swe_agent-1.13.0.dist-info/entry_points.txt,sha256=d1_yRbTaGjs1UXHa6JQK0sKDGBIVGm8oeW0k2kfbJgQ,182
-mini_swe_agent-1.13.0.dist-info/top_level.txt,sha256=zKF4t8bFpV87fdVABZt2Da-vnb4Vkh_CxkwQx5YT4Ew,13
-mini_swe_agent-1.13.0.dist-info/RECORD,,
+mini_swe_agent-1.13.2.dist-info/METADATA,sha256=9kkyymOo_pfywhGT-wYZ0jeU84Hn8DBmwWTOjkgbdGw,14150
+mini_swe_agent-1.13.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mini_swe_agent-1.13.2.dist-info/entry_points.txt,sha256=d1_yRbTaGjs1UXHa6JQK0sKDGBIVGm8oeW0k2kfbJgQ,182
+mini_swe_agent-1.13.2.dist-info/top_level.txt,sha256=zKF4t8bFpV87fdVABZt2Da-vnb4Vkh_CxkwQx5YT4Ew,13
+mini_swe_agent-1.13.2.dist-info/RECORD,,

minisweagent/__init__.py

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

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!"
             )