chore(ai-monitoring): Add a unit to total cost (getsentry#70484)

uyefe · May 8, 2024 · 6b530ff · 6b530ff
1 parent 50b4ed3
commit 6b530ff
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 15 deletions.
diff --git a/static/app/views/aiMonitoring/PipelinesTable.tsx b/static/app/views/aiMonitoring/PipelinesTable.tsx
@@ -39,15 +39,15 @@ type Row = Pick<
   | 'avg(span.duration)'
   | 'sum(span.duration)'
   | 'ai_total_tokens_used()'
-  | 'ai_total_tokens_used(c:spans/ai.total_cost@none)'
+  | 'ai_total_tokens_used(c:spans/ai.total_cost@usd)'
 >;
 
 type Column = GridColumnHeader<
   | 'span.description'
   | 'spm()'
   | 'avg(span.duration)'
   | 'ai_total_tokens_used()'
-  | 'ai_total_tokens_used(c:spans/ai.total_cost@none)'
+  | 'ai_total_tokens_used(c:spans/ai.total_cost@usd)'
 >;
 
 const COLUMN_ORDER: Column[] = [
@@ -62,7 +62,7 @@ const COLUMN_ORDER: Column[] = [
     width: 180,
   },
   {
-    key: 'ai_total_tokens_used(c:spans/ai.total_cost@none)',
+    key: 'ai_total_tokens_used(c:spans/ai.total_cost@usd)',
     name: t('Total cost'),
     width: 180,
   },
@@ -130,24 +130,24 @@ export function PipelinesTable() {
     fields: [
       'span.ai.pipeline.group',
       'ai_total_tokens_used()',
-      'ai_total_tokens_used(c:spans/ai.total_cost@none)',
+      'ai_total_tokens_used(c:spans/ai.total_cost@usd)',
     ],
   });
 
   const rows: Row[] = (data as Row[]).map(baseRow => {
     const row: Row = {
       ...baseRow,
       'ai_total_tokens_used()': 0,
-      'ai_total_tokens_used(c:spans/ai.total_cost@none)': 0,
+      'ai_total_tokens_used(c:spans/ai.total_cost@usd)': 0,
     };
     if (!tokensUsedLoading) {
       const tokenUsedDataPoint = tokensUsedData.find(
         tokenRow => tokenRow['span.ai.pipeline.group'] === row['span.group']
       );
       if (tokenUsedDataPoint) {
         row['ai_total_tokens_used()'] = tokenUsedDataPoint['ai_total_tokens_used()'];
-        row['ai_total_tokens_used(c:spans/ai.total_cost@none)'] =
-          tokenUsedDataPoint['ai_total_tokens_used(c:spans/ai.total_cost@none)'];
+        row['ai_total_tokens_used(c:spans/ai.total_cost@usd)'] =
+          tokenUsedDataPoint['ai_total_tokens_used(c:spans/ai.total_cost@usd)'];
       }
     }
     return row;
@@ -237,8 +237,8 @@ function renderBodyCell(
       </Link>
     );
   }
-  if (column.key === 'ai_total_tokens_used(c:spans/ai.total_cost@none)') {
-    const cost = row['ai_total_tokens_used(c:spans/ai.total_cost@none)'];
+  if (column.key === 'ai_total_tokens_used(c:spans/ai.total_cost@usd)') {
+    const cost = row['ai_total_tokens_used(c:spans/ai.total_cost@usd)'];
     if (cost) {
       if (cost < 0.01) {
         return <span>US {cost * 100}¢</span>;

diff --git a/static/app/views/aiMonitoring/aiMonitoringDetailsPage.tsx b/static/app/views/aiMonitoring/aiMonitoringDetailsPage.tsx
@@ -73,10 +73,7 @@ export default function AiMonitoringPage({params}: Props) {
       'span.category': 'ai',
       'span.ai.pipeline.group': groupId,
     }),
-    fields: [
-      'ai_total_tokens_used()',
-      'ai_total_tokens_used(c:spans/ai.total_cost@none)',
-    ],
+    fields: ['ai_total_tokens_used()', 'ai_total_tokens_used(c:spans/ai.total_cost@usd)'],
     enabled: Boolean(groupId),
     referrer: 'api.ai-pipelines.view',
   });
@@ -137,7 +134,7 @@ export default function AiMonitoringPage({params}: Props) {
                             title={t('Total Cost')}
                             value={
                               tokenUsedMetric[
-                                'ai_total_tokens_used(c:spans/ai.total_cost@none)'
+                                'ai_total_tokens_used(c:spans/ai.total_cost@usd)'
                               ]
                             }
                             unit={CurrencyUnit.USD}

diff --git a/static/app/views/starfish/types.tsx b/static/app/views/starfish/types.tsx
@@ -132,7 +132,7 @@ export type SpanMetricsResponse = {
   'http_response_rate(4)': number;
   'http_response_rate(5)': number;
 } & {
-  'ai_total_tokens_used(c:spans/ai.total_cost@none)': number;
+  'ai_total_tokens_used(c:spans/ai.total_cost@usd)': number;
 } & {
   ['project']: string;
   ['project.id']: number;