Making sure responses from tool use work the same in streaming and normal messages

Author: antonpirker

sentry_sdk/integrations/anthropic.py

@@ -76,6 +76,21 @@ def _calculate_token_usage(result, span):
     record_token_usage(span, input_tokens, output_tokens, total_tokens)
 
 
+def _get_responses(content):
+    # type: (list[dict[str, str]]) -> dict[str, str]
+    """Get JSON of a Anthropic responses."""
+    responses = []
+    for item in content:
+        if hasattr(item, "text"):
+            responses.append(
+                {
+                    "type": item.type,
+                    "text": item.text,
+                }
+            )
+    return responses
+
+
 def _wrap_message_create(f):
     # type: (Any) -> Any
     @wraps(f)
@@ -115,18 +130,7 @@ def _sentry_patched_create(*args, **kwargs):
                 span.set_data(SPANDATA.AI_INPUT_MESSAGES, messages)
             if hasattr(result, "content"):
                 if should_send_default_pii() and integration.include_prompts:
-                    span.set_data(
-                        SPANDATA.AI_RESPONSES,
-                        list(
-                            map(
-                                lambda message: {
-                                    "type": message.type,
-                                    "text": message.text,
-                                },
-                                result.content,
-                            )
-                        ),
-                    )
+                    span.set_data(SPANDATA.AI_RESPONSES, _get_responses(result.content))
                 _calculate_token_usage(result, span)
                 span.__exit__(None, None, None)
             elif hasattr(result, "_iterator"):
@@ -149,8 +153,6 @@ def new_iterator():
                                 elif event.type == "content_block_delta":
                                     if hasattr(event.delta, "text"):
                                         content_blocks.append(event.delta.text)
-                                    elif hasattr(event.delta, "partial_json"):
-                                        content_blocks.append(event.delta.partial_json)
                                 elif event.type == "content_block_stop":
                                     pass
                                 elif event.type == "message_delta":