add provider_details field to parts

Author: ajac-zero

pydantic_ai_slim/pydantic_ai/messages.py

@@ -1003,6 +1003,9 @@ class TextPart:
     part_kind: Literal['text'] = 'text'
     """Part type identifier, this is available on all parts as a discriminator."""
 
+    provider_details: dict[str, Any] | None = None
+    """Additional provider-specific details in a serializable format."""
+
     def has_content(self) -> bool:
         """Return `True` if the text content is non-empty."""
         return bool(self.content)
@@ -1042,6 +1045,9 @@ class ThinkingPart:
     part_kind: Literal['thinking'] = 'thinking'
     """Part type identifier, this is available on all parts as a discriminator."""
 
+    provider_details: dict[str, Any] | None = None
+    """Additional provider-specific details in a serializable format."""
+
     def has_content(self) -> bool:
         """Return `True` if the thinking content is non-empty."""
         return bool(self.content)
@@ -1068,6 +1074,9 @@ class FilePart:
     part_kind: Literal['file'] = 'file'
     """Part type identifier, this is available on all parts as a discriminator."""
 
+    provider_details: dict[str, Any] | None = None
+    """Additional provider-specific details in a serializable format."""
+
     def has_content(self) -> bool:
         """Return `True` if the file content is non-empty."""
         return bool(self.content.data)
@@ -1101,6 +1110,9 @@ class BaseToolCallPart:
 
     This is used by some APIs like OpenAI Responses."""
 
+    provider_details: dict[str, Any] | None = None
+    """Additional provider-specific details in a serializable format."""
+
     def args_as_dict(self) -> dict[str, Any]:
         """Return the arguments as a Python dictionary.
 
@@ -1436,6 +1448,9 @@ class TextPartDelta:
     part_delta_kind: Literal['text'] = 'text'
     """Part delta type identifier, used as a discriminator."""
 
+    provider_details: dict[str, Any] | None = None
+    """Additional provider-specific details in a serializable format."""
+
     def apply(self, part: ModelResponsePart) -> TextPart:
         """Apply this text delta to an existing `TextPart`.
 
@@ -1477,6 +1492,9 @@ class ThinkingPartDelta:
     part_delta_kind: Literal['thinking'] = 'thinking'
     """Part delta type identifier, used as a discriminator."""
 
+    provider_details: dict[str, Any] | None = None
+    """Additional provider-specific details in a serializable format."""
+
     @overload
     def apply(self, part: ModelResponsePart) -> ThinkingPart: ...
 
@@ -1540,6 +1558,9 @@ class ToolCallPartDelta:
     part_delta_kind: Literal['tool_call'] = 'tool_call'
     """Part delta type identifier, used as a discriminator."""
 
+    provider_details: dict[str, Any] | None = None
+    """Additional provider-specific details in a serializable format."""
+
     def as_part(self) -> ToolCallPart | None:
         """Convert this delta to a fully formed `ToolCallPart` if possible, otherwise return `None`.
 

tests/test_agent.py

@@ -3733,7 +3733,9 @@ def test_binary_content_serializable():
                 'metadata': None,
             },
             {
-                'parts': [{'content': 'success (no tool calls)', 'id': None, 'part_kind': 'text'}],
+                'parts': [
+                    {'content': 'success (no tool calls)', 'id': None, 'part_kind': 'text', 'provider_details': None}
+                ],
                 'usage': {
                     'input_tokens': 56,
                     'cache_write_tokens': 0,
@@ -3793,7 +3795,9 @@ def test_image_url_serializable_missing_media_type():
                 'metadata': None,
             },
             {
-                'parts': [{'content': 'success (no tool calls)', 'id': None, 'part_kind': 'text'}],
+                'parts': [
+                    {'content': 'success (no tool calls)', 'id': None, 'part_kind': 'text', 'provider_details': None}
+                ],
                 'usage': {
                     'input_tokens': 51,
                     'cache_write_tokens': 0,
@@ -3860,7 +3864,9 @@ def test_image_url_serializable():
                 'metadata': None,
             },
             {
-                'parts': [{'content': 'success (no tool calls)', 'id': None, 'part_kind': 'text'}],
+                'parts': [
+                    {'content': 'success (no tool calls)', 'id': None, 'part_kind': 'text', 'provider_details': None}
+                ],
                 'usage': {
                     'input_tokens': 51,
                     'cache_write_tokens': 0,