From 42df97db3c058270fc69280d590cd4d51102ef83 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 7 Jun 2024 12:50:50 -0700 Subject: [PATCH] fix importing litellm --- litellm/proxy/litellm_pre_call_utils.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/litellm_pre_call_utils.py b/litellm/proxy/litellm_pre_call_utils.py index ddc2eb5528e..1614a6dfe66 100644 --- a/litellm/proxy/litellm_pre_call_utils.py +++ b/litellm/proxy/litellm_pre_call_utils.py @@ -1,9 +1,15 @@ import copy from fastapi import Request -from typing import Any, Dict, Optional +from typing import Any, Dict, Optional, TYPE_CHECKING from litellm.proxy._types import UserAPIKeyAuth from litellm._logging import verbose_proxy_logger, verbose_logger -from litellm.proxy.proxy_server import ProxyConfig + +if TYPE_CHECKING: + from litellm.proxy.proxy_server import ProxyConfig as _ProxyConfig + + ProxyConfig = _ProxyConfig +else: + ProxyConfig = Any def parse_cache_control(cache_control):