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.py): fix casting

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-04 13:54:16 -07:00
parent 4dbb46cf42
commit 2e40ab959d
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

@ -1005,7 +1005,7 @@ def test_reading_keys_os_environ():
assert client.max_retries == int( assert client.max_retries == int(
os.environ["AZURE_MAX_RETRIES"] os.environ["AZURE_MAX_RETRIES"]
), f"{client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" ), f"{client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}"
assert client.timeout == ( assert client.timeout == int(
os.environ["AZURE_TIMEOUT"] os.environ["AZURE_TIMEOUT"]
), f"{client.timeout} vs {os.environ['AZURE_TIMEOUT']}" ), f"{client.timeout} vs {os.environ['AZURE_TIMEOUT']}"
print("sync client set correctly!") print("sync client set correctly!")
@ -1017,7 +1017,7 @@ def test_reading_keys_os_environ():
assert stream_client.max_retries == int( assert stream_client.max_retries == int(
os.environ["AZURE_MAX_RETRIES"] os.environ["AZURE_MAX_RETRIES"]
), f"{stream_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" ), f"{stream_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}"
assert stream_client.timeout == ( assert stream_client.timeout == int(
os.environ["AZURE_STREAM_TIMEOUT"] os.environ["AZURE_STREAM_TIMEOUT"]
), f"{stream_client.timeout} vs {os.environ['AZURE_TIMEOUT']}" ), f"{stream_client.timeout} vs {os.environ['AZURE_TIMEOUT']}"
print("sync stream client set correctly!") print("sync stream client set correctly!")
@ -1102,7 +1102,7 @@ def test_reading_openai_keys_os_environ():
assert client.max_retries == int( assert client.max_retries == int(
os.environ["AZURE_MAX_RETRIES"] os.environ["AZURE_MAX_RETRIES"]
), f"{client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" ), f"{client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}"
assert client.timeout == ( assert client.timeout == int(
os.environ["AZURE_TIMEOUT"] os.environ["AZURE_TIMEOUT"]
), f"{client.timeout} vs {os.environ['AZURE_TIMEOUT']}" ), f"{client.timeout} vs {os.environ['AZURE_TIMEOUT']}"
print("sync client set correctly!") print("sync client set correctly!")
@ -1113,7 +1113,7 @@ def test_reading_openai_keys_os_environ():
assert stream_client.max_retries == int( assert stream_client.max_retries == int(
os.environ["AZURE_MAX_RETRIES"] os.environ["AZURE_MAX_RETRIES"]
), f"{stream_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" ), f"{stream_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}"
assert stream_client.timeout == ( assert stream_client.timeout == int(
os.environ["AZURE_STREAM_TIMEOUT"] os.environ["AZURE_STREAM_TIMEOUT"]
), f"{stream_client.timeout} vs {os.environ['AZURE_TIMEOUT']}" ), f"{stream_client.timeout} vs {os.environ['AZURE_TIMEOUT']}"
print("sync stream client set correctly!") print("sync stream client set correctly!")