scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

fix(langfuse_otel): type metadata helper and drop null fields so they cannot clobber caller trace metadata

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-23 07:43:48 +00:00
parent 1201677f17
commit 732d2c58e6
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/integrations/langfuse/langfuse_otel.py
View file

@ -134,7 +134,7 @@ class LangfuseOtelLogger(OpenTelemetry):
safe_set_attribute(span, enum_attr.value, value)
@staticmethod
def _set_request_metadata_attributes(span: Span, kwargs: dict) -> None:
def _set_request_metadata_attributes(span: Span, kwargs: dict[str, object]) -> None:
from litellm.integrations.arize._utils import safe_set_attribute
from litellm.integrations.langfuse.langfuse import log_requester_metadata
from litellm.litellm_core_utils.redact_messages import redact_user_api_key_info
@ -146,12 +146,15 @@ class LangfuseOtelLogger(OpenTelemetry):
)
if not isinstance(request_metadata, dict):
return
observation_metadata: Final = log_requester_metadata(redact_user_api_key_info(metadata=request_metadata))
observation_metadata: Final = {
key: value
for key, value in log_requester_metadata(redact_user_api_key_info(metadata=request_metadata)).items()
if value is not None
}
safe_set_attribute(span, LangfuseSpanAttributes.OBSERVATION_METADATA.value, safe_dumps(observation_metadata))
trace_prefix: Final = LangfuseSpanAttributes.TRACE_METADATA.value
for field in _TRACE_IDENTITY_FIELDS:
value = observation_metadata.get(field)
if value is not None:
if (value := observation_metadata.get(field)) is not None:
safe_set_attribute(span, f"{trace_prefix}.{field}", value)
@staticmethod