Skip to content

Commit 51b8ccd

Browse files
committed
Add support for ImageGenerationTool output_compression and output_format for Vertex AI Gemini image models
1 parent acde3ec commit 51b8ccd

File tree

3 files changed

+267
-76
lines changed

3 files changed

+267
-76
lines changed

pydantic_ai_slim/pydantic_ai/builtin_tools.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -301,6 +301,7 @@ class ImageGenerationTool(AbstractBuiltinTool):
301301
Supported by:
302302
303303
* OpenAI Responses. Only supported for 'png' and 'webp' output formats.
304+
* Google (Vertex AI only). Only supported for 'jpeg' output format.
304305
"""
305306

306307
output_format: Literal['png', 'webp', 'jpeg'] | None = None
@@ -309,6 +310,7 @@ class ImageGenerationTool(AbstractBuiltinTool):
309310
Supported by:
310311
311312
* OpenAI Responses. Default: 'png'.
313+
* Google (Vertex AI only).
312314
"""
313315

314316
partial_images: int = 0

0 commit comments

Comments
 (0)