.

Author: alexander-alderman-webb

sentry_sdk/integrations/anthropic.py

@@ -395,12 +395,11 @@ def _set_output_data(
 def _patch_streaming_response_iterator(
     result: "AsyncStream[RawMessageStreamEvent]",
     span: "sentry_sdk.tracing.Span",
-):
+    integration: "AnthropicIntegration",
+) -> None:
     """
     Responsible for closing the `gen_ai.chat` span and setting attributes acquired during response consumption.
     """
-    integration = sentry_sdk.get_client().get_integration(AnthropicIntegration)
-
     old_iterator = result._iterator
 
     def new_iterator() -> "Iterator[MessageStreamEvent]":
@@ -513,7 +512,7 @@ def _sentry_patched_create_common(f: "Any", *args: "Any", **kwargs: "Any") -> "A
 
     is_streaming_response = kwargs.get("stream", False)
     if is_streaming_response:
-        _patch_streaming_response_iterator(result, span)
+        _patch_streaming_response_iterator(result, span, integration)
         return result
 
     with capture_internal_exceptions():