set_data to set_attribute

Author: antonpirker

sentry_sdk/integrations/openai_agents/spans/ai_client.py

@@ -25,7 +25,7 @@ def ai_client_span(agent, get_response_kwargs):
         origin=SPAN_ORIGIN,
     )
     # TODO-anton: remove hardcoded stuff and replace something that also works for embedding and so on
-    span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "chat")
+    span.set_attribute(SPANDATA.GEN_AI_OPERATION_NAME, "chat")
 
     return span
 

sentry_sdk/integrations/openai_agents/spans/execute_tool.py

@@ -20,17 +20,17 @@ def execute_tool_span(tool, *args, **kwargs):
         origin=SPAN_ORIGIN,
     )
 
-    span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "execute_tool")
+    span.set_attribute(SPANDATA.GEN_AI_OPERATION_NAME, "execute_tool")
 
     if tool.__class__.__name__ == "FunctionTool":
-        span.set_data(SPANDATA.GEN_AI_TOOL_TYPE, "function")
+        span.set_attribute(SPANDATA.GEN_AI_TOOL_TYPE, "function")
 
-    span.set_data(SPANDATA.GEN_AI_TOOL_NAME, tool.name)
-    span.set_data(SPANDATA.GEN_AI_TOOL_DESCRIPTION, tool.description)
+    span.set_attribute(SPANDATA.GEN_AI_TOOL_NAME, tool.name)
+    span.set_attribute(SPANDATA.GEN_AI_TOOL_DESCRIPTION, tool.description)
 
     if should_send_default_pii():
         input = args[1]
-        span.set_data(SPANDATA.GEN_AI_TOOL_INPUT, input)
+        span.set_attribute(SPANDATA.GEN_AI_TOOL_INPUT, input)
 
     return span
 
@@ -40,4 +40,4 @@ def update_execute_tool_span(span, agent, tool, result):
     _set_agent_data(span, agent)
 
     if should_send_default_pii():
-        span.set_data(SPANDATA.GEN_AI_TOOL_OUTPUT, result)
+        span.set_attribute(SPANDATA.GEN_AI_TOOL_OUTPUT, result)

sentry_sdk/integrations/openai_agents/spans/handoff.py

@@ -16,4 +16,4 @@ def handoff_span(context, from_agent, to_agent_name):
         name=f"handoff from {from_agent.name} to {to_agent_name}",
         origin=SPAN_ORIGIN,
     ) as span:
-        span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "handoff")
+        span.set_attribute(SPANDATA.GEN_AI_OPERATION_NAME, "handoff")

sentry_sdk/integrations/openai_agents/spans/invoke_agent.py

@@ -20,7 +20,7 @@ def invoke_agent_span(context, agent):
     )
     span.__enter__()
 
-    span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "invoke_agent")
+    span.set_attribute(SPANDATA.GEN_AI_OPERATION_NAME, "invoke_agent")
 
     _set_agent_data(span, agent)
 

sentry_sdk/integrations/openai_agents/utils.py

@@ -31,60 +31,60 @@ def _capture_exception(exc):
 
 def _set_agent_data(span, agent):
     # type: (sentry_sdk.tracing.Span, agents.Agent) -> None
-    span.set_data(
+    span.set_attribute(
         SPANDATA.GEN_AI_SYSTEM, "openai"
     )  # See footnote for  https://opentelemetry.io/docs/specs/semconv/registry/attributes/gen-ai/#gen-ai-system for explanation why.
 
-    span.set_data(SPANDATA.GEN_AI_AGENT_NAME, agent.name)
+    span.set_attribute(SPANDATA.GEN_AI_AGENT_NAME, agent.name)
 
     if agent.model_settings.max_tokens:
-        span.set_data(
+        span.set_attribute(
             SPANDATA.GEN_AI_REQUEST_MAX_TOKENS, agent.model_settings.max_tokens
         )
 
     if agent.model:
-        span.set_data(SPANDATA.GEN_AI_REQUEST_MODEL, agent.model)
+        span.set_attribute(SPANDATA.GEN_AI_REQUEST_MODEL, agent.model)
 
     if agent.model_settings.presence_penalty:
-        span.set_data(
+        span.set_attribute(
             SPANDATA.GEN_AI_REQUEST_PRESENCE_PENALTY,
             agent.model_settings.presence_penalty,
         )
 
     if agent.model_settings.temperature:
-        span.set_data(
+        span.set_attribute(
             SPANDATA.GEN_AI_REQUEST_TEMPERATURE, agent.model_settings.temperature
         )
 
     if agent.model_settings.top_p:
-        span.set_data(SPANDATA.GEN_AI_REQUEST_TOP_P, agent.model_settings.top_p)
+        span.set_attribute(SPANDATA.GEN_AI_REQUEST_TOP_P, agent.model_settings.top_p)
 
     if agent.model_settings.frequency_penalty:
-        span.set_data(
+        span.set_attribute(
             SPANDATA.GEN_AI_REQUEST_FREQUENCY_PENALTY,
             agent.model_settings.frequency_penalty,
         )
 
     if len(agent.tools) > 0:
-        span.set_data(
+        span.set_attribute(
             SPANDATA.GEN_AI_REQUEST_AVAILABLE_TOOLS,
             safe_serialize([vars(tool) for tool in agent.tools]),
         )
 
 
 def _set_usage_data(span, usage):
     # type: (sentry_sdk.tracing.Span, Usage) -> None
-    span.set_data(SPANDATA.GEN_AI_USAGE_INPUT_TOKENS, usage.input_tokens)
-    span.set_data(
+    span.set_attribute(SPANDATA.GEN_AI_USAGE_INPUT_TOKENS, usage.input_tokens)
+    span.set_attribute(
         SPANDATA.GEN_AI_USAGE_INPUT_TOKENS_CACHED,
         usage.input_tokens_details.cached_tokens,
     )
-    span.set_data(SPANDATA.GEN_AI_USAGE_OUTPUT_TOKENS, usage.output_tokens)
-    span.set_data(
+    span.set_attribute(SPANDATA.GEN_AI_USAGE_OUTPUT_TOKENS, usage.output_tokens)
+    span.set_attribute(
         SPANDATA.GEN_AI_USAGE_OUTPUT_TOKENS_REASONING,
         usage.output_tokens_details.reasoning_tokens,
     )
-    span.set_data(SPANDATA.GEN_AI_USAGE_TOTAL_TOKENS, usage.total_tokens)
+    span.set_attribute(SPANDATA.GEN_AI_USAGE_TOTAL_TOKENS, usage.total_tokens)
 
 
 def _set_input_data(span, get_response_kwargs):
@@ -118,7 +118,9 @@ def _set_input_data(span, get_response_kwargs):
         if len(messages) > 0:
             request_messages.append({"role": role, "content": messages})
 
-    span.set_data(SPANDATA.GEN_AI_REQUEST_MESSAGES, safe_serialize(request_messages))
+    span.set_attribute(
+        SPANDATA.GEN_AI_REQUEST_MESSAGES, safe_serialize(request_messages)
+    )
 
 
 def _set_output_data(span, result):
@@ -143,12 +145,12 @@ def _set_output_data(span, result):
                     output_messages["response"].append(output_message.dict())
 
     if len(output_messages["tool"]) > 0:
-        span.set_data(
+        span.set_attribute(
             SPANDATA.GEN_AI_RESPONSE_TOOL_CALLS, safe_serialize(output_messages["tool"])
         )
 
     if len(output_messages["response"]) > 0:
-        span.set_data(
+        span.set_attribute(
             SPANDATA.GEN_AI_RESPONSE_TEXT, safe_serialize(output_messages["response"])
         )