From 842625a6f092bfc0929a8289f2d97c5bed20c028 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 18 Mar 2025 12:25:32 -0700 Subject: [PATCH] :test_completion_azure_ad_toke --- litellm/llms/openai/common_utils.py | 29 +++++++++++++---------------- 1 file changed, 13 insertions(+), 16 deletions(-) diff --git a/litellm/llms/openai/common_utils.py b/litellm/llms/openai/common_utils.py index 99a6c8837a..649ce2e0f1 100644 --- a/litellm/llms/openai/common_utils.py +++ b/litellm/llms/openai/common_utils.py @@ -102,22 +102,19 @@ class BaseOpenAILLM: @staticmethod def _get_async_http_client() -> Optional[httpx.AsyncClient]: - if litellm.ssl_verify: - return httpx.AsyncClient( - limits=httpx.Limits( - max_connections=1000, max_keepalive_connections=100 - ), - verify=litellm.ssl_verify, - ) - return litellm.aclient_session + if litellm.aclient_session is not None: + return litellm.aclient_session + + return httpx.AsyncClient( + limits=httpx.Limits(max_connections=1000, max_keepalive_connections=100), + verify=litellm.ssl_verify, + ) @staticmethod def _get_sync_http_client() -> Optional[httpx.Client]: - if litellm.ssl_verify: - return httpx.Client( - limits=httpx.Limits( - max_connections=1000, max_keepalive_connections=100 - ), - verify=litellm.ssl_verify, - ) - return litellm.client_session + if litellm.client_session is not None: + return litellm.client_session + return httpx.Client( + limits=httpx.Limits(max_connections=1000, max_keepalive_connections=100), + verify=litellm.ssl_verify, + )