pydantic-ai-slim 0.2.6__py3-none-any.whl → 0.2.8__py3-none-any.whl

pydantic_ai/messages.py

@@ -13,6 +13,7 @@ import pydantic_core
 from opentelemetry._events import Event  # pyright: ignore[reportPrivateImportUsage]
 from typing_extensions import TypeAlias
 
+from . import _utils
 from ._utils import generate_tool_call_id as _generate_tool_call_id, now_utc as _now_utc
 from .exceptions import UnexpectedModelBehavior
 from .usage import Usage
@@ -50,7 +51,7 @@ DocumentFormat: TypeAlias = Literal['csv', 'doc', 'docx', 'html', 'md', 'pdf', '
 VideoFormat: TypeAlias = Literal['mkv', 'mov', 'mp4', 'webm', 'flv', 'mpeg', 'mpg', 'wmv', 'three_gp']
 
 
-@dataclass
+@dataclass(repr=False)
 class SystemPromptPart:
     """A system prompt, generally written by the application developer.
 
@@ -75,8 +76,10 @@ class SystemPromptPart:
     def otel_event(self, _settings: InstrumentationSettings) -> Event:
         return Event('gen_ai.system.message', body={'content': self.content, 'role': 'system'})
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
 
-@dataclass
+
+@dataclass(repr=False)
 class VideoUrl:
     """A URL to an video."""
 
@@ -116,8 +119,10 @@ class VideoUrl:
         """
         return _video_format_lookup[self.media_type]
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
-@dataclass
+@dataclass(repr=False)
 class AudioUrl:
     """A URL to an audio file."""
 
@@ -142,8 +147,10 @@ class AudioUrl:
         """The file format of the audio file."""
         return _audio_format_lookup[self.media_type]
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
-@dataclass
+@dataclass(repr=False)
 class ImageUrl:
     """A URL to an image."""
 
@@ -175,8 +182,10 @@ class ImageUrl:
         """
         return _image_format_lookup[self.media_type]
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
 
-@dataclass
+
+@dataclass(repr=False)
 class DocumentUrl:
     """The URL of the document."""
 
@@ -206,8 +215,10 @@ class DocumentUrl:
         except KeyError as e:
             raise ValueError(f'Unknown document media type: {media_type}') from e
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
-@dataclass
+@dataclass(repr=False)
 class BinaryContent:
     """Binary content, e.g. an audio or image file."""
 
@@ -255,6 +266,8 @@ class BinaryContent:
         except KeyError as e:
             raise ValueError(f'Unknown media type: {self.media_type}') from e
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
 UserContent: TypeAlias = 'str | ImageUrl | AudioUrl | DocumentUrl | VideoUrl | BinaryContent'
 
@@ -292,7 +305,7 @@ _video_format_lookup: dict[str, VideoFormat] = {
 }
 
 
-@dataclass
+@dataclass(repr=False)
 class UserPromptPart:
     """A user prompt, generally written by the end user.
 
@@ -329,11 +342,13 @@ class UserPromptPart:
                     content.append({'kind': part.kind})  # pragma: no cover
         return Event('gen_ai.user.message', body={'content': content, 'role': 'user'})
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
 tool_return_ta: pydantic.TypeAdapter[Any] = pydantic.TypeAdapter(Any, config=pydantic.ConfigDict(defer_build=True))
 
 
-@dataclass
+@dataclass(repr=False)
 class ToolReturnPart:
     """A tool return message, this encodes the result of running a tool."""
 
@@ -373,11 +388,13 @@ class ToolReturnPart:
             body={'content': self.content, 'role': 'tool', 'id': self.tool_call_id, 'name': self.tool_name},
         )
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
 error_details_ta = pydantic.TypeAdapter(list[pydantic_core.ErrorDetails], config=pydantic.ConfigDict(defer_build=True))
 
 
