Fix AttributeError when image_url is string shorthand with base64 data

Handle both string shorthand and dict formats for image_url in
_is_image_type_with_blob_content and redact_blob_message_parts functions.
The OpenAI API supports both formats:
- {"type": "image_url", "image_url": "data:..."}
- {"type": "image_url", "image_url": {"url": "data:..."}}

Previously, calling .get("url", "") on a string raised AttributeError.

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>

Author: cursoragent

sentry_sdk/ai/utils.py

@@ -597,7 +597,14 @@ def _is_image_type_with_blob_content(item: "Dict[str, Any]") -> bool:
     if item.get("type") != "image_url":
         return False
 
-    image_url = item.get("image_url", {}).get("url", "")
+    image_url_data = item.get("image_url")
+    if isinstance(image_url_data, str):
+        image_url = image_url_data
+    elif isinstance(image_url_data, dict):
+        image_url = image_url_data.get("url", "")
+    else:
+        return False
+
     data_url_match = DATA_URL_BASE64_REGEX.match(image_url)
 
     return bool(data_url_match)
@@ -682,7 +689,11 @@ def redact_blob_message_parts(
                     if item.get("type") == "blob":
                         item["content"] = BLOB_DATA_SUBSTITUTE
                     elif _is_image_type_with_blob_content(item):
-                        item["image_url"]["url"] = BLOB_DATA_SUBSTITUTE
+                        image_url_data = item.get("image_url")
+                        if isinstance(image_url_data, str):
+                            item["image_url"] = BLOB_DATA_SUBSTITUTE
+                        elif isinstance(image_url_data, dict):
+                            item["image_url"]["url"] = BLOB_DATA_SUBSTITUTE
 
     return messages_copy
 

tests/test_ai_monitoring.py

@@ -845,6 +845,38 @@ def test_redacts_single_blob_within_image_url_content(self):
         assert result[0]["content"][1]["type"] == "image_url"
         assert result[0]["content"][1]["image_url"]["url"] == BLOB_DATA_SUBSTITUTE
 
+    def test_redacts_image_url_string_shorthand_with_blob(self):
+        """Test redacting image_url using string shorthand format with base64 data"""
+        messages = [
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "text": "How many ponies do you see in the image?",
+                        "type": "text",
+                    },
+                    {
+                        "type": "image_url",
+                        "image_url": "data:image/jpeg;base64,/9j/4AAQSkZJRg==",
+                    },
+                ],
+            }
+        ]
+
+        original_blob_content = messages[0]["content"][1]
+
+        result = redact_blob_message_parts(messages)
+
+        assert messages[0]["content"][1] == original_blob_content
+
+        assert (
+            result[0]["content"][0]["text"]
+            == "How many ponies do you see in the image?"
+        )
+        assert result[0]["content"][0]["type"] == "text"
+        assert result[0]["content"][1]["type"] == "image_url"
+        assert result[0]["content"][1]["image_url"] == BLOB_DATA_SUBSTITUTE
+
     def test_does_not_redact_image_url_content_with_non_blobs(self):
         messages = [
             {