Merge branch 'main' into vercel-ai-chat

Author: DouweM

.github/workflows/after-ci.yml

@@ -65,6 +65,7 @@ jobs:
           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

.github/workflows/ci.yml

@@ -142,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:
@@ -198,7 +198,7 @@ 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:
@@ -263,7 +263,7 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
           enable-cache: true
-          cache-suffix: all-extras
+          cache-suffix: examples
 
       - name: cache HuggingFace models
         uses: actions/cache@v4

pyproject.toml

@@ -54,6 +54,11 @@ examples = ["pydantic-ai-examples=={{ version }}"]
 a2a = ["fasta2a>=0.4.1"]
 dbos = ["pydantic-ai-slim[dbos]=={{ version }}"]
 prefect = ["pydantic-ai-slim[prefect]=={{ version }}"]
+outlines-transformers = ["pydantic-ai-slim[outlines-transformers]=={{ version }}"]
+outlines-llamacpp = ["pydantic-ai-slim[outlines-llamacpp]=={{ version }}"]
+outlines-mlxlm = ["pydantic-ai-slim[outlines-mlxlm]=={{ version }}"]
+outlines-sglang = ["pydantic-ai-slim[outlines-sglang]=={{ version }}"]
+outlines-vllm-offline = ["pydantic-ai-slim[outlines-vllm-offline]=={{ version }}"]
 
 [project.urls]
 Homepage = "https://ai.pydantic.dev"
@@ -105,11 +110,6 @@ dev = [
     "pip>=25.2",
     "genai-prices>=0.0.28",
     "mcp-run-python>=0.0.20",
-    # Needed to test Outlines (not included in the default installation)
-    "pydantic-ai-slim[outlines-transformers]",
-    "pydantic-ai-slim[outlines-llamacpp]",
-    "pydantic-ai-slim[outlines-mlxlm]",
-    "pydantic-ai-slim[outlines-sglang]",
 ]
 lint = ["mypy>=1.11.2", "pyright>=1.1.390", "ruff>=0.6.9"]
 docs = [

tests/models/test_outlines.py

@@ -56,10 +56,10 @@
     import vllm  # type: ignore[reportMissingImports]
 
     # We try to load the vllm model to ensure it is available
-    try:  # pragma: no cover
+    try:  # pragma: no lax cover
         vllm.LLM('microsoft/Phi-3-mini-4k-instruct')  # type: ignore
-    except RuntimeError as e:  # pragma: no cover
-        if 'Found no NVIDIA driver' in str(e):
+    except RuntimeError as e:  # pragma: lax no cover
+        if 'Found no NVIDIA driver' in str(e) or 'Device string must not be empty' in str(e):
             # Treat as import failure
             raise ImportError('CUDA/NVIDIA driver not available') from e
         raise