-@dataclass
+@dataclass(repr=False)
 class RetryPromptPart:
     """A message back to a model asking it to try again.
 
@@ -438,6 +455,8 @@ class RetryPromptPart:
                 },
             )
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
 ModelRequestPart = Annotated[
     Union[SystemPromptPart, UserPromptPart, ToolReturnPart, RetryPromptPart], pydantic.Discriminator('part_kind')
@@ -445,7 +464,7 @@ ModelRequestPart = Annotated[
 """A message part sent by PydanticAI to a model."""
 
 
-@dataclass
+@dataclass(repr=False)
 class ModelRequest:
     """A request generated by PydanticAI and sent to a model, e.g. a message from the PydanticAI app to the model."""
 
@@ -463,8 +482,10 @@ class ModelRequest:
         """Create a `ModelRequest` with a single user prompt as text."""
         return cls(parts=[UserPromptPart(user_prompt)], instructions=instructions)
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
 
-@dataclass
+
+@dataclass(repr=False)
 class TextPart:
     """A plain text response from a model."""
 
@@ -478,15 +499,17 @@ class TextPart:
         """Return `True` if the text content is non-empty."""
         return bool(self.content)
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
-@dataclass
+@dataclass(repr=False)
 class ToolCallPart:
     """A tool call from a model."""
 
     tool_name: str
     """The name of the tool to call."""
 
-    args: str | dict[str, Any]
+    args: str | dict[str, Any] | None = None
     """The arguments to pass to the tool.
 
     This is stored either as a JSON string or a Python dictionary depending on how data was received.
@@ -506,10 +529,10 @@ class ToolCallPart:
 
         This is just for convenience with models that require dicts as input.
         """
+        if not self.args:
+            return {}
         if isinstance(self.args, dict):
             return self.args
-        if isinstance(self.args, str) and not self.args:
-            return {}
         args = pydantic_core.from_json(self.args)
         assert isinstance(args, dict), 'args should be a dict'
         return cast(dict[str, Any], args)
@@ -519,6 +542,8 @@ class ToolCallPart:
 
         This is just for convenience with models that require JSON strings as input.
         """
+        if not self.args:
+            return '{}'
         if isinstance(self.args, str):
             return self.args
         return pydantic_core.to_json(self.args).decode()
@@ -532,12 +557,14 @@ class ToolCallPart:
         else:
             return bool(self.args)
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
 ModelResponsePart = Annotated[Union[TextPart, ToolCallPart], pydantic.Discriminator('part_kind')]
 """A message part returned by a model."""
 
 
-@dataclass
+@dataclass(repr=False)
 class ModelResponse:
     """A response from a model, e.g. a message from the model to the PydanticAI app."""
 
@@ -602,6 +629,8 @@ class ModelResponse:
 
         return result
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
 ModelMessage = Annotated[Union[ModelRequest, ModelResponse], pydantic.Discriminator('kind')]
 """Any message sent to or returned by a model."""
