From d17816dd209e5316ac16e1c04de111da7f308004 Mon Sep 17 00:00:00 2001
From: Indigo <119925802+ubirdio@users.noreply.github.com>
Date: Mon, 1 Jul 2024 15:54:02 +1000
Subject: [PATCH] Fix usage of parameter-based credentials when using
 vertex_ai_beta route

---
 litellm/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/litellm/utils.py b/litellm/utils.py
index 4b80d203b4..7302bea753 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -2406,7 +2406,7 @@ def get_optional_params(
         elif k == "hf_model_name" and custom_llm_provider != "sagemaker":
             continue
         elif (
-            k.startswith("vertex_") and custom_llm_provider != "vertex_ai"
+            k.startswith("vertex_") and custom_llm_provider != "vertex_ai" and custom_llm_provider != "vertex_ai_beta"
         ):  # allow dynamically setting vertex ai init logic
             continue
         passed_params[k] = v