Merge branch 'main' into openai-response-model-base-url

Author: DouweM

.github/workflows/after-ci.yml

@@ -16,7 +16,6 @@ jobs:
     steps:
       - uses: astral-sh/setup-uv@v5
         with:
-          enable-cache: true
           python-version: "3.12"
 
       - uses: dawidd6/action-download-artifact@v6
@@ -46,6 +45,10 @@ jobs:
       name: deploy-docs-preview
 
     steps:
+      - run: echo "$GITHUB_EVENT_JSON"
+        env:
+          GITHUB_EVENT_JSON: ${{ toJSON(github.event) }}
+
       - uses: actions/checkout@v4
 
       - uses: actions/setup-node@v4
@@ -54,20 +57,24 @@ jobs:
 
       - uses: astral-sh/setup-uv@v5
         with:
-          enable-cache: true
           python-version: "3.12"
+          enable-cache: true
+          cache-suffix: deploy-docs-preview
 
-      - uses: dawidd6/action-download-artifact@v6
+      - id: download-artifact
+        uses: dawidd6/action-download-artifact@v6
         with:
           workflow: ci.yml
           name: site
           path: site
           commit: ${{ github.event.workflow_run.head_sha }}
           allow_forks: true
           workflow_conclusion: completed
+          if_no_artifact_found: warn
 
       - uses: cloudflare/wrangler-action@v3
         id: deploy
+        if: steps.download-artifact.outputs.found_artifact == 'true'
         with:
           apiToken: ${{ secrets.CLOUDFLARE_API_TOKEN }}
           environment: previews
@@ -77,12 +84,9 @@ jobs:
             --var GIT_COMMIT_SHA:${{ github.event.workflow_run.head_sha }}
             --var GIT_BRANCH:${{ github.event.workflow_run.head_branch }}
 
-      - run: echo "$GITHUB_EVENT_JSON"
-        env:
-          GITHUB_EVENT_JSON: ${{ toJSON(github.event) }}
-
       - name: Set preview URL
         run: uv run --no-project --with httpx .github/set_docs_pr_preview_url.py
+        if: steps.deploy.outcome == 'success'
         env:
           DEPLOY_OUTPUT: ${{ steps.deploy.outputs.command-output }}
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

.github/workflows/ci.yml

@@ -26,14 +26,15 @@ jobs:
         # Typecheck both Python 3.10 and 3.13. We've had issues due to not checking against both.
         python-version: ["3.10", "3.13"]
     env:
-      UV_PYTHON: ${{ matrix.python-version }}
       PYRIGHT_PYTHON: ${{ matrix.python-version }}
     steps:
       - uses: actions/checkout@v4
 
       - uses: astral-sh/setup-uv@v5
         with:
+          python-version: ${{ matrix.python-version }}
           enable-cache: true
+          cache-suffix: lint
 
       - name: Install dependencies
         run: uv sync --all-extras --all-packages --group lint
@@ -56,6 +57,7 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
+          cache-suffix: mypy
 
       - name: Install dependencies
         run: uv sync --no-dev --group lint
@@ -70,6 +72,7 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
+          cache-suffix: docs
 
       - run: uv sync --group docs
 
@@ -107,6 +110,7 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
+          cache-suffix: live
 
       - uses: pydantic/ollama-action@v3
         with:
@@ -138,7 +142,7 @@ jobs:
   test:
     name: test on ${{ matrix.python-version }} (${{ matrix.install.name }})
     runs-on: ubuntu-latest
-    timeout-minutes: 15
+    timeout-minutes: 20
     strategy:
       fail-fast: false
       matrix:
@@ -151,26 +155,27 @@ jobs:
           - name: all-extras
             command: "--all-extras"
     env:
-      UV_PYTHON: ${{ matrix.python-version }}
       CI: true
       COVERAGE_PROCESS_START: ./pyproject.toml
     steps:
       - uses: actions/checkout@v4
 
       - uses: astral-sh/setup-uv@v5
         with:
+          python-version: ${{ matrix.python-version }}
           enable-cache: true
-          prune-cache: false
+          cache-suffix: ${{ matrix.install.name }}
 
       - uses: denoland/setup-deno@v2
         with:
           deno-version: v2.x
 
       - run: mkdir .coverage
 
-      - run: uv run mcp-run-python example --deps=numpy
       - run: uv sync --only-dev
 
+      - run: uv run mcp-run-python example --deps=numpy
+
       - name: cache HuggingFace models
         uses: actions/cache@v4
         with:
@@ -193,22 +198,22 @@ jobs:
   test-lowest-versions:
     name: test on ${{ matrix.python-version }} (lowest-versions)
     runs-on: ubuntu-latest
-    timeout-minutes: 15
+    timeout-minutes: 20
     strategy:
       fail-fast: false
       matrix:
         python-version: ["3.10", "3.11", "3.12", "3.13"]
     env:
-      UV_PYTHON: ${{ matrix.python-version }}
       CI: true
       COVERAGE_PROCESS_START: ./pyproject.toml
     steps:
       - uses: actions/checkout@v4
 
       - uses: astral-sh/setup-uv@v5
         with:
+          python-version: ${{ matrix.python-version }}
           enable-cache: true
-          prune-cache: false
+          cache-suffix: lowest-versions
 
       - uses: denoland/setup-deno@v2
         with:
