From 4d0d6127d82dc00ece7c6d56bb8394b6e17615d3 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Tue, 16 Apr 2024 18:50:27 -0700
Subject: [PATCH] fix(utils.py): fix get_api_base

---
 litellm/utils.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/litellm/utils.py b/litellm/utils.py
index 3a319f24b..e10a44e1a 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -5407,7 +5407,9 @@ def get_api_base(model: str, optional_params: dict) -> Optional[str]:
     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
     try:
         model, custom_llm_provider, dynamic_api_key, api_base = get_llm_provider(