(Gj\t|UdZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl Z ddlmZmZmZmZmZmZmZejeZdZddd d d d d ddddd dddddd dZded<dZddddZdZd Zhd!Z d"Z!d#Z"d$Z#d%Z$e%hd&Z&hd'Z'd9d*Z(d:d;d.Z)d/Z*dd8Z.dS)?uKrea 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 completedreturnDict[str, Any]c ddlm}|}t|tr|dnd}t|tr|niS#t $r'}t d|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 E/home/rurouni/.hermes/hermes-agent/plugins/image_gen/krea/__init__.py_load_krea_configr<ws111111kmm*4S$*?*?I#''+&&&T$Wd33;ww;  :C@@@ sAA B!B=BBexplicit Optional[str]Tuple[str, Dict[str, Any]]c.t|trN|tvr3|t|fStjd}|r|tvr|t|fSt}t|dtr|dni}d}t|tr5|d}t|tr |tvr|}|5|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) r2strstriprosenvironr4r<r3 DEFAULT_MODEL)r= env_overrider8krea_cfg candidatevaluetops r;_resolve_modelrNs`(C  ;X^^%5%5%@%@~~)9)9!::::>>"455L3 //W\222   C",SWWV__d"C"CKswwvH#I(D!! W%% eS ! ! ew&6&6Iggg c3   C7NNI'),,, '-0 00cZ ddlm}ddlm}n3#t$r&}t d|Yd}~dSd}~wwxYwtj dr |dsdS |dS#t$r&}t d |Yd}~dSd}~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_KEYr0rBz*Managed Krea gateway resolution failed: %s) tools.managed_tool_gatewayrQtools.tool_backend_helpersrRr5r6r7rFrGr4)rQrRr:s r;_resolve_managed_krea_gatewayrVsKKKKKK>>>>>>>  FLLLttttt z~~n%%ook.J.Jt++F333  A3GGGttttts(  ?:?/ A:: B*B%%B*boolc ddlm}n#t$rYdSwxYw t|dS#t$rYdSwxYw)z2233T7R7T7TTrOList[Dict[str, Any]]cHdtDS)NcTg|]%\}}||d|d|d|dd&S)rrrr)idrrrrrf).0model_idmetas r; z4KreaImageGenProvider.list_models..sU    $  ?g!+.g       rO)ritemsrgs r; list_modelsz KreaImageGenProvider.list_modelss-   #*--//    rOr>ctSN)rHrgs r; default_modelz"KreaImageGenProvider.default_modelsrOr-cdddddddgdS) NrlpaiduKrea 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.rSz Krea API keyz'https://www.krea.ai/settings/api-tokens)keyprompturl)ribadgetagenv_varsrfrgs r;get_setup_schemaz%KreaImageGenProvider.get_setup_schemas7Q*,D   rOcddgddS)Ntextimage ) modalitiesmax_reference_imagesrfrgs r; capabilitiesz!KreaImageGenProvider.capabilities s &w/LLLrON) image_urlreference_image_urlsr aspect_ratiorrOptional[List[str]]kwargsrc |pd}t|}t|d}g}t |t r;|r'||t|pgD]} || |d} t | trm| D]j} t | t r<| r'|| S| r|| kt} g} |D]F} t | t r| | vr| | | | G| dd}|rdnd}|stddd | St}|"|j d }|j}n;t }t"jd }|std dd | St'|d\}}t)|d}|t |dtr*|drtddd |||St |dtr*|drtddd |||S||t*|d}|d}t |t,r||d<|d}t |tr|r||d<|rTg}|D]J} t | t r|| t.d5|| K||d<|d}t |tr|r |dd|d<d|ddd}|#t t1j|d<|d |d!} t5j|||d"#}|n#t4j$r~}|j}||jnd$} || ni} t | d%tBr)| d%id&n)| d&p| d'p||j"dd(nt |}!n0#tF$r#||j"dd(nt |}!YnwxYwtH%d)||!|Dd*|cxkrd+kr7nn4|d,krd-nd.|d/}"td0|d1|d2|!d3|"d4d |||cYd}~Std5|d2|!d4d |||cYd}~Sd}~wt4j&$rtd6d7d |||cYSt4j'$r"}td8|d9d |||cYd}~Sd}~wwxYw | }#n/#tF$r"}td:|d;d |||cYd}~Sd}~wwxYw|#d<}$t |$t r|$std=d;d |||S|d>|$}%d|dd?}&tP}'tSj*tVz}(d}) tSj,|'t[|'t\zt^}' t5j|%|&d"A}*|*n#t4j$rx}|j}||jnd$}tH%dB||$|t`vstSj*|(kr tdC|dD|$d4d |||cYd}~SYd}~d}~wt4j&t4j'f$r^}tH1dE|$|tSj*|(kr tdF|$dG|d7d |||cYd}~SYd}~Yd}~wwxYw |* }+nh#tF$r[}tH1dH|$|tSj*|(krtdI|d;d |||cYd}~SYd}~d}~wwxYwt |+tBr|+dJnd},t |,t r |,})|,tdvrnwt |+tBr|+dKrnLtSj*|(kr3tdL|$dMt-tVdN|)pdOdPd7d |||St |+tBstdQd;d |||S|)dRkrqt |+dStBr*|+dSpid%nd}-tdL|$dT|-pdUd4d |||S|)dVkrtdL|$dWdVd |||S|+dS}.t |.tBstdXdYd |||Sd}/|.dZ}0t |0trF|0rD|0D]A}1t |1t r*|1r|1}/nB|/R|.d[}2t |2t r(|2r|2}/|/td\dYd |||S tg|/d]|^}3t |3}4n5#tF$r(}tH1d_|/||/}4Yd}~nd}~wwxYw|t*||$d`}5t |+dKt r |+dK|5dK<ti|4|||d ||5aS)bNrimage_style_referencesrrrz1Prompt is required and must be a non-empty stringinvalid_argumentrB)error error_typeproviderr/rSuKREA_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_requiredrCr]styleszManaged Krea (Nous Subscription) does not support trained styles (LoRAs). Set KREA_API_KEY to use Krea directly, or omit `styles`.unsupported_argument)rrrrCrr moodboardszzManaged Krea (Nous Subscription) does not support moodboards. Set KREA_API_KEY to use Krea directly, or omit `moodboards`.)rr resolutionr]seed)rstrengthzBearer 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/)rrT)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_responseurlsrz"Krea result contained no image URLkrea_)prefixzEKrea image URL %s could not be cached (%s); falling back to bare URL.)krea_aspect_ratiorr]r)rrCrrrmodalityextra)5rEr _ASPECT_MAPr4r2rDappendr listsetaddr rVgateway_originrstripnous_user_tokenBASE_URLrFrGrNrbDEFAULT_RESOLUTIONint!_DEFAULT_STYLE_REFERENCE_STRENGTHuuiduuid4requestspostraise_for_status HTTPErrorresponse status_coderr3rr5r6rTimeoutConnectionError_POLL_INITIAL_INTERVALtime monotonic_POLL_TIMEOUT_SECONDSsleepmin _POLL_BACKOFF_POLL_MAX_INTERVAL_RETRYABLE_POLL_STATUSESwarning_TERMINAL_STATESrr)6rhrrrrraspectkrea_ar style_refsref legacy_refsseendedupedrrmanagedbase_url auth_tokenrurvr]payloadrrnormalized_refsrr submit_urlrr:resprbodyerr_msghint submit_bodyrjob_url poll_headersintervaldeadline last_status poll_respjob status_strerrrresult_image_urlrrKsingle saved_path image_refrs6 r;generatezKreaImageGenProvider.generates;,B%%''%l33//&%00!# i % % 1)//*;*; 1   ioo// 0 0 0./CDDJ # #C   c " " " "jj!9:: k4 ( ( +" + +c3''+yy{{7"))#))++666+%%c***EE  A!S!! 99 NN1    SbS\ (477f !I-#  011  -44S99H 0JJH77J %, /#!'    ( 7(;(;<<$(L)A)ABB  &**X..55 &**X:N:N %6 6#"!!'    &**\22D99 fjj>V>V %- 6#"!!'    #,$ # # zz&!! dC  #"GFOH%% fd # # ' ' &GH   @ *,O! 0 0c3''0#** #1RSS$**3////0?G, -ZZ -- j$ ' ' 3J 3$.rrNGL !4z33.=     ,/tz||+<+H>>!'>>,3>>7;>> +#"!!'         "KvKK'KK&#          !3$#    '   !555-#         "--//KK   !DsDD-#        **&#&& f !;-#  --V--3z33=  *>##&;;%) O  Jx 8m35GHHH" $L,PRSSS **,,,,%   |-1-=))1 ?PPP !999T^=M=MQY=Y=Y)M6MMVMM#.!'&%%+ $h&>?   I6SVWWW>##x//)LVLLsLL#,!'&%%+   nn&&   OQWY\]]]>##x//)G#GG#5!'&%%+  /9d.C.CM***J*c** ( !111 #t$$ )@)@ ~8++%cFcc455ccGRG_V_ccc )#"!!'    KO d#t$$ !7-#   ( " "> 2#)<<>>  #!:+#   ('(8ASASASTTTJJII ) ) ) NNW     )IIIIII  )"),$ ! !  cggn--s 3 3 8$'$7E. !    s",RY2 X3B3U'&X'*VXVAX3Y29XY2'Y2Y2Y-'Y2-Y26Z Z7Z2,Z72Z7!+^b A(`b b .A b;b b b%% d /A d9d d .o po??p)r,rDr,rW)r,rp)r,r>r,r-) rrDrrDrr>rrrrr,r-)__name__ __module__ __qualname____doc__propertyrirmroryr|rrr rrfrOr;rdrdsJJ XXUUUU          MMMM1I $(48 I I I I I I I I rOrdNonecH|tdS)uGPlugin entry point — wire ``KreaImageGenProvider`` into the registry.N)register_image_gen_providerrd)ctxs r;registerrs###$8$:$:;;;;;rOrr{)r=r>r,r?r)rLr>r,rD)r,r)/r __future__rloggingrFrrtypingrrrrrragent.image_gen_providerr r r r r rr getLoggerrr6rr__annotations__rHrrrr_rrrr frozensetrrr<rNrVr[rbrdrrfrOr;rs,#""""" 33333333333333  8 $ $ ! #ZL "`L )Q7 &&0   %(!544   %9%M%M%MNN777    11111@4        &@ @ @ @ @ +@ @ @ P<<<<<