From 49464df216d136c22e8c877bfcb72fad253f1569 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 22 Nov 2024 17:33:22 -0800 Subject: [PATCH] fix ensure async clients test --- tests/code_coverage_tests/ensure_async_clients_test.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/code_coverage_tests/ensure_async_clients_test.py b/tests/code_coverage_tests/ensure_async_clients_test.py index a4203d881..0565de9b3 100644 --- a/tests/code_coverage_tests/ensure_async_clients_test.py +++ b/tests/code_coverage_tests/ensure_async_clients_test.py @@ -5,9 +5,19 @@ ALLOWED_FILES = [ # local files "../../litellm/__init__.py", "../../litellm/llms/custom_httpx/http_handler.py", + "../../litellm/router_utils/client_initalization_utils.py", + "../../litellm/llms/custom_httpx/http_handler.py", + "../../litellm/llms/huggingface_restapi.py", + "../../litellm/llms/base.py", + "../../litellm/llms/custom_httpx/httpx_handler.py", # when running on ci/cd "./litellm/__init__.py", "./litellm/llms/custom_httpx/http_handler.py", + "./litellm/router_utils/client_initalization_utils.py", + "./litellm/llms/custom_httpx/http_handler.py", + "./litellm/llms/huggingface_restapi.py", + "./litellm/llms/base.py", + "./litellm/llms/custom_httpx/httpx_handler.py", ] warning_msg = "this is a serious violation that can impact latency. Creating Async clients per request can add +500ms per request"