From de9f10195fd0800efab73c0355a8b90a8ae67609 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 7 Jun 2024 09:38:37 -0700 Subject: [PATCH] fix import Span --- litellm/proxy/_types.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py index e089c0429..05efc7495 100644 --- a/litellm/proxy/_types.py +++ b/litellm/proxy/_types.py @@ -1,13 +1,20 @@ from pydantic import BaseModel, Extra, Field, model_validator, Json, ConfigDict from dataclasses import fields import enum -from typing import Optional, List, Union, Dict, Literal, Any +from typing import Optional, List, Union, Dict, Literal, Any, TYPE_CHECKING from datetime import datetime import uuid, json, sys, os from litellm.types.router import UpdateRouterConfig from litellm.types.utils import ProviderField from typing_extensions import Annotated -from opentelemetry.trace import Span + + +if TYPE_CHECKING: + from opentelemetry.trace import Span as _Span + + Span = _Span +else: + Span = Any class LitellmUserRoles(str, enum.Enum):