diff --git a/litellm/main.py b/litellm/main.py index 44bebb351..433e7be31 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -1468,11 +1468,13 @@ def completion( elif custom_llm_provider == "vertex_ai": vertex_ai_project = ( optional_params.pop("vertex_project", None) + or optional_params.pop("vertex_ai_project", None) or litellm.vertex_project or get_secret("VERTEXAI_PROJECT") ) vertex_ai_location = ( optional_params.pop("vertex_location", None) + or optional_params.pop("vertex_ai_location", None) or litellm.vertex_location or get_secret("VERTEXAI_LOCATION") ) @@ -2567,11 +2569,13 @@ def embedding( elif custom_llm_provider == "vertex_ai": vertex_ai_project = ( optional_params.pop("vertex_project", None) + or optional_params.pop("vertex_ai_project", None) or litellm.vertex_project or get_secret("VERTEXAI_PROJECT") ) vertex_ai_location = ( optional_params.pop("vertex_location", None) + or optional_params.pop("vertex_ai_location", None) or litellm.vertex_location or get_secret("VERTEXAI_LOCATION") )