@@ -218,6 +223,8 @@ jobs:
 
       - run: uv sync --group dev
 
+      - run: uv run mcp-run-python example --deps=numpy
+
       - name: cache HuggingFace models
         uses: actions/cache@v4
         with:
@@ -226,8 +233,6 @@ jobs:
           restore-keys: |
             hf-${{ runner.os }}-
 
-      - run: uv run mcp-run-python example --deps=numpy
-
       - run: unset UV_FROZEN
 
       - run: uv run --all-extras --resolution lowest-direct coverage run -m pytest --durations=100 -n auto --dist=loadgroup
@@ -250,15 +255,15 @@ jobs:
       matrix:
         python-version: ["3.11", "3.12", "3.13"]
     env:
-      UV_PYTHON: ${{ matrix.python-version }}
       CI: true
     steps:
       - uses: actions/checkout@v4
 
       - uses: astral-sh/setup-uv@v5
         with:
+          python-version: ${{ matrix.python-version }}
           enable-cache: true
-          prune-cache: false
+          cache-suffix: examples
 
       - name: cache HuggingFace models
         uses: actions/cache@v4
@@ -288,6 +293,7 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
+          cache-suffix: dev
 
       - run: uv sync --group dev
       - run: uv run coverage combine
@@ -343,6 +349,7 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
+          cache-suffix: docs-upload
 
       - uses: actions/download-artifact@v4
         with:
@@ -384,6 +391,7 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
+          cache-suffix: deploy-docs-preview
 
       - uses: actions/download-artifact@v4
         with:
@@ -431,6 +439,7 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
+          cache-suffix: release
 
       - run: uv build --all-packages
 

.github/workflows/claude.yml

@@ -37,6 +37,7 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
+          cache-suffix: claude-code
 
       - uses: denoland/setup-deno@v2
         with:

.github/workflows/manually-deploy-docs.yml

@@ -13,6 +13,7 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
+          cache-suffix: docs
 
       - run: uv sync --group docs
 
@@ -54,6 +55,7 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
+          cache-suffix: docs-upload
 
       - uses: actions/download-artifact@v4
         with:

.gitignore

@@ -10,7 +10,7 @@ env*/
 /TODO.md
 /postgres-data/
 .DS_Store
-examples/pydantic_ai_examples/.chat_app_messages.sqlite
+.chat_app_messages.sqlite
 .cache/
 .vscode/
 /question_graph_history.json

README.md

@@ -50,8 +50,8 @@ Designed to give your IDE or AI coding agent as much context as possible for aut
 5. **Powerful Evals**:
 Enables you to systematically test and [evaluate](https://ai.pydantic.dev/evals) the performance and accuracy of the agentic systems you build, and monitor the performance over time in Pydantic Logfire.
 
-6. **MCP, A2A, and AG-UI**:
-Integrates the [Model Context Protocol](https://ai.pydantic.dev/mcp/client), [Agent2Agent](https://ai.pydantic.dev/a2a), and [AG-UI](https://ai.pydantic.dev/ag-ui) standards to give your agent access to external tools and data, let it interoperate with other agents, and build interactive applications with streaming event-based communication.
+6. **MCP, A2A, and UI**:
+Integrates the [Model Context Protocol](https://ai.pydantic.dev/mcp/overview), [Agent2Agent](https://ai.pydantic.dev/a2a), and various [UI event stream](https://ai.pydantic.dev/ui/overview) standards to give your agent access to external tools and data, let it interoperate with other agents, and build interactive applications with streaming event-based communication.
 
 7. **Human-in-the-Loop Tool Approval**:
 Easily lets you flag that certain tool calls [require approval](https://ai.pydantic.dev/deferred-tools#human-in-the-loop-tool-approval) before they can proceed, possibly depending on tool call arguments, conversation history, or user preferences.

docs-site/src/index.ts

@@ -55,6 +55,7 @@ const redirect_lookup: Record<string, string> = {
   '/examples': 'examples/setup/',
   '/mcp': '/mcp/overview/',
   '/models': '/models/overview/',
+  '/ag-ui': '/ui/ag-ui/'
 }
 
 function redirect(pathname: string): string | null {

docs-site/wrangler.toml

@@ -4,6 +4,7 @@ compatibility_date = "2025-06-17"
 routes = ["ai.pydantic.dev/*"]
 main = "src/index.ts"
 workers_dev = false
+preview_urls = true
 compatibility_flags = [ "nodejs_compat_v2" ]
 
 [vars]

docs/agents.md

@@ -103,6 +103,9 @@ async def main():
         FinalResultEvent(tool_name=None, tool_call_id=None),
         PartDeltaEvent(index=0, delta=TextPartDelta(content_delta='Mexico is Mexico ')),
         PartDeltaEvent(index=0, delta=TextPartDelta(content_delta='City.')),
+        PartEndEvent(
+            index=0, part=TextPart(content='The capital of Mexico is Mexico City.')
+        ),
         AgentRunResultEvent(
             result=AgentRunResult(output='The capital of Mexico is Mexico City.')
         ),

docs/api/ui/ag_ui.md

@@ -0,0 +1,5 @@
+# `pydantic_ai.ui.ag_ui`
+
+::: pydantic_ai.ui.ag_ui
+
+::: pydantic_ai.ui.ag_ui.app