Fix usage of parameter-based credentials when using vertex_ai_beta route

This commit is contained in:
Indigo 2024-07-01 15:54:02 +10:00 committed by GitHub
parent 46fe089204
commit d17816dd20
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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