From 27e94cc89cd48716d9949fd402e04dda0f932e98 Mon Sep 17 00:00:00 2001 From: Fabian Schindler Date: Tue, 30 Sep 2025 15:33:53 +0200 Subject: [PATCH] fix(openai-agents): Move _set_agent_data call to ai_client_span function --- sentry_sdk/integrations/openai_agents/spans/ai_client.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sentry_sdk/integrations/openai_agents/spans/ai_client.py b/sentry_sdk/integrations/openai_agents/spans/ai_client.py index d325ae86e3..e215edfd26 100644 --- a/sentry_sdk/integrations/openai_agents/spans/ai_client.py +++ b/sentry_sdk/integrations/openai_agents/spans/ai_client.py @@ -28,12 +28,13 @@ def ai_client_span(agent, get_response_kwargs): # 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") + _set_agent_data(span, agent) + return span def update_ai_client_span(span, agent, get_response_kwargs, result): # type: (sentry_sdk.tracing.Span, Agent, dict[str, Any], Any) -> None - _set_agent_data(span, agent) _set_usage_data(span, result.usage) _set_input_data(span, get_response_kwargs) _set_output_data(span, result)