From aa6a01904d1ecb285c30ca1c56e7fdcd1419bd35 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 20 Oct 2023 15:00:21 -0700 Subject: [PATCH] (fix) get_llm_provider --- litellm/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index b900a14247..05bc0d4fa2 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -1550,7 +1550,7 @@ def get_llm_provider(model: str, custom_llm_provider: Optional[str] = None, api_ dynamic_api_key = None # check if llm provider provided 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 if model.split("/",1)[0] in litellm.provider_list: