"jl4vSSKJrJrJr SSKJr SSKJr SSKJ r SSK J r J r J r SSKJr SSKJr SSKJr SS KJr SS KJrJr S S KJrJr S S KJrJr S SKJ r J!r! S SK"J#r#J$r$J%r%J&r& S SK'J(r( S SK)J*r* /SQr+\"S\ S5r,\S.S\\,-\--S\\-/\\ 4S\4Sjjr.\ "SSS9"SS55r/g))Callable GeneratorSequence)contextmanager) ContextVar) dataclass)AnyClassVarLiteral) TypeAliasType)_utils) UserError)InstrumentationSettings)Providerinfer_provider)ImageGenerationInputImageGenerationModel) InstrumentedImageGenerationModel!instrument_image_generation_model)GeneratedImageImageGenerationResult)ImageDimensionsImageGenerationAspectRatioImageGenerationSettingsmerge_image_generation_settings)TestImageGenerationModel)WrapperImageGenerationModel)rrrrrrrImageGeneratorrKnownImageGenerationModelNamerrinfer_image_generation_modelrrr )z#google-cloud:gemini-2.5-flash-imagezgoogle-cloud:gemini-3-pro-imagez#google-cloud:gemini-3.1-flash-imagez(google-cloud:gemini-3.1-flash-lite-imagezgoogle:gemini-2.5-flash-imagezgoogle:gemini-3-pro-imagezgoogle:gemini-3.1-flash-imagez"google:gemini-3.1-flash-lite-imagezopenai:gpt-image-1zopenai:gpt-image-1-minizopenai:gpt-image-1.5zopenai:gpt-image-2zxai:grok-imagine-imagezxai:grok-imagine-image-2.0zxai:grok-imagine-image-quality)provider_factorymodelr"returnc[U[5(aU$URSSS9up#UnUR S5(a$SSKJn U"U5nUS :wa[S U<S 35eU"U5nUS :Xa SS KJ n U"X7S9$US;a SSK J n U "X7S9$US:Xa SSK J n U "X7S9$[SU<S35e![an[S5UeSnAff=f)z/Infer the image generation model from the name.:r)maxsplitzQYou must provide a provider prefix when specifying an image generation model nameNzgateway/)normalize_gateway_provider google-cloudzImage generation provider zc cannot be routed through the Pydantic AI Gateway. The supported gateway route is `gateway/google`.openai)OpenAIImageGenerationModel)provider)googler*)GoogleImageGenerationModelxai)XaiImageGenerationModelz Provider z* does not support direct image generation.) isinstancersplit ValueError startswithproviders.gatewayr)rr+r,r.r/r0r1) r#r" provider_name model_namee model_kindr)r-r,r/r1s X/home/mande/repo/quber/.venv/lib/python3.13/site-packages/pydantic_ai/images/__init__.pyr!r!Hs  %-.. u$)KKaK$@! JZ((B 0 ;  ',],=>CC   .HX6)*HH 1 16)*HH u 0&zEE)M#44^_``E ulmsttusB55 C? C  CF)initeqc \rSrSr%Sr\\-S-\S'Sr\ \\-\S'SSSS.S \ \ -\ -S \ S-S \S\\-S-S S4 S jjr\SS\\-S S4Sjj5r\S \ \ -\ -4Sj5r\\R*S.S \ \ -\ -\R,-S \S4Sjj5rSSS.S\ S\\S-S \ S-S \4SjjrSSS.S\ S\\S-S \ S-S \4SjjrS \ 4SjrSrg)r{a1High-level interface for generating images. The `ImageGenerator` class provides a convenient way to generate images from a prompt, and to edit or transform reference images, using dedicated image models. It handles model inference, settings management, and optional OpenTelemetry instrumentation. Example: ```python from pydantic_ai import ImageGenerator generator = ImageGenerator('openai:gpt-image-2') async def main(): result = await generator.generate('A watercolor map of a floating city.') print(result.image.media_type) #> image/png ``` N instrumentF_instrument_defaultT)settingsdefer_model_checkr@r#rBrCr$clU(aUO [U5UlX lX@l[ SSS9Ulg)a$Initialize an ImageGenerator. Args: model: The image generation model to use. Can be specified as: - A model name string in the format `'provider:model-name'` (e.g., `'openai:gpt-image-2'`) - An [`ImageGenerationModel`][pydantic_ai.images.ImageGenerationModel] instance settings: Optional [`ImageGenerationSettings`][pydantic_ai.images.ImageGenerationSettings] to use as defaults for all generate calls. defer_model_check: Whether to defer resolving the model name to a model instance, and the provider authentication that resolution requires, until the first generate call. Set to `False` to resolve the model immediately on construction. instrument: OpenTelemetry instrumentation settings. Set to `True` to enable with defaults, or pass an [`InstrumentationSettings`][pydantic_ai.models.instrumented.InstrumentationSettings] instance to customize. If `None`, uses the value from [`ImageGenerator.instrument_all()`][pydantic_ai.images.ImageGenerator.instrument_all]. _override_modelN)default)r!_model _settingsr@rrE)selfr#rBrCr@s r;__init__ImageGenerator.__init__s14 1e6RSX6Y !$HRSdnrHscU[lg)aSet the default instrumentation options for all image generators where `instrument` is not explicitly set. This is useful for enabling instrumentation globally without modifying each generator individually. Args: instrument: Instrumentation settings to use as the default. Set to `True` for default settings, `False` to disable, or pass an [`InstrumentationSettings`][pydantic_ai.models.instrumented.InstrumentationSettings] instance to customize. N)rrA)r@s r;instrument_allImageGenerator.instrument_alls .8*rLcUR$)z2The image generation model used by this generator.)rG)rIs r;r#ImageGenerator.models{{rL)r#c# # [R"U5(a%URR[ U55nOSnSv UbURR U5 gg!UbURR U5 ff=f7f)aHContext manager to temporarily override the image generation model. Useful for testing or dynamically switching models. Args: model: The image generation model to use within this context. Example: ```python from pydantic_ai import ImageGenerator generator = ImageGenerator('openai:gpt-image-2') async def main(): # Temporarily use a different model with generator.override(model='google:gemini-3.1-flash-image'): result = await generator.generate('A watercolor map of a floating city.') print(result.model_name) #> gemini-3.1-flash-image ``` N)r is_setrEsetr!reset)rIr# model_tokens r;overrideImageGenerator.overrides~8 ==  ..223OPU3VWKK 8 &$$**;7'{&$$**;7'sABA* B*!B  BimagesrBpromptrZc# UR5n[URU5nURXUS9IShvN $N7f)a}Generate images from a prompt and optional reference images. Args: prompt: The text prompt describing the image to generate. images: Optional reference images to edit or transform. Passing reference images sends the request to the provider's image-editing path, preserving the order of the images. Each item can be a [`BinaryImage`][pydantic_ai.messages.BinaryImage], [`ImageUrl`][pydantic_ai.messages.ImageUrl], or [`UploadedFile`][pydantic_ai.messages.UploadedFile]; see the [Image Generation guide](../image-generation.md#editing-images) for the reference-input types each provider accepts. settings: Optional settings to override the generator's default settings for this call. Returns: An [`ImageGenerationResult`][pydantic_ai.images.ImageGenerationResult] containing the generated images and metadata about the operation. Raises: ContentFilterError: If the provider blocked the request, or every generated image, for content moderation. UserError: If the prompt is empty, a setting is invalid, or the model cannot produce the requested dimensions. rYN) _get_modelrrHgenerate)rIr[rZrBr#s r;r^ImageGenerator.generates<<!24>>8L^^FH^MMMMs:AAAcJ[R"URXUS95$)aSynchronous version of [`generate()`][pydantic_ai.images.ImageGenerator.generate]. Args: prompt: The text prompt describing the image to generate. images: Optional reference images to edit or transform. settings: Optional settings to override the generator's default settings for this call. Returns: An [`ImageGenerationResult`][pydantic_ai.images.ImageGenerationResult] containing the generated images and metadata about the operation. Raises: ContentFilterError: If the provider blocked the request, or every generated image, for content moderation. UserError: If the prompt is empty, a setting is invalid, or the model cannot produce the requested dimensions. rY)r run_until_completer^)rIr[rZrBs r; generate_syncImageGenerator.generate_syncs#0((vW_)`aarLcURR5=n(aUnO[UR5=o lUR nUc UR n[X#5$)z-Create a model configured for this generator.)rEgetr!r#rGr@rAr)rI some_modelmodel_r@s r;r]ImageGenerator._get_model3s[--113 3: 3F#? #K KF[__  11J0DDrL)rGrErHr@)T) __name__ __module__ __qualname____firstlineno____doc__rbool__annotations__rAr rr strrrJ staticmethodrNpropertyr#rr UNSETUnsetrrWrrrr^rbr]__static_attributes__rLr;rr{s(($.55EJ"9D"@AI 48"&<@ t#&CCcIt*D0 t  t ,d2T9 t t@ 8#:T#A 8T 8 8+.KKcQ\b[g[g$8$&CCcIFLLX$8 4 $8$8T9=37 N N-.5 N *D0 N  NL9=37 bb-.5 b *D0 b  b4 E0 ErLrN)0collections.abcrrr contextlibr contextvarsr dataclassesrtypingr r r typing_extensionsr pydantic_air pydantic_ai.exceptionsrpydantic_ai.models.instrumentedrpydantic_ai.providersrrbaserr instrumentedrrresultrrrBrrrrtestrwrapperr__all__r rpr!rrvrLr;rs99%"!))+,C:<]9 +0 $!.#  *!(8F-a "? ?# E-auhsm34-a -af % CECE!CErL