`gj UdZddlZddlZddlZddlZddlZddlZddlmZm Z m Z da ee d<defdZ ddlmZddlmZmZmZdd lmZdd lmZmZmZmZej6eZd 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;fdZFedk(r@eGdeGdeDs eGdeGdeGdeHd#eGd ddl a eGde5\ZJZKeGdeKjdeJdeJdeGdeKjdddeKjddeGdeKjdrdnd›eGdëejD]C\ZNZOeNeJk(rdndZPeGdeNdǛdeOjdddțdeOjddePEe,jreGde,jddlSmTZTmUZUddddddМ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 eeffdZaeTjddeVe^eFgddea y#eI$reGdeHd#wxYw)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_clientreturnc@ttSddlm}|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. rimport_fal_client)rtools.fal_commonr r s N/root/.hermes/venv/lib/python3.12/site-packages/tools/image_generation_tool.py_load_fal_clientr ,s!2"$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_varcDtr tdsytdS)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_gatewayres{'C ' 44rc4|jjd|jf}t5tt |k(rtcdddSt tt|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_configr rr)managed_gateway client_configs r _get_managed_fal_clientrrs &&--c2''M " #  */I]/Z& # # 3 //,;;  &3"" # # #sB 7BBmodel argumentsc tdttji}t }|t j |||St|} |j |||S#t$rL}t|}|9d|cxkrdkr.nd}|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 ,=>O24O  )_UU,_=N$$ #%   &c*  #"5#"5$ #!O(;6$( >Cc:d} ddlm}|}t|tr|j dnd}t|tr1|j d}t|t r|j }|s$tjddj }|stttfS|tvr.tjd |ttttfS|t|fS#t$r }tjd|Yd}~d}~wwxYw) 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_modelrs H L1m*4S$*?#''+&T gt $++g&C#s#99; 99.399; j777z! B m j777 Z) ))# L EsKKLsA1C11 D:DDrr%r$ overridesc t|}|d}|d}|xstjj}||vrt}t |j di} |xsdj| d<|dvr ||| d<n|dk(r ||| d<nt d ||t|tr|| d <|r |jD] \} } |  | | | <|d } | jD cic]\} } | | vs| dk(r| | c} } Scc} } w) 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: r$r0) r]DEFAULT_ASPECT_RATIOlowerrrrrrintitems) rr%r>r$rmetar-r.aspectpayloadkvr0s r _build_fal_payloadr5sC h Dl#J ME22 9 9 ; A A CF U%"488J#;?? JtS1OO% DAq}  JH ! A =AM 1  s0D r'cdt|}|jdxs t}|d}|d} |xstj j } | | vrt} t |jdi} |xsdj | d<t|| d<|dvr d |vr | | | d <n|d k(r d |vr| | | d <|t|tr|| d <|r |jD] \} } |  | | | <ddh}| jD cic]\} } | |vs| |vr| | c} } Scc} } w) 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>r$) r]rsetrrrrlistrrr)rr%r'r>r$rrr3r-r.rrrr _requireds r _build_fal_edit_payloadrfsc" h DHH_-6Ml#J ME22 9 9 ; A A CF U%"488J#;[ %f   ~ %.M*I"'- JtS1OO% DAq}  <(I  A i 1  sD, image_urloriginal_promptc < tjd|td|ttt t tttd }tt|}|j}|rod|vrk|d}tjd|jdd|jd d|d |jdd |jd d d td Stjdy#t$r"}tjd|d Yd}~yd}~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_imagers % >?#01O3DE-7-/5#?%<  &n@RS g'#G_N KK6""7I6""8Y7  &e,'++GQ7(,,Xq9 "1    9:  0!d CsCC0C00 D9DDvaluec|rt|tsy|j}|jdrytj j |ryt|dk\xr|ddk(xr|ddvS) NF)zhttp://zhttps://zdata:TrYr6:rC>rg\)rr}r startswithrpathisabslen)rrs r _looks_like_absolute_file_pathrsj  5#. KKME 89 ww}}U u:? JuQx3 J58{3JJrtask_idc ddlm}||xsdS#t$r }tj d|Yd}~yd}~wwxYw)Nr)get_active_envdefaultz1Could not inspect active terminal environment: %s)tools.terminal_toolrrrr)rrrs r _active_terminal_envrs<6g233  H#Ns =8=envc|t|dd}t|r% |}|rt|jdS t|dd}|rt|jddS|jj}|dvrytjdxsd jj}|d vry|d k(ry y#t$r }t j d|Yd}~d}~wwxYw) 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_envrs  3 :DA H  T u:,,S11 c>48 +&--c238< <==)) Z Z" yy(3G::<BBDG44% ) T MsSS Ts"C C1C,,C1 host_pathct|syt|}|sy ddlm}|||S#t$r }t j d|Yd}~yd}~wwxYw)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 sZ )) 4*3/J RF*9ZPP R KSQQ Rs, AAAct|dd}|y |jdy#t$r }tj d|Yd}~yd}~wwxYw)N _sync_managerTforcez1Could not force-sync generated image artifact: %s)rsyncrrr)r sync_managerrs r _force_artifact_syncrsT36LQ% QJCPPQs% AA  Arc t|trtj|n|}t|t r|j ds|S|j d}t|tr t|s|St|}t||}|r||k(r|S| t||jd||jd|tj|dS#t$r|cYSwxYw)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. successr host_imageagent_visible_imageF) ensure_ascii) rr}jsonloadsrrrrrrr setdefaultdumps)rrrrr agent_paths r "_postprocess_image_generate_resultr &s%/S%9$**S/s gt $GKK ,B KK E eS !)G)N w 'C*5#6J u,  S! |U+ ,j9 ::gE 22)  s'C"" C0/C0r!r8r9r"reference_image_urlsc  t\} } g} t|tr/|jr| j |jt|t t frH|D]C} t| ts| js%| j | jE| jd} t| xr t| }|rdnd}| ||||||||t| d ddddd}tjj} |r,t|trt|jdk(r td tststt| r$| s"td | jd | d | d |xst j#j}|t$vr!t&j)d|t t }i}|||d<|||d<|||d<|||d<|rst+| jdxsd}|dkDr| d|n| }t-| |||||}| }t&j/d| jd | |t||ddn=t1| ||||}| }t&j/d| jd | | |ddt3||}|j}tjj|z j5}|rd|vr td|jdg}|s tdt| jddxr| }g}|D]}t|t6rd|vs|d|jdd|jd dd!}|rFt9|d|j} | r|j | tt&j)d"d|d#<|j ||s td$t;d%|D}!t&j/d&t|||!||d'|r|ddnd|d(}"d'|d)<t||d*<||d+<t<j?d,|t<jAtCjD|"d-d.S#tF$r}#tjj|z j5}d/t|#}$t&jId0|$d'1ddt|#tK|#jLd2}"|$|d3<||d+<t<j?d,|t<jAtCjD|"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|]}|jdsdyw)rr6N)r).0imgs r z&image_generate_tool..sR3cggj>QQRs  z8Generated %s image(s) in %.1fs (%s upscaled) via %s [%s]T)rrrrrrimage_generate_toolrC)indentrzError generating image: z%srrrr error_typer)'rrr}rappendrtuplerboolrdatetimenowrrre_build_no_backend_setup_messagerrVALID_ASPECT_RATIOSrrrrrrr total_secondsrrsum_debuglog_callsaverr rrtyper)%r%r>r!r8r9r"r$rr rrrrefr2use_editrdebug_call_data start_time aspect_lcrmax_refsclamped_sourcesrtendpointrrrrshould_upscaleformatted_imagesroriginal_imagerupscaled_count response_datar error_msgs% r rrFs}4()NHdM)S!ioo&7Y__./&u 6' 2C#s# $$SYY[1 2HH_-MM":tM':H"wH(#6,$*  /  !O&""&&(JIGZ4FLLN8Kq8PPQ Q%'+G+I<>? ? $((9h78H:F;< "9%9@@BHHJ / / NN?2 -I$&  */BI+ ,  %*8I& '  !&0Il #  $)6Io & 488$:;@qAH:BQ,mIX6MO/&/9YI%H KKOH-x_9Ms   +&)$)I H KK>H-x  &h)D#,,002Z?NNP/VW WHb)78 8dhhy%89J(l 4CsD)esl5z!,''(A.N !/E FLLN!K!$++N;RS).N: &  # #N 3# 4& DE ER*:RR F  !?NH  3C%a(/  &* ".12B.C*+-<)*-? zz-FF G#,,002Z?NNP.s1vh7  T9t 4Vq'**  $- -<)*-? zz-FF#Gs NR)) V2CU;5V;Vc@ttxs tS)zDTrue if the FAL.ai API key (direct or managed gateway) is available.)r!rrerdrr check_fal_api_keyr;s %'I+G+I JJrcddg}|jdtr|jdn2|jdtd}|r|jd||jd|jd|jd tr|jd |jd d j|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) )rrrjoin)linesrs r r$r$sDR HE LL()!# O   BC? *   LL401 2 LL LL9: LL 8"#  /  LL  99U rc tr ty t}|r|dk(ry ddlm}ddlm}|||}t|xr|jS#t$rYVwxYw#t$rYywxYw)zDTrue if FAL or the explicitly configured image backend is available.TfalFr get_provider_ensure_plugins_discovered) r;r ImportError_read_configured_image_provideragent.image_gen_registryrChermes_cli.pluginsrEr! is_availabler) configuredrCrEproviders r #check_image_generation_requirementsrM>s      12J u,9A"$ +H8!6!6!899    s"A#7A2# A/.A/2 A>=A>__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+enumrXrudOptional 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+rrXr%r>rr )r+ propertiesrequired)namerXrc` ddlm}|}t|tr|j dnd}t|trA|j d}t|t r |j r|j Sy#t$r }tjd|Yd}~yd}~wwxYw)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_modelrcs @1m*4S$*?#''+&T gt $KK(E%%%++-{{}$  @ 93?? @BB B- B((B-c` ddlm}|}t|tr|j dnd}t|trA|j d}t|t r |j r|j Sy#t$r }tjd|Yd}~yd}~wwxYw)uReturn ``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). rrrcNrLz%Could not read image_gen.provider: %sr`ras r rGrGs C1m*4S$*?#''+&T gt $KK +E%%%++-{{}$  C rsrnormalize_reference_imagesr zQimage_gen provider '%s' rejected image-to-image kwargs (signature too narrow): %sr^rPz 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)rGrcrHrCrIrErrrrr rr}ragent.image_gen_providerrigenerate TypeErrorrrr)r%r>rr rKconfigured_modelrCrErLrkwargs norm_refsrirs r _dispatch_to_plugin_providerrss*12J u,45  :A"$ +  L 'T 2#J/Hzz&zl3014   )/ MF8 .F7O i %)//*;"+//"3F;   + K23GHI -6F) *""",V,Z fd #zz:-    ::f w   krea-2-large krea-2-mediumkrea-2-medium-turbocZt|tsy|j}|tvr|Sy)zIReturn the native Krea plugin model id when ``model_id`` is ``krea-2-*``.N)rr}r_KREA_NATIVE_MODELS)r candidates r _normalize_krea_modelrz|s- h $ I'' rct|duS)zATrue when ``model_id`` is a native Krea plugin id (``krea-2-*``).N)rz)rs r is_krea_modelr|s  *$ 66rc2tdk(rytt}|y ddlm}|y ddlm }ddl m }||d} | y|||d } t|tr#|jr|j| d <d} |dd lm} | |} | r| | d <| j"di| } t| t*st'j(dddddSt'j(| S#t $r }t jd|Yd}~yd}~wwxYw#t $r }t jd|Yd}~yd}~wwxYw#t $r=}t j%d |t'j(ddd|ddcYd}~Sd}~wwxYw)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: %srBrDz.Managed Krea routing: provider unavailable: %s)r%r>rsrrhr zManaged Krea routing failed: %sFzManaged Krea generation error: rkrz(Krea provider returned a non-dict resultrlrd)rGrzrcplugins.image_gen.krearrrrrHrCrIrErr}rrmrirnrrr r)r%r>rr  normalizedrrrCrErLrqrrrirs r _maybe_route_managed_krears&'(F2&'C'EFJH ( * 2 3 9A"$'$F  i %)//*;"+//"3F;   + K23GHI -6F) *""",V, fd #zz?-    ::f _  zz6se<.     sM C8D$AE8 D!DD!$ E -EE  F2F FFc |jdd}|s tdS|jdt}|jd}|jd}|jd}t||||}| t || St ||||}| t || St |||| } t | | S) Nr%rwz'prompt is required for image generationr>rr r)rr )rr[)rrTrrsr rr) argskwr%r>rr r dispatched krea_routedrs r _handle_image_generaters XXh #F CDD88N,@AL%I88$:;ffYG . 1J 1*gNN, 1K 1+wOO !1  C .c7 CCrrXcdgdd}t}|r|dk7r ddlm}ddlm}|||}|i} |j xsi}|j|d<txs|jxsd|d <|jd rt|d |d <|jd rt|d |d <|S t\}}d |d<|jd ||d <|jdr*ddg|d <t|jd xsd|d <|Sdg|d <d|d < |S#t $ri}YwxYw#t $rYwxYw#t $rY|SwxYw)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) modalitiesr4rArBrDrLrwrsrr4zFAL.air)r2rr6)rGrHrCrIrE capabilitiesr display_namerc default_modelrrrr)rconfigured_providerrCrErLcapsrrs r _active_image_capabilitiesrs,2(AND9;2e;  = E & (#$78H##0028bD$,#8#8Z < > b8CYCYC[Ca_aW 88L))-d<.@)AD&882336td|vr:|jd xsd }|r |d kDrd |dnd}|j d|dn+d |vrd|vr|j dn|j dddj |iS#t$r dticYSwxYw)zIBuild a description reflecting whether the active model supports editing.rXrLrsrrz 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_DESCRIPTIONrrrrrr>)partsrrLrsrliner1ref_notes r _build_dynamic_image_schemarKsV ' (E;)+xx #H HHW ETXXl+7x8J D "XJ  +eW%% LL*:!588238qHqLxj L M    88@zB# # J 6#; R      499U+ ,,M ;9::;s C;;DDrcu🎨) r^toolsetschemarcheck_fn requires_envis_asyncemojidynamic_schema_overrides)N)NN)c__doc__rloggingrr" threadingr~typingrrrr__annotations__r tools.debug_helpersrr rrrtools.managed_tool_gatewayrtools.tool_backend_helpersrrrr getLoggerrrr]r}rrr%rrrrrrrrrr(rnroLockrmrerrrr rrrrrrr!rrrrrr floatrr;r$rMprint SystemExitrFrrrrmidmmarkeractive session_idtools.registryrSrTIMAGE_GENERATE_SCHEMArcrGrsrxrzr|rrrrrregisterrdrr rs *   && C # - D   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) Dd38n$ %CL ) "9 +>K! m-@ A!)9>>+ 5#0(s(tCH~(\ *E *L-*. .. .. 3- . S#X' .  #s(^ .j-*. 66 66 6 3- 6 S#X' 6 #s(^6x+c+C+HT#s(^ -.m $% 01 ()NHd H =>b ! LM Jtxx-.mDHHWc