"jSSKJr SSKrSSKJrJrJr SSKJrJ r SSK J r SSK J r JrJrJr SSKJr SSKJr SS KJr SS KJrJrJrJr SS KJrJrJrJ r J!r! SS K"J#r# SS K$J%r% SSK&J'r'J(r(J)r) SSK*J+r+J,r,J-r-J.r.J/r/ SSK0J1r1J2r2J3r3J4r4 SSK5J6r6 SSK7J8r8 SSK9J:r: SSK;Jr> \ (aSSK?J@r@JArA \B"\"\/55rC\B"\"\,55rDSrESrF\"SS9"SS55rG\"SS9"S S!\>\155rHg)") annotationsN) AwaitableCallableSequence) dataclassreplace)cached_property) TYPE_CHECKINGAnyLiteralget_args) deprecated)replace_no_init)PydanticAIDeprecationWarning)ContentFilterError ModelRetryUnexpectedModelBehavior UserError)ImageDimensionsImageGenerationAspectRatioImageGenerationModelImageGenerationSettingsImageGenerator) DIMENSIONS_ASPECT_RATIO_CONFLICT) BinaryImage) AbstractModelKnownModelNameModel)SUPPORTED_NATIVE_TOOLSImageAspectRatioImageGenerationModelNameImageGenerationTool ImageSize) AgentDepsT RunContextToolToolDefinition)AbstractToolset)PreparedToolset)check_deprecated_fallback_model)AbstractCapability)NativeOrLocalTool)ImageGenerationFallbackModelImageGenerationNativeToolzThe direct `ImageGeneration` fallback cannot honor `action="edit"` because the `generate_image` tool does not receive reference images. Use `ImageGenerator.generate(..., images=...)` directly for image editing.zThe direct `ImageGeneration` fallback ignored native-tool setting(s): {settings}. Configure provider-specific direct settings on the `ImageGenerator` or `ImageGenerationModel` instead.T)kw_onlycL\rSrSr%SrS\S'S\S'S\S'S \S 'SS jrS rg )_DirectImageGenerationTool>zBLocal capability tool backed directly by the image generation API.%ImageGenerator | ImageGenerationModel generatorrsettings*Literal['generate', 'edit', 'auto'] | NoneactionImageGenerationModelName | None image_modelc# URS:Xa[[5eURb[R "S[ SS9 URRXRS9IShvN n[UR5S:wa"[S[UR5S35eUR $NK![an[[U55UeSnAff=f7f) NeditzDirect `ImageGeneration` fallback ignored `image_model`; the direct image model is already selected by `local` or `fallback_image_model` stacklevelr6r*z*Direct image generation fallback returned z images; expected exactly one. If the generator asks for more than one image per call through a provider-specific image count setting, call `ImageGenerator.generate()` directly instead.)r8r_EDIT_ACTION_UNSUPPORTEDr:warningswarn UserWarningr5generater6rrstrlenimagesrimage)selfpromptresultes f/home/mande/repo/quber/.venv/lib/python3.13/site-packages/pydantic_ai/capabilities/image_generation.py__call__#_DirectImageGenerationTool.__call__Gs ;;& 45 5    ' MMb   ,>>226MM2RRF v}}  ")0B<1B>5AC%<B>> C"CC""C%N)rKrFreturnr)__name__ __module__ __qualname____firstlineno____doc____annotations__rO__static_attributes__rQrNr2r2>s"L44%% 6600rZr2F)initc.^\rSrSr%SrSrS\S'S\S'SrS\S 'S \S 'S \S 'S\S'S\S'S\S'S\S'S\S'S\S'S\S'S\S'S\S'S rS!\S"'S#SSSSSSSSSSSSSSS S$SSS%.S=S&jjr S>U4S'jjr \ \ "S(\ S)9S?S*j55r\R\ "S(\ S)9S@S+j55r\SAS,j5r\ SBS-j5rSBU4S.jjr\SCU4S/jj5r\S#SSSSSSSSSSSSSSS S$SSS%.SDS0jj5rSES1jrSES2jrSFS3jrSGS4jrSHS5jrSIS6jrSJS7jrSKS8jrSLS9jrSMS:jr SNU4S;jjr!SSUR4SUR4SUR44VVs/sH upUcM UPM nnn[U5S:a%[ SSR SU55S35eUR S LaU(d [ S 5e[UR[5(a [ S 5e[UR[5(a [ S 5e[UR[5(a*S UR;a[ SUR<S35eUR(aURbURb[ [5eUR S LapURS:Xa[ [ 5eUR#5=n(a:[$R&"[(R+SR U5S9[,SS9 /nUR S LdURc URbUR/5upeO!UR(dUR15n[2TU]i5 U(a.[$R&"SSR U5S3[,SS9 ggs snnf)Nr_r`rbr*z1ImageGeneration: cannot specify more than one of , c3.# UH nSUS3v M g7f)`NrQ).0names rN 0ImageGeneration.__post_init__..wsMmfl^bPQRVQWWXkflsu — use `local` for an `ImageGenerator` or a custom tool, `fallback_image_model` for a direct image model, or `fallback_subagent_model` for the subagent fallbackFuImageGeneration(native=False) requires an explicit fallback — pass `local` for an `ImageGenerator` or a custom tool, `fallback_image_model` for a direct image model, or `fallback_subagent_model` for the subagent fallbackzImageGeneration: a bare image model goes to `fallback_image_model`, not `local`, which takes an `ImageGenerator`, a custom `Tool`, `AbstractToolset`, or callable.zImageGeneration: an `ImageGenerator` goes to `local`, not `fallback_image_model`, which takes an `ImageGenerationModel` or a `'provider:model'` name.:z'ImageGeneration: `fallback_image_model=z=` is not a direct image model. Name it as `'provider:model'`.r<r@r>z2`ImageGeneration` ignored direct-only setting(s): z. Only a direct generator applies them: use `native=False` with `fallback_image_model='provider:image-model'` or `local=ImageGenerator(...)`.)r_r`rbrGrjoinrx isinstancerrrF_has_direct_generatorrrrtrr8rA_native_only_settingsrBrC_NATIVE_ONLY_SETTINGS_DROPPEDformatrD_native_geometry_direct_only_geometrysuperr)rJrvaluestated native_onlyignored_ __class__s rNrImageGeneration.__post_init__fs$**%*D,H,HI')B)BC          v;?CDIIMmflMmDmCnoVV  ;;% V  djj"6 7 7d  d// @ @V  d// 5 5#TE^E^:^ 9$:S:S9VW>>   % %*t/@/@/L @AA{{e#;;&(#$<=="&"<"<">>;>MM5<[TU]5=(d URSL$r)r_has_local_fallbackrb)rJrs rNr#ImageGeneration._has_local_fallbacks$ w*,U0I0IQU0UUrZc>[TU]U5n[X 5(deUVs/sH5n[U[5(dMURcM)URPM7 nnU(a [ X$SS9$U$s snf)aqMerge like `NativeOrLocalTool`, except that `dimensions` is one value, not a collection. The default merge unions two sequences, and a `(width, height)` pair's entries are not independent: `(1024, 1024)` beside `(1536, 1024)` unions to `(1024, 1536)`, a flipped orientation neither instance asked for, and two disjoint pairs union to a three-element tuple that is no size at all. It takes the later stated value instead, the rule the scalar fields already get. Applied after the base merge because `__post_init__` reads only whether `dimensions` is set, never what it is, and a merge never turns a stated pair into `None`. )rr)rcombinerr]rrr)cls capabilitiesmerged capabilityrrs rNrImageGeneration.combines.&&&&&+ * *o6 ";E;P;P "J ! !*  BHv*=SVS  sA: A:A:c[U[5(a$[U5S:wa [S5eUSUS4nU"S0SU_SU_SU_SU_S U_S U_S U_S U_S U_SU _SU _SU _SU _SU _SU_SU_SU_SU_SU_6$)aConstruct from the JSON/YAML-serializable subset of the runtime API. Runtime objects, such as the `ImageGenerationModel` that `fallback_image_model` also takes and the `ImageGenerator`, `Tool`, toolset and callables `local` takes, can be passed to `ImageGeneration(...)` directly but cannot be represented in an agent spec. A direct image model name is serializable and can be passed as `fallback_image_model='provider:model'`. r=z?Image generation `dimensions` must contain exactly two integersrr*rxr_r`r{rbr8rdrfrhr:rjrlrnrprrrtrwryrzrQ)rlistrGr)rrxr_r`rbr8rdrfrhr:rjrlrnrprrrtrwryrzr{s rN from_specImageGeneration.from_specsJ j$ ' ':!# abb$Q-A7J   %< *  "6    " * " $  2 (   " &! "# $(% &$'  rZc/nURbURS5 URb%UR[;aURS5 U$)z8Geometry settings the native tool has no way to express.rrrt)rrappendrt_NATIVE_IMAGE_ASPECT_RATIOS)rJ direct_onlys rNr%ImageGeneration._direct_only_geometry]sM!# ?? &   | ,    (T->->Fa-a   ~ .rZcSUR4SUR4SUR4SUR4SUR4SUR 4SUR 44VVs/sH upUcM UPM snn$s snnf)zJSettings only the native tool can express, which a direct generator drops.rdrfrhrjrlrnrp)rdrfrhrjrlrnrp)rJrrs rNr%ImageGeneration._native_only_settingsfst/!4#6#67t/%t'>'>? $"4"45DLL)#          s ! A8.A8cT0n/nURb5UR[;aURUS'OURS5 URb#UR[;aURUS'UR (dUR UR55 X4$)uThe geometry settings the native tool can express, and the ones it can't. `dimensions` and `aspect_ratio` are only reported as ignored when no direct generator is configured, since the `generate_image` tool built for one forwards both. `size` has no direct counterpart, so it is dropped whichever path runs. That suppression is a construction-time approximation: this runs from `__post_init__`, before a model exists, while native-vs-local is decided per request in `models.resolve_request_tools`. Warning here regardless would fire on every configuration whose model has no native image generation, where the settings *are* applied. The case it misses — `native=True` plus a natively capable model, which drops the direct generator — is warned about per request instead, from the prepare function `get_toolset` installs. Split out of `_image_gen_kwargs` only to keep that method under the complexity limit. rprt)rp_NATIVE_IMAGE_SIZESrrtrrextendr)rJkwargsrs rNr ImageGeneration._native_geometryxs "$ 99 yy//!%vv&    (T->->B]-]%)%6%6F> ")) NN4557 8rZc0nURbURUS'URbURUS'URbURUS'URbURUS'URbURUS'UR bUR US'UR 5up#URU5 URbURUS'URbURUS'U$) zDCollect settings supported by the native `ImageGenerationTool` path.rdrfrhrjrlrnr8model) rdrfrhrjrlrnrupdater8r:)rJrgeometryrs rN_image_gen_kwargs!ImageGeneration._image_gen_kwargss!# ?? &#'??F<    *'+':':F# $ ?? &#'??F<  " " .+/+B+BF' (    )&*&8&8F? # << # $ F9 ++-  h ;; "#{{F8     '"..F7O rZc4[S0UR5D6$)NrQ)r"rrs rN_default_nativeImageGeneration._default_natives">T%;%;%=>>rZcL[[U5RSU<S35e)Nz : `local=z` is not supported. Pass an `ImageGenerator`, `Tool`, `AbstractToolset`, or callable directly, or name a direct image model as `fallback_image_model='provider:model'`.)rrrS)rJrs rN_resolve_local_strategy'ImageGeneration._resolve_local_strategys1Dz""#9TH57 7  rZc d0nURbURUS'URnUcBURc5[UR[5(aURRnUbX2S'[ [ "[UUURURS9RSSS9$)aPBuild the `generate_image` tool from the capability's current settings. Derived when the toolset is requested rather than stored at construction, so what the tool carries is always what the capability declares: `dataclasses.replace` and `combine` both produce an instance whose fields no longer match a tool built from an earlier one. rrrt)r5r6r8r:generate_imagez,Generate an image based on the given prompt.)rrz) rrrtrrxr"r&r r2r8r:rO)rJr5r6rts rN_direct_local_tool"ImageGeneration._direct_local_tools-/ ?? &%)__H\ "((  DOO$; 4;;Xk@l@l;;33L  #'3^ $Cy &#!{{ ,,   h!F  rZc"[R$r)r"kindrs rN_native_unique_id!ImageGeneration._native_unique_ids"'''rZcJUR[UR55$)zVGet the ImageGenerationTool for the fallback, with capability-level overrides applied.)_resolve_native_with_overridesr"rrs rN_resolved_native ImageGeneration._resolved_natives223FH^H^H`aarZchURcgSSKJn U"URUR5S9$)Nr)image_generation_tool)r native_tool)r`)pydantic_ai.common_tools.image_generationrr)rJrs rN_default_localImageGeneration._default_locals1  ' ' /S$4+G+GUYUjUjUlmmrZc>^^^UnUR=nb[XRU5S9n[[U]5n[ U[5(a0[ UR[5(aUR(dU$UR5mUR5mT(d T(dU$URmSUUU4Sjjn[X4S9$)N)r_c>URRS5n[U[5(a[UR RS[ 5;nU(aBT(a;[R"SURSSRT5S3[SS9 OHU(dAT(a:[R"[RSRT5S 9[SS9 T"X5$) Nrsupported_native_toolszEThe `ImageGeneration` native tool supersedes the direct generator on z*, so direct-only setting(s) go unapplied: rz(. Pass `native=False` to guarantee them.r=r>r@)__dict__getrrr"profilerrBrC model_namerrDrr)ctx tool_defsrnative_supersedesadd_unless_nativerrs rN*_warn_about_settings_the_chosen_path_dropsOImageGeneration.get_toolset.._warn_about_settings_the_chosen_path_dropss+.,,*:*:7*CE%'' %85==;L;L,.D<%!%MM_`e`p`p_qrCCG99[CYBZ[AA$#$ +{MM5<??--N002 002 ;N#00 5' 54H 5 C 5 5DwXXrZ)r8rtrdryrzrrrbr`rwr:rfr_rhrxrjrlrnrp)(rxzImageGenerationTool | Callable[[RunContext[AgentDepsT]], Awaitable[ImageGenerationTool | None] | ImageGenerationTool | None] | boolr_zlImageGenerator | Tool[AgentDepsT] | Callable[..., Any] | AbstractToolset[AgentDepsT] | Literal[False] | Noner`Model | KnownModelName | str | Callable[[RunContext[AgentDepsT]], Awaitable[Model | KnownModelName | str] | Model | KnownModelName | str] | Nonerbrar8r7rdrcrfrerhrgr:r9rjrirlrkrnrmrprorrrqrtrsrwrvryboolrzrvr{rrRNone)rRr)rRr.)rr.rRr)rRz,ImageGenerator | ImageGenerationModel | None)rRr)rz(Sequence[AbstractCapability[AgentDepsT]]rRzAbstractCapability[AgentDepsT])(rxzImageGenerationTool | boolr_zLiteral[False] | Noner`KnownModelName | str | Nonerbrvr8r7rdrcrfrerhrgr:r9rjrirlrkrnrmrprorrrqrtrsrwrvryrrzrvr{rrRzImageGeneration[AgentDepsT])rRz list[str])rRz tuple[dict[str, Any], list[str]])rRzdict[str, Any])rRr")rz str | boolrRz.Tool[AgentDepsT] | AbstractToolset[AgentDepsT])r5r4rRz Tool[Any])rRrF)rRz%ImageGenerationNativeTool[AgentDepsT])rRz5Tool[AgentDepsT] | AbstractToolset[AgentDepsT] | None)rRz"AbstractToolset[AgentDepsT] | None)$rSrTrUrVrWr_rXrbrwrrpropertyrrr{setterr rrr classmethodrrrrrrrrrrrrrrY __classcell__)rs@rNr]r]gs 2  s :9 ?C;B& 76@?21.-10#"98=< '& 43 (B '  BF=AFJ8<487;)-?CCG!%-1:>+#"& KAA  A"A"@#A$;%A&D'A(6)A*2+A,5-A.'/A0=1A2A3A45A6+7A889A: ;A<=A> ?ABCAL MAFj^P[w,,P[w--))"22VTT*.2'+?C+/=AFJ8<487;)-?CCG!%-1:>+#"&7;1= += % = "= = ) = ;= D= 6= 2= 5= '= == A= = +!= "8#= $ %= &'= ( )= 041= 2 %3= = ~ $:2?  @(bn=Y=YrZr])I __future__rrBcollections.abcrrr dataclassesrr functoolsr typingr r r r typing_extensionsrpydantic_ai._utilsrpydantic_ai._warningsrpydantic_ai.exceptionsrrrrpydantic_ai.imagesrrrrrpydantic_ai.images._validationrpydantic_ai.messagesrpydantic_ai.modelsrrrpydantic_ai.native_toolsrr r!r"r#pydantic_ai.toolsr$r%r&r'pydantic_ai.toolsetsr(pydantic_ai.toolsets.preparedr)_deprecated_fallback_modelr+abstractr,native_or_localr-rr.r/ frozensetrrrArr2r]rQrZrNrs"99*%88(.>eeL,CCKJ09G(.q  34'1A(BCM& 4%%%P } Y' 3} Y} YrZ