fix tests

Author: alexander-alderman-webb

tests/tracing/test_decorator.py

@@ -122,7 +122,10 @@ async def _some_function_traced(a, b, c):
 
 
 def test_span_templates_ai_dicts(sentry_init, capture_items):
-    sentry_init(traces_sample_rate=1.0)
+    sentry_init(
+        traces_sample_rate=1.0,
+        _experiments={"gen_ai_as_v2_spans": True},
+    )
     items = capture_items("span")
 
     @sentry_sdk.trace(template=SPANTEMPLATE.AI_TOOL)
@@ -241,7 +244,10 @@ def my_agent():
 
 
 def test_span_templates_ai_objects(sentry_init, capture_items):
-    sentry_init(traces_sample_rate=1.0)
+    sentry_init(
+        traces_sample_rate=1.0,
+        _experiments={"gen_ai_as_v2_spans": True},
+    )
     items = capture_items("span")
 
     @sentry_sdk.trace(template=SPANTEMPLATE.AI_TOOL)
@@ -365,7 +371,11 @@ def my_agent():
 
 @pytest.mark.parametrize("send_default_pii", [True, False])
 def test_span_templates_ai_pii(sentry_init, capture_items, send_default_pii):
-    sentry_init(traces_sample_rate=1.0, send_default_pii=send_default_pii)
+    sentry_init(
+        traces_sample_rate=1.0,
+        send_default_pii=send_default_pii,
+        _experiments={"gen_ai_as_v2_spans": True},
+    )
     items = capture_items("span")
 
     @sentry_sdk.trace(template=SPANTEMPLATE.AI_TOOL)

tests/tracing/test_misc.py

@@ -650,7 +650,10 @@ def test_conversation_id_propagates_to_span_with_gen_ai_op(
         self, sentry_init, capture_items
     ):
         """Span with gen_ai.* op should get conversation_id."""
-        sentry_init(traces_sample_rate=1.0)
+        sentry_init(
+            traces_sample_rate=1.0,
+            _experiments={"gen_ai_as_v2_spans": True},
+        )
         items = capture_items("span")
 
         scope = sentry_sdk.get_current_scope()