scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-02 07:13:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 49

fix(databricks): use native tuple annotations in cost_calculator

Browse source 
Replace typing.Tuple with tuple[] to satisfy the strict ruff UP006/UP035 budget on this path.
This commit is contained in:
michaelxer 2026-08-03 13:16:07 +07:00
parent 7bf4f89313
commit 6fcf7e8d1b
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/databricks/cost_calculator.py
View file

@ -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")