From d7afeee71cc867834107ae4ce84d8a2a086c496a Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 10 Sep 2024 13:15:50 -0700 Subject: [PATCH] fix test --- .../gemini/vertex_and_google_ai_studio_gemini.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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,