Add Cloudflare AI Gateway provider

docs/api/providers.md

@@ -34,6 +34,8 @@
 
 ::: pydantic_ai.providers.vercel.VercelProvider
 
+::: pydantic_ai.providers.cloudflare.CloudflareProvider
+
 ::: pydantic_ai.providers.huggingface.HuggingFaceProvider
 
 ::: pydantic_ai.providers.moonshotai.MoonshotAIProvider

docs/models/openai.md

@@ -441,6 +441,54 @@ agent = Agent(model)
 ...
 ```
 
+### Cloudflare AI Gateway
+
+To use [Cloudflare AI Gateway](https://developers.cloudflare.com/ai-gateway/), first set up a gateway in your [Cloudflare dashboard](https://dash.cloudflare.com/?to=/:account/ai/ai-gateway) and obtain your account ID and gateway ID.
+
+!!! note
+    This provider uses Cloudflare's [unified API endpoint](https://developers.cloudflare.com/ai-gateway/usage/chat-completion/) for routing requests to multiple AI providers. For the full list of supported providers, see [Cloudflare's documentation](https://developers.cloudflare.com/ai-gateway/usage/chat-completion/#supported-providers).
+
+You can set the `CLOUDFLARE_ACCOUNT_ID`, `CLOUDFLARE_GATEWAY_ID`, and optionally `CLOUDFLARE_AI_GATEWAY_AUTH` environment variables and use [`CloudflareProvider`][pydantic_ai.providers.cloudflare.CloudflareProvider]:
+
+```python
+from pydantic_ai import Agent
+from pydantic_ai.models.openai import OpenAIChatModel
+from pydantic_ai.providers.cloudflare import CloudflareProvider
+
+model = OpenAIChatModel(
+    'openai/gpt-4o',
+    provider=CloudflareProvider(
+        account_id='your-account-id',
+        gateway_id='your-gateway-id',
+        api_key='your-openai-api-key',
+    ),
+)
+agent = Agent(model)
+...
+```
+
+For authenticated gateways with stored API keys in Cloudflare's dashboard:
+
+```python
+from pydantic_ai import Agent
+from pydantic_ai.models.openai import OpenAIChatModel
+from pydantic_ai.providers.cloudflare import CloudflareProvider
+
+model = OpenAIChatModel(
+    'anthropic/claude-3-5-sonnet',
+    provider=CloudflareProvider(
+        account_id='your-account-id',
+        gateway_id='your-gateway-id',
+        cf_aig_authorization='your-gateway-token',
+        use_gateway_keys=True,
+    ),
+)
+agent = Agent(model)
+...
+```
+
+See [`CloudflareProvider`][pydantic_ai.providers.cloudflare.CloudflareProvider] for additional configuration options including BYOK modes and authenticated gateways.
+
 ### Grok (xAI)
 
 Go to [xAI API Console](https://console.x.ai/) and create an API key.

docs/models/overview.md

@@ -20,6 +20,7 @@ In addition, many providers are compatible with the OpenAI API, and can be used
 - [Ollama](openai.md#ollama)
 - [OpenRouter](openai.md#openrouter)
 - [Vercel AI Gateway](openai.md#vercel-ai-gateway)
+- [Cloudflare AI Gateway](openai.md#cloudflare-ai-gateway)
 - [Perplexity](openai.md#perplexity)
 - [Fireworks AI](openai.md#fireworks-ai)
 - [Together AI](openai.md#together-ai)

pydantic_ai_slim/pydantic_ai/providers/__init__.py

@@ -69,6 +69,10 @@ def infer_provider_class(provider: str) -> type[Provider[Any]]:  # noqa: C901
         from .vercel import VercelProvider
 
         return VercelProvider
+    elif provider == 'cloudflare':
+        from .cloudflare import CloudflareProvider
+
+        return CloudflareProvider
     elif provider == 'azure':
         from .azure import AzureProvider