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

(fix) get_llm_provider

Browse source
This commit is contained in:
ishaan-jaff 2023-10-20 15:00:21 -07:00
parent c8f89f3484
commit 02a20218ab
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/utils.py
View file

@ -1550,7 +1550,7 @@ def get_llm_provider(model: str, custom_llm_provider: Optional[str] = None, api_
dynamic_api_key = None dynamic_api_key = None
# check if llm provider provided # check if llm provider provided
if custom_llm_provider: if custom_llm_provider:
return model, custom_llm_provider, dynamic_api_key return model, custom_llm_provider, dynamic_api_key, api_base
# check if llm provider part of model name # check if llm provider part of model name
if model.split("/",1)[0] in litellm.provider_list: if model.split("/",1)[0] in litellm.provider_list: