From 90787076723f9911cd6426f04ba551e111abc836 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 28 Feb 2024 08:13:13 -0800 Subject: [PATCH] (fix) vertex ai project/location --- litellm/main.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/litellm/main.py b/litellm/main.py index 1a6c8e178..44bebb351 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -1467,12 +1467,12 @@ def completion( response = model_response elif custom_llm_provider == "vertex_ai": vertex_ai_project = ( - optional_params.pop("vertex_ai_project", None) + optional_params.pop("vertex_project", None) or litellm.vertex_project or get_secret("VERTEXAI_PROJECT") ) vertex_ai_location = ( - optional_params.pop("vertex_ai_location", None) + optional_params.pop("vertex_location", None) or litellm.vertex_location or get_secret("VERTEXAI_LOCATION") ) @@ -2566,12 +2566,12 @@ def embedding( ) elif custom_llm_provider == "vertex_ai": vertex_ai_project = ( - optional_params.pop("vertex_ai_project", None) + optional_params.pop("vertex_project", None) or litellm.vertex_project or get_secret("VERTEXAI_PROJECT") ) vertex_ai_location = ( - optional_params.pop("vertex_ai_location", None) + optional_params.pop("vertex_location", None) or litellm.vertex_location or get_secret("VERTEXAI_LOCATION") )