phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +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