.

Author: alexander-alderman-webb

sentry_sdk/integrations/anthropic.py

@@ -75,127 +75,6 @@ def setup_once() -> None:
         Messages.create = _wrap_message_create(Messages.create)
         AsyncMessages.create = _wrap_message_create_async(AsyncMessages.create)
 
-        Stream.__iter__ = _wrap_stream_iter(Stream.__iter__)
-        AsyncStream.__aiter__ = _wrap_async_stream_aiter(AsyncStream.__aiter__)
-
-
-def _wrap_stream_iter(
-    f: "Callable[..., Iterator[RawMessageStreamEvent]]",
-) -> "Callable[..., Iterator[RawMessageStreamEvent]]":
-    """
-    Sets information received while iterating the response stream on the AI Client Span.
-    Responsible for closing the AI Client Span.
-    """
-
-    @wraps(f)
-    def _patched_iter(self: "Stream") -> "Iterator[RawMessageStreamEvent]":
-        if not hasattr(self, "_sentry_span"):
-            for event in f(self):
-                yield event
-            return
-
-        model = None
-        usage = _RecordedUsage()
-        content_blocks: "list[str]" = []
-
-        for event in f(self):
-            (
-                model,
-                usage,
-                content_blocks,
-            ) = _collect_ai_data(
-                event,
-                model,
-                usage,
-                content_blocks,
-            )
-            yield event
-
-        # Anthropic's input_tokens excludes cached/cache_write tokens.
-        # Normalize to total input tokens for correct cost calculations.
-        total_input = (
-            usage.input_tokens
-            + (usage.cache_read_input_tokens or 0)
-            + (usage.cache_write_input_tokens or 0)
-        )
-
-        span = self._sentry_span
-        integration = self._integration
-
-        _set_output_data(
-            span=span,
-            integration=integration,
-            model=model,
-            input_tokens=total_input,
-            output_tokens=usage.output_tokens,
-            cache_read_input_tokens=usage.cache_read_input_tokens,
-            cache_write_input_tokens=usage.cache_write_input_tokens,
-            content_blocks=[{"text": "".join(content_blocks), "type": "text"}],
-            finish_span=True,
-        )
-
-    return _patched_iter
-
-
-def _wrap_async_stream_aiter(
-    f: "Callable[..., AsyncIterator[RawMessageStreamEvent]]",
-) -> "Callable[..., AsyncIterator[RawMessageStreamEvent]]":
-    """
-    Sets information received while iterating the response stream on the AI Client Span.
-    Responsible for closing the AI Client Span.
-    """
-
-    @wraps(f)
-    async def _patched_aiter(
-        self: "AsyncStream",
-    ) -> "AsyncIterator[RawMessageStreamEvent]":
-        if not hasattr(self, "_sentry_span"):
-            async for event in f(self):
-                yield event
-            return
-
-        model = None
-        usage = _RecordedUsage()
-        content_blocks: "list[str]" = []
-
-        async for event in f(self):
-            (
-                model,
-                usage,
-                content_blocks,
-            ) = _collect_ai_data(
-                event,
-                model,
-                usage,
-                content_blocks,
-            )
-            yield event
-
-        # Anthropic's input_tokens excludes cached/cache_write tokens.
-        # Normalize to total input tokens for correct cost calculations.
-        total_input = (
-            usage.input_tokens
-            + (usage.cache_read_input_tokens or 0)
-            + (usage.cache_write_input_tokens or 0)
-        )
-
-        span = self._sentry_span
-        integration = self._integration
-
-        _set_output_data(
-            span=span,
-            integration=integration,
-            model=model,
-            input_tokens=total_input,
-            output_tokens=usage.output_tokens,
-            cache_read_input_tokens=usage.cache_read_input_tokens,
-            cache_write_input_tokens=usage.cache_write_input_tokens,
-            content_blocks=[{"text": "".join(content_blocks), "type": "text"}],
-            finish_span=True,
-        )
-
-    return _patched_aiter
-
 
 def _capture_exception(exc: "Any") -> None:
     set_span_errored()