߲j\UdZddlZddlZddlZddlZddlZddlZddlmZm Z m Z da ee d<defdZ ddlmZddlmZmZmZmZmZdd lmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#dd l$m%Z%dd l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,ejZe.Z/ed d Z0da1da2ejfZ4dZ5dZ6Gdde7Z8ddde9fdZ:de;de e;effdZdZ?de e;fdZ@deAfdZBd d d!d"ZCdqde e;effd#ZDeddfd$ZEeddfd%ZFd&e;d'e;de e e;effd(ZGhd)ZHd*d*d*d+d,ZId-e;deJfd.ZKd/e;dzfd0ZLd1ede;dzfd2ZMdqd3e;d/e;dzde;fd4ZNd5eOd6eJd7e;deOfd8ZPd9ZQeddddddddf d7e;d!e;d:e eRd;e e9de eRd&e e;d?e eOd@e eJde;fdAZSdeJfdBZTde;fdCZUdDdEdFe;dGeJfdHZVdeJfdIZWddJlXmYZYmZZZdKdLdMdNdOdPdNeOe#dQedRdSd7gdTdUZ[dVe;dWe;de;fdXZ\dYe;de;fdZZ]dqde e;effd[Z^ drd7e;d!e;d&e e;d?e eOd@e eJf d\Z_d]e e;de e;fd^Z`de eAfd_Za drd7e;d!e;d&e e;d?e eOd@e eJde e;f d`ZbdadbdceAfddZcdeZddfgddDdgZede e;effdhZfdNdidPZgdjdkdPZhde e;effdlZieYjdKdme[edeWgdDdneio d]e e;deJfdpZky)savImage generation via FAL.ai (model picked in ``hermes tools``, persisted to ``image_gen.model``). ``_build_fal_payload()`` / ``_build_fal_edit_payload()`` translate unified inputs into the ``FAL_MODELS`` payload filtered to its ``supports`` whitelist so models never receive rejected keys. Clarity upscaling is strictly per-call opt-in: default-on degraded text/CJK/faces. N)AnyDictOptional fal_clientreturnc4t ddlm}|atS)zZLazily import fal_client into the module global (idempotent; keeps a test-installed mock).rimport_fal_client)rtools.fal_commonr r s @/home/ubuntu/.hermes/hermes-agent/tools/image_generation_tool.py_load_fal_clientr s6&( ) DebugSession)_ManagedFalSyncClient_extract_http_status_managed_fal_billing_error_normalize_fal_queue_url_format(submit_managed_fal_with_rate_limit_retry) DEFAULT_ASPECT_RATIO DEFAULT_MODEL FAL_MODELSUPSCALER_CREATIVITYUPSCALER_DEFAULT_PROMPTUPSCALER_FACTORUPSCALER_GUIDANCE_SCALEUPSCALER_MODELUPSCALER_NEGATIVE_PROMPTUPSCALER_NUM_INFERENCE_STEPSUPSCALER_RESEMBLANCEUPSCALER_SAFETY_CHECKERVALID_ASPECT_RATIOS)resolve_managed_tool_gateway)NOUS_MANAGED_PROVIDERfal_key_is_configuredmanaged_nous_tools_enabled%nous_tool_gateway_unavailable_messageread_selectionselection_error image_toolsIMAGE_TOOLS_DEBUG)env_varctd}|tk(r)td}|tt dtd|S|!t rytt d|dt rdStdS)uManaged gateway config for the stored `hermes tools` selection, or ``None`` for direct FAL. ``"nous"`` (or legacy ``use_gateway: true``) → managed ONLY (unreachable = selection-naming error, never a silent FAL_KEY fallback). Other stored provider → direct ONLY (missing FAL_KEY = error naming the selection). Never configured → autodetect: direct if FAL_KEY, else managed. image_genz fal-queueNzDthe Nous Tool Gateway is not available (not entitled or unreachable)zFAL_KEY is not set)r'r#r" ValueErrorr(r$)selectedgateways r _resolve_managed_fal_gatewayr17sk*H((.{; ?_2VXY Y "h@TUVV(*4Y0L[0YYrc |jjd|jf}t5t t |k7r,t t|j|ja|atcdddS#1swYyxYw)zQReuse the managed FAL client so its internal httpx.Client is not leaked per call./N)keyqueue_run_origin)gateway_originrstripnous_user_token_managed_fal_client_lock_managed_fal_client_managed_fal_client_configrr )managed_gateway client_configs r _get_managed_fal_clientr>Nsy%33::3?A`A`aM !#  &*D *U"7 "(G(G!0!?!?#A *7 &"###s AA99BceZdZdZy)ImageGenerationInterruptedz=Raised when the user interrupts while a FAL job is in flight.N)__name__ __module__ __qualname____doc__rr r@r@\sGrr@g?) poll_secondsrFc$ddlm}ggfd}tj|dd}|j |j r5|r t d|j||j r5rdrdSd S) zInterrupt-aware ``handler.get()``: the SDK blocks 30-60s, hiding user interrupts. The get runs on a daemon worker; the interrupt bit is polled between join slices and on interrupt the worker is abandoned (remote job keeps running). r)is_interruptedc jjy#t$r}j|Yd}~yd}~wwxYwN)appendget BaseException)exc error_boxhandler result_boxs r _getz_wait_fal_result.._getis= "   gkkm , "   S ! ! "s# AAATzfal-result-wait)targetdaemonnameuIImage generation interrupted by user — abandoned the in-flight FAL job.)timeoutN)tools.interruptrH threadingThreadstartis_aliver@join)rPrFrHrRworkerrOrQs` @@r _wait_fal_resultr^`s /JI"   T$=N OF LLN //   ,[] ] L ) //  l&:a=0D0rmodel argumentsc tdttji}t t j |S tfddS#t$ro}t|}|\d|cxkrdkrQnt|d }|td d |d ||d }|dvrdtddz}td d |d||d}~wwxYw)zKSubmit a FAL request using direct credentials or the managed queue gateway.zx-idempotency-keyNr`headersc>tj|S)Nrb)r>submit)rcr`r<r_s r z%_submit_fal_request..s%3ODKKGL=rz image model)whatrUiir_z*Nous Subscription gateway rejected model 'z' (HTTP z): >z managed FAL image generationT) force_freshu). This model may not yet be enabled on the Nous Portal's FAL proxy. Either: • Set FAL_KEY in your environment to use FAL.ai directly, or • Pick a different model via `hermes tools` → Image Generation.) r struuiduuid4r1rrer Exceptionrrr.r&)r_r`request_headersrNstatusbillinggateway_messager<s`` @r _submit_fal_requestrvzs3*C ,=>O24O  )_ UU7 =U, , &c*  #"5#"5 1g>G" @xPVxWZ[bZcdfkno O("(+P2,F#F`` string from config.yaml, or None.r) load_configr-NzCould not read image_gen.%s: %s) hermes_cli.configrx isinstancedictrLrnstriprqloggerdebug)r4rxcfgsectionvaluerNs r _read_image_gen_keyrsB1m*4S$*?#''+&T$.w$= C 4 eS !ekkm;;=   B 6SAA BsBB B0B++B0ctdS)z.``image_gen.model`` from config.yaml, or None.r_rrErr _read_configured_image_modelrs w ''rctdS)u}``image_gen.provider`` from config.yaml, or None (unset keeps the in-tree FAL fallback even when other providers are registered; ``"fal"`` routes via ``plugins/image_gen/fal/``). We only consult the plugin registry when this is explicitly set — an unset value keeps users on the in-tree FAL fallback even when other providers happen to be registered (e.g. a user has OPENAI_API_KEY set for other features but never asked for OpenAI image gen). ``"fal"`` explicitly routes through ``plugins/image_gen/fal/`` (which delegates back into this module's pipeline via call-time indirection — see issue #26241). providerrrErr _read_configured_image_providerrs z **rc4t}|r |dtfvry|S)zVConfigured provider that must go through the plugin registry; None for unset/fal/nous.falN)rr#) configureds r _plugin_provider_namers#02J /D'EE rctdxs$tjddj}|r%|tvrt j d|td}|xst}|t|fS)znReturn ``(model_id, meta)`` for the configured FAL model, falling back to DEFAULT_MODEL (warned) when unknown.r_FAL_IMAGE_MODELrhz4Unknown FAL model '%s' in config; falling back to %sN)rosgetenvr|rr}warningrmodel_ids r _resolve_fal_modelrsc#7+Wryy9JB/O/U/U/WHHJ.MxYfg(=H Z) ))r image_size aspect_ratio)image_size_preset gpt_literalrcTt|}|du}|r|jdxstn|d}|d} |xstj j } | | vrt} t |jdi} |xsdj | d<dh} |rG|jdxsd } | d k7rt|d n t|| | <| j| tj|d }||std |d ||r||vr| | | |<t|tr|| d <| j|xsijDcic] \}}| || c}}| jDcic]\}}||vs|| vs||c}}Scc}}wcc}}w)aText-to-image / edit payload (``image_urls`` selects edit mode): defaults + native size spec + overrides, filtered to the model whitelist. Edit endpoints mostly auto-infer size, so the size key is sent only when ``edit_supports`` lists it. ``prompt`` (and the source-image key on edits) survive a whitelist gap: every FAL endpoint requires them, so a catalog mistake can't send a broken request. N edit_supportssupportssizesdefaultsrhpromptedit_image_param image_urlsr size_stylezUnknown size_style: seed)rrLsetrlowerr|r{listadd_SIZE_KEY_BY_STYLEr.rzintupdateitems)rrrr overridesrmetaeditrraspectpayloadrequired image_paramsize_keykvs r _build_payloadrs h D T !D7;)2SUjAQH ME22 9 9 ; A A CF U%"488J#;PTQ!-AqDPQ$]]_ OTQX hAqD OOQ Os F $F F$F$c t|||||S)z?FAL text-to-image payload for ``model_id`` from unified inputs.r)rrrrrs r _build_fal_payloadrs (FL$ JJrc$t||||||S)z\FAL *edit* (image-to-image) payload: ``image_urls`` + prompt, filtered to ``edit_supports``.)rr)rrrrrrs r _build_fal_edit_payloadrs (FL$ V` aar image_urloriginal_promptc @ tjdtt|td|t t tttttd }t|}|rod|vrk|d}tjd|jdd|jd d|d |jdd |jd d d t d Stjdy#t$rt $r"}tjd|d Yd}~yd}~wwxYw)zPUpscale via FAL's Clarity Upscaler; None on failure (caller keeps the original).z(Upscaling image with Clarity Upscaler...z, ) rrupscale_factornegative_prompt creativity resemblanceguidance_scalenum_inference_stepsenable_safety_checkerr`imagez$Image upscaled successfully to %sx%swidthunknownheighturlrT)rrrupscaledrz"Upscaler returned invalid responseNzError upscaling image: %sexc_info)r}inforvrrrrrrrrr r^rLerrorr@rq)rrrPresultupes r _upscale_imagers >?%n"1H0IOK\.]-BZ->R5#?%< A>?"'* g'B KK>w 2BFF8Y4O Q%y266'1+=PXZ[I\ OE E  9: %  0!d CsCC)C))D;DD>ModalEnvironmentDockerEnvironmentSingularityEnvironment /root/.hermesz ~/.hermes)docker singularitymodalsshrc|r/t|tr|jjdrytj j |xs!t|dk\xr|ddk(xr|ddvS)N)zhttp://zhttps://zdata:F:>r3\)rzrnr startswithrpathisabslen)rs r _looks_like_absolute_file_pathr&sb  5#.%++-2J2JKk2l 77==  dCJ!O$caC$cERSHXcLcdrtask_idc ddlm}||xsdS#t$r }tj d|Yd}~yd}~wwxYw)Nr)get_active_envdefaultz1Could not inspect active terminal environment: %s)tools.terminal_tool_lifecyclerrqr}r~)rrrNs r _active_terminal_envr,s<@g233  H#Ns =8=envc|t|dd}t|r% |}|rt|jdS t|dd}|rt|jddS|jjtvryddl m }|d xsd jj}tj|S#t$r }t j d|Yd}~d}~wwxYw) Nagent_visible_cache_baser3z.active env agent_visible_cache_base failed: %s _remote_homez/.hermesrr terminal_env TERMINAL_ENVlocal)getattrcallablernr7rqr}r~ __class__rA_CONTAINER_HOME_ENVStools.terminal_scoperr|r_CACHE_BASE_BY_BACKENDrL)rexplicitrrN remote_homerbackends r _agent_cache_base_for_envr5s 3 :DA H  T u:,,S11c>48 +&--c238< < == ! !%9 9"1N+6w==?EEGG ! % %g .. T MsSS Ts"C C5C00C5rawc t|trtj|n|}t|t r|j ds|S|j d}t|tr t|s|St|}t|}|s|S ddl m }|||}|r||k(r|St|dd} |  | jd |j#d ||j#d |tj$|dS#t$r|cYSwxYw#t$r"}tjd||cYd}~Sd}~wwxYw#t$r }tj!d |Yd}~d}~wwxYw)zAnnotate successful local results: ``image`` stays the host/gateway-deliverable path; ``agent_visible_image`` is the same file as seen by a non-local terminal backend.successrr)map_cache_path_to_container)container_basez4Could not translate image cache path for backend: %sN _sync_managerTforcez1Could not force-sync generated image artifact: %s host_imageagent_visible_imageF) ensure_ascii)rzrnjsonloadsrqr{rLrrrtools.credential_filesrr}r~rsyncr setdefaultdumps) rrrrr cache_baser agent_pathrN sync_managers r "_postprocess_image_generate_resultrJsa%/S%9$**S/s gt $GKK ,B KK E eS !)G)N w 'C*3/J  F0zR  u, 36L U   D  ) |U+ ,j9 ::gE 227    KSQ  U NNNPS T T UsG'D D>E DD E%E<EE E3E..E3imagesshould_upscalerc Dg}|D]}t|trd|vs|rFt|d|j}|r|j |Kt j d|j |d|jdd|jdddd|S)zfNormalize FAL result images, optionally chaining the upscaler (falls back to the original on failure).rz1Using original image as fallback (upscale failed)rrrF)rrrr)rzr{rr|rKr}rrL)rrr formattedimgrs r _format_imagesrnsI  3%%3,  %c%j&,,.AH  * NNN Ou:(;swwxYZG[    rc t|tr|js tdt st stt |jd|jd|}}|r|std|d|d|xstjj} | tvr!tjd|tt} |rdt|jdxsd } | d kDr|d | n|} t||| | || } tjd ||t!| |d d|| fSt#||| || } tjd|||d d|| fS)zeValidate inputs and return ``(endpoint, arguments)``; raises ValueError with the user-facing message.z1Prompt is required and must be a non-empty string edit_endpointdisplayzModel 'z' (u) is not capable of image-to-image / editing. Provide a text-only prompt (omit image_url), or switch to an edit-capable model via `hermes tools` → Image Generation.z-Invalid aspect_ratio '%s', defaulting to '%s'max_reference_imagesrrN)rru=Editing image with %s (%s) — %d source image(s), prompt: %sPu,Generating image with %s (%s) — prompt: %s)rzrnr|r.r$r1_build_no_backend_setup_messagerLrrr!r}rrrrrr) rrrrrr source_imagesrr aspect_lcmax_refsclamped_sourcesr`s r _prepare_fal_requestr$st fc "&,,.LMM " #'C'E8:;;!XXo6H8U7M]gYc(,7 89 95!5<<>DDFI++F Vjk( txx 67<1=6>l- 2 + foytyZ  S]C,@&"+ Oi''"8VYTU^_I KK>SYZ][]S^_ Y rrr num_images output_formatrreference_image_urlsupscalec & t\} } t|ttfr|ng} |g| D cgc]4} t| ts| j s%| j 6}} t |xrt | jd}|rdnd}||||d}| ||d|||t|dddd d d  tjj}d td tttfd tf fd } t| | ||||jDcic] \}}| || c}}|\}}t!t#||}tjj|z j%}|rd|vr t'd|jdg}|s t'd| t | }n t | jddxr| }t)|||}|s t'dt+d|D}t,j/dt|||||d d<t| d<||d|d d|t |d jddScc} wcc}}w#t0$r}dt |}t,j3d|d| d <tjj|z j%}||ddt |t5|j6d!cYd}~Sd}~wwxYw)"a;Generate (or, with source images + an ``edit_endpoint`` model, edit) an image via FAL. Extra kwargs are overrides filtered per-model via ``supports`` / ``edit_supports`` (dropped silently so callers survive model switches). Returns JSON ``{"success", "image", "modality", "error", "error_type"}``. rrtext)rrr%r&rr)rmodalityr NFr)r_ parametersrrimages_generatedgeneration_timer/responserc|d<tjdtjtj|ddS)Nr/image_generate_toolrF)indentr )_debuglog_callsaver r)r/r0debug_call_datas r finishz#image_generate_tool..finishs9-<)*-? zz(15AArrru7Invalid response from FAL.ai API — no images returnedzNo images were generatedr(z%No valid image URLs returned from APIc3DK|]}|jdsdyw)rrN)rL).0rs r z&image_generate_tool..sR3cggj>QQRs  z8Generated %s image(s) in %.1fs (%s upscaled) via %s [%s]Trr.rr)rrr,rzError generating image: z%srrrrr error_type)rrzrtuplernr|boolrLrdatetimenowfloatrrr$rr^rv total_secondsr.rsumr}rrqrtyperA)!rrrrr%r&rrr'r(rrrefscr use_editr,r start_timer8rrendpointr`rr/rrformatted_imagesupscaled_countr error_msgr7s! @r r2r2s7()NHd#-.BT5M#R XZD)2(:T(:_1jC>PUVU\U\U^QWWY_M_M"FtDHH_,E'FH"wH2n =!BI!'RR\`#+c->PR%QST VO ""&&(JBBc3hBCB $j2 dFL$'oo/ Ada1=QT A=R)""5h)"TU#,,002Z?NNP/VW WHb)78 8  !']N!$((9e"<=Nh,N)&.&IDE ER*:RR N()?NHV^ `%) ".12B.C*+o%a(/ -a044Z@A (CD DW`* B6 j.s1vh7  T9t 4#, #,,002Z?NNPo"'$QW[\]W^WgWghj j jsII.I.I.I9 I3)I3.D?I93I99 LBL L Lcd t}t|xs tS#t$rYywxYw)zTrue if the selected FAL backend (never configured: any FAL backend) is available. A stored-but-broken selection reports False here (registry gating); the naming error surfaces at call time from ``_resolve_managed_fal_gateway``. F)r1r.r?r$)r0s r check_fal_api_keyrOs8 .0 = 3133 s # //c"t}gd}|r|jdn2|jdtdx}r|jd||gdz }|r|jd|jdd j|S) zXActionable no-backend error: FAL_KEY signup, managed-gateway status, plugin alternative.)z4Image generation is unavailable in this environment.rhzMissing requirements:zA - FAL_KEY is not set and the managed FAL gateway is unreachablez+ - FAL_KEY environment variable is not setrlz - )rhz&To enable image generation, do one of:z_ 1. Get a free API key at https://fal.ai and set FAL_KEY= (then restart the session)zX 2. Sign in to a Nous account that has the managed FAL gateway enabled (`hermes setup`)u 3. Configure a different image_gen provider via `hermes tools` → Image Generation (run `hermes plugins list` to see installed backends) )r%rKr&r\)managedlinesrus r rrs(*G aE XY BCCDbc c? c LL401 2 ,,E ( ) LLIJ 99U rFrrUrcPddlm}ddlm}|r |dn|||S)zpDiscover plugins (local import: importing this module must not trigger discovery) and return the named provider.r) get_provider)_ensure_plugins_discoveredTr)agent.image_gen_registryrUhermes_cli.pluginsrV)rUrrUrVs r _get_plugin_providerrY s$5= "."$  rc tr ty t}|yt |}t |xr|j S#t$rY@wxYw)zDTrue if FAL or the explicitly configured image backend is available.TF)rOr ImportErrorrrYr? is_available)rrs r #check_image_generation_requirementsr]se      '(J#J/H 4X224 55    sA AA)registry tool_errorimage_generatezoGenerate images from text prompts. The active model's edit/reference capabilities are rendered at serving time.objectstringzThe text prompt describing the desired image (text-to-image) or the edit to apply (image-to-image). Be detailed and descriptive.)rE descriptionzlThe aspect ratio of the generated image. 'landscape' is 16:9 wide, 'portrait' is 16:9 tall, 'square' is 1:1.)rEenumrcrr+rE propertiesr)rUrcr-rr=c6tjdd||dS)zCJSON error envelope shared by every provider-dispatch failure path.FNr<)r r)rr=s r _provider_errorrhMs ::%$V`a bbrcontract_errorcdt|ts t|dStj|S)zLJSON-encode a provider's dict result; anything else is a contract violation.provider_contract)rzr{rhr r)rris r _provider_resultrlRs) fd #~/BCC ::f rc|r||d<t|tr#|jr|j|d<|ddlm}||}|r||d<|t ||d<|S)z^Add the optional ``provider.generate(**kwargs)`` args in place (edit args only when supplied).r_rr)normalize_reference_imagesr'r()rzrnr|agent.image_gen_providerrnr?)kwargsrr'r(r_rn norm_refss r _add_provider_kwargsrrYsm w)S!ioo&7'oo/{'G./CD -6F) * My Mrct}|y t|}| t|d}|t d|ddSt |d d }||d } t| |||t |jdi| } t| dS#t$r }tj d|Yd}~yd}~wwxYw#t$r }tj d|Yd}~d}~wwxYw#t$r~}t|t} | r4d | vsd| vr,tjd||t d|ddcYd}~Stjd|| rdnd|t d|d|dcYd}~Sd}~wwxYw)zJSON result from the selected plugin provider, or ``None`` to fall through to in-tree FAL (provider unset / ``"fal"`` / ``"nous"``). Providers without ``upscale`` ignore it via ``**kwargs``.Nz%image_gen plugin dispatch skipped: %sTrz*image_gen plugin force-refresh skipped: %szimage_gen.provider='zk' is set but no plugin registered that name. Run `hermes plugins list` to see available image gen backends.provider_not_registeredrU?r+)r_rr'zQimage_gen provider '%s' rejected image-to-image kwargs (signature too narrow): %sz Provider 'u' does not support image-to-image / editing (its generate() signature is out of date with the image_generate schema). Omit image_url for text-to-image, or pick a backend that supports editing via `hermes tools` → Image Generation.modality_unsupportedz$Image gen provider '%s' raised%s: %sz TypeErrorrhz ' error: provider_exceptionz#Provider returned a non-dict resultrE)rrYrqr}r~rhrrrrgeneraterz TypeErrorrrl) rrrr'r(rrrNpnamerpr is_type_errors r _dispatch_to_plugin_providerr|is '(J' 3 L+JdCH":,/M NOhj j Hfc *E(. MFYVY0Dg#?#A C""",V, F$I JJG   E"UG$$%'= > > =u'4|"c CE7)C5ACWXXYsY B B1*C B.B))B.1 C:CC E$&AE&E$,-EE$E$rcbddlm}t|tr|j nd}||vr|SdS)zQReturn ``model_id`` when it is one of the Krea plugin's model ids, else ``None``.r)KREA_MODEL_IDSN)plugins.image_gen.krear~rzrnr|)rr~ candidates r _normalize_krea_modelrs.5$.x$= 4I!^39==rcddlm}m}m}t }| |t k7ryt }||}||k(rd|fS||k(r |t k(rd|fSy)u``(plugin_name, model_id)`` when the managed selection stores a Krea or Portal model, else ``None``. The managed row writes ``provider: nous`` for three gateways; the model id says which. FAL models (and an unset model) return ``None`` so the in-tree FAL path handles them. Only the ``nous``/unset selection qualifies — a direct/BYO provider pick dispatches normally. r)KREAPORTALmanaged_backend_for_modelNkreanous)tools.image_generation_managedrrrrr#r)rrrconfigured_providerrrs r _managed_model_pluginrsiWV9;&+>BW+W+-H'1G$x&04IIx rct}|y|\}} |dk(rddlm}|yt|} | |dk(rytd|ddS|||d } t| |||| jdi| } t| | jdS#t$r#} t j d|| d} Yd} ~ d} ~ wwxYw#t$r=} t jd || td | jd | d cYd} ~ Sd} ~ wwxYw)u1JSON result from the Krea or Portal gateway the stored model belongs to, or ``None`` to fall through to FAL. A Krea model with no reachable Krea gateway falls through (direct/BYO users keep their pipeline); a Portal model never does — falling through would silently bill a FAL default. Nrr)_resolve_managed_krea_gatewayz"Managed %s routing unavailable: %szimage_gen.model='u' is a Nous Portal model but the Portal image backend is not available. Pick another model via `hermes tools` → Image Generation.rt)rrr_zManaged %s routing failed: %szManaged z generation error: rwz$ provider returned a non-dict resultrE) rrrrYrqr}r~rhrrrxr display_namerl) rrrr'r(rS plugin_namerrrrNrprs r _maybe_route_managed_modelrsD# $F ~"K & L,.6' 4 & z*U VWpr r)/ W_`FqVY0DgN""",V, Fx'<'<&==a$b cc!  9;L q6 SI(*?*?)@@STWSXY[oppqs:B B  B9 B6B11B69 C?2C:4C?:C?)r permittedrc  ddlm}|dxsdjjdvr||dfSddlm ddlm}m   fd } t|tr|jr||}t|ttfr'|Dcgc]}t|tr||n|}}||dfScc}w#|$r/}||td |t|jfcYd}~Sd}~wwxYw) arResolve path-like sources to ``data:`` URLs under a non-local terminal backend. Routes through ``tools.image_source`` (in-sandbox exec-read, media-cache host reads, credential guard) so generation obeys the same confinement as vision. URLs/data: pass through; local backend is a no-op. Returns ``(image_url, reference_image_urls, error_json_or_None)``. rrrr)rhrN) _run_async)ImageResolutionError resolve_local_source_to_data_urlc&|S)NrrE)refrrrrs r resolvez'_confine_source_images..resolves:3S\]^^rzCould not read source image: )rrr|r model_toolsrtools.image_sourcerrrzrnrr>rhrErA) rr'rrrrrrrNrrs `` @@r _confine_source_imagesrs2^$/668>>@MQ.44&Y_G i %)//*; *I *T5M :Ui#jPQ*Q2DGAJ!$K#j #j *D 00 $k G.+C5 1493E3E1GG GGs1AC!B;4C;CC4$C/)C4/C4c |jdd}|s tdS|jdt}|jd}|jd}t|jd|jd|\}}}||St ||t |t r|nd } d} tttfD]} | ||fi| } | nt| | S) Nrrhz'prompt is required for image generationrr(rrr')rr'r()r) rLr_rrr{rzr?r|rr2r) argskwrrr(rrr' confine_errorsourcesrroutes r _handle_image_generaters XXh #F CDD88N,@ALhhy!GffYG5K txx(>?6J2I#] Y=Q&0$&?7TKG C.0JL_`FL4G4 ?  .c7 CCrr*) modalitiesrsupports_upscalectt}t}t}||d}n |r|dk7r|}nd}|r t |}| |j xsi}|j|d<txs|jxsd|d<|jdrt|d|d<|jdrt|d|d<t|jd |d <|S t\}}t|jd }d |d<|jd ||d<|rd dgnd g|d<|rt|jdxsdnd|d<d|d <|S#t $ri}YwxYw#t $rYwxYw)a;Best-effort capabilities of the active backend/model; never raises. Mirrors runtime dispatch: a Krea or Portal model id under the managed selection asks that plugin, a set ``image_gen.provider`` asks that plugin, else the FAL catalog. Fail-closed: an undeclared capability is advertised as absent. Nrrrrhr_rrrrzFAL.airr*rrT)r{_NO_CAPABILITIESrrrY capabilitiesrqrr default_modelrLrrr?r) rrrRrrcapsrrcan_edits r _active_image_capabilitiesrs 01D9;#%Gaj !4!=)   +K8H##0028bD$,#8#8Z < > b8CYCYC[Ca_aW 88L))-d<.@)AD&882336t E,BE>, E;7E>:E;;E>> F  F zSource image to edit/transform (image-to-image). A public URL or an absolute local file path from the conversation. Omit for text-to-image.booleanuPost-generation high-resolution pass (~2x, extra cost/latency), off by default. A creative enhancer that can alter fine detail (rendered text, faces) — use only when resolution matters more than fidelity.cxd}t}t|jdxsd}dt|jdxsdgv}tdd}|d |d d }|r!d }t |d <|dkDrdddi|d|dd|d<nd}|jdr t |d<|j|d|d gddS)zRender description AND params from the active model's capabilities; args it cannot honor are NOT advertised (the handler still accepts them for replay compat).uGenerate high-quality images from text prompts{edit_clause}. Returns the result in the `image` field — a URL or an absolute file path; reference it in your response using the current platform's file-delivery convention.rrrrr*r-rfrrr+z<, or edit / transform an existing image by passing image_urlrrarrayrErbzUp to zn additional reference images (style, character, or composition) guiding an edit. URLs or absolute local paths.)rErmaxItemsrcr'uF (text-to-image only — the active model cannot edit existing images)rr() edit_clauserare)rcr-)rrrLrIMAGE_GENERATE_SCHEMA_IMAGE_URL_PARAM_UPSCALE_PARAMformat) base_descrr"r static_propsrfrs r _build_dynamic_image_schemarXs / & 'D488238q9H#dhh|4@AAH(6|DLx(,~:V"XJT "2 ; a< (+$XJ',, 2J- ._  xx"# . 9$++ +D#+:T\S]^ ``rr-u🎨) rUtoolsetschemarPcheck_fn requires_envis_asyncemojidynamic_schema_overridesct|duS)zATrue when ``model_id`` is a native Krea plugin id (``krea-2-*``).N)rrs r is_krea_modelrs  *$ 66rrJ)NNN)lrDr loggingrr@rXrotypingrrrr__annotations__r tools.debug_helpersrr rrrrrtools.image_generation_catalogrrrrrrrrrrrr r!tools.managed_tool_gatewayr"tools.tool_backend_helpersr#r$r%r&r'r( getLoggerrAr}r4r:r;Lockr9r1r>rqr@rBr^rnrvrrrrr>rrrrrrrrr?rrrrrrr$rr2rOrrYr]tools.registryr^r_rrhrlrrr|rrrrrrrrrrregisterrrErr rsw   && C#- DLL   8 $ m-@ A!)9>>+Z. #HH8;1u14stCH~F S Xc] ( +x}*E*,8 &46PX\]`be]eXfPB7KQUaeK H\!%bcCHT#s(^c]Dj 3- Dj,4C= Dj #4. Dj ;C4. Dj UX DjN 44 4*6;sd6T6$0   5!'!01 N/  $J) Hc3cCcCc SS Z^_bdg_gZh "@DKO-K -K"-K/7}-K"4.-K:B4.-K`>HSM>hsm>x,@DKO#d #d"#d/7}#d"4.#d:B4.#dT\]`Ta#dL\f1SX14D6$*(A[`a,DcN,b    "`T#s(^"`J ;7L "-P_a  +F 7HSM7d7r