@@ -89,7 +89,7 @@
     from sentry_sdk._types import TextPart
 
     class _PatchedRawMessageStream(Stream[RawMessageStreamEvent]):
-        _span: Span
+        _span: Union[Span, StreamedSpan]
         _integration: "AnthropicIntegration"
 
         _model: Optional[ModelParam]
@@ -99,7 +99,7 @@ class _PatchedRawMessageStream(Stream[RawMessageStreamEvent]):
         _finish_reason: Optional[str]
 
     class _PatchedMessageStream(MessageStream):
-        _span: Span
+        _span: Union[Span, StreamedSpan]
         _integration: "AnthropicIntegration"
 
         _model: Optional[ModelParam]
@@ -109,7 +109,7 @@ class _PatchedMessageStream(MessageStream):
         _finish_reason: Optional[str]
 
     class _PatchedRawAsyncMessageStream(AsyncStream[RawMessageStreamEvent]):
-        _span: Span
+        _span: Union[Span, StreamedSpan]
         _integration: "AnthropicIntegration"
 
         _model: Optional[ModelParam]
@@ -119,7 +119,7 @@ class _PatchedRawAsyncMessageStream(AsyncStream[RawMessageStreamEvent]):
         _finish_reason: Optional[str]
 
     class _PatchedAsyncMessageStream(AsyncMessageStream):
-        _span: Span
+        _span: Union[Span, StreamedSpan]
         _integration: "AnthropicIntegration"
 
         _model: Optional[ModelParam]
@@ -966,7 +966,7 @@ def _accumulate_event_data(
 
 
 def _set_streaming_output_data(
-    span: "Span",
+    span: "Union[Span, StreamedSpan]",
     integration: "AnthropicIntegration",
     model: "Optional[str]",
     usage: "_RecordedUsage",