߲ja UdZddlmZddlZddlZddlZddlmZmZm Z m Z m Z m Z ddl Z ddlmZddlmZmZmZmZddlmZmZmZmZmZmZmZmZej<eZ dZ!d d d d d dddddddddddddddddZ"de#d<e$e"Z%dZ&ddd d!Z'd"Z(d#Z)d$Z*hd%Z+d&Z,d'Z-d(Z.d)Z/e$hd*Z0hd+Z1d,Z2d-Z3d.Z4d/d0fd1d2fd3d4fd1d5fd6Z5d7e#d8<d9Z6dTd:Z7dTd;Z8dUdVd<Z9d=Z:dWd>Z;dXd?Zd[dBZ?e/ddC d\dDZ@d]dEZA d^dFZB d_dGZC d`dHZD dadIZE dbdJZFdcdKZGGdLdMeZHdddNZIddlJZJdOdPdQdRZKdSZLy)euKrea image generation backend (``Krea 2`` Medium / Large / Medium Turbo). Krea's API is asynchronous: submit returns a ``job_id`` polled at ``GET /jobs/{job_id}``; ``generate()`` hides that (submit, poll every 2s with light backoff, cache the result URL locally). Selection: ``model`` kwarg → ``KREA_IMAGE_MODEL`` → ``image_gen.krea.model`` → ``image_gen.model`` (when one of our IDs) → :data:`DEFAULT_MODEL`. Docs: https://docs.krea.ai/developers/krea-2/overview ) annotationsN)AnyCallableDictListOptionalTuple) get_secret)DEFAULT_ASPECT_RATIOresolve_aspect_ratiosave_url_imagesuccess_response)ErrorFnStaticImageGenProvidercollect_source_images error_factoryload_image_gen_config post_jsonprompt_required_errorresolve_static_modelzhttps://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)mediumF)displayspeed strengthspricepathupscalez 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 completedz/generate/enhance/krea/enhancez!Hermes-Agent/1.0 (krea-image-gen) api_errorcd|d|S)NzKrea poll failed (z ) for job job_iddetails D/home/ubuntu/.hermes/hermes-agent/plugins/image_gen/krea/__init__.pyr=Ns3EfXZX^W_1`timeoutcd|d|S)NzKrea poll timed out for job z: r8r9s r<r=r=Os4PQWPXXZ[aZb2cr>invalid_responsec d|S)Nz!Krea poll returned invalid JSON: r8r9s r<r=r=PsBcdjck@lr>c<d|dttd|xsddS)N Krea job z did not complete within zs (last status: unknown))int_POLL_TIMEOUT_SECONDSr9s r<r=r=Qs2 F84 $ % &&6v7J6K1 Nr>)httpr? invalid_jsondeadlinez0Dict[str, Tuple[str, Callable[[str, Any], str]]]_POLL_FAILURES))ztrained styles (LoRAs)styles) moodboardsrNctS)zERead ``image_gen`` (the krea section lives under ``image_gen.krea``).)rr8r>r<_load_krea_configrPYs ""r>c\tjd}t|tr|SiS)Nkrea)rPget isinstancedict)sections r< _krea_sectionrW^s)!%%f-G $/77R7r>c Dtttdd|tS)NKREA_IMAGE_MODELrR)env_var config_keyexplicitconfig)rr DEFAULT_MODELrP)r\s r<_resolve_modelr_cs" (:vX` " $$r>cB ddlm}ddlm}m} |d}|||k7ry| tdry |dS#t $r }t jd|Yd}~yd}~wwxYw#t $rd}YXwxYw#t $r }t jd |Yd}~yd}~wwxYw) aManaged gateway config on the managed path, else ``None``. Managed when the stored ``image_gen`` selection is ``nous`` (or legacy ``use_gateway: true``), or never-configured with no ``KREA_API_KEY``; an explicit vendor selection pins direct. Never raises (discovery scans).r)resolve_managed_tool_gateway)NOUS_MANAGED_PROVIDERread_selectionz/Managed Krea gateway resolution unavailable: %sN image_gen KREA_API_KEYrRz*Managed Krea gateway resolution failed: %s) tools.managed_tool_gatewayratools.tool_backend_helpersrbrc Exceptionloggerdebugr )rarbrcexcselecteds r<_resolve_managed_krea_gatewayrmisKT!+.,A AJ~6+F33  FL   A3Gs=8A$A5 A!AA!$ A21A25 B>BBcP ddlm}t|dS#t$rYywxYw)z?- i %)//*;*A*A*CGX*X??$**, ,- r>cpd|td}|rd|d<|r ttj|d<|S)NzBearer ) Authorizationz User-Agentzapplication/jsonz Content-Typezx-idempotency-key) _USER_AGENTruuuiduuid4) auth_tokenmanaged json_bodyheaderss r<_headersrs>")* 6kRG"4'*4::<'8#$ Nr>c@||jddn t|} ||jni}|jd}t |t r|jd}n$|jdxs|jd}|xs|S#t $r|cYSwxYw)Ni,errormessager;)textrujsonrSrTrUrh)resprkfallbackbodyrrs r<_submit_error_messagers"&"2tyy#CH ".tyy{B! eT "ii *Ghhy)?TXXh-?G"(" sA0B BBct|txr3|jdtvxst |jdS)u``completed_at`` is a backstop terminal marker for unfamiliar ``status`` enums (Krea adds pending states — backlogged/scheduled/sampling — over time).status completed_at)rTrUrS_TERMINAL_STATESrq)jobs r< _is_terminalrs< c4 mcggh&7;K&K<TWT[T[\jTkOlmr>)timeout_secondson_errorc `|d|}t|dd}t}tj|z}d} du} dfd } tj|t |t zt} tj||d} | j | j+}t-|t.r$t-|jdt0r|d} t3|r|Stj|k\r| d| d|t5|S#tj$rx} | j| jjnd}| stjd |||t vstj|k\r| d |d ||cYd} ~ SYd} ~ [d} ~ wtj"tj$f$rL} | stj'd || tj|k\r| d | d|| cYd} ~ SYd} ~ d} ~ wt($rA} | stj|k\rtj'd|| Yd} ~ yYd} ~ d} ~ wwxYw#t($rL} | stj'd|| tj|k\r| d| d|| cYd} ~ SYd} ~ hd} ~ wwxYw)uuPoll ``/jobs/{job_id}`` until terminal; returns the job dict or ``None`` when it gave up. With ``on_error(kind, detail)`` (main path) a fatal poll failure returns that callback's result; without it (best-effort Enhance) failures only log. ``kind`` ∈ ``http`` (detail = status) / ``timeout`` / ``invalid_json`` / ``deadline`` (detail = last status seen). z/jobs/FrrNcHtj|g|y||SN)riwarning)kindr;r warn_argsrs r<give_upz_poll_krea_job..give_ups*   NN7 /Y /f%%r>)rr?rz Krea poll failed (%d) for job %srIz(Krea enhance poll failed (%d) for job %sz(Krea poll transient error for job %s: %sr?z(Krea enhance poll gave up for job %s: %sz.Krea poll returned invalid JSON for job %s: %srJrrKz)Krea enhance job %s did not finish in %ds) rrur;rrrurrreturnr)r_POLL_INITIAL_INTERVALtime monotonicsleepmin _POLL_BACKOFF_POLL_MAX_INTERVALrequestsrSraise_for_status HTTPErrorresponse status_coderir_RETRYABLE_POLL_STATUSESTimeoutConnectionErrorrrhrrTrUrurrG)base_urlrr:rrjob_urlrintervalrK last_statusenhancerrrkrrs ` r<_poll_krea_jobrs^ &)Gz5EBG%H~~/1H!%K$G&  8x-/1CD <<"ED  ! ! #. ))+C c4 Z0A3%Gh-K  J >> x 'K;VSEY[ [U  !! 141IS\\--qF ?P559IX9Uvv/Y[acijj   (":":; I6SVW~~8+y#/Y[acfgg  ~~8+I6SVW    OQWY\]~~8+~s4^`fhkll   s[.(DII(A'FI&I;H<I I1II J-!;J(J-(J-cLt|tr|jdnd}t|tsy|jd}gt|tr|ng|jdD]5}t|ts|j s%|j cSy)zMFirst result URL: ``result.urls[]`` per Krea's job docs, else ``result.url``.resultNurlsurl)rTrUrSlistrurv)rrrrzs r<_extract_result_urlrs",S$"7SWWX TF fd # ::f DS 4 6BSEARS% i %)//*;??$ $% r>c$|t|d} tj|tt ||d|d}|j |j xsijd}t|tr|stjd yt|||} t| tr| jd d vrtjd |yt| S#t$r }tjd|Yd}~yd}~wwxYw) uKrea Enhance on ``image_url`` → enhanced URL, or ``None`` on any failure (best-effort: an upscale failure must never destroy an already-successful generation).) image_urlimage_scaling_factorpromptTrr)rrr?r:zKrea Enhance submit failed: %sNz+Krea Enhance submit response missing job_idr>r2r3z1Krea Enhance job %s did not complete successfully)_ENHANCE_SCALE_FACTORrpost _ENHANCE_PATHrrrrSrhrirrTrurrUr) rrrrrpayloadrr:rkrs r<_enhance_imagers &?T`fgG}}j ((:wbf2g"& ))+#((2 fc "&DE :v 6C c4 CGGH$59P$PJFS s ## 7=sA C&& D/D  Dct||}t|tr|ngD]W}t|tr1|j s$|j |j D|sG|j |Yt }g}|D]9}t|tr||vr|j||j |;|dtS)z``image_url`` + ``reference_image_urls`` first, then legacy ``image_style_references`` (URL strings or Krea ref objects, passed through); strings deduped in order; capped at 10.N) rrTrrurvappendsetadd_MAX_STYLE_REFERENCES)rreference_image_urls legacy_refsrefsrefseendedupedrs r<_collect_style_refsrs ,I7KLD(d;{ c3 yy{ CIIK(  KK   DG  a Dy HHQKq  )) **r>cv||t|d}t|jdtr|d|d<|jd|jd}}t|tr|r||d<|r,|Dcgc]}t|t r |t dn|c}|d<t|tr |r|dd|d<|Scc}w)N)r aspect_ratio resolutionrtseedrMrN)rstrengthimage_style_references)DEFAULT_RESOLUTIONrTrSrGrru!_DEFAULT_STYLE_REFERENCE_STRENGTH) rkrea_arrt style_refskwargsrrMrNrs r<_build_payloadr5s'AScmG&**V$c* .H-vzz,/GJF&$F""- LVVY[^K_C%F Geh h- ()*d# *2A  N - s2"B6c t|d|t||d|ddt\}}||jdk(rx|j|j } } t jd | | |r2d | cxkrd kr'nn$| d k(rd nd|d} d|d|d| d| d| dfSd||jdfS|jdk(r d|ddfS|jdk(rd|d|j dfSd||j|jfS|jd} t| tr| s d|ddfS| dfS)zEPOST the generation request; ``(job_id, None)`` or ``(None, error)``.z/generate/image/krea/krea-2/TrrKrea)rrr?label error_messageNrIzKrea submit failed (%d): %sir1r0u<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. r6r?zKrea submit timed out (30s)rJz&Krea returned invalid JSON on submit: rAr:z#Krea submit response missing job_id) rrrrrrrir error_typerSrTru) rr model_pathrrmodel_idfail submit_bodyfailurererr_msghintr:s r< _submit_jobrKs % *0 =WE6AVXK <<6 !%nngooGF LL6 H3&.3.V\`cVcRhZ(;; T?zJ#HCy4&:!!!gmm[99 9 <<9 $;YGG G <<> ) FwFWXZlmm mT'--););<<< __X &F fc "&T?ASTTT 4<r>cp|jd}|jddk(r8t|tr|jdnd}d|d|d|xsdd fS|jdd k(rd|d|d d fSt|ts d|d d fSt|}| d|dd fS|dfS)zCResult URL of a terminal job; ``(url, None)`` or ``(None, error)``.rrr2rNrDz failed: z unknown errorr6r3z was cancelledz)Krea job completed but result was missingempty_responsez"Krea result contained no image URL)rSrTrUr)rr:rrerrresult_image_urls r<_terminal_result_urlrosWWX F wwxH$%/%=fjj!4TIfXYs7Mo6NOQ\]]] wwxK'TIfX^@PQQQ T !!r>ct|tr|Stjd}t|tr|St|jddS)z[Precedence: explicit kwarg > ``image_gen.krea.upscale`` config > per-model catalog default.rF)rTrqrWrS)r\meta cfg_upscales r<_upscale_requestedrsH(D!/%%i0K$[$7;]T$((9V[B\=]]r>cveZdZdZdZdZeZeZ e dddddd Z dd Z dd Z efd d d  ddZy )KreaImageGenProviderz@Krea ``Krea 2`` foundation image model backend (Medium + Large).rRrpaiduKrea 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.rez Krea API keyz'https://www.krea.ai/settings/api-tokens)namebadgetagkeyrrcBttdxs tS)Nre)rqr rrselfs r< is_availablez!KreaImageGenProvider.is_availablesJ~./P3N3PPr>cddgtddS)NrimageT) modalitiesmax_reference_imagessupports_upscale)rrs r< capabilitiesz!KreaImageGenProvider.capabilitiess!7+EZ $  r>N)rrc  |xsdj}t|}tj|d}t |||jd}|s t d|St } | (| jjd} | j} n&t} td} | std|ddSt|jd \} } t|jd }td|| | t|||||}| $t D]\}}||vs d |d |ddcSt#| | | d|| du| \}||Sgd$fd }t%| | |}rdSt'|t(s ddSt+|\}}||Sd}t-|jd| r.t/| | ||| du}|r|}d}nt0j3d t5t7|d| }|t:||d}|r t<|d<t'|jd t4r|d |d <t?|| ||d|rd!|#Sd"|#S#t8$r#}t0j3d|||}Yd}~d}~wwxYw)%Nr!rrR/reuKREA_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_requiredmodelrt)r rz2Managed Krea (Nous Subscription) does not support z2. Set KREA_API_KEY to use Krea directly, or omit `z`.unsupported_argumentrctj|td\}}j|||dS)NrK)rLrSr)rr;rbuildrr: poll_errorss r< poll_errorz1KreaImageGenProvider.generate..poll_errorsE . 2 24 9S T J   tE&&$9:F Gr? "r>)rrzKrea returned non-dict job bodyrAFr)rTu>Krea Enhance pass failed — returning native-resolution imagekrea_)prefixzEKrea image URL %s could not be cached (%s); falling back to bare URL.)krea_aspect_ratiorrtr:upscaledupscale_factorrrr)rr rrprovidermodalityextra)rrur;rrDict[str, Any]) rvr _ASPECT_MAPrSrrrmgateway_originrstripnous_user_tokenBASE_URLr rr_r{r_MANAGED_UNSUPPORTEDrrrTrUrrrrirrur rhrrr)rrrrrraspectrrrrrrrrtrwhatargrrrrr enhanced_url image_refrkrrr:rs @@@r<generatezKreaImageGenProvider.generates ,B%%'%l3//&%0( +VZZ8P-QS (8 801  --44S9H 00JH#N3J4}VV4( $ %%( 7(;<$(L)AB VV8FK *j&Q  1 0 c'>LTFSKKN%rS.00 0" j$v,9LhX\^  ?J-/  # Xz6JO q> !#t$9;MN N!5S&$ G# ?J fjj3T :)*&6W[H[]L#/ _` )N+;eH:DVWXI")8JZd(!  &;E" # cggn-s 3$'$7E. !8FZ` *WeE E06eE E ) NNWYikn )I  )s.I!! J *JJ rrqrr) rrurrur Optional[str]rOptional[List[str]]rrrr)__name__ __module__ __qualname____doc__ provider_idrr modelsr^default_model_idrUsetuprrr r%r8r>r<rrsJK E F$  6 J >7` bE Q 0DdE#'UYdEdE),dE dE?RdEdE  dEr>rc6|jty)uGPlugin entry point — wire ``KreaImageGenProvider`` into the registry.N)register_image_gen_providerr)ctxs r<registerr5s##$8$:;r>)agent.image_gen_providerImageGenProvider)r6error_response)r6normalize_reference_images)r7r8r9ctj|}|tdtd|ddl}ddlm}|t|g|t|j|d|dS)Nzmodule z has no attribute r) warn_oncer) _PLUGIN_COMPAT_LAZYrSAttributeErrorr* importlibhermes_cli.plugin_compatr;getattr import_module)rtargetr>r;s r< __getattr__rCse $ $T *F ~wxl2DTHMNN2 h&v& 9**6!95vay AAr>r'r)r\r(rzTuple[str, Dict[str, Any]]r&)ryr(rru)rrurrqrrqrzDict[str, str])rrrkrhrru)rrrrq) rrurrur:rurfloatrz Optional[Any]rr)rzOptional[Dict[str, Any]]rr() rrurrurrurrurrqrr()rr(rr)rrr List[Any]) rrurrurtrurrErrrr)rrurrurrurrrrqrrurrr.Tuple[Optional[str], Optional[Dict[str, Any]]])rrr:rurrrrF)r\rrrrrq)rNone)Mr- __future__rloggingrrtypingrrrrrr ragent.secret_scoper agent.image_gen_providerr r r rplugins.image_gen._commonrrrrrrrr getLoggerr*rirr __annotations__ frozensetKREA_MODEL_IDSr^rrrrrxrrrrHrrrrr~rLrrPrWr_rmrrr{rrrrrrrrrrrrr5osr<rCr8r>r<rSs# ==)qq111   8 $  #YZLU "I`LE )8Q7\a& "&7# #eH $'!4 $%MN71 1 ` acd')lmOP D@\# 8 $ 2 nNc"C[C["C[,/C[EJC[C[ C[L $$"$/2$L!W[!gj! !4!H" "!$",3"3"$^zE1zEz< ID"\Br>