agent name works

Author: nina-kollman

packages/opentelemetry-instrumentation-openai/opentelemetry/instrumentation/openai/shared/chat_wrappers.py

@@ -268,11 +268,6 @@ async def _handle_request(span, kwargs, instance):
     _set_request_attributes(span, kwargs, instance)
     _set_client_attributes(span, instance)
 
-    # Set agent name attribute if available in context
-    agent_name = get_value("agent_name")
-    if agent_name is not None:
-        _set_span_attribute(span, GEN_AI_AGENT_NAME, str(agent_name))
-
     if should_emit_events():
         for message in kwargs.get("messages", []):
             emit_event(

packages/traceloop-sdk/traceloop/sdk/tracing/__init__.py

@@ -1,2 +1,2 @@
 from traceloop.sdk.tracing.context_manager import get_tracer
-from traceloop.sdk.tracing.tracing import set_workflow_name
+from traceloop.sdk.tracing.tracing import set_workflow_name, set_agent_name

packages/traceloop-sdk/traceloop/sdk/tracing/tracing.py

@@ -35,6 +35,7 @@
 from traceloop.sdk.utils import is_notebook
 from traceloop.sdk.utils.package_check import is_package_installed
 from typing import Callable, Dict, List, Optional, Set, Union
+from opentelemetry.semconv._incubating.attributes.gen_ai_attributes import GEN_AI_AGENT_NAME
 
 
 TRACER_NAME = "traceloop.tracer"
@@ -322,7 +323,7 @@ def default_span_processor_on_start(span: Span, parent_context: Context | None =
 
     agent_name = get_value("agent_name")
     if agent_name is not None:
-        span.set_attribute(SpanAttributes.TRACELOOP_ENTITY_NAME, str(agent_name))
+        span.set_attribute(GEN_AI_AGENT_NAME, str(agent_name))
 
     entity_path = get_value("entity_path")
     if entity_path is not None:
@@ -1177,10 +1178,6 @@ def metrics_common_attributes():
     if workflow_name is not None:
         common_attributes[SpanAttributes.TRACELOOP_WORKFLOW_NAME] = workflow_name
 
-    agent_name = get_value("agent_name")
-    if agent_name is not None:
-        common_attributes[SpanAttributes.TRACELOOP_ENTITY_NAME] = agent_name
-
     entity_name = get_value("entity_name")
     if entity_name is not None:
         common_attributes[SpanAttributes.TRACELOOP_ENTITY_NAME] = entity_name