phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(main.py): fix model alias map logic

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-20 20:49:02 -08:00
parent b752ba8a46
commit b2e52eedf8
2 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/utils.py
View file

@ -2247,6 +2247,12 @@ def get_llm_provider(model: str, custom_llm_provider: Optional[str] = None, api_
try:
dynamic_api_key = None
# check if llm provider provided
if litellm.model_alias_map and model in litellm.model_alias_map:
model = litellm.model_alias_map[
model
] # update the model to the actual value if an alias has been passed in
if custom_llm_provider:
return model, custom_llm_provider, dynamic_api_key, api_base