Merge pull request #4486 from ubirdio/vertex-ai-beta-fix

Fix usage of parameter-based credentials when using vertex_ai_beta route
This commit is contained in:
Krish Dholakia
2024-07-01 08:15:49 -07:00
committed by GitHub
+1 -1
View File
@@ -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