phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

build(model_prices_and_context_window.json): add new dbrx llama 3.3 model
All checks were successful
Read Version from pyproject.toml / read-version (push) Successful in 1m9s
Details

Browse source 
fixes llama cost calc on databricks
This commit is contained in:
Krrish Dholakia 2024-12-11 13:01:22 -08:00
parent 5885ee5e14
commit 5ec649b512
3 changed files with 29 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -4377,6 +4377,9 @@ def _strip_model_name(model: str, custom_llm_provider: Optional[str]) -> str:
):
strip_version = _strip_stable_vertex_version(model_name=model)
return strip_version
elif custom_llm_provider and (custom_llm_provider == "databricks"):
strip_version = _strip_stable_vertex_version(model_name=model)
return strip_version
else:
strip_finetune = _strip_openai_finetune_model_name(model_name=model)
return strip_finetune
@ -4542,7 +4545,6 @@ def get_model_info( # noqa: PLR0915
)
#########################
supported_openai_params = litellm.get_supported_openai_params(
model=model, custom_llm_provider=custom_llm_provider
)