move _get_parent_otel_span_from_kwargs to otel.py

This commit is contained in:
Ishaan Jaff 2024-07-27 11:12:13 -07:00
parent cde46a4a09
commit 40f9e67be4
2 changed files with 22 additions and 10 deletions

View file

@ -21,6 +21,7 @@ from openai._models import BaseModel as OpenAIObject
import litellm
from litellm._logging import verbose_logger
from litellm.integrations.opentelemetry import _get_parent_otel_span_from_kwargs
from litellm.types.services import ServiceLoggerPayload, ServiceTypes
@ -33,16 +34,6 @@ def print_verbose(print_statement):
pass
def _get_parent_otel_span_from_kwargs(kwargs: Optional[dict] = None):
try:
if kwargs is None:
return None
_metadata = kwargs.get("metadata") or {}
return _metadata.get("litellm_parent_otel_span")
except:
return None
class BaseCache:
def set_cache(self, key, value, **kwargs):
raise NotImplementedError