@@ -612,7 +641,7 @@ ModelMessagesTypeAdapter = pydantic.TypeAdapter(
 """Pydantic [`TypeAdapter`][pydantic.type_adapter.TypeAdapter] for (de)serializing messages."""
 
 
-@dataclass
+@dataclass(repr=False)
 class TextPartDelta:
     """A partial update (delta) for a `TextPart` to append new text content."""
 
@@ -638,8 +667,10 @@ class TextPartDelta:
             raise ValueError('Cannot apply TextPartDeltas to non-TextParts')  # pragma: no cover
         return replace(part, content=part.content + self.content_delta)
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
-@dataclass
+@dataclass(repr=False)
 class ToolCallPartDelta:
     """A partial update (delta) for a `ToolCallPart` to modify tool name, arguments, or tool call ID."""
 
@@ -666,9 +697,9 @@ class ToolCallPartDelta:
         """Convert this delta to a fully formed `ToolCallPart` if possible, otherwise return `None`.
 
         Returns:
-            A `ToolCallPart` if both `tool_name_delta` and `args_delta` are set, otherwise `None`.
+            A `ToolCallPart` if `tool_name_delta` is set, otherwise `None`.
         """
-        if self.tool_name_delta is None or self.args_delta is None:
+        if self.tool_name_delta is None:
             return None
 
         return ToolCallPart(self.tool_name_delta, self.args_delta, self.tool_call_id or _generate_tool_call_id())
@@ -728,7 +759,7 @@ class ToolCallPartDelta:
             delta = replace(delta, tool_call_id=self.tool_call_id)
 
         # If we now have enough data to create a full ToolCallPart, do so
-        if delta.tool_name_delta is not None and delta.args_delta is not None:
+        if delta.tool_name_delta is not None:
             return ToolCallPart(delta.tool_name_delta, delta.args_delta, delta.tool_call_id or _generate_tool_call_id())
 
         return delta
@@ -741,12 +772,12 @@ class ToolCallPartDelta:
             part = replace(part, tool_name=tool_name)
 
         if isinstance(self.args_delta, str):
-            if not isinstance(part.args, str):
+            if isinstance(part.args, dict):
                 raise UnexpectedModelBehavior(f'Cannot apply JSON deltas to non-JSON tool arguments ({part=}, {self=})')
-            updated_json = part.args + self.args_delta
+            updated_json = (part.args or '') + self.args_delta
             part = replace(part, args=updated_json)
         elif isinstance(self.args_delta, dict):
-            if not isinstance(part.args, dict):
+            if isinstance(part.args, str):
                 raise UnexpectedModelBehavior(f'Cannot apply dict deltas to non-dict tool arguments ({part=}, {self=})')
             updated_dict = {**(part.args or {}), **self.args_delta}
             part = replace(part, args=updated_dict)
@@ -755,12 +786,14 @@ class ToolCallPartDelta:
             part = replace(part, tool_call_id=self.tool_call_id)
         return part
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
 ModelResponsePartDelta = Annotated[Union[TextPartDelta, ToolCallPartDelta], pydantic.Discriminator('part_delta_kind')]
 """A partial update (delta) for any model response part."""
 
 
-@dataclass
+@dataclass(repr=False)
 class PartStartEvent:
     """An event indicating that a new part has started.
 
@@ -777,8 +810,10 @@ class PartStartEvent:
     event_kind: Literal['part_start'] = 'part_start'
     """Event type identifier, used as a discriminator."""
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
 
-@dataclass
+
+@dataclass(repr=False)
 class PartDeltaEvent:
     """An event indicating a delta update for an existing part."""
 
@@ -791,8 +826,10 @@ class PartDeltaEvent:
     event_kind: Literal['part_delta'] = 'part_delta'
     """Event type identifier, used as a discriminator."""
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
-@dataclass
+@dataclass(repr=False)
 class FinalResultEvent:
     """An event indicating the response to the current model request matches the output schema and will produce a result."""
 
@@ -803,6 +840,8 @@ class FinalResultEvent:
     event_kind: Literal['final_result'] = 'final_result'
     """Event type identifier, used as a discriminator."""
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
 ModelResponseStreamEvent = Annotated[Union[PartStartEvent, PartDeltaEvent], pydantic.Discriminator('event_kind')]
 """An event in the model response stream, either starting a new part or applying a delta to an existing one."""
@@ -813,7 +852,7 @@ AgentStreamEvent = Annotated[
 """An event in the agent stream."""
 
 
-@dataclass
+@dataclass(repr=False)
 class FunctionToolCallEvent:
     """An event indicating the start to a call to a function tool."""
 
@@ -827,8 +866,10 @@ class FunctionToolCallEvent:
     def __post_init__(self):
         self.call_id = self.part.tool_call_id or str(uuid.uuid4())
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
 
-@dataclass
+
+@dataclass(repr=False)
 class FunctionToolResultEvent:
     """An event indicating the result of a function tool call."""
 
@@ -839,6 +880,8 @@ class FunctionToolResultEvent:
     event_kind: Literal['function_tool_result'] = 'function_tool_result'
     """Event type identifier, used as a discriminator."""
 
+    __repr__ = _utils.dataclasses_no_defaults_repr
+
 
 HandleResponseEvent = Annotated[
     Union[FunctionToolCallEvent, FunctionToolResultEvent], pydantic.Discriminator('event_kind')