`gj\trUdZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl Z ddlmZmZmZmZmZmZmZej,eZdZddd d d d d ddddd dddddd dZded<dZddddZdZd Zhd!Z d"Z!d#Z"d$Z#d%Z$e%hd&Z&hd'Z'd0d(Z(d1d2d)Z)d*Z*d3d+Z+d4d,Z,Gd-d.eZ-d5d/Z.y)6uKrea image generation backend. Exposes Krea's `Krea 2` foundation image model family — Krea 2 Medium and Krea 2 Large — as an :class:`ImageGenProvider` implementation. Krea's API is asynchronous: the generate endpoint returns a ``job_id`` that you poll at ``GET /jobs/{job_id}``. This provider hides that roundtrip behind the synchronous ``generate()`` contract: submit, poll every 2s with light backoff, materialise the result URL to local cache, return the success/error dict like every other backend. Selection precedence (first hit wins): 1. ``KREA_IMAGE_MODEL`` env var (escape hatch for scripts / tests) 2. ``image_gen.krea.model`` in ``config.yaml`` 3. ``image_gen.model`` in ``config.yaml`` (when it's one of our IDs) 4. :data:`DEFAULT_MODEL` — ``krea-2-medium`` (Krea's "start here" recommendation) Docs: https://docs.krea.ai/developers/krea-2/overview API: https://docs.krea.ai/api-reference/krea/krea-2-large ) annotationsN)AnyDictListOptionalTuple)DEFAULT_ASPECT_RATIOImageGenProvidererror_responsenormalize_reference_imagesresolve_aspect_ratiosave_url_imagesuccess_responsezhttps://api.krea.aiz Krea 2 Mediumz~15-25szCIllustration, anime, painting, expressive styles. Faster + cheaper.z9$0.030 (text) / $0.035 (style refs) / $0.040 (moodboards)medium)displayspeed strengthspricepathz Krea 2 Largez~25-60szIPhotorealism, raw textured looks (motion blur, grain), expressive styles.z9$0.060 (text) / $0.065 (style refs) / $0.070 (moodboards)largezKrea 2 Medium Turboz~8-15su:Fastest Krea 2 — medium quality at lower latency / cost.z$$0.015 (text) / $0.0175 (style refs)z medium-turbo) krea-2-mediumz krea-2-largezkrea-2-medium-turbozDict[str, Dict[str, Any]]_MODELSrz16:91:1z9:16) landscapesquareportrait1Kg333333?>lowrawhighrg@g@g?gf@>>failed cancelled completedc ddlm}|}t|tr|j dnd}t|tr|SiS#t $r"}t jd|icYd}~Sd}~wwxYw)zMRead ``image_gen.krea`` (with fallthrough to ``image_gen``) from config.yaml.r) load_config image_genNz#Could not load image_gen config: %s)hermes_cli.configr- isinstancedictget Exceptionloggerdebug)r-cfgsectionexcs R/root/.hermes/venv/lib/python3.12/site-packages/plugins/image_gen/krea/__init__.py_load_krea_configr:wsb1m*4S$*?#''+&T$Wd3w;;  :C@ s$AAA A1A,&A1,A1ct|tr=|jtvr'|jt|jfStj j d}|r|tvr |t|fSt}t|j dtr|j dni}d}t|tr+|j d}t|tr |tvr|}|+|j d}t|tr |tvr|}| |t|fStttfS)uDecide which model to use and return ``(model_id, meta)``. Precedence: explicit caller override (e.g. managed-mode routing or a direct ``model`` kwarg) → ``KREA_IMAGE_MODEL`` env → ``image_gen.krea.model`` → ``image_gen.model`` → :data:`DEFAULT_MODEL`. KREA_IMAGE_MODELkreaNmodel) r0strstriprosenvironr2r:r1 DEFAULT_MODEL)explicit env_overrider6krea_cfg candidatevaluetops r9_resolve_modelrJs(C X^^%5%@~~)9!:::::>>"45L /W\222  C",SWWV_d"CswwvH#I(D! W% eS !ew&6Iggg c3 C7NI'),,, '-0 00c. ddlm}ddlm}tjjdr |dsy |dS#t$r }t j d|Yd}~yd}~wwxYw#t$r }t j d |Yd}~yd}~wwxYw) uReturn managed Krea gateway config when the user is on the managed path. Mirrors ``_resolve_managed_fal_gateway`` in ``tools/image_generation_tool.py``: the Nous-hosted Krea gateway wins when it is resolvable AND either no direct ``KREA_API_KEY`` is configured or the user explicitly opted into the gateway for ``image_gen``. Returns ``None`` (direct/BYO path) otherwise, and never raises — plugin discovery and availability scans must stay robust. r)resolve_managed_tool_gateway)prefers_gatewayz/Managed Krea gateway resolution unavailable: %sN KREA_API_KEYr.r=z*Managed Krea gateway resolution failed: %s) tools.managed_tool_gatewayrMtools.tool_backend_helpersrNr3r4r5rArBr2)rMrNr8s r9_resolve_managed_krea_gatewayrRsK>  zz~~n%ok.J+F33  FL  A3Gs, ?A+ A(A##A(+ B4BBcp ddlm} t|dS#t$rYywxYw#t$rYywxYw)z23T7R7TTrKc tjDcgc]\}}||d|d|d|ddc}}Scc}}w)Nrrrr)idrrrr)ritems)rdmodel_idmetas r9 list_modelsz KreaImageGenProvider.list_modelssR#*--/  $  ?g!+.g     s:ctSN)rCrcs r9 default_modelz"KreaImageGenProvider.default_modelsrKcdddddddgdS) NrhpaiduKrea 2 foundation model — Medium ($0.03), Large ($0.06), Medium Turbo ($0.015). Style transfer, moodboards, reference-guided generation. Direct key or managed Nous Subscription gateway.rOz Krea API keyz'https://www.krea.ai/settings/api-tokens)keyprompturl)rebadgetagenv_varsrbrcs r9get_setup_schemaz%KreaImageGenProvider.get_setup_schemas-Q*,D  rKcddgddS)Ntextimage ) modalitiesmax_reference_imagesrbrcs r9 capabilitiesz!KreaImageGenProvider.capabilities s &w/LLrKN) image_urlreference_image_urlsc r|xsdj}t|}tj|d}g}t |t r/|jr|j |jt|xsgD]} |j | |jd} t | tr\| D]W} t | t r1| js$|j | jD| sG|j | Yt} g} |D]9} t | t r| | vr| j| | j | ;| dd}|rdnd}|stddd | St}|(|jjd }|j}n6t }t"j$jd }|std dd | St'|jd\}}t)|jd}|t |jdtr"|jdrtddd |||St |jdtr"|jdrtddd |||S||t*|d}|jd}t |t,r||d<|jd}t |tr|r||d<|rHg}|D]<} t | t r|j | t.d,|j | >||d<|jd}t |tr |r|dd|d<d|ddd}| t t1j2|d<|d |d!} t5j6|||d"#}|j9 |jA}#|#jd<}$t |$t r|$std=d;d |||S|d>|$}%d|dd?}&tP}'tSjTtVz}(d}) tSjX|'t[|'t\zt^}' t5j|%|&d"@}*|*j9 |*jA}+t |+tBr|+jdInd},t |,t r |,})|,tdvrngt |+tBr|+jdJrnEtSjT|(k\r,tdK|$dLt-tVdM|)xsdNdOd7d |||S t |+tBstdPd;d |||S|)dQk(r`t |+jdRtBr$|+jdRxsijd%nd}-tdK|$dS|-xsdTd4d |||S|)dUk(rtdK|$dVdUd |||S|+jdR}.t |.tBstdWdXd |||Sd}/|.jdY}0t |0tr<|0r:|0D]5}1t |1t s|1js%|1j}/n|/A|.jdZ}2t |2t r |2jr|2j}/|/td[dXd |||S tg|/d\|]}3t |3}4|t*||$d_}5t |+jdJt r|+dJ|5dJ<ti|4|||d ||5`S#t4j:$rV}|j<}| |j>nd$} ||jAni} t | jd%tBr!| jd%ijd&n#| jd&xs| jd'xs||jDdd(n t |}!n+#tF$r||jDdd(n t |}!YnwxYwtHjKd)||!|=d*|cxkrd+kr2nn/|d,k(rd-nd.|d/}"td0|d1|d2|!d3|"d4d |||cYd}~Std5|d2|!d4d |||cYd}~Sd}~wt4jL$rtd6d7d |||cYSt4jN$r}td8|d9d |||cYd}~Sd}~wwxYw#tF$r}td:|d;d |||cYd}~Sd}~wwxYw#t4j:$ry}|j<}| |j>nd$}tHjKdA||$|t`vstSjT|(k\rtdB|dC|$d4d |||cYd}~SYd}~ed}~wt4jLt4jNf$rU}tHjcdD|$|tSjT|(k\rtdE|$dF|d7d |||cYd}~SYd}~d}~wwxYw#tF$rR}tHjcdG|$|tSjT|(k\rtdH|d;d |||cYd}~SYd}~9d}~wwxYw#tF$r$}tHjcd^|/||/}4Yd}~d}~wwxYw)aNrimage_style_referencesrrrz1Prompt is required and must be a non-empty stringinvalid_argumentr=)error error_typeprovider aspect_ratio/rOuKREA_API_KEY not set. Run `hermes tools` → Image Generation → Krea to configure, get a key at https://www.krea.ai/settings/api-tokens, or sign in to a Nous account with the managed Krea gateway enabled (`hermes setup`). auth_requiredr>rYstyleszManaged Krea (Nous Subscription) does not support trained styles (LoRAs). Set KREA_API_KEY to use Krea directly, or omit `styles`.unsupported_argument)rrrr>rxr moodboardszzManaged Krea (Nous Subscription) does not support moodboards. Set KREA_API_KEY to use Krea directly, or omit `moodboards`.)rxr resolutionrYseed)rystrengthzBearer zapplication/jsonz!Hermes-Agent/1.0 (krea-image-gen)) Authorizationz Content-Type User-Agentzx-idempotency-keyz/generate/image/krea/krea-2/r)headersjsontimeoutrrmessagedetaili,zKrea submit failed (%d): %sir(r'u<Krea's shared-key concurrency cap was hit — retry shortly.zModel 'u' may not be enabled/priced on the Nous Portal's Krea gateway. Set KREA_API_KEY to use Krea directly, or pick a different model via `hermes tools` → Image Generation.z)Nous Subscription Krea gateway rejected 'z' (HTTP z): z. api_errorzKrea image generation failed (zKrea submit timed out (30s)rzKrea connection error: connection_errorz&Krea returned invalid JSON on submit: invalid_responsejob_idz#Krea submit response missing job_idz/jobs/)rr)rrz Krea poll failed (%d) for job %szKrea poll failed (z ) for job z(Krea poll transient error for job %s: %szKrea poll timed out for job z: z.Krea poll returned invalid JSON for job %s: %sz!Krea poll returned invalid JSON: status completed_atz Krea job z did not complete within zs (last status: unknown)zKrea returned non-dict job bodyr)resultz failed: z unknown errorr*z was cancelledz)Krea job completed but result was missingempty_responseurlsryz"Krea result contained no image URLkrea_)prefixzEKrea image URL %s could not be cached (%s); falling back to bare URL.)krea_aspect_ratiorrYr)rr>rxrrmodalityextra)5r@r _ASPECT_MAPr2r0r?appendr listsetaddr rRgateway_originrstripnous_user_tokenBASE_URLrArBrJr^DEFAULT_RESOLUTIONint!_DEFAULT_STYLE_REFERENCE_STRENGTHuuiduuid4requestspostraise_for_status HTTPErrorresponse status_coderr1rr3r4rTimeoutConnectionError_POLL_INITIAL_INTERVALtime monotonic_POLL_TIMEOUT_SECONDSsleepmin _POLL_BACKOFF_POLL_MAX_INTERVAL_RETRYABLE_POLL_STATUSESwarning_TERMINAL_STATESrr)6rdrxrrrkwargsaspectkrea_ar style_refsref legacy_refsseendedupedrrmanagedbase_url auth_tokenrorprYpayloadrrnormalized_refsrr submit_urlrr8resprbodyerr_msghint submit_bodyrjob_url poll_headersintervaldeadline last_status poll_respjob status_strerrrresult_image_urlrrGsingle saved_path image_refrs6 r9generatezKreaImageGenProvider.generatesp ,B%%'%l3//&%0!# i %)//*;   ioo/ 0./CDJ #C   c " #jj!9: k4 (" +c3'yy{"))#))+6%%c*  +E A!S!9 NN1    Sb\ (7f!I-#  01  --44S9H 00JH7J%, /#!'  ( 7(;<$(L)AB  &**X.5&**X:N%6 6#"!!'  &**\2D9fjj>V%- 6#"!!'  #,$ # zz&! dC "GFOH% fd # &GH   *,O! 0c3'#** #1RS$**3/  01@G, -ZZ - j$ 'J$.rNGL ! 'zl3.=    ,/tzz|+>#&;;%)  JJx 8m35GHH" $LL,PRS **,D nn&/9d.C*J*c*( !11 #t$)@~~8+%#F8+D4566F{G_V_F``ac )#"!!'  Kd#t$!7-#   ( "#)<<>  #!:+#   ('(85 ASTJJI"),$ !  cggn-s 3$'$7E. !  e!!. < +#"!!'  "6vhc'K&#   !3$#  '' !/u5-#    !>seD-#   P%% ||-1-=))1 ?P !99T^^=MQY=Y) 26(*VHM#.!'&%%+ $$h&>&>? I6SVW>>#x/) >#x/) A#G#5!'&%%+  ` ) NNW   )I  )s()\;c<6(d&h+j ;c9b ,B`b %`+(b *`++Ab >c9b c9 'c9 c9c4.c94c9< d#dd#d#&h(9A(f-!h(-&h(Ah#h(#h(+ j4Aj5jj j6j11j6)returnr?rrV)rzList[Dict[str, Any]])r Optional[str]rDict[str, Any]) rxr?rr?rrrzOptional[List[str]]rrrr)__name__ __module__ __qualname____doc__propertyrerirkrqrtr}rr rrbrKr9r`r`sJ U    M1I $(48 I I I ! I 2 I I  I rKr`c6|jty)uGPlugin entry point — wire ``KreaImageGenProvider`` into the registry.N)register_image_gen_providerr`)ctxs r9registerrs##$8$:;rKrrs)rDrrzTuple[str, Dict[str, Any]]r)rHrrr?)rNone)/r __future__rloggingrArrtypingrrrrrragent.image_gen_providerr r r r r rr getLoggerrr4rr__annotations__rCrrrr[rrrr frozensetrrr:rJrRrWr^r`rrbrKr9r s,,# 33   8 $ ! #ZL "`L )Q7 & "0   %(!5   %%MN7 1@4  &@ +@ P