fix imports

litellm/_service_logger.py

@@ -4,7 +4,6 @@ from typing import TYPE_CHECKING, Any, Optional, Union
 
 import litellm
 from litellm._logging import verbose_logger
-from litellm_proxy._types import UserAPIKeyAuth
 
 from .integrations.custom_logger import CustomLogger
 from .integrations.datadog.datadog import DataDogLogger
@@ -15,11 +14,14 @@ from .types.services import ServiceLoggerPayload, ServiceTypes
 if TYPE_CHECKING:
     from opentelemetry.trace import Span as _Span
 
+    from litellm_proxy._types import UserAPIKeyAuth
+
     Span = Union[_Span, Any]
     OTELClass = OpenTelemetry
 else:
     Span = Any
     OTELClass = Any
+    UserAPIKeyAuth = Any
 
 
 class ServiceLogging(CustomLogger):

tests/litellm/proxy/test_common_request_processing.py

@@ -1,11 +1,19 @@
 import copy
+import os
+import sys
 import uuid
 from unittest.mock import AsyncMock, MagicMock
 
 import pytest
 from fastapi import Request
 
+sys.path.insert(
+    0, os.path.abspath("../../../..")
+)  # Adds the parent directory to the system path
+
+
 import litellm
+import litellm_proxy
 from litellm.integrations.opentelemetry import UserAPIKeyAuth
 from litellm_proxy.common_request_processing import (
     ProxyBaseLLMRequestProcessing,