phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

_arize_otel_logger

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-18 22:19:51 -07:00
parent 43eedf26f7
commit a27782e657
2 changed files with 12 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/integrations/opentelemetry.py
View file

@ -884,15 +884,20 @@ class OpenTelemetry(CustomLogger):
return BatchSpanProcessor(ConsoleSpanExporter())
@staticmethod
def _get_headers_dictionary(headers: Optional[Union[str, dict]]) -> dict:
def _get_headers_dictionary(headers: Optional[Union[str, dict]]) -> Dict[str, str]:
"""
Convert a string or dictionary of headers into a dictionary of headers.
"""
_split_otel_headers = {}
_split_otel_headers: Dict[str, str] = {}
if headers:
if isinstance(headers, str):
_split_otel_headers = headers.split("=")
_split_otel_headers = {_split_otel_headers[0]: _split_otel_headers[1]}
# when passed HEADERS="x-honeycomb-team=B85YgLm96******"
# Split only on first '=' occurrence
parts = headers.split("=", 1)
if len(parts) == 2:
_split_otel_headers = {parts[0]: parts[1]}
else:
_split_otel_headers = {}
elif isinstance(headers, dict):
_split_otel_headers = headers
return _split_otel_headers

6
litellm/litellm_core_utils/litellm_logging.py
View file

@ -2663,9 +2663,9 @@ def _init_custom_logger_compatible_class( # noqa: PLR0915
and callback.callback_name == "arize"
):
return callback # type: ignore
_otel_logger = ArizeLogger(config=otel_config, callback_name="arize")
_in_memory_loggers.append(_otel_logger)
return _otel_logger # type: ignore
_arize_otel_logger = ArizeLogger(config=otel_config, callback_name="arize")
_in_memory_loggers.append(_arize_otel_logger)
return _arize_otel_logger # type: ignore
elif logging_integration == "arize_phoenix":
from litellm.integrations.opentelemetry import (
OpenTelemetry,