tunacode-cli 0.0.46__py3-none-any.whl → 0.0.47__py3-none-any.whl

tunacode/constants.py

@@ -7,7 +7,7 @@ Centralizes all magic strings, UI text, error messages, and application constant
 
 # Application info
 APP_NAME = "TunaCode"
-APP_VERSION = "0.0.46"
+APP_VERSION = "0.0.47"
 
 # File patterns
 GUIDE_FILE_PATTERN = "{name}.md"

tunacode/core/token_usage/cost_calculator.py

@@ -49,6 +49,14 @@ class CostCalculator:
 
         pricing = model_config.pricing
 
+        # Safety check for None pricing
+        if not pricing:
+            return 0.0
+
+        # Safety check for None pricing attributes
+        if pricing.input is None or pricing.output is None:
+            return 0.0
+
         input_cost = (prompt_tokens / TOKENS_PER_MILLION) * pricing.input
 
         output_cost = (completion_tokens / TOKENS_PER_MILLION) * pricing.output

tunacode/core/token_usage/usage_tracker.py

@@ -60,18 +60,27 @@ class UsageTracker(UsageTrackerProtocol):
             if not api_model_name.startswith(provider_prefix + ":"):
                 final_model_name = f"{provider_prefix}:{api_model_name}"
 
-        return self.calculator.calculate_cost(
+        cost = self.calculator.calculate_cost(
             prompt_tokens=parsed_data.get("prompt_tokens", 0),
             completion_tokens=parsed_data.get("completion_tokens", 0),
             model_name=final_model_name,
         )
 
+        # Ensure cost is never None
+        return cost if cost is not None else 0.0
+
     def _update_state(self, parsed_data: dict, cost: float):
         """Updates the last_call and session_total usage in the state."""
         session = self.state_manager.session
         prompt_tokens = parsed_data.get("prompt_tokens", 0)
         completion_tokens = parsed_data.get("completion_tokens", 0)
 
+        # Initialize usage dicts if they're None
+        if session.last_call_usage is None:
+            session.last_call_usage = {"prompt_tokens": 0, "completion_tokens": 0, "cost": 0.0}
+        if session.session_total_usage is None:
+            session.session_total_usage = {"prompt_tokens": 0, "completion_tokens": 0, "cost": 0.0}
+
         # Update last call usage
         session.last_call_usage["prompt_tokens"] = prompt_tokens
         session.last_call_usage["completion_tokens"] = completion_tokens
@@ -85,6 +94,13 @@ class UsageTracker(UsageTrackerProtocol):
     async def _display_summary(self):
         """Formats and prints the usage summary to the console."""
         session = self.state_manager.session
+
+        # Initialize usage dicts if they're None
+        if session.last_call_usage is None:
+            session.last_call_usage = {"prompt_tokens": 0, "completion_tokens": 0, "cost": 0.0}
+        if session.session_total_usage is None:
+            session.session_total_usage = {"prompt_tokens": 0, "completion_tokens": 0, "cost": 0.0}
+
         prompt = session.last_call_usage["prompt_tokens"]
         completion = session.last_call_usage["completion_tokens"]
         last_cost = session.last_call_usage["cost"]

{tunacode_cli-0.0.46.dist-info → tunacode_cli-0.0.47.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tunacode-cli
-Version: 0.0.46
+Version: 0.0.47
 Summary: Your agentic CLI developer.
 Author-email: larock22 <noreply@github.com>
 License: MIT

{tunacode_cli-0.0.46.dist-info → tunacode_cli-0.0.47.dist-info}/RECORD

@@ -1,5 +1,5 @@
 tunacode/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tunacode/constants.py,sha256=o1NSdboUcXjPgeYQJ7sCC7HtStJP2e8fE9lo0Jio0N4,5168
+tunacode/constants.py,sha256=833fgfAmkcJmU1RmSYDhGCOzoU-eUMdgbcfBBwUQOZ4,5168
 tunacode/context.py,sha256=_gXVCyjU052jlyRAl9tklZSwl5U_zI_EIX8XN87VVWE,2786
 tunacode/exceptions.py,sha256=oDO1SVKOgjcKIylwqdbqh_g5my4roU5mB9Nv4n_Vb0s,3877
 tunacode/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -46,8 +46,8 @@ tunacode/core/setup/coordinator.py,sha256=oVTN2xIeJERXitVJpkIk9tDGLs1D1bxIRmaogJ
 tunacode/core/setup/environment_setup.py,sha256=n3IrObKEynHZSwtUJ1FddMg2C4sHz7ca42awemImV8s,2225
 tunacode/core/setup/git_safety_setup.py,sha256=CRIqrQt0QUJQRS344njty_iCqTorrDhHlXRuET7w0Tk,6714
 tunacode/core/token_usage/api_response_parser.py,sha256=CTtqGaFaxpkzkW3TEbe00QJzyRULpWN1EQxIYMleseg,1622
-tunacode/core/token_usage/cost_calculator.py,sha256=oQPMphGhqIt7NKdOg1o5Zbo59_nwFWmRJMQ30ViiCWs,1835
-tunacode/core/token_usage/usage_tracker.py,sha256=uKCpdZgmDfLauwsawSCifMu0kJE3lAnK7Sjd-0KYUgA,3894
+tunacode/core/token_usage/cost_calculator.py,sha256=RjO-O0JENBuGOrWP7QgBZlZxeXC-PAIr8tj_9p_BxOU,2058
+tunacode/core/token_usage/usage_tracker.py,sha256=kuAjUCyQkFykPy5mqsLRbKhZW298pyiCuFGn-ptBpy4,4657
 tunacode/prompts/system.md,sha256=hXpjZ8Yiv2Acr2_6EmC2uOklP8FbmvyYR9oais-1KLk,16290
 tunacode/services/__init__.py,sha256=w_E8QK6RnvKSvU866eDe8BCRV26rAm4d3R-Yg06OWCU,19
 tunacode/services/mcp.py,sha256=R48X73KQjQ9vwhBrtbWHSBl-4K99QXmbIhh5J_1Gezo,3046
@@ -91,9 +91,9 @@ tunacode/utils/system.py,sha256=FSoibTIH0eybs4oNzbYyufIiV6gb77QaeY2yGqW39AY,1138
 tunacode/utils/text_utils.py,sha256=6YBD9QfkDO44-6jxnwRWIpmfIifPG-NqMzy_O2NAouc,7277
 tunacode/utils/token_counter.py,sha256=lLbkrNUraRQn5RMhwnGurqq1RHFDyn4AaFhruONWIxo,2690
 tunacode/utils/user_configuration.py,sha256=Ilz8dpGVJDBE2iLWHAPT0xR8D51VRKV3kIbsAz8Bboc,3275
-tunacode_cli-0.0.46.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
-tunacode_cli-0.0.46.dist-info/METADATA,sha256=kO5Xol8ooArWqyfluRNjVq9t3MmZh9J02-ySYlpW_WA,5902
-tunacode_cli-0.0.46.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-tunacode_cli-0.0.46.dist-info/entry_points.txt,sha256=hbkytikj4dGu6rizPuAd_DGUPBGF191RTnhr9wdhORY,51
-tunacode_cli-0.0.46.dist-info/top_level.txt,sha256=lKy2P6BWNi5XSA4DHFvyjQ14V26lDZctwdmhEJrxQbU,9
-tunacode_cli-0.0.46.dist-info/RECORD,,
+tunacode_cli-0.0.47.dist-info/licenses/LICENSE,sha256=Btzdu2kIoMbdSp6OyCLupB1aRgpTCJ_szMimgEnpkkE,1056
+tunacode_cli-0.0.47.dist-info/METADATA,sha256=oHfTMPtEUgbgVd7Nm0bb39wAKAuPKRHskBpZmDnBxMM,5902
+tunacode_cli-0.0.47.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tunacode_cli-0.0.47.dist-info/entry_points.txt,sha256=hbkytikj4dGu6rizPuAd_DGUPBGF191RTnhr9wdhORY,51
+tunacode_cli-0.0.47.dist-info/top_level.txt,sha256=lKy2P6BWNi5XSA4DHFvyjQ14V26lDZctwdmhEJrxQbU,9
+tunacode_cli-0.0.47.dist-info/RECORD,,