phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix test

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-10 13:15:50 -07:00
parent 1e8cf9f2a6
commit d7afeee71c
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/llms/vertex_ai_and_google_ai_studio/gemini/vertex_and_google_ai_studio_gemini.py
View file

@ -1293,7 +1293,10 @@ class VertexLLM(BaseLLM):
_async_client_params = {} _async_client_params = {}
if timeout: if timeout:
_async_client_params["timeout"] = timeout _async_client_params["timeout"] = timeout
if client is None or not isinstance(client, AsyncHTTPHandler):
client = _get_async_httpx_client(params=_async_client_params) client = _get_async_httpx_client(params=_async_client_params)
else:
client = client # type: ignore
## LOGGING ## LOGGING
logging_obj.pre_call( logging_obj.pre_call(
input=messages, input=messages,