From 8e961ccad6985c166ea6c2bfd8acda9fbb7c57b9 Mon Sep 17 00:00:00 2001 From: Alexander Alderman Webb Date: Mon, 5 Oct 2026 09:10:11 +0200 Subject: [PATCH] chore(pydantic-ai): Step setting gen_ai.provider.name on Invoke Agent spans --- .../integrations/pydantic_ai/spans/ai_client.py | 11 +++++++++++ sentry_sdk/integrations/pydantic_ai/utils.py | 4 ---- 2 files changed, 11 insertions(+), 4 deletions(-) diff --git a/sentry_sdk/integrations/pydantic_ai/spans/ai_client.py b/sentry_sdk/integrations/pydantic_ai/spans/ai_client.py index bd9137c39c..ac2489a4bd 100644 --- a/sentry_sdk/integrations/pydantic_ai/spans/ai_client.py +++ b/sentry_sdk/integrations/pydantic_ai/spans/ai_client.py @@ -290,6 +290,17 @@ def ai_client_context( ) _set_agent_data(context.span, agent) + + # Extract model information + model_obj = model + if not model_obj and agent and hasattr(agent, "model"): + model_obj = agent.model + + if model_obj: + # Set system from model + if hasattr(model_obj, "system"): + context.span.set_attribute(SPANDATA.GEN_AI_PROVIDER_NAME, model_obj.system) + _set_model_data(context.span, agent, model, model_settings) _set_available_tools(context.span, agent) diff --git a/sentry_sdk/integrations/pydantic_ai/utils.py b/sentry_sdk/integrations/pydantic_ai/utils.py index ad9fd62c05..235dc58ae2 100644 --- a/sentry_sdk/integrations/pydantic_ai/utils.py +++ b/sentry_sdk/integrations/pydantic_ai/utils.py @@ -74,10 +74,6 @@ def _set_model_data( model_obj = agent.model if model_obj: - # Set system from model - if hasattr(model_obj, "system"): - span.set_attribute(SPANDATA.GEN_AI_PROVIDER_NAME, model_obj.system) - # Set model name model_name = _get_model_name(model_obj) if model_name: