From fec670c34117cd61a6e450fb9e20bc03ac12959c Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Wed, 23 Apr 2025 16:17:42 -0700
Subject: [PATCH] fix import loc2

---
 litellm/integrations/custom_logger.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/litellm/integrations/custom_logger.py b/litellm/integrations/custom_logger.py
index 580381710a..977a408c26 100644
--- a/litellm/integrations/custom_logger.py
+++ b/litellm/integrations/custom_logger.py
@@ -25,14 +25,16 @@ from litellm.types.utils import (
     StandardCallbackDynamicParams,
     StandardLoggingPayload,
 )
-from litellm_proxy._types import UserAPIKeyAuth
 
 if TYPE_CHECKING:
     from opentelemetry.trace import Span as _Span
 
+    from litellm_proxy._types import UserAPIKeyAuth
+
     Span = Union[_Span, Any]
 else:
     Span = Any
+    UserAPIKeyAuth = Any
 
 
 class CustomLogger:  # https://docs.litellm.ai/docs/observability/custom_callback#callback-class