fix import loc2

This commit is contained in:
Ishaan Jaff 2025-04-23 16:17:42 -07:00
parent 1b6b05e8d4
commit fec670c341

View file

@ -25,14 +25,16 @@ from litellm.types.utils import (
StandardCallbackDynamicParams, StandardCallbackDynamicParams,
StandardLoggingPayload, StandardLoggingPayload,
) )
from litellm_proxy._types import UserAPIKeyAuth
if TYPE_CHECKING: if TYPE_CHECKING:
from opentelemetry.trace import Span as _Span from opentelemetry.trace import Span as _Span
from litellm_proxy._types import UserAPIKeyAuth
Span = Union[_Span, Any] Span = Union[_Span, Any]
else: else:
Span = Any Span = Any
UserAPIKeyAuth = Any
class CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callback#callback-class class CustomLogger: # https://docs.litellm.ai/docs/observability/custom_callback#callback-class