wip: remove id generator and BuiltinToolReturnPart - fix tests using dumping and IsStr - add dump_messages to base adapter class

Author: dsfaccini

pydantic_ai_slim/pydantic_ai/ui/_adapter.py

@@ -143,6 +143,12 @@ def load_messages(cls, messages: Sequence[MessageT]) -> list[ModelMessage]:
         """Transform protocol-specific messages into Pydantic AI messages."""
         raise NotImplementedError
 
+    @classmethod
+    @abstractmethod
+    def dump_messages(cls, messages: Sequence[ModelMessage]) -> list[MessageT]:
+        """Transform Pydantic AI messages into protocol-specific messages."""
+        raise NotImplementedError
+
     @abstractmethod
     def build_event_stream(self) -> UIEventStream[RunInputT, EventT, AgentDepsT, OutputDataT]:
         """Build a protocol-specific event stream transformer."""

pydantic_ai_slim/pydantic_ai/ui/ag_ui/_adapter.py

@@ -109,6 +109,11 @@ def state(self) -> dict[str, Any] | None:
         """Frontend state from the AG-UI run input."""
         return self.run_input.state
 
+    @classmethod
+    def dump_messages(cls, messages: Sequence[ModelMessage]) -> list[Message]:
+        """Transform Pydantic AI messages into AG-UI messages."""
+        raise NotImplementedError('TODO: implement dump_messages method')  # TODO: implement dump_messages method
+
     @classmethod
     def load_messages(cls, messages: Sequence[Message]) -> list[ModelMessage]:
         """Transform AG-UI messages into Pydantic AI messages."""

tests/test_ui.py

@@ -87,6 +87,10 @@ class DummyUIAdapter(UIAdapter[DummyUIRunInput, ModelMessage, str, AgentDepsT, O
     def build_run_input(cls, body: bytes) -> DummyUIRunInput:
         return DummyUIRunInput.model_validate_json(body)
 
+    @classmethod
+    def dump_messages(cls, messages: Sequence[ModelMessage]) -> list[ModelMessage]:
+        return list(messages)
+
     @classmethod
     def load_messages(cls, messages: Sequence[ModelMessage]) -> list[ModelMessage]:
         return list(messages)