Merge branch 'webb/anthropic/streaming-reasoning' of github.com:getsentry/sentry-python into webb/anthropic/streaming-reasoning

Author: alexander-alderman-webb

sentry_sdk/integrations/anthropic.py

@@ -57,6 +57,8 @@
         MessageDeltaEvent,
         MessageStartEvent,
         MessageStopEvent,
+        TextBlock,
+        ToolUseBlock,
     )
 
     if TYPE_CHECKING:
@@ -65,10 +67,10 @@
     raise DidNotEnable("Anthropic not installed")
 
 try:
-    from anthropic.types import ThinkingBlock
+    from anthropic.types import ThinkingBlock, ThinkingDelta
 except ImportError:
     ThinkingBlock = None  # type: ignore[assignment,misc]
-
+    ThinkingDelta = None  # type: ignore[assignment,misc]
 
 if TYPE_CHECKING:
     from typing import (
@@ -385,18 +387,16 @@ def _collect_ai_data(
                     event.content_block, ThinkingBlock
                 ):
                     content_blocks.append({"type": "thinking", "thinking": ""})
-                else:
+                elif isinstance(event.content_block, (TextBlock, ToolUseBlock)):
                     content_blocks.append({"type": "text", "text": ""})
             elif event.type == "content_block_delta" and content_blocks:
-                block = content_blocks[-1]
-                if block["type"] == "thinking":
-                    block["thinking"] += event.delta.thinking
-                else:
-                    chunk = getattr(event.delta, "text", None) or getattr(
-                        event.delta, "partial_json", None
-                    )
-                    if chunk:
-                        block["text"] += chunk
+                delta = event.delta
+                if ThinkingDelta is not None and isinstance(delta, ThinkingDelta):
+                    content_blocks[-1]["thinking"] += delta.thinking
+                elif hasattr(event.delta, "text"):
+                    content_blocks[-1]["text"] += event.delta.text
+                elif hasattr(event.delta, "partial_json"):
+                    content_blocks[-1]["text"] += event.delta.partial_json
             elif event.type == "content_block_stop":
                 pass
 
@@ -754,9 +754,7 @@ def _set_output_data(
             if output["type"] == "text":
                 parts.append({"type": "text", "content": output["text"]})
             elif output["type"] == "thinking":
-                parts.append(
-                    {"type": "reasoning", "content": output.get("thinking", "")}
-                )
+                parts.append({"type": "reasoning", "content": output.get("thinking")})
             elif output["type"] == "tool_use":
                 tool_calls.append(output)