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

test(test_router.py): fix test to check cast

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-04 13:31:39 -07:00
parent 1119cc49a8
commit c372c873a0
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_router.py
View file

@ -980,7 +980,7 @@ def test_reading_keys_os_environ():
assert async_client.max_retries == int( assert async_client.max_retries == int(
os.environ["AZURE_MAX_RETRIES"] os.environ["AZURE_MAX_RETRIES"]
), f"{async_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" ), f"{async_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}"
assert async_client.timeout == ( assert async_client.timeout == int(
os.environ["AZURE_TIMEOUT"] os.environ["AZURE_TIMEOUT"]
), f"{async_client.timeout} vs {os.environ['AZURE_TIMEOUT']}" ), f"{async_client.timeout} vs {os.environ['AZURE_TIMEOUT']}"
print("async client set correctly!") print("async client set correctly!")
@ -993,7 +993,7 @@ def test_reading_keys_os_environ():
assert stream_async_client.max_retries == int( assert stream_async_client.max_retries == int(
os.environ["AZURE_MAX_RETRIES"] os.environ["AZURE_MAX_RETRIES"]
), f"{stream_async_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" ), f"{stream_async_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}"
assert stream_async_client.timeout == ( assert stream_async_client.timeout == int(
os.environ["AZURE_STREAM_TIMEOUT"] os.environ["AZURE_STREAM_TIMEOUT"]
), f"{stream_async_client.timeout} vs {os.environ['AZURE_TIMEOUT']}" ), f"{stream_async_client.timeout} vs {os.environ['AZURE_TIMEOUT']}"
print("async stream client set correctly!") print("async stream client set correctly!")
@ -1079,7 +1079,7 @@ def test_reading_openai_keys_os_environ():
assert async_client.max_retries == int( assert async_client.max_retries == int(
os.environ["AZURE_MAX_RETRIES"] os.environ["AZURE_MAX_RETRIES"]
), f"{async_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" ), f"{async_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}"
assert async_client.timeout == ( assert async_client.timeout == int(
os.environ["AZURE_TIMEOUT"] os.environ["AZURE_TIMEOUT"]
), f"{async_client.timeout} vs {os.environ['AZURE_TIMEOUT']}" ), f"{async_client.timeout} vs {os.environ['AZURE_TIMEOUT']}"
print("async client set correctly!") print("async client set correctly!")
@ -1091,7 +1091,7 @@ def test_reading_openai_keys_os_environ():
assert stream_async_client.max_retries == int( assert stream_async_client.max_retries == int(
os.environ["AZURE_MAX_RETRIES"] os.environ["AZURE_MAX_RETRIES"]
), f"{stream_async_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" ), f"{stream_async_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}"
assert stream_async_client.timeout == ( assert stream_async_client.timeout == int(
os.environ["AZURE_STREAM_TIMEOUT"] os.environ["AZURE_STREAM_TIMEOUT"]
), f"{stream_async_client.timeout} vs {os.environ['AZURE_TIMEOUT']}" ), f"{stream_async_client.timeout} vs {os.environ['AZURE_TIMEOUT']}"
print("async stream client set correctly!") print("async stream client set correctly!")