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

fix(utils.py): fix get_api_base

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-16 18:50:27 -07:00
parent 7656bd1c3f
commit 4d0d6127d8
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -5407,7 +5407,9 @@ def get_api_base(model: str, optional_params: dict) -> Optional[str]:
get_api_base(model="gemini/gemini-pro") get_api_base(model="gemini/gemini-pro")
``` ```
""" """
_optional_params = LiteLLM_Params(**optional_params) # convert to pydantic object _optional_params = LiteLLM_Params(
model=model, **optional_params
) # convert to pydantic object
# get llm provider # get llm provider
try: try:
model, custom_llm_provider, dynamic_api_key, api_base = get_llm_provider( model, custom_llm_provider, dynamic_api_key, api_base = get_llm_provider(