fix import Span

This commit is contained in:
Ishaan Jaff 2024-06-07 09:38:37 -07:00
parent c7dda09773
commit de9f10195f

View file

@ -1,13 +1,20 @@
from pydantic import BaseModel, Extra, Field, model_validator, Json, ConfigDict from pydantic import BaseModel, Extra, Field, model_validator, Json, ConfigDict
from dataclasses import fields from dataclasses import fields
import enum 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 from datetime import datetime
import uuid, json, sys, os import uuid, json, sys, os
from litellm.types.router import UpdateRouterConfig from litellm.types.router import UpdateRouterConfig
from litellm.types.utils import ProviderField from litellm.types.utils import ProviderField
from typing_extensions import Annotated 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): class LitellmUserRoles(str, enum.Enum):