tunacode-cli 0.0.46__tar.gz → 0.0.47__tar.gz

{tunacode_cli-0.0.46/src/tunacode_cli.egg-info → tunacode_cli-0.0.47}/PKG-INFO

@@ -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 → tunacode_cli-0.0.47}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tunacode-cli"
-version = "0.0.46"
+version = "0.0.47"
 description = "Your agentic CLI developer."
 keywords = ["cli", "agent", "development", "automation"]
 readme = "README.md"

{tunacode_cli-0.0.46 → tunacode_cli-0.0.47}/src/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_cli-0.0.46 → tunacode_cli-0.0.47}/src/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_cli-0.0.46 → tunacode_cli-0.0.47}/src/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 → tunacode_cli-0.0.47/src/tunacode_cli.egg-info}/PKG-INFO

@@ -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 → tunacode_cli-0.0.47}/tests/characterization/repl/test_input_handling.py

@@ -22,6 +22,23 @@ async def test_repl_input_validation(inputs, expected_process_calls):
     state_manager.session.current_model = "gpt-test"
     state_manager.session.input_sessions = {}
     state_manager.session.show_thoughts = False
+    state_manager.session.user_config = {"context_window_size": 100000}
+    state_manager.session.max_tokens = 100000
+    state_manager.session.total_tokens = 0
+    state_manager.session.update_token_count = MagicMock()
+    state_manager.session._startup_shown = True
+    state_manager.session.current_task = None
+    state_manager.session.spinner = None
+    state_manager.session.session_total_usage = {
+        "prompt_tokens": 0,
+        "completion_tokens": 0,
+        "cost": 0.0,
+    }
+    state_manager.session.last_call_usage = {
+        "prompt_tokens": 0,
+        "completion_tokens": 0,
+        "cost": 0.0,
+    }
 
     # Patch UI and agent
     with (
@@ -39,20 +56,41 @@ async def test_repl_input_validation(inputs, expected_process_calls):
         agent_instance.run_mcp_servers = MagicMock(return_value=mcp_context)
         get_agent.return_value = agent_instance
 
+        # Track which input we're on
+        input_index = [0]
+
         # Simulate multiline_input yielding from the inputs list
         async def fake_multiline_input(*a, **kw):
-            return inputs.pop(0)
+            if input_index[0] < len(inputs):
+                val = inputs[input_index[0]]
+                input_index[0] += 1
+                # Give time for background tasks to complete
+                await asyncio.sleep(0.01)
+                return val
+            else:
+                raise Exception("No more inputs")
+
+        # Mock process_request to prevent actual processing
+        async def mock_process_request(text, state_manager, *args, **kwargs):
+            try:
+                await asyncio.sleep(0)  # Yield to event loop
+            finally:
+                # Mimic the real process_request clearing current_task
+                state_manager.session.current_task = None
 
         with (
             patch.object(repl_mod.ui, "multiline_input", new=fake_multiline_input),
             patch("tunacode.cli.repl.get_app") as get_app,
+            patch.object(repl_mod, "process_request", new=mock_process_request),
         ):
-            # Mock background task creation
-            # Mock a task-like object that is awaitable and has a .done() method
-            bg_task = asyncio.create_task(asyncio.sleep(0))
-            # bg_task.done = MagicMock(return_value=True)
+            # Mock background task creation to properly handle coroutines
+            def mock_create_background_task(coro):
+                # Create and return a task from the coroutine
+                return asyncio.create_task(coro)
 
-            get_app.return_value.create_background_task = MagicMock(return_value=bg_task)
+            get_app.return_value.create_background_task = MagicMock(
+                side_effect=mock_create_background_task
+            )
 
             await repl_mod.repl(state_manager)