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

fix: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-05 22:20:33 -07:00
parent c7cbe16324
commit 5c93a52567
1 changed files with 5 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/tests/test_router_init.py
View file

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