scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-15 23:31:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 165

fix(langfuse): preserve existing trace tags

Browse source
This commit is contained in:
Devin AI 2026-07-21 06:37:50 +00:00
parent 2c1dfdb4f0
commit 5be1e40a84
2 changed files with 28 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/integrations/langfuse/langfuse.py
View file

@ -716,7 +716,7 @@ class LangFuseLogger:
propagated_version = trace_version if isinstance(trace_version, str) else None
trace_release = trace_params.get("release")
propagated_release = trace_release if isinstance(trace_release, str) else None
propagated_tags = tags if tags else None
propagated_tags = tags if tags and existing_trace_id is None else None
trace_name_value = trace_params.get("name")
propagated_trace_name = (
trace_name_value if isinstance(trace_name_value, str) and trace_name_value else trace_name

27
tests/test_litellm/integrations/test_langfuse.py
View file

@ -482,6 +482,33 @@ class TestLangfuseUsageDetails(unittest.TestCase):
assert self.last_trace_kwargs.get("id") == "call-id-xyz"
def test_log_langfuse_v2_does_not_propagate_tags_to_existing_trace(self):
payload = self._build_standard_logging_payload()
payload["request_tags"] = ["request-tag"]
kwargs = self._build_langfuse_kwargs(payload)
with patch(
"litellm.integrations.langfuse.langfuse._add_prompt_to_generation_params",
side_effect=lambda generation_params, **kwargs: generation_params,
create=True,
):
self.logger._log_langfuse_v2(
user_id="user-1",
metadata={"existing_trace_id": "existing-trace-id"},
litellm_params={"metadata": {}},
output=None,
start_time=datetime.datetime.utcnow(),
end_time=datetime.datetime.utcnow(),
kwargs=kwargs,
optional_params={},
input=None,
response_obj=None,
level="INFO",
litellm_call_id="call-id-xyz",
)
assert self.mock_langfuse.propagate_attributes.call_args.kwargs["tags"] is None
def test_log_langfuse_v2_uses_litellm_trace_id_fallback_over_call_id(self):
"""
When standard_logging_object has no trace_id, but kwargs contains