Skip to content

Commit 19950e1

Browse files
authored
Make BinaryImage work with inline-snapshot (#3125)
1 parent 3909999 commit 19950e1

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

pydantic_ai_slim/pydantic_ai/messages.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -595,9 +595,13 @@ def __init__(
595595
media_type: str,
596596
identifier: str | None = None,
597597
vendor_metadata: dict[str, Any] | None = None,
598+
# Required for inline-snapshot which expects all dataclass `__init__` methods to take all field names as kwargs.
598599
kind: Literal['binary'] = 'binary',
600+
_identifier: str | None = None,
599601
):
600-
super().__init__(data=data, media_type=media_type, identifier=identifier, vendor_metadata=vendor_metadata)
602+
super().__init__(
603+
data=data, media_type=media_type, identifier=identifier or _identifier, vendor_metadata=vendor_metadata
604+
)
601605

602606
if not self.is_image:
603607
raise ValueError('`BinaryImage` must be have a media type that starts with "image/"') # pragma: no cover

0 commit comments

Comments
 (0)