feat(ag-ui): preserve thinking signatures, files and tool returns in roundtrip; fix stray TOOL_CALL_ARGS (#4733)

pydantic_ai_slim/pydantic_ai/ag_ui.py

@@ -29,7 +29,7 @@
     from starlette.responses import Response
 
     from .ui import SSE_CONTENT_TYPE, OnCompleteFunc, StateDeps, StateHandler
-    from .ui.ag_ui import AGUIAdapter
+    from .ui.ag_ui import AGUIAdapter, AGUIVersion
     from .ui.ag_ui.app import AGUIApp
 except ImportError as e:  # pragma: no cover
     raise ImportError(
@@ -53,6 +53,7 @@ async def handle_ag_ui_request(
     agent: AbstractAgent[AgentDepsT, Any],
     request: Request,
     *,
+    ag_ui_version: AGUIVersion = '0.1.10',
     output_type: OutputSpec[Any] | None = None,
     message_history: Sequence[ModelMessage] | None = None,
     deferred_tool_results: DeferredToolResults | None = None,
@@ -71,6 +72,7 @@ async def handle_ag_ui_request(
     Args:
         agent: The agent to run.
         request: The Starlette request (e.g. from FastAPI) containing the AG-UI run input.
+        ag_ui_version: AG-UI protocol version controlling thinking/reasoning event format.
 
         output_type: Custom output type to use for this run, `output_type` may only be used if the agent has no
             output validators since output validators would expect an argument that matches the agent's output type.
@@ -94,6 +96,7 @@ async def handle_ag_ui_request(
     return await AGUIAdapter[AgentDepsT].dispatch_request(
         request,
         agent=agent,
+        ag_ui_version=ag_ui_version,
         deps=deps,
         output_type=output_type,
         message_history=message_history,
@@ -114,6 +117,7 @@ def run_ag_ui(
     run_input: RunAgentInput,
     accept: str = SSE_CONTENT_TYPE,
     *,
+    ag_ui_version: AGUIVersion = '0.1.10',
     output_type: OutputSpec[Any] | None = None,
     message_history: Sequence[ModelMessage] | None = None,
     deferred_tool_results: DeferredToolResults | None = None,
@@ -133,6 +137,7 @@ def run_ag_ui(
         agent: The agent to run.
         run_input: The AG-UI run input containing thread_id, run_id, messages, etc.
         accept: The accept header value for the run.
+        ag_ui_version: AG-UI protocol version controlling thinking/reasoning event format.
 
         output_type: Custom output type to use for this run, `output_type` may only be used if the agent has no
             output validators since output validators would expect an argument that matches the agent's output type.
@@ -153,7 +158,7 @@ def run_ag_ui(
     Yields:
         Streaming event chunks encoded as strings according to the accept header value.
     """
-    adapter = AGUIAdapter(agent=agent, run_input=run_input, accept=accept)
+    adapter = AGUIAdapter(agent=agent, run_input=run_input, accept=accept, ag_ui_version=ag_ui_version)
     return adapter.encode_stream(
         adapter.run_stream(
             output_type=output_type,

pydantic_ai_slim/pydantic_ai/ui/ag_ui/__init__.py

@@ -1,9 +1,10 @@
 """AG-UI protocol integration for Pydantic AI agents."""
 
 from ._adapter import AGUIAdapter
-from ._event_stream import AGUIEventStream
+from ._event_stream import AGUIEventStream, AGUIVersion
 
 __all__ = [
     'AGUIAdapter',
     'AGUIEventStream',
+    'AGUIVersion',
 ]