phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_router_init.py): cleanup tests

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-05 21:48:43 -07:00
parent ce39649b2a
commit ea902f40de
1 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/tests/test_router_init.py
View file

@ -59,7 +59,9 @@ def test_init_clients():
assert router.cache.get_cache(f"{model_id}_stream_async_client") is not None
# check if timeout for stream/non stream clients is set correctly
async_client = router.cache.get_cache(f"{model_id}_async_client")
from openai import AsyncAzureOpenAI
async_client: AsyncAzureOpenAI = router.cache.get_cache(f"{model_id}_async_client") # type: ignore
stream_async_client = router.cache.get_cache(
f"{model_id}_stream_async_client"
)
@ -70,12 +72,12 @@ def test_init_clients():
print()
print(async_client._base_url)
assert (
async_client._base_url
== "https://openai-gpt-4-test-v-1.openai.azure.com//openai/"
str(async_client._base_url)
== "https://openai-gpt-4-test-v-1.openai.azure.com/openai/"
) # openai python adds the extra /
assert (
stream_async_client._base_url
== "https://openai-gpt-4-test-v-1.openai.azure.com//openai/"
str(stream_async_client._base_url)
== "https://openai-gpt-4-test-v-1.openai.azure.com/openai/"
)
print("PASSED !")