Image Generation

The ImageGeneration capability lets your agent generate images. Like all provider-adaptive tools, it uses the provider’s native image generation when available, with an optional subagent fallback for other models.

ImageGeneration defaults to native-only. Backed by ImageGenerationTool on the native side (see Image Generation Tool for provider support and configuration) — pass native=ImageGenerationTool(...) directly for full control.

For the local side, pass fallback_model='…' to delegate unsupported requests to a subagent running an image-generation-capable model (e.g. openai-responses:gpt-5.4), or local= with any callable, Tool, or AbstractToolset for a custom generator.

image_generation.py

from pydantic_ai.capabilities import ImageGeneration

# Native-only — raises on models without native image generation
ImageGeneration()

# Native preferred; subagent fallback for unsupported models
ImageGeneration(fallback_model='openai-responses:gpt-5.4')

# Native preferred; custom callable as fallback
def my_generator(prompt: str) -> bytes: ...
ImageGeneration(local=my_generator)