test(pydantic-ai): Add tests for data URLs with optional parameters

ericapisani · claude · ericapisani · commit eb4393dd25c7 · 2026-03-10T14:08:52.000-04:00
Cover the case where data URLs include optional parameters between the
MIME type and base64 encoding, e.g. `data:image/png;name=file.png;base64,...`
and `data:text/plain;charset=utf-8;name=hello.txt;base64,...`. These should
be matched and redacted by DATA_URL_BASE64_REGEX.

Co-Authored-By: Claude &lt;noreply@anthropic.com&gt;
diff --git a/sentry_sdk/integrations/pydantic_ai/consts.py b/sentry_sdk/integrations/pydantic_ai/consts.py
@@ -5,5 +5,5 @@
 # Matches data URLs with base64-encoded content, e.g. "data:image/png;base64,iVBORw0K..."
 # Group 1: MIME type (e.g. "image/png"), Group 2: base64 data
 DATA_URL_BASE64_REGEX = re.compile(
-    r"^data:([a-zA-Z0-9][a-zA-Z0-9.+\-]*/[a-zA-Z0-9][a-zA-Z0-9.+\-]*);base64,([A-Za-z0-9+/\-_]+={0,2})$"
+    r"^data:([a-zA-Z0-9][a-zA-Z0-9.+\-]*/[a-zA-Z0-9][a-zA-Z0-9.+\-]*)(?:;[a-zA-Z0-9\-]+=[^;,]*)*;base64,([A-Za-z0-9+/\-_]+={0,2})$"
 )
diff --git a/tests/integrations/pydantic_ai/test_pydantic_ai.py b/tests/integrations/pydantic_ai/test_pydantic_ai.py
@@ -2959,7 +2959,6 @@ def test_image_url_http_url_with_base64_data_in_query_param_is_not_redacted_no_m
 def test_image_url_redacts_base64_data_url_with_complex_mime_type(
     sentry_init, capture_events
 ):
-    """Test that ImageUrl with a data URL using a complex MIME type (e.g. image/svg+xml) is redacted."""
     sentry_init(
         integrations=[PydanticAIIntegration()],
         traces_sample_rate=1.0,
@@ -2987,6 +2986,68 @@ def test_image_url_redacts_base64_data_url_with_complex_mime_type(
     assert _find_image_content(messages_data, "image/svg+xml")
 
 
+def test_image_url_redacts_base64_data_url_with_optional_parameters(
+    sentry_init, capture_events
+):
+    sentry_init(
+        integrations=[PydanticAIIntegration()],
+        traces_sample_rate=1.0,
+        send_default_pii=True,
+    )
+
+    events = capture_events()
+
+    with sentry_sdk.start_transaction(op="test", name="test"):
+        span = sentry_sdk.start_span(op="test_span")
+        # Data URL with a charset parameter before the base64 marker
+        image_url = ImageUrl(
+            url="data:image/png;name=file.png;base64,iVBORw0KGgoAAAANSUhEUgAAAAoAAAAKCAYAAACNMs"
+        )
+        user_part = UserPromptPart(content=["Look at this image:", image_url])
+        mock_msg = MagicMock()
+        mock_msg.parts = [user_part]
+        mock_msg.instructions = None
+
+        _set_input_messages(span, [mock_msg])
+        span.finish()
+
+    (event,) = events
+    span_data = event["spans"][0]["data"]
+    messages_data = _get_messages_from_span(span_data)
+    assert _find_image_content(messages_data, "image/png")
+
+
+def test_image_url_redacts_base64_data_url_with_multiple_optional_parameters(
+    sentry_init, capture_events
+):
+    sentry_init(
+        integrations=[PydanticAIIntegration()],
+        traces_sample_rate=1.0,
+        send_default_pii=True,
+    )
+
+    events = capture_events()
+
+    with sentry_sdk.start_transaction(op="test", name="test"):
+        span = sentry_sdk.start_span(op="test_span")
+        # Data URL with multiple parameters before the base64 marker
+        image_url = ImageUrl(
+            url="data:text/plain;charset=utf-8;name=hello.txt;base64,SGVsbG8sIFdvcmxkIQ=="
+        )
+        user_part = UserPromptPart(content=["Look at this text:", image_url])
+        mock_msg = MagicMock()
+        mock_msg.parts = [user_part]
+        mock_msg.instructions = None
+
+        _set_input_messages(span, [mock_msg])
+        span.finish()
+
+    (event,) = events
+    span_data = event["spans"][0]["data"]
+    messages_data = _get_messages_from_span(span_data)
+    assert _find_image_content(messages_data, "text/plain")
+
+
 @pytest.mark.asyncio
 async def test_invoke_agent_image_url_http_url_no_redaction(
     sentry_init, capture_events

Original file line number	Diff line number	Diff line change
`@@ -5,5 +5,5 @@`
`5`	`5`	`# Matches data URLs with base64-encoded content, e.g. "data:image/png;base64,iVBORw0K..."`
`6`	`6`	`# Group 1: MIME type (e.g. "image/png"), Group 2: base64 data`
`7`	`7`	`DATA_URL_BASE64_REGEX = re.compile(`
`8`		`- r"^data:([a-zA-Z0-9][a-zA-Z0-9.+\-]/[a-zA-Z0-9][a-zA-Z0-9.+\-]);base64,([A-Za-z0-9+/\-_]+={0,2})$"`
	`8`	`+ r"^data:([a-zA-Z0-9][a-zA-Z0-9.+\-]/[a-zA-Z0-9][a-zA-Z0-9.+\-])(?:;[a-zA-Z0-9\-]+=[^;,]);base64,([A-Za-z0-9+/\-_]+={0,2})$"`
`9`	`9`	`)`