fix test?

getsentry · May 21, 2024 · f54236c · f54236c
1 parent 997e5a2
commit f54236c
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/relay-event-schema/src/protocol/span.rs b/relay-event-schema/src/protocol/span.rs
@@ -616,6 +616,9 @@ mod tests {
         insta::assert_debug_snapshot!(data, @r###"
         SpanData {
             app_start_type: ~,
+            ai_total_tokens_used: ~,
+            ai_prompt_tokens_used: ~,
+            ai_completion_tokens_used: ~,
             browser_name: ~,
             code_filepath: String(
                 "task.py",

diff --git a/relay-event-schema/src/protocol/span/convert.rs b/relay-event-schema/src/protocol/span/convert.rs
@@ -269,6 +269,9 @@ mod tests {
             ),
             data: SpanData {
                 app_start_type: ~,
+                ai_total_tokens_used: ~,
+                ai_prompt_tokens_used: ~,
+                ai_completion_tokens_used: ~,
                 browser_name: "Chrome",
                 code_filepath: ~,
                 code_lineno: ~,

diff --git a/...napshots/relay_server__metrics_extraction__event__tests__extract_span_metrics_mobile.snap b/...napshots/relay_server__metrics_extraction__event__tests__extract_span_metrics_mobile.snap
@@ -77,6 +77,9 @@ expression: "(&event.value().unwrap().spans, metrics)"
             profile_id: ~,
             data: SpanData {
                 app_start_type: ~,
+                ai_total_tokens_used: ~,
+                ai_prompt_tokens_used: ~,
+                ai_completion_tokens_used: ~,
                 browser_name: ~,
                 code_filepath: ~,
                 code_lineno: ~,
@@ -395,6 +398,9 @@ expression: "(&event.value().unwrap().spans, metrics)"
                 app_start_type: String(
                     "cold",
                 ),
+                ai_total_tokens_used: ~,
+                ai_prompt_tokens_used: ~,
+                ai_completion_tokens_used: ~,
                 browser_name: ~,
                 code_filepath: ~,
                 code_lineno: ~,
@@ -491,6 +497,9 @@ expression: "(&event.value().unwrap().spans, metrics)"
                 app_start_type: String(
                     "cold",
                 ),
+                ai_total_tokens_used: ~,
+                ai_prompt_tokens_used: ~,
+                ai_completion_tokens_used: ~,
                 browser_name: ~,
                 code_filepath: ~,
                 code_lineno: ~,
@@ -633,6 +642,9 @@ expression: "(&event.value().unwrap().spans, metrics)"
             profile_id: ~,
             data: SpanData {
                 app_start_type: ~,
+                ai_total_tokens_used: ~,
+                ai_prompt_tokens_used: ~,
+                ai_completion_tokens_used: ~,
                 browser_name: ~,
                 code_filepath: ~,
                 code_lineno: ~,
@@ -729,6 +741,9 @@ expression: "(&event.value().unwrap().spans, metrics)"
             profile_id: ~,
             data: SpanData {
                 app_start_type: ~,
+                ai_total_tokens_used: ~,
+                ai_prompt_tokens_used: ~,
+                ai_completion_tokens_used: ~,
                 browser_name: ~,
                 code_filepath: ~,
                 code_lineno: ~,