Merge branch 'main' into deploy-docs-preview

Author: DouweM

.github/workflows/ci.yml

@@ -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