tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-21 02:20:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix test

Browse Source
This commit is contained in:
Krrish Dholakia
2024-08-05 22:20:33 -07:00
parent ea902f40de
commit 4603eedcc1
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_router_init.py
+5 -7
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 !")