From 72dd21dc472b23beafc60700b69575676397eba8 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 10 Sep 2024 13:29:02 -0700 Subject: [PATCH] fix linting error --- litellm/llms/custom_httpx/http_handler.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/litellm/llms/custom_httpx/http_handler.py b/litellm/llms/custom_httpx/http_handler.py index ed5773be83..e63bd3f545 100644 --- a/litellm/llms/custom_httpx/http_handler.py +++ b/litellm/llms/custom_httpx/http_handler.py @@ -1,7 +1,7 @@ import asyncio import os import traceback -from typing import Any, Mapping, Optional, Union +from typing import TYPE_CHECKING, Any, Mapping, Optional, Union import httpx @@ -9,6 +9,11 @@ import litellm from .types import httpxSpecialProvider +if TYPE_CHECKING: + from litellm import LlmProviders +else: + LlmProviders = Any + try: from litellm._version import version except: @@ -381,7 +386,7 @@ class HTTPHandler: def get_async_httpx_client( - llm_provider: Union[litellm.LlmProviders, httpxSpecialProvider], + llm_provider: Union[LlmProviders, httpxSpecialProvider], params: Optional[dict] = None, ) -> AsyncHTTPHandler: """