Skip to content

Commit c4ce725

Browse files
committed
fix(genai): use standard baggage key for agent name
1 parent 6a71890 commit c4ce725

5 files changed

Lines changed: 10 additions & 17 deletions

File tree

util/opentelemetry-util-genai/CHANGELOG-loongsuite.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
99

1010
### Added
1111

12-
- Propagate agent name through the internal `traffic.llm_sdk.gen_ai.agent.name`
13-
Baggage key during `start_invoke_agent` and automatically apply it as
14-
`gen_ai.agent.name` to nested GenAI child span attributes, including LLM,
15-
embedding, tool, retrieval, rerank, memory, entry, and ReAct step invocations.
12+
- Propagate agent name through the `gen_ai.agent.name` Baggage key during
13+
`start_invoke_agent` and automatically apply it to nested GenAI child span
14+
attributes, including LLM, embedding, tool, retrieval, rerank, memory, entry,
15+
and ReAct step invocations.
1616

1717
## Version 0.6.0 (2026-06-03)
1818

util/opentelemetry-util-genai/README-loongsuite.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@ OpenTelemetry Util for GenAI - LoongSuite 扩展
2929

3030
- **llm**:聊天/补全类调用;支持多模态消息的**外置存储与 URI 替换**(见第 4 节),减轻 Trace 体积。
3131
- **invoke_agent / create_agent**:Agent 调用与创建;``invoke_agent`` 可将
32-
Agent 名称写入内部 Baggage key ``traffic.llm_sdk.gen_ai.agent.name``,
33-
使其下游 LLM、工具、检索、ReAct step 等 GenAI 子 Span 自动带上
32+
Agent 名称写入 Baggage key ``gen_ai.agent.name``,使其下游 LLM、
33+
工具、检索、ReAct step 等 GenAI 子 Span 自动带上
3434
``gen_ai.agent.name`` 属性。
3535
- **embedding**:向量嵌入。
3636
- **execute_tool**:工具/函数执行。

util/opentelemetry-util-genai/src/opentelemetry/util/genai/extended_handler.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -147,9 +147,6 @@ class ExtendedTelemetryHandler(MultimodalProcessingMixin, TelemetryHandler): #
147147
- Async multimodal processing (via MultimodalProcessingMixin)
148148
"""
149149

150-
# Aliyun Python Agent Extension
151-
_AUTO_INJECT_BAGGAGE_PREFIX = "traffic.llm_sdk."
152-
153150
def __init__(
154151
self,
155152
tracer_provider: TracerProvider | None = None,
@@ -490,9 +487,8 @@ def start_invoke_agent(
490487
current_context = _current_context(context)
491488
ctx = set_span_in_context(span, current_context)
492489
if invocation.agent_name:
493-
# Aliyun Python Agent Extension
494490
ctx = baggage.set_baggage(
495-
f"{self._AUTO_INJECT_BAGGAGE_PREFIX}{GenAI.GEN_AI_AGENT_NAME}",
491+
GenAI.GEN_AI_AGENT_NAME,
496492
invocation.agent_name,
497493
ctx,
498494
)

util/opentelemetry-util-genai/src/opentelemetry/util/genai/handler.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -103,11 +103,8 @@
103103
# LoongSuite Extension
104104
logger = logging.getLogger(__name__)
105105

106-
# Aliyun Python Agent Extension
107-
_AUTO_INJECT_BAGGAGE_PREFIX = "traffic.llm_sdk."
108-
_AGENT_NAME_BAGGAGE_KEY = (
109-
f"{_AUTO_INJECT_BAGGAGE_PREFIX}{GenAI.GEN_AI_AGENT_NAME}"
110-
)
106+
# LoongSuite Extension
107+
_AGENT_NAME_BAGGAGE_KEY = GenAI.GEN_AI_AGENT_NAME
111108

112109

113110
class _InvocationWithAttributes(Protocol):

util/opentelemetry-util-genai/tests/test_extended_handler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@
104104
Uri,
105105
)
106106

107-
_AGENT_NAME_BAGGAGE_KEY = "traffic.llm_sdk.gen_ai.agent.name"
107+
_AGENT_NAME_BAGGAGE_KEY = "gen_ai.agent.name"
108108

109109

110110
def patch_env_vars(

0 commit comments

Comments
 (0)