(fix) get_llm_provider

2023-10-20 15:00:21 -07:00 · 2023-10-20 15:00:21 -07:00 · aa6a01904d
commit aa6a01904d
parent 0b0564167c
1 changed files with 1 additions and 1 deletions
--- 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: