diff --git a/litellm/llms/vertex_ai_and_google_ai_studio/gemini/vertex_and_google_ai_studio_gemini.py b/litellm/llms/vertex_ai_and_google_ai_studio/gemini/vertex_and_google_ai_studio_gemini.py index abe8b29edf..000512d945 100644 --- a/litellm/llms/vertex_ai_and_google_ai_studio/gemini/vertex_and_google_ai_studio_gemini.py +++ b/litellm/llms/vertex_ai_and_google_ai_studio/gemini/vertex_and_google_ai_studio_gemini.py @@ -1293,7 +1293,10 @@ class VertexLLM(BaseLLM): _async_client_params = {} if timeout: _async_client_params["timeout"] = timeout - client = _get_async_httpx_client(params=_async_client_params) + if client is None or not isinstance(client, AsyncHTTPHandler): + client = _get_async_httpx_client(params=_async_client_params) + else: + client = client # type: ignore ## LOGGING logging_obj.pre_call( input=messages,