Document Agent.__aenter__

Author: DouweM

pydantic_ai_slim/pydantic_ai/agent.py

@@ -1788,6 +1788,7 @@ def is_end_node(
         return isinstance(node, End)
 
     async def __aenter__(self) -> Self:
+        """Enter the agent. This will start all [`MCPServerStdio`s][pydantic_ai.mcp.MCPServerStdio] registered with the agent so they can be used in a run."""
         if self._running_count == 0:
             self._exit_stack = AsyncExitStack()
             await self._exit_stack.enter_async_context(self._toolset)
@@ -1828,6 +1829,9 @@ async def run_mcp_servers(
     ) -> AsyncIterator[None]:
         """Run [`MCPServerStdio`s][pydantic_ai.mcp.MCPServerStdio] so they can be used by the agent.
 
+        Deprecated: use [`async with agent`][pydantic_ai.agent.Agent.__aenter__] instead.
+        If you need to set a sampling model on all MCP servers, use [`agent.set_mcp_sampling_model(...)`][pydantic_ai.agent.Agent.set_mcp_sampling_model].
+
         Returns: a context manager to start and shutdown the servers.
         """
         try: