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 48

style: ruff format databricks cost_calculator

Browse source 
CI lint failed because ruff format --check wanted single-line startswith/generic_cost_per_token calls.
This commit is contained in:
michaelxer 2026-08-03 11:12:22 +07:00
parent 0576edf003
commit 7bf4f89313
1 changed files with 3 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -21,13 +21,9 @@ def _resolve_databricks_base_model(model: str) -> str:
"""
if model.startswith("databricks/dbrx-instruct") or model.startswith("dbrx-instruct"):
return "databricks-dbrx-instruct"
if model.startswith("databricks/meta-llama-3.1-70b-instruct") or model.startswith(
"meta-llama-3.1-70b-instruct"
):
if model.startswith("databricks/meta-llama-3.1-70b-instruct") or model.startswith("meta-llama-3.1-70b-instruct"):
return "databricks-meta-llama-3-1-70b-instruct"
if model.startswith("databricks/meta-llama-3.1-405b-instruct") or model.startswith(
"meta-llama-3.1-405b-instruct"
):
if model.startswith("databricks/meta-llama-3.1-405b-instruct") or model.startswith("meta-llama-3.1-405b-instruct"):
return "databricks-meta-llama-3-1-405b-instruct"
if (
model.startswith("databricks/mixtral-8x7b-instruct-v0.1")
@ -57,6 +53,4 @@ def cost_per_token(model: str, usage: Usage) -> tuple[float, float]:
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"
)
return generic_cost_per_token(model=base_model, usage=usage, custom_llm_provider="databricks")