diff --git a/litellm/litellm_core_utils/llm_cost_calc/tool_call_cost_tracking.py b/litellm/litellm_core_utils/llm_cost_calc/tool_call_cost_tracking.py index 4ecc4a33702..75bb699292e 100644 --- a/litellm/litellm_core_utils/llm_cost_calc/tool_call_cost_tracking.py +++ b/litellm/litellm_core_utils/llm_cost_calc/tool_call_cost_tracking.py @@ -248,7 +248,7 @@ class StandardBuiltInToolCostTracking: ) @staticmethod - def _extract_token_counts(computer_use_usage: Any) -> tuple[Optional[int], Optional[int]]: + def _extract_token_counts(computer_use_usage: Any) -> Tuple[Optional[int], Optional[int]]: """Extract and convert token counts safely.""" input_tokens = None output_tokens = None