diff --git a/tests/code_coverage_tests/ensure_async_clients_test.py b/tests/code_coverage_tests/ensure_async_clients_test.py index 0565de9b38..db47973b69 100644 --- a/tests/code_coverage_tests/ensure_async_clients_test.py +++ b/tests/code_coverage_tests/ensure_async_clients_test.py @@ -10,6 +10,7 @@ ALLOWED_FILES = [ "../../litellm/llms/huggingface_restapi.py", "../../litellm/llms/base.py", "../../litellm/llms/custom_httpx/httpx_handler.py", + "../../litellm/llms/openai/common_utils.py", # when running on ci/cd "./litellm/__init__.py", "./litellm/llms/custom_httpx/http_handler.py", @@ -18,6 +19,7 @@ ALLOWED_FILES = [ "./litellm/llms/huggingface_restapi.py", "./litellm/llms/base.py", "./litellm/llms/custom_httpx/httpx_handler.py", + "./litellm/llms/openai/common_utils.py", ] warning_msg = "this is a serious violation that can impact latency. Creating Async clients per request can add +500ms per request"