limit span.__enter__ to legacy branches

Author: alexander-alderman-webb

sentry_sdk/integrations/anthropic.py

@@ -654,7 +654,7 @@ def _sentry_patched_create_sync(f: "Any", *args: "Any", **kwargs: "Any") -> "Any
             origin=AnthropicIntegration.origin,
         )
 
-    span.__enter__()
+        span.__enter__()
 
     _set_create_input_data(span, kwargs, integration)
 
@@ -751,7 +751,7 @@ async def _sentry_patched_create_async(
             origin=AnthropicIntegration.origin,
         )
 
-    span.__enter__()
+        span.__enter__()
 
     _set_create_input_data(span, kwargs, integration)
 
@@ -985,21 +985,19 @@ def _sentry_patched_enter(self: "MessageStreamManager") -> "MessageStream":
                 attributes={
                     "sentry.op": OP.GEN_AI_CHAT,
                     "sentry.origin": AnthropicIntegration.origin,
+                    SPANDATA.GEN_AI_RESPONSE_STREAMING: True,
                 },
             )
-
-            set_on_span = span.set_attribute
         else:
             span = get_start_span_function()(
                 op=OP.GEN_AI_CHAT,
                 name="chat" if self._model is None else f"chat {self._model}".strip(),
                 origin=AnthropicIntegration.origin,
             )
 
-            set_on_span = span.set_data
-        span.__enter__()
+            span.set_data(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)
+            span.__enter__()
 
-        set_on_span(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)
         _set_common_input_data(
             span=span,
             integration=integration,
@@ -1094,21 +1092,19 @@ async def _sentry_patched_aenter(
                 attributes={
                     "sentry.op": OP.GEN_AI_CHAT,
                     "sentry.origin": AnthropicIntegration.origin,
+                    SPANDATA.GEN_AI_RESPONSE_STREAMING: True,
                 },
             )
-
-            set_on_span = span.set_attribute
         else:
             span = get_start_span_function()(
                 op=OP.GEN_AI_CHAT,
                 name="chat" if self._model is None else f"chat {self._model}".strip(),
                 origin=AnthropicIntegration.origin,
             )
 
-            set_on_span = span.set_data
-        span.__enter__()
+            span.set_data(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)
+            span.__enter__()
 
-        set_on_span(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)
         _set_common_input_data(
             span=span,
             integration=integration,