diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py index 5cfe42457..966969cc9 100644 --- a/litellm/tests/test_router.py +++ b/litellm/tests/test_router.py @@ -705,7 +705,7 @@ def test_reading_keys_os_environ(): assert float(model["litellm_params"]["stream_timeout"]) == float(os.environ["AZURE_STREAM_TIMEOUT"]), f"{model['litellm_params']['stream_timeout']} vs {os.environ['AZURE_STREAM_TIMEOUT']}" assert int(model["litellm_params"]["max_retries"]) == int(os.environ["AZURE_MAX_RETRIES"]), f"{model['litellm_params']['max_retries']} vs {os.environ['AZURE_MAX_RETRIES']}" print("passed testing of reading keys from os.environ") - async_client: openai.AsyncAzureOpenAI = model["async_client"] + async_client: openai.AsyncAzureOpenAI = model["async_client"] # type: ignore assert async_client.api_key == os.environ["AZURE_API_KEY"] assert async_client.base_url == os.environ["AZURE_API_BASE"] assert async_client.max_retries == (os.environ["AZURE_MAX_RETRIES"]), f"{async_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" @@ -714,7 +714,7 @@ def test_reading_keys_os_environ(): print("\n Testing async streaming client") - stream_async_client: openai.AsyncAzureOpenAI = model["stream_async_client"] + stream_async_client: openai.AsyncAzureOpenAI = model["stream_async_client"] # type: ignore assert stream_async_client.api_key == os.environ["AZURE_API_KEY"] assert stream_async_client.base_url == os.environ["AZURE_API_BASE"] assert stream_async_client.max_retries == (os.environ["AZURE_MAX_RETRIES"]), f"{stream_async_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" @@ -722,7 +722,7 @@ def test_reading_keys_os_environ(): print("async stream client set correctly!") print("\n Testing sync client") - client: openai.AzureOpenAI = model["client"] + client: openai.AzureOpenAI = model["client"] # type: ignore assert client.api_key == os.environ["AZURE_API_KEY"] assert client.base_url == os.environ["AZURE_API_BASE"] assert client.max_retries == (os.environ["AZURE_MAX_RETRIES"]), f"{client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}" @@ -730,7 +730,7 @@ def test_reading_keys_os_environ(): print("sync client set correctly!") print("\n Testing sync stream client") - stream_client: openai.AzureOpenAI = model["stream_client"] + stream_client: openai.AzureOpenAI = model["stream_client"] # type: ignore assert stream_client.api_key == os.environ["AZURE_API_KEY"] assert stream_client.base_url == os.environ["AZURE_API_BASE"] assert stream_client.max_retries == (os.environ["AZURE_MAX_RETRIES"]), f"{stream_client.max_retries} vs {os.environ['AZURE_MAX_RETRIES']}"