phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +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
# check if timeout for stream/non stream clients is set correctly
from openai import AsyncAzureOpenAI
async_client: AsyncAzureOpenAI = router.cache.get_cache(f"{model_id}_async_client") # type: ignore
async_client = router.cache.get_cache(f"{model_id}_async_client")
stream_async_client = router.cache.get_cache(
f"{model_id}_stream_async_client"
)
@ -72,12 +70,12 @@ def test_init_clients():
print()
print(async_client._base_url)
assert (
str(async_client._base_url)
== "https://openai-gpt-4-test-v-1.openai.azure.com/openai/"
async_client._base_url
== "https://openai-gpt-4-test-v-1.openai.azure.com//openai/"
) # openai python adds the extra /
assert (
str(stream_async_client._base_url)
== "https://openai-gpt-4-test-v-1.openai.azure.com/openai/"
stream_async_client._base_url
== "https://openai-gpt-4-test-v-1.openai.azure.com//openai/"
)
print("PASSED !")