From ab43b80bb91dd7144b84474337db133aad4d2320 Mon Sep 17 00:00:00 2001 From: Dvir Rezenman Date: Thu, 18 Jun 2026 10:30:38 +0300 Subject: [PATCH] fix(sdk): @agent decorator no longer overwrites enclosing @workflow name --- .../traceloop-sdk/traceloop/sdk/decorators/base.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/packages/traceloop-sdk/traceloop/sdk/decorators/base.py b/packages/traceloop-sdk/traceloop/sdk/decorators/base.py index b54dbc84b1..84583aaadf 100644 --- a/packages/traceloop-sdk/traceloop/sdk/decorators/base.py +++ b/packages/traceloop-sdk/traceloop/sdk/decorators/base.py @@ -137,14 +137,12 @@ def _is_async_method(fn): def _setup_span(entity_name, tlp_span_kind, version): """Sets up the OpenTelemetry span and context""" - if tlp_span_kind in [ - TraceloopSpanKindValues.WORKFLOW, - TraceloopSpanKindValues.AGENT, - ]: + if tlp_span_kind == TraceloopSpanKindValues.WORKFLOW: set_workflow_name(entity_name) - - if tlp_span_kind == TraceloopSpanKindValues.AGENT: + elif tlp_span_kind == TraceloopSpanKindValues.AGENT: set_agent_name(entity_name) + if context_api.get_value("workflow_name") is None: + set_workflow_name(entity_name) span_name = f"{entity_name}.{tlp_span_kind.value}"