From 6fcf7e8d1b598470b23d22e75a8ea5731fe7cd31 Mon Sep 17 00:00:00 2001 From: michaelxer Date: Mon, 3 Aug 2026 13:16:07 +0700 Subject: [PATCH] fix(databricks): use native tuple annotations in cost_calculator Replace typing.Tuple with tuple[] to satisfy the strict ruff UP006/UP035 budget on this path. --- litellm/llms/databricks/cost_calculator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/llms/databricks/cost_calculator.py b/litellm/llms/databricks/cost_calculator.py index 471f8e08977..17b59531b0c 100644 --- a/litellm/llms/databricks/cost_calculator.py +++ b/litellm/llms/databricks/cost_calculator.py @@ -50,7 +50,7 @@ def cost_per_token(model: str, usage: Usage) -> tuple[float, float]: - usage: LiteLLM Usage block, containing anthropic caching information Returns: - Tuple[float, float] - prompt_cost_in_usd, completion_cost_in_usd + tuple[float, float] - prompt_cost_in_usd, completion_cost_in_usd """ base_model = _resolve_databricks_base_model(model) return generic_cost_per_token(model=base_model, usage=usage, custom_llm_provider="databricks")