From c372c873a0666a476026cd34867d8afac9711009 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 4 Apr 2024 13:31:39 -0700 Subject: [PATCH] test(test_router.py): fix test to check cast --- litellm/tests/test_router.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py index 811704bd1..82547e5a3 100644 --- a/litellm/tests/test_router.py +++ b/litellm/tests/test_router.py @@ -980,7 +980,7 @@ def test_reading_keys_os_environ(): assert async_client.max_retries == int( 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"] ), f"{async_client.timeout} vs {os.environ['AZURE_TIMEOUT']}" print("async client set correctly!") @@ -993,7 +993,7 @@ def test_reading_keys_os_environ(): assert stream_async_client.max_retries == int( 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"] ), f"{stream_async_client.timeout} vs {os.environ['AZURE_TIMEOUT']}" print("async stream client set correctly!") @@ -1079,7 +1079,7 @@ def test_reading_openai_keys_os_environ(): assert async_client.max_retries == int( 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"] ), f"{async_client.timeout} vs {os.environ['AZURE_TIMEOUT']}" print("async client set correctly!") @@ -1091,7 +1091,7 @@ def test_reading_openai_keys_os_environ(): assert stream_async_client.max_retries == int( 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"] ), f"{stream_async_client.timeout} vs {os.environ['AZURE_TIMEOUT']}" print("async stream client set correctly!")