mini-swe-agent 1.14.3__tar.gz → 1.14.4__tar.gz

{mini_swe_agent-1.14.3/src/mini_swe_agent.egg-info → mini_swe_agent-1.14.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini-swe-agent
-Version: 1.14.3
+Version: 1.14.4
 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.14.3 → mini_swe_agent-1.14.4/src/mini_swe_agent.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mini-swe-agent
-Version: 1.14.3
+Version: 1.14.4
 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.14.3 → mini_swe_agent-1.14.4}/src/minisweagent/__init__.py

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

{mini_swe_agent-1.14.3 → mini_swe_agent-1.14.4}/src/minisweagent/models/portkey_model.py

@@ -96,14 +96,25 @@ class PortkeyModel:
             if response_for_cost_calc.model:
                 response_for_cost_calc.model = self.config.litellm_model_name_override
         prompt_tokens = response_for_cost_calc.usage.prompt_tokens
+        if prompt_tokens is None:
+            logger.warning(
+                f"Prompt tokens are None for model {self.config.model_name}. Setting to 0. Full response: {response_for_cost_calc.model_dump()}"
+            )
+            prompt_tokens = 0
         total_tokens = response_for_cost_calc.usage.total_tokens
         completion_tokens = response_for_cost_calc.usage.completion_tokens
+        if completion_tokens is None:
+            logger.warning(
+                f"Completion tokens are None for model {self.config.model_name}. Setting to 0. Full response: {response_for_cost_calc.model_dump()}"
+            )
+            completion_tokens = 0
         if total_tokens - prompt_tokens - completion_tokens != 0:
             # This is most likely related to how portkey treats cached tokens: It doesn't count them towards the prompt tokens (?)
             logger.warning(
                 f"WARNING: Total tokens - prompt tokens - completion tokens != 0: {response_for_cost_calc.model_dump()}."
                 " This is probably a portkey bug or incompatibility with litellm cost tracking. "
-                "Setting prompt tokens based on total tokens and completion tokens. You might want to double check your costs."
+                "Setting prompt tokens based on total tokens and completion tokens. You might want to double check your costs. "
+                "Full response: {response_for_cost_calc.model_dump()}"
             )
             response_for_cost_calc.usage.prompt_tokens = total_tokens - completion_tokens
         try: