fix docs

Author: DouweM

docs/install.md

@@ -43,7 +43,7 @@ pip/uv-add "pydantic-ai-slim[openai]"
 * `logfire` — installs [Pydantic Logfire](logfire.md) dependency `logfire` [PyPI ↗](https://pypi.org/project/logfire){:target="_blank"}
 * `evals` — installs [Pydantic Evals](evals.md) dependency `pydantic-evals` [PyPI ↗](https://pypi.org/project/pydantic-evals){:target="_blank"}
 * `openai` — installs [OpenAI Model](models/openai.md) dependency `openai` [PyPI ↗](https://pypi.org/project/openai){:target="_blank"}
-* `vertexai` — installs [Google Vertex AI Model](models/google.md#vertex-ai) dependencies `google-auth` [PyPI ↗](https://pypi.org/project/google-auth){:target="_blank"} and `requests` [PyPI ↗](https://pypi.org/project/requests){:target="_blank"}
+* `vertexai` — installs `GoogleVertexProvider` dependencies `google-auth` [PyPI ↗](https://pypi.org/project/google-auth){:target="_blank"} and `requests` [PyPI ↗](https://pypi.org/project/requests){:target="_blank"}
 * `google` — installs [Google Model](models/google.md) dependency `google-genai` [PyPI ↗](https://pypi.org/project/google-genai){:target="_blank"}
 * `anthropic` — installs [Anthropic Model](models/anthropic.md) dependency `anthropic` [PyPI ↗](https://pypi.org/project/anthropic){:target="_blank"}
 * `groq` — installs [Groq Model](models/groq.md) dependency `groq` [PyPI ↗](https://pypi.org/project/groq){:target="_blank"}

pydantic_ai_slim/pydantic_ai/ui/_adapter.py

@@ -144,7 +144,7 @@ def load_messages(cls, messages: Sequence[MessageT]) -> list[ModelMessage]:
 
     @abstractmethod
     def build_event_stream(self) -> UIEventStream[RunInputT, EventT, AgentDepsT, OutputDataT]:
-        """Build a protocol-specific event stream."""
+        """Build a protocol-specific event stream transformer."""
         raise NotImplementedError
 
     @cached_property

pydantic_ai_slim/pydantic_ai/ui/_event_stream.py

@@ -240,7 +240,7 @@ async def handle_event(self, event: NativeEvent) -> AsyncIterator[EventT]:
         - [`FinalResultEvent`][pydantic_ai.messages.FinalResultEvent] -> `handle_final_result`
         - [`FunctionToolCallEvent`][pydantic_ai.messages.FunctionToolCallEvent] -> `handle_function_tool_call`
         - [`FunctionToolResultEvent`][pydantic_ai.messages.FunctionToolResultEvent] -> `handle_function_tool_result`
-        - [`AgentRunResultEvent`][pydantic_ai.messages.AgentRunResultEvent] -> `handle_run_result`
+        - [`AgentRunResultEvent`][pydantic_ai.run.AgentRunResultEvent] -> `handle_run_result`
 
         Subclasses are encouraged to override the individual `handle_*` methods rather than this one.
         If you need specific behavior for all events, make sure you call the super method.

pydantic_ai_slim/pydantic_ai/ui/ag_ui/_adapter.py

@@ -85,27 +85,33 @@ class AGUIAdapter(UIAdapter[RunAgentInput, Message, BaseEvent, AgentDepsT, Outpu
 
     @classmethod
     def build_run_input(cls, body: bytes) -> RunAgentInput:
+        """Build an AG-UI run input object from the request body."""
         return RunAgentInput.model_validate_json(body)
 
     def build_event_stream(self) -> UIEventStream[RunAgentInput, BaseEvent, AgentDepsT, OutputDataT]:
+        """Build an AG-UI event stream transformer."""
         return AGUIEventStream(self.run_input, accept=self.accept)
 
+    @cached_property
+    def messages(self) -> list[ModelMessage]:
+        """Pydantic AI messages from the AG-UI run input."""
+        return self.load_messages(self.run_input.messages)
+
     @cached_property
     def toolset(self) -> AbstractToolset[AgentDepsT] | None:
+        """Toolset representing frontend tools from the AG-UI run input."""
         if self.run_input.tools:
             return _AGUIFrontendToolset[AgentDepsT](self.run_input.tools)
         return None
 
     @cached_property
     def state(self) -> dict[str, Any] | None:
+        """Frontend state from the AG-UI run input."""
         return self.run_input.state
 
-    @cached_property
-    def messages(self) -> list[ModelMessage]:
-        return self.load_messages(self.run_input.messages)
-
     @classmethod
     def load_messages(cls, messages: Sequence[Message]) -> list[ModelMessage]:
+        """Transform AG-UI messages into Pydantic AI messages."""
         builder = MessagesBuilder()
         tool_calls: dict[str, str] = {}  # Tool call ID to tool name mapping.
 

pydantic_ai_slim/pydantic_ai/ui/vercel_ai/_adapter.py

@@ -65,17 +65,21 @@ class VercelAIAdapter(UIAdapter[RequestData, UIMessage, BaseChunk, AgentDepsT, O
 
     @classmethod
     def build_run_input(cls, body: bytes) -> RequestData:
+        """Build a Vercel AI run input object from the request body."""
         return request_data_ta.validate_json(body)
 
     def build_event_stream(self) -> UIEventStream[RequestData, BaseChunk, AgentDepsT, OutputDataT]:
+        """Build a Vercel AI event stream transformer."""
         return VercelAIEventStream(self.run_input, accept=self.accept)
 
     @cached_property
     def messages(self) -> list[ModelMessage]:
+        """Pydantic AI messages from the Vercel AI run input."""
         return self.load_messages(self.run_input.messages)
 
     @classmethod
     def load_messages(cls, messages: Sequence[UIMessage]) -> list[ModelMessage]:  # noqa: C901
+        """Transform Vercel AI messages into Pydantic AI messages."""
         builder = MessagesBuilder()
 
         for msg in messages: