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

fix test test_partner_models_httpx

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-14 17:52:50 -08:00
parent 6f1466a26b
commit 9d53b8836e
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/local_testing/test_amazing_vertex_completion.py
View file

@ -975,6 +975,7 @@ async def test_partner_models_httpx(model, sync_mode):
data = { data = {
"model": model, "model": model,
"messages": messages, "messages": messages,
"timeout": 10,
} }
if sync_mode: if sync_mode:
response = litellm.completion(**data) response = litellm.completion(**data)
@ -988,6 +989,8 @@ async def test_partner_models_httpx(model, sync_mode):
assert isinstance(response._hidden_params["response_cost"], float) assert isinstance(response._hidden_params["response_cost"], float)
except litellm.RateLimitError as e: except litellm.RateLimitError as e:
pass pass
except litellm.Timeout as e:
pass
except litellm.InternalServerError as e: except litellm.InternalServerError as e:
pass pass
except Exception as e: except Exception as e: