fix importing Span

This commit is contained in:
Ishaan Jaff 2024-06-07 09:55:59 -07:00
parent ca99456e04
commit 4828e2426f
3 changed files with 26 additions and 8 deletions

View file

@ -17,14 +17,20 @@ from litellm.proxy._types import (
LiteLLM_OrganizationTable,
LitellmUserRoles,
)
from typing import Optional, Literal, Union
from typing import Optional, Literal, TYPE_CHECKING, Any
from litellm.proxy.utils import PrismaClient, ProxyLogging, log_to_opentelemetry
from litellm.caching import DualCache
import litellm
from opentelemetry.trace import Span
from litellm.types.services import ServiceLoggerPayload, ServiceTypes
from datetime import datetime
if TYPE_CHECKING:
from opentelemetry.trace import Span as _Span
Span = _Span
else:
Span = Any
all_routes = LiteLLMRoutes.openai_routes.value + LiteLLMRoutes.management_routes.value