ref(ai-agents): Prefer gen_ai.response.model over gen_ai.request.model (#112462)

Author: obostjancic

static/app/views/dashboards/utils/prebuiltConfigs/ai/aiAgentsModels.ts

@@ -164,8 +164,8 @@ export const AI_AGENTS_MODELS_PREBUILT_CONFIG: PrebuiltDashboard = {
       {
         dataset: WidgetType.SPANS,
         tag: {
-          key: 'gen_ai.response.model',
-          name: 'gen_ai.response.model',
+          key: SpanFields.GEN_AI_RESPONSE_MODEL,
+          name: SpanFields.GEN_AI_RESPONSE_MODEL,
           kind: FieldKind.TAG,
         },
         value: '',

static/app/views/insights/pages/agents/components/llmCallsWidget.tsx

@@ -42,7 +42,7 @@ export function LLMCallsWidget() {
 
   const generationsRequest = useSpans(
     {
-      fields: ['gen_ai.request.model', 'count()'],
+      fields: [SpanFields.GEN_AI_RESPONSE_MODEL, 'count()'],
       sorts: [{field: 'count()', kind: 'desc'}],
       search: fullQuery,
       limit: 3,
@@ -54,7 +54,7 @@ export function LLMCallsWidget() {
     {
       ...pageFilterChartParams,
       query: fullQuery,
-      groupBy: [SpanFields.GEN_AI_REQUEST_MODEL],
+      groupBy: [SpanFields.GEN_AI_RESPONSE_MODEL],
       yAxis: ['count(span.duration)'],
       sort: {field: 'count(span.duration)', kind: 'desc'},
       topEvents: 3,
@@ -109,7 +109,7 @@ export function LLMCallsWidget() {
   const footer = hasData && (
     <WidgetFooterTable>
       {models?.map((item, index) => {
-        const modelId = item['gen_ai.request.model'];
+        const modelId = item[SpanFields.GEN_AI_RESPONSE_MODEL];
         return (
           <Fragment key={modelId}>
             <div>
@@ -149,7 +149,7 @@ export function LLMCallsWidget() {
                   yAxes: ['count(span.duration)'],
                 },
               ],
-              groupBy: ['gen_ai.request.model'],
+              groupBy: [SpanFields.GEN_AI_RESPONSE_MODEL],
               query: fullQuery,
               sort: '-count(span.duration)',
               interval: pageFilterChartParams.interval,

static/app/views/insights/pages/agents/components/modelCostWidget.tsx

@@ -41,7 +41,7 @@ export function ModelCostWidget() {
 
   const tokensRequest = useSpans(
     {
-      fields: ['gen_ai.request.model', 'sum(gen_ai.cost.total_tokens)'],
+      fields: [SpanFields.GEN_AI_RESPONSE_MODEL, 'sum(gen_ai.cost.total_tokens)'],
       sorts: [{field: 'sum(gen_ai.cost.total_tokens)', kind: 'desc'}],
       search: fullQuery,
       limit: 3,
@@ -53,7 +53,7 @@ export function ModelCostWidget() {
     {
       ...pageFilterChartParams,
       query: fullQuery,
-      groupBy: [SpanFields.GEN_AI_REQUEST_MODEL],
+      groupBy: [SpanFields.GEN_AI_RESPONSE_MODEL],
       yAxis: ['sum(gen_ai.cost.total_tokens)'],
       sort: {field: 'sum(gen_ai.cost.total_tokens)', kind: 'desc'},
       topEvents: 3,
@@ -111,7 +111,7 @@ export function ModelCostWidget() {
   const footer = hasData && (
     <WidgetFooterTable>
       {tokens?.map((item, index) => {
-        const modelId = `${item['gen_ai.request.model']}`;
+        const modelId = `${item[SpanFields.GEN_AI_RESPONSE_MODEL]}`;
         return (
           <Fragment key={modelId}>
             <div>
@@ -149,7 +149,7 @@ export function ModelCostWidget() {
                   yAxes: ['sum(gen_ai.cost.total_tokens)'],
                 },
               ],
-              groupBy: ['gen_ai.request.model'],
+              groupBy: [SpanFields.GEN_AI_RESPONSE_MODEL],
               query: fullQuery,
               sort: '-sum(gen_ai.cost.total_tokens)',
               interval: pageFilterChartParams.interval,

static/app/views/insights/pages/agents/components/modelsTable.tsx

@@ -38,6 +38,7 @@ import {getAIGenerationsFilter} from 'sentry/views/insights/pages/agents/utils/q
 import {Referrer} from 'sentry/views/insights/pages/agents/utils/referrers';
 import {DurationCell} from 'sentry/views/insights/pages/platform/shared/table/DurationCell';
 import {NumberCell} from 'sentry/views/insights/pages/platform/shared/table/NumberCell';
+import {SpanFields} from 'sentry/views/insights/types';
 
 interface TableData {
   avg: number;
@@ -100,7 +101,7 @@ export function ModelsTable() {
   const modelsRequest = useSpans(
     {
       fields: [
-        'gen_ai.request.model',
+        SpanFields.GEN_AI_RESPONSE_MODEL,
         'sum(gen_ai.usage.input_tokens)',
         'sum(gen_ai.usage.output_tokens)',
         'sum(gen_ai.usage.output_tokens.reasoning)',
@@ -126,7 +127,7 @@ export function ModelsTable() {
     }
 
     return modelsRequest.data.map(span => ({
-      model: span['gen_ai.request.model'],
+      model: span[SpanFields.GEN_AI_RESPONSE_MODEL],
       requests: span['count()'] ?? 0,
       avg: span['avg(span.duration)'] ?? 0,
       p95: span['p95(span.duration)'] ?? 0,
@@ -223,9 +224,9 @@ const BodyCell = memo(function BodyCell({
         yAxes: ['avg(span.duration)'],
       },
     ],
-    query: `gen_ai.request.model:${dataRow.model}`,
+    query: `${SpanFields.GEN_AI_RESPONSE_MODEL}:${dataRow.model}`,
     field: [
-      'gen_ai.request.model',
+      SpanFields.GEN_AI_RESPONSE_MODEL,
       'gen_ai.operation.name',
       'gen_ai.usage.input_tokens',
       'gen_ai.usage.output_tokens',
@@ -268,7 +269,7 @@ const BodyCell = memo(function BodyCell({
         <ErrorCell
           value={dataRow.errors}
           target={getExploreUrl({
-            query: `${query} span.status:internal_error gen_ai.request.model:"${dataRow.model}"`,
+            query: `${query} span.status:internal_error ${SpanFields.GEN_AI_RESPONSE_MODEL}:"${dataRow.model}"`,
             organization,
             selection,
             referrer: Referrer.MODELS_TABLE,

static/app/views/insights/pages/agents/components/tokenUsageWidget.tsx

@@ -42,7 +42,7 @@ export function TokenUsageWidget() {
 
   const tokensRequest = useSpans(
     {
-      fields: ['gen_ai.request.model', 'sum(gen_ai.usage.total_tokens)'],
+      fields: [SpanFields.GEN_AI_RESPONSE_MODEL, 'sum(gen_ai.usage.total_tokens)'],
       sorts: [{field: 'sum(gen_ai.usage.total_tokens)', kind: 'desc'}],
       search: fullQuery,
       limit: 3,
@@ -54,7 +54,7 @@ export function TokenUsageWidget() {
     {
       ...pageFilterChartParams,
       query: fullQuery,
-      groupBy: [SpanFields.GEN_AI_REQUEST_MODEL],
+      groupBy: [SpanFields.GEN_AI_RESPONSE_MODEL],
       yAxis: ['sum(gen_ai.usage.total_tokens)'],
       sort: {field: 'sum(gen_ai.usage.total_tokens)', kind: 'desc'},
       topEvents: 3,
@@ -112,7 +112,7 @@ export function TokenUsageWidget() {
   const footer = hasData && (
     <WidgetFooterTable>
       {tokens?.map((item, index) => {
-        const modelId = `${item['gen_ai.request.model']}`;
+        const modelId = `${item[SpanFields.GEN_AI_RESPONSE_MODEL]}`;
         return (
           <Fragment key={modelId}>
             <div>
@@ -152,7 +152,7 @@ export function TokenUsageWidget() {
                   yAxes: ['sum(gen_ai.usage.total_tokens)'],
                 },
               ],
-              groupBy: ['gen_ai.request.model'],
+              groupBy: [SpanFields.GEN_AI_RESPONSE_MODEL],
               query: fullQuery,
               sort: '-sum(gen_ai.usage.total_tokens)',
               interval: pageFilterChartParams.interval,