(Gj UdZddlZddlZddlZddlZddlZddlZddlmZm Z m Z da ee d<defdZ ddlmZddlmZmZmZdd lmZdd lmZmZmZmZejeZd d d ddddddddddhdddhddd dddd dddddd!d"d#ddd$d%d&hd'd%d(hd)dd d*d+d,d-dddddd.d#dddd/hd0dd1 d2d3d4d5d6d7d8d9dd#dd$d:d;hd<dd=hd>d?d d@dAdBdCdDdEdFdGddHd#ddIhdJddKhdLdMd dNdOdPdQddRddSddHd#ddIhdTddUhdVdMd dWdXdYdZdddddd[d%d\d]hd^dd_hd`d#d dad3dbdcddddddddihdedd1 dfdgdhdiddddddjdkd#ddldmhdnddohdpdqd drdsdtdud6d7d8d9ddvdHihdwdd1 dxdydzd{d6d7d8d9ddvdHihdwdd1 d| Ze ee eeffe d}<d~Z dZ!dZ"dZ#d?Z$dZ%dZ&dZ'dZ(dZ)dZ*dZ+eddZ,da-da.ej/Z0dZ1dZ2dede eeffdZ3de4fdZ5e!ddfdeded6ede e6de e eefde eeff dZ7e!ddfdedede8d6ede e6de e eefde eeffdZ9dedede e eeffdZ:dede;fdZdedededzfdZ?deddfdZ@ddededzdefdZAe!dddddddfded6ede e6de eBde e6de ede e6de ede e8defdZCde;fdZDdefdZEde;fdZFedkreGdeGdeDs,eGdeGdeGdeHd#eGd ddl a eGdn#eI$reGdeHd#wxYwe5\ZJZKeGdeKLdeJdeJdeGdeKLdddeKLddeGdeKLdrdnd›eGdæeMD]O\ZNZOeNeJkrdndZPeGdeNdǛdeOLdddțdeOLddePPe,jQreGde,jRddlSmTZTmUZUddddddМde8e"de!dҜdddМdddiddלd؜dgdٜdڜZVdۄZWd܄ZX dded6ede ede e8fd݄ZYhdޣZZde ede efd߄Z[de ede;fdZ\ dded6ede ede e8de ef dZ]dZ^eVdZ_de eeffdZ`de eeffdZaeTjbddeVe^eFgddea dS)u Image Generation Tools Module Provides image generation via FAL.ai. Multiple FAL models are supported and selectable via ``hermes tools`` → Image Generation; the active model is persisted to ``image_gen.model`` in ``config.yaml``. Architecture: - ``FAL_MODELS`` is a catalog of supported models with per-model metadata (size-style family, defaults, ``supports`` whitelist, upscaler flag). - ``_build_fal_payload()`` translates the agent's unified inputs (prompt + aspect_ratio) into the model-specific payload and filters to the ``supports`` whitelist so models never receive rejected keys. - Upscaling via FAL's Clarity Upscaler is gated per-model via the ``upscale`` flag — on for FLUX 2 Pro (backward-compat), off for all faster/newer models where upscaling would either hurt latency or add marginal quality. Pricing shown in UI strings is as-of the initial commit; we accept drift and update when it's noticed. N)AnyDictOptional fal_clientreturncLttSddlm}|atS)u'Lazily import fal_client and rebind the module global on first use. Idempotent. Returns the (now-loaded) ``fal_client`` module reference. Skips the import if the global is already truthy — this preserves the test pattern of monkeypatching the module global to install a mock. Nrimport_fal_client)rtools.fal_commonr r s A/home/rurouni/.hermes/hermes-agent/tools/image_generation_tool.py_load_fal_clientr ,s6222222""$$J ) DebugSession)_ManagedFalSyncClient_extract_http_status_normalize_fal_queue_url_format)resolve_managed_tool_gateway)fal_key_is_configuredmanaged_nous_tools_enabled%nous_tool_gateway_unavailable_messageprefers_gatewayzFLUX 2 Klein 9Bz<1szFast, crisp textz $0.006/MPimage_size_presetlandscape_16_9 square_hd portrait_16_9) landscapesquareportraitpngF)num_inference_steps output_formatenable_safety_checker>seedprompt image_sizer"r!r#zfal-ai/flux-2/klein/9b/edit>r$r% image_urlsr"r!r# ) displayspeed strengthsprice size_stylesizesdefaultssupportsupscale edit_endpoint edit_supportsmax_reference_imagesz FLUX 2 Proz~6szStudio photorealismz$0.03/MP2g@5T)r!guidance_scale num_imagesr"r#safety_tolerance sync_mode> r$r%r;r&r9r"r8r:r!r#zfal-ai/flux-2-pro/edit> r$r%r;r'r9r"r8r:r!r#z Z-Image Turboz~2szBilingual EN/CN, 6Bz $0.005/MP)r!r9r"r#enable_prompt_expansion>r$r%r&r9r"r!r#r=) r)r*r+r,r-r.r/r0r1z$Nano Banana Pro (Gemini 3 Pro Image)z~8sz-Gemini 3 Pro, reasoning depth, text renderingz$0.15/image (1K) aspect_ratioz16:9z1:1z9:161K)r9r"r: resolution> r$r%r;r9r@r>r"r:enable_web_searchlimit_generationszfal-ai/nano-banana-pro/edit> r$r%r;r'r9r@r>r"r:rArBz GPT Image 1.5z~15szPrompt adherencez $0.034/image gpt_literal 1536x1024 1024x1024 1024x1536medium)qualityr9r">r%rIr; backgroundr&r9r"zfal-ai/gpt-image-1.5/edit>r%rIr;r&r'r9r"z GPT Image 2z~20sz3SOTA text rendering + CJK, world-aware photorealismu$0.04–0.06/image landscape_4_3 portrait_4_3>r%rIr;r&r9r"zopenai/gpt-image-2/edit>r%rIr;r'r9r"mask_image_urlz Ideogram V3z~5szBest typographyz$0.03-0.09/imageBALANCEDAUTO)rendering_speed expand_promptstyle>r$rSr%r&rRrQzfal-ai/ideogram/v3/edit>r$rSr%r'rRrQzRecraft V4 Proz'Design, brand systems, production-readyz $0.25/imager#>colorsr%r&background_colorr#z Qwen Imagez~12szLLM-based, complex textz$0.02/MPg@regular)r!r8r9r" acceleration> r$r%r;r&r9rXr"r8r!zfal-ai/qwen-image-2/pro/edit> r$r%r;r'r9rXr"r8r!z Krea 2 Mediumz~15-25sz9Illustration, anime, painting, expressive/artistic stylesz#$0.030 (text) / $0.035 (style refs) creativity>r$r%rZr>image_style_referencesz Krea 2 Largez~25-60sz;Photorealism, raw textured looks (motion blur, grain, film)z#$0.060 (text) / $0.065 (style refs)) fal-ai/flux-2/klein/9bzfal-ai/flux-2-prozfal-ai/z-image/turbozfal-ai/nano-banana-prozfal-ai/gpt-image-1.5zfal-ai/gpt-image-2zfal-ai/ideogram/v3z#fal-ai/recraft/v4/pro/text-to-imagezfal-ai/qwen-imagez#fal-ai/krea/v2/medium/text-to-imagez"fal-ai/krea/v2/large/text-to-image FAL_MODELSr\rzfal-ai/clarity-upscalerz"masterpiece, best quality, highresz.(worst quality, low quality, normal quality:2)gffffff?g333333? image_toolsIMAGE_TOOLS_DEBUG)env_varc^trtdsdStdS)zsReturn managed fal-queue gateway config when the user prefers the gateway or direct FAL credentials are absent. image_genNz fal-queue)rrrrr _resolve_managed_fal_gatewayres4{'C'Ct ' 4 44rc@|jd|jf}t5tt |krtcdddSt tt|j|ja|atcdddS#1swxYwYdS)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_configr rr)managed_gateway client_configs r _get_managed_fal_clientrrs  &--c22'M " # #  */I]/Z/Z& # # # # # # # # 3 /,;    &3"" # # # # # # # # # # # # # # # # # #sB7BBBmodel argumentsc tdttji}t }|t j|||St|} ||||S#t$rZ}t|}|Cd|cxkrdkr6nn3d}|dvrdtd d z}td |d |d||d}~wwxYw)zKSubmit a FAL request using direct credentials or the managed queue gateway.zx-idempotency-keyN)rtheadersii>z managed FAL image generationT) force_freshz*Nous Subscription gateway rejected model 'z' (HTTP u). 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 struuiduuid4rersubmitrr Exceptionrr ValueError)rsrtrequest_headersrpmanaged_clientexcstatusgateway_messages r _submit_fal_requestrs`*C ,=,=>O244O )_UUUU,_==N$$ #%    &c**  #"5"5"5"5#"5"5"5"5"5 O((;6$( %U%%%% # %%  1s(B C$ ACC$cd} ddlm}|}t|tr|dnd}t|tr>|d}t|t r|}n2#t$r%}t d|Yd}~nd}~wwxYw|s'tj dd}|stttfS|tvr:td |ttttfS|t|fS) zResolve the active FAL model from config.yaml (primary) or default. Returns (model_id, metadata_dict). Falls back to DEFAULT_MODEL if the configured model is unknown (logged as a warning). rwr load_configrcNrsz.Could not load image_gen.model from config: %sFAL_IMAGE_MODELz4Unknown FAL model '%s' in config; falling back to %s)hermes_cli.configr isinstancedictgetr}striprloggerdebugosgetenv DEFAULT_MODELr]warning)model_idrcfgimg_cfgrawrs r _resolve_fal_modelrsf H L111111kmm*4S$*?*?I#''+&&&T gt $ $ '++g&&C#s## '99;; LLL EsKKKKKKKKL <9.3399;; 8j777z!! B m   j777 Z) ))sBB CB>>Crr%r$ overridesc\ t|}|d}|d}|pt}||vrt}t |di} |pd| d<|dvr ||| d<n$|dkr ||| d<nt d ||t|tr|| d <|r!| D] \} } | | | | < |d  fd | DS)a)Build a FAL request payload for `model_id` from unified inputs. Translates aspect_ratio into the model's native size spec (preset enum, aspect-ratio enum, or GPT literal string), merges model defaults, applies caller overrides, then filters to the model's ``supports`` whitelist. r-r.r/rwr%rDrr&r>zUnknown size_style: Nr$r0c0i|]\}}|vs|dk||S)r%rd).0kvr0s r z&_build_fal_payload..`s8   A ==AMM 1)MMr) r]DEFAULT_ASPECT_RATIOlowerrrrrrintitems) rr%r>r$rmetar-r.aspectpayloadrrr0s @r _build_fal_payloadr5si h Dl#J ME22 9 9 ; ; A A C CF U%"488J#;#;< >>??? JtS11OO%%  DAq} JH        rr'c t|}|dp t|d}|d}|pt} | |vrt} t |di} |pd| d<t|| d<|dvrd vr || | d <n|d krd vr || | d <|t|tr|| d <|r!| D] \} } | | | | < ddh fd | DS)aBuild a FAL *edit* request payload (image-to-image) from unified inputs. Every FAL edit endpoint takes ``image_urls`` (a list of source/reference image URLs) plus the prompt. Size handling differs from text-to-image: most edit endpoints auto-infer output dimensions from the input image, so we only send ``image_size`` / ``aspect_ratio`` when the edit endpoint's ``edit_supports`` whitelist accepts it. Keys outside ``edit_supports`` are stripped before submission. r3r-r.r/rwr%r'rr&r>Nr$c,i|]\}}|vs|v ||Srdrd)rrr _requiredr3s r rz+_build_fal_edit_payload..s:   A  i 1!/r) r]rsetrrrrlistrrr)rr%r'r>r$rrr-r.rrrrrr3s @@r _build_fal_edit_payloadrfs" h DHH_--6Ml#J ME22 9 9 ; ; A A C CF U%"488J#;#;<[>[ %f   ~ % %.M*I*I"'- JtS11OO%%  DAq} <(I         r image_urloriginal_promptc  td|td|ttt t tttd }tt|}| }|rd|vr|d}td| dd| d d|d | dd | d d d td St ddS#t$r(}t d|d Yd}~dSd}~wwxYw)zUpscale an image using FAL.ai's Clarity Upscaler. Returns upscaled image dict, or None on failure (caller falls back to the original image). z(Upscaling image with Clarity Upscaler...z, ) rr%upscale_factornegative_promptrZ resemblancer8r!r#rtimagez$Image upscaled successfully to %sx%swidthunknownheighturlrT)rrrupscaledrz"Upscaler returned invalid responseNzError upscaling image: %sexc_info)rinfoUPSCALER_DEFAULT_PROMPTUPSCALER_FACTORUPSCALER_NEGATIVE_PROMPTUPSCALER_CREATIVITYUPSCALER_RESEMBLANCEUPSCALER_GUIDANCE_SCALEUPSCALER_NUM_INFERENCE_STEPSUPSCALER_SAFETY_CHECKERrUPSCALER_MODELrerrorr)rrupscaler_argumentshandlerresultupscaled_imagees r _upscale_imagersr % >???#0EEOEE-7-/5#?%<   &n@RSSS  g''#G_N KK6""7I66""8Y77    &e,'++GQ77(,,Xq99 "1    9:::t  0!d CCCtttttsD D( D(( E2EEvaluec|rt|tsdS|}|drdStj|rdSt|dko|ddko |ddvS) NF)zhttp://zhttps://zdata:TrYr6:rC>rg\)rr}r startswithrpathisabslen)rrs r _looks_like_absolute_file_pathrs  5#..u KKMME 899u w}}Ut u::? JuQx3 J58{3JJrtask_idc ddlm}||pdS#t$r&}td|Yd}~dSd}~wwxYw)Nr)get_active_envdefaultz1Could not inspect active terminal environment: %s)tools.terminal_toolrrrr)rrrs r _active_terminal_envrso666666~g2333  H#NNNttttts AAAenvc2|t|dd}t|rb |}|r"t|dSn2#t$r%}t d|Yd}~nd}~wwxYwt|dd}|r%t|ddS|jj}|dvrdStj dpd }|d vrdS|d krd SdS) Nagent_visible_cache_basergz.active env agent_visible_cache_base failed: %s _remote_homez/.hermes>ModalEnvironmentDockerEnvironmentSingularityEnvironmentz /root/.hermes TERMINAL_ENVlocal>modaldocker singularitysshz ~/.hermes) getattrcallabler}rkrrr __class____name__rrrr)rexplicitrr remote_homeenv_namebackends r _agent_cache_base_for_envrsZ  3 :DAA H   T T  2u::,,S1112 T T T MsSSSSSSSS Tc>488  =+&&--c22<<< <=) Z Z Z"? y((3G::<<BBDDG444%{ 4s-A BA==B host_pathct|sdSt|}|sdS ddlm}|||S#t$r%}t d|Yd}~nd}~wwxYwdS)Nr)map_cache_path_to_container)container_basez4Could not translate image cache path for backend: %s)rrtools.credential_filesrrrr)rr cache_baserrs r _agent_visible_cache_pathr s )) 4 4t*3//J tRFFFFFF**9ZPPPP RRR KSQQQQQQQQR 4s9 A(A##A(ct|dd}|dS |ddS#t$r&}td|Yd}~dSd}~wwxYw)N _sync_managerTforcez1Could not force-sync generated image artifact: %s)rsyncrrr)r sync_managerrs r _force_artifact_syncrs366LQ%%%%% QQQJCPPPPPPPPPQs/ AAArcF t|trtj|n|}n#t$r|cYSwxYwt|t r|ds|S|d}t|trt|s|St|}t||}|r||kr|S|t|| d|| d|tj |dS)aAnnotate successful local image results with backend-visible paths. ``image`` remains the host/gateway-deliverable path. When the active terminal backend has a different filesystem, ``agent_visible_image`` gives the path the agent can use with terminal/file tools. successrN host_imageagent_visible_imageF) ensure_ascii) rr}jsonloadsrrrrrrr setdefaultdumps)rrrrr agent_paths r "_postprocess_image_generate_resultr&s9%/S%9%9B$*S///s   gt $ $GKK ,B,B KK E eS ! !)G)N)N w ' 'C*5#66J u,,  S!!! |U+++ ,j999 :gE 2 2 22s +. ==r!r8r9r"reference_image_urlsc V t\} } g} t|tr;|r'| |t|t t frU|D]R} t| tr;| r'| | S| d} t| ot| }|rdnd}| ||||||||t| d ddddd}tj } |r:t|tr%t|dkrtd ts)tstt| r,| s*td | d | d | d |pt }|t$vr(t&d|t t }i}|||d<|||d<|||d<|||d<|rt+| dpd}|dkr | d|n| }t-| |||||}| }t&d| d | |t||ddnOt1| ||||}| }t&d| d | | |ddt3||}|}tj |z }|rd|vrtd|dg}|stdt| ddo| }g}|D]}t|t6rd|vs|d|dd|d dd!}|rZt9|d|} | r|| t&d"d|d#<|||std$t;d%|D}!t&d&t|||!||d'|r|ddnd|d(}"d'|d)<t||d*<||d+<t<d,|t< tCj"|"d-d.S#tF$r}#tj |z }d/t|#}$t&$d0|$d'1ddt|#tK|#j&d2}"|$|d3<||d+<t<d,|t< tCj"|"d-d.cYd}#~#Sd}#~#wwxYw)4a-Generate an image from a text prompt, or edit a source image, via FAL. Routing: when ``image_url`` (or ``reference_image_urls``) is provided AND the configured model declares an ``edit_endpoint``, the call routes to that image-to-image / edit endpoint; otherwise it's plain text-to-image. The agent-facing schema exposes ``prompt``, ``aspect_ratio``, ``image_url`` and ``reference_image_urls``; the remaining kwargs are overrides for direct Python callers and are filtered per-model via the ``supports`` / ``edit_supports`` whitelist (unsupported overrides are silently dropped so legacy callers don't break when switching models). Returns a JSON string with ``{"success": bool, "image": url | None, "modality": "text" | "image", "error": str, "error_type": str}``. r2rtext) r%r>r!r8r9r"r$modality source_imagesNFr)rs parametersrrimages_generatedgeneration_timez1Prompt is required and must be a non-empty stringzModel 'r)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'r!r8r9r"r4r6)r$ru=Editing image with %s (%s) — %d source image(s), prompt: %sPu,Generating image with %s (%s) — prompt: %srimagesu7Invalid response from FAL.ai API — no images returnedzNo images were generatedr1rrr)rrrz1Using original image as fallback (upscale failed)rz%No valid image URLs returned from APIc3DK|]}|ddVdS)rr6N)r)rimgs r z&image_generate_tool..s3RR3cggj>Q>QRQRRRRRRrz8Generated %s image(s) in %.1fs (%s upscaled) via %s [%s]T)rrrrrrimage_generate_toolrC)indentr zError generating image: z%srrrr error_typer)'rrr}rappendrtuplerboolrdatetimenowrrre_build_no_backend_setup_messagerrVALID_ASPECT_RATIOSrrrrrrr total_secondsrrsum_debuglog_callsaver rrrtyper)%r%r>r!r8r9r"r$rrrrrrefr2use_editrdebug_call_data start_time aspect_lcrmax_refsclamped_sourcesrtendpointrrrrshould_upscaleformatted_imagesroriginal_imagerupscaled_count response_datar error_msgs% r rrFs\4())NHdM)S!!0ioo&7&70Y__..///&u 662' 2 2C#s## 2  2$$SYY[[111HH_--MM"":tM':':H".wwH(#6,$*  //   !O&"&&((JIG RZ44 RFLLNN8K8Kq8P8PPQQ Q%'' @+G+I+I @<>>?? ?   <$((9h77<H--x    &h)DDD#,0022Z?NNPP X//VWW WHb)) 9788 8dhhy%8899J(l 4 4CsD)) esll5z!,,''(A..N  T!/E FLLNN!K!K!$++N;;;RSSS).N: &  # #N 3 3 3 3 FDEE ERR*:RRRRR F  ! !?NH    3CM%a(//   &* ".12B.C.C*+-<)*-??? z-FFFF GGG#,0022Z?NNPP7s1vv77  T9t 444VVq''*   $- -<)*-??? z-FFFFFFFFF#Gs Q)V77 Z(CZ#Z(#Z(cTttp tS)zDTrue if the FAL.ai API key (direct or managed gateway) is available.)r$rrerdrr check_fal_api_keyr>s# %''I+G+I+I J JJrcddg}|dtr|dn>|dtd}|r|d||d|d|d tr|d |d d |S) aRBuild an actionable error string when no FAL backend is reachable. Used by the in-tree FAL path. Mentions: - FAL_KEY signup link - managed-gateway status (if Nous tools are enabled) - plugin alternative pointer (so users on a stale ``image_gen.provider`` know the registry exists and how to inspect it) z4Image generation is unavailable in this environment.rwzMissing requirements:zA - FAL_KEY is not set and the managed FAL gateway is unreachablez+ - FAL_KEY environment variable is not setr{z - z&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"rrjoin)linesrs r r'r's6DR HE LL()))!## 3 O      BCCC? *    3 LL111 2 2 2 LL LL9::: LL 8"##   /    LL  99U  rc  trtdSn#t$rYnwxYw ddlm}ddlm}||D]*} |rdS#t$rY'wxYwn#t$rYnwxYwdS)aTrue if any image gen backend is available. Providers are considered in this order: 1. The in-tree FAL backend (FAL_KEY or managed gateway). 2. Any plugin-registered provider whose ``is_available()`` returns True. Plugins win only when the in-tree FAL path is NOT ready, which matches the historical behavior: shipping hermes with a FAL key configured should still expose the tool. The active selection among ready providers is resolved per-call by ``image_gen.provider``. Tr)list_providers_ensure_plugins_discoveredF) r>r ImportErroragent.image_gen_registryrDhermes_cli.pluginsrF is_availabler)rDrFproviders r #check_image_generation_requirementsrL>s         4          ;;;;;;AAAAAA""$$$&((  H ((** 44              5sG! .."A>A-)A>,A>- A:7A>9A::A>> B  B __main__u:🎨 Image Generation Tools — FAL.ai multi-model supportz<============================================================u(❌ FAL_KEY environment variable not setz- Set it via: export FAL_KEY='your-key-here'z Get a key: https://fal.ai/u✅ FAL.ai API key foundu ✅ fal_client library availableu;❌ fal_client library not found — pip install fal-clientu🤖 Active model: r)z ()z Speed: r*?u · Price: r,z Upscaler: r1onoffz Available models:u ← activerwz z<32z<6u% 🐛 Debug mode enabled — session )registry tool_errorimage_generateuBGenerate high-quality images from text prompts (text-to-image), or edit / transform an existing image (image-to-image) when the active model supports it. Pass `image_url` to edit that image; add `reference_image_urls` for style/composition references; omit both for text-to-image. The underlying backend (FAL, OpenAI, xAI, etc.) and model are user-configured and not selectable by the agent. Returns the result in the `image` field — either a URL or an absolute file path. To show it to the user, reference that path/URL in your response using the file-delivery convention for the current platform (your platform guidance describes how files are delivered here). When the active terminal backend has a different filesystem, successful local-file results may also include `agent_visible_image` for follow-up terminal/file operations.objectstringzThe text prompt describing the desired image (text-to-image) or the edit to apply (image-to-image). Be detailed and descriptive.)r. descriptionzlThe aspect ratio of the generated image. 'landscape' is 16:9 wide, 'portrait' is 16:9 tall, 'square' is 1:1.)r.enumrWrudOptional source image to edit/transform (image-to-image). When provided, the active backend routes to its image editing endpoint; when omitted, it generates from text alone. Pass a public URL or an absolute local file path from the conversation. Only honored by models that support editing — the description above indicates whether the active model does.arrayr.zOptional list of additional reference image URLs / paths (style, character, or composition references) to guide an image-to-image edit. Supported only by some models and capped per-model; the description above indicates the max.)r.rrWr%r>rr)r. propertiesrequired)namerWrc ddlm}|}t|tr|dnd}t|trR|d}t|t r(|r|Sn2#t$r%}t d|Yd}~nd}~wwxYwdS)zBReturn the value of ``image_gen.model`` from config.yaml, or None.rrrcNrsz"Could not read image_gen.model: %s rrrrrr}rrrrrrsectionrrs r _read_configured_image_modelrbs @111111kmm*4S$*?*?I#''+&&&T gt $ $ %KK((E%%% %%++-- %{{}}$ @@@ 93????????@ 4B"B&& C0CCc ddlm}|}t|tr|dnd}t|trR|d}t|t r(|r|Sn2#t$r%}t d|Yd}~nd}~wwxYwdS)uReturn the value of ``image_gen.provider`` from config.yaml, or None. 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). rrrcNrKz%Could not read image_gen.provider: %sr_r`s r _read_configured_image_providerres C111111kmm*4S$*?*?I#''+&&&T gt $ $ %KK ++E%%% %%++-- %{{}}$ CCC rsrnormalize_reference_imagesrzQimage_gen provider '%s' rejected image-to-image kwargs (signature too narrow): %sr]rOz 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 TypeError: %sz ' error: provider_exceptionz"Image gen provider '%s' raised: %sz#Provider returned a non-dict resultprovider_contractrd)rerbrHrhrIrFrrrr rrr}ragent.image_gen_providerrkgenerate TypeErrorrrr)r%r>rr configuredconfigured_modelrhrFrKrkwargs norm_refsrkrs r _dispatch_to_plugin_providerrvs}*122J t455  :99999AAAAAA""$$$< ++   krea-2-large krea-2-mediumkrea-2-medium-turbocrt|tsdS|}|tvr|SdS)zIReturn the native Krea plugin model id when ``model_id`` is ``krea-2-*``.N)rr}r_KREA_NATIVE_MODELS)r candidates r _normalize_krea_modelr}s? h $ $t  I''' 4rc$t|duS)zATrue when ``model_id`` is a native Krea plugin id (``krea-2-*``).N)r})rs r is_krea_modelrs  * *$ 66rctdkrdStt}|dS ddlm}|dSn3#t $r&}t d|Yd}~dSd}~wwxYw ddlm }ddl m }||d} n3#t $r&}t d|Yd}~dSd}~wwxYw| dS|||d } t|tr+|r|| d <d} |dd lm} | |} | r| | d <| jdi| } nN#t $rA}t d |t'jddd|ddcYd}~Sd}~wwxYwt| t*st'jdddddSt'j| S)aORoute a native ``krea-2-*`` model to the managed Krea gateway, in managed mode. Returns a JSON result string when handled by the Krea managed gateway, or ``None`` to fall through to the normal plugin/FAL pipeline. Fires only when all hold: - the configured image model is a native ``krea-2-*`` id, AND - the user isn't already routed to the Krea plugin via ``image_gen.provider`` (that path dispatches normally), AND - the managed Krea gateway is resolvable (portal/managed mode). Direct/BYO users (no managed gateway) fall through untouched. kreaNr)_resolve_managed_krea_gatewayz%Managed Krea routing probe failed: %srgrEz.Managed Krea routing: provider unavailable: %s)r%r>rsrrjrzManaged Krea routing failed: %sFzManaged Krea generation error: rmr z(Krea provider returned a non-dict resultrnrd)rer}rbplugins.image_gen.krearrrrrHrhrIrFrr}rrorkrprr rr)r%r>rr normalizedrrrhrFrKrtrurkrs r _maybe_route_managed_krears&'((F22t&'C'E'EFFJtHHHHHH ( ( * * 24 3  >>zrrr)rr)rrZ)rrSrrvrrr) argskwr%r>rrr dispatched krea_routedrs r _handle_image_generaters# XXh # #F ECDDD88N,@AAL%%I88$:;;ffYG . 1J 1*gNNNN, 1K 1+wOOOO !1    C .c7 C C CCrrWc:dgdd}t}|r|dkr ddlm}ddlm}|||}|i} |pi}n#t $ri}YnwxYw|j|d<tp| pd |d <| d rt|d |d <| d rt|d |d <|Sn#t $rYnwxYw t\}}d |d<| d||d <| dr/ddg|d <t| d pd|d <n dg|d <d|d <n#t $rYnwxYw|S)aBest-effort: return the active backend/model's image capabilities. Resolution order mirrors the runtime dispatch: 1. If ``image_gen.provider`` is set, ask that plugin provider. 2. Otherwise inspect the in-tree FAL model catalog for the active model. Returns a dict like ``{"modalities": [...], "max_reference_images": N, "model": "...", "provider": "..."}``. Never raises. rr) modalitiesr4falrgrENrKrwrsrr4zFAL.air)r2rr6)rerHrhrIrF capabilitiesr display_namerb default_modelrrrr)rconfigured_providerrhrFrKcapsrrs r _active_image_capabilitiesr"s@,2(ANND9;;2e;;  = = = = = = E E E E E E & & ( ( (#|$788H##00228bDD DDD#+#8Z < > > b8CYCYC[C[Ca_aW 88L))B)-d<.@)A)AD&88233U36tF FFc~tg} t}n#t$r dticYSwxYw|d}|d}t |dpdg}d}|r|d|z }|r|d|z }||d |vrEd|vrA|d pd }|r |d krd |dnd}|d|dn3d |vrd|vr|dn|ddd|iS)zIBuild a description reflecting whether the active model supports editing.rWrKrsrrz Active backendz: u · model: rr4rr6z; up to z, reference image(s) via reference_image_urlsrwz\- supports both text-to-image (omit image_url) and image-to-image / editing (pass image_url)u — routes automaticallyuD- this model is image-to-image / edit only — image_url is REQUIREDu- this model is text-to-image only — it is NOT capable of image-to-image / editing; do not pass image_url or reference_image_urls (they will be rejected). Provide a text-only prompt.r@)_GENERIC_IMAGE_DESCRIPTIONrrrrr"rA)partsrrKrsrliner4ref_notes r _build_dynamic_image_schemarWs ' (E;)++ ;;;9::::;xx ##H HHW  ETXXl++7x88J D  X & %e%%% LL*:!5!5882338q $qLL Nx M M M M    #8@ # # #    J  6#;#; R           499U++ ,,s //rcu🎨) r]toolsetschemarcheck_fn requires_envis_asyncemojidynamic_schema_overrides)N)NN)c__doc__r loggingrr% threadingr~typingrrrr__annotations__r tools.debug_helpersrr rrrtools.managed_tool_gatewayrtools.tool_backend_helpersrrrr getLoggerrrr]r}rrr(rrrrrrrrrr+rnroLockrmrerrrr#rrrrrrr$rrrrrrfloatrr>r'rLprint SystemExitrGrrrrmidmmarkeractive session_idtools.registryrRrSIMAGE_GENERATE_SCHEMArbrervr{r}rrrrrrregisterrdrr rs*   &&&&&&&&&& C #    -,,,,, DCCCCC  8 $ $4%'))!'  $%"%*     7   !"9> *))!'  $&!"%* #     1   !"C""H#*))!'  $%"%*',      /4:D#$  " #      7   !"C""H#'#$!#   "      4   !#;@!J% *(!&   "      3   !#O((T!&#))!'  *!     2   !"7<$>))!'   $U    ',,, .))!'  $&!"%      8   !"=H#P6$  (    %,,*"R6$  (    %++_ C)C) Dd38n$ %CCCL ) "9 +>K! m-@ A A A!)9>++ 555###0(s(tCH~((((\ *E * * * *L-*. ... .. 3- . S#X' .  #s(^ ....j-*. 666 66 6 3- 6 S#X' 6 #s(^6666x+c+C+HT#s(^>> -...jmm E $%%% 01111  KLLLjmm('))NHd E LH = = L L L L LMMM E Ttxx-- T TDHHWc