jONUdZddlmZddlZddlZddlZddlmZmZm Z m Z m Z ddl Z ddl mZmZmZmZmZmZejeZdZddd d d d d ddddd dZded<dZddddZdZhdZdZdZdZ dZ!e"hd Z#hd!Z$d/d$Z%d0d&Z&d1d*Z'Gd+d,eZ(d2d.Z)dS)3uKrea 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_responseresolve_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)large) krea-2-mediumz krea-2-largezDict[str, Dict[str, Any]]_MODELSrz16:91:1z9:16) landscapesquareportrait1K>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 >/usr/local/lib/hermes-agent/plugins/image_gen/krea/__init__.py_load_krea_configr;is111111kmm*4S$*?*?I#''+&&&T$Wd33;ww;  :C@@@ sAA B!B=BBTuple[str, Dict[str, Any]]chtjd}|r|tvr|t|fSt }t |dt r|dni}d}t |t r5|d}t |tr |tvr|}|5|d}t |tr |tvr|}||t|fStttfS)z:Decide which model to use and return ``(model_id, meta)``.KREA_IMAGE_MODELkreaNmodel) osenvironr3rr;r1r2str DEFAULT_MODEL) env_overrider7krea_cfg candidatevaluetops r:_resolve_modelrJvs:>>"455L3 //W\222   C",SWWV__d"C"CKswwvH#I(D!! W%% eS ! ! ew&6&6Iggg c3   C7NNI'),,, '-0 00rH Optional[str]rCcTt|tr1|}|tvr|St }t|dtr|dni}t|tr|dnd}t|trS|tvr&|SdS)zGCoerce ``creativity`` kwarg to a valid Krea value (default ``medium``).r? creativityNr)r1rCstriplower_VALID_CREATIVITYr;r3r2)rHvr7rF cfg_values r:_resolve_creativityrTs% KKMM   ! ! ! ! !H   C",SWWV__d"C"CKswwvH.84.H.HR \***dI)S!!)ioo&7&7&=&=&?&?CT&T&T  &&((( 8rKcneZdZdZeddZeddZddZdd Zdd Z dd Z e fddZ dS)KreaImageGenProviderz@Krea ``Krea 2`` foundation image model backend (Medium + Large).r+rCcdS)Nr?selfs r:namezKreaImageGenProvider.namevrKcdS)NKrearXrYs r: display_namez!KreaImageGenProvider.display_namer\rKboolcZttjdS)N KREA_API_KEY)r`rArBr3rYs r: is_availablez!KreaImageGenProvider.is_availablesBJNN>22333rKList[Dict[str, Any]]cHdtDS)NcTg|]%\}}||d|d|d|dd&S)rrrr)idrrrrrX).0model_idmetas r: z4KreaImageGenProvider.list_models..sU    $  ?g!+.g       rK)ritemsrYs r: list_modelsz KreaImageGenProvider.list_modelss-   #*--//    rKrLctS)N)rDrYs r: default_modelz"KreaImageGenProvider.default_modelsrKr,cdddddddgdS) Nr^paidu_Krea 2 foundation model — Medium ($0.03) + Large ($0.06). Strong style transfer + moodboards.rbz Krea API keyz'https://www.krea.ai/settings/api-tokens)keyprompturl)r[badgetagenv_varsrXrYs r:get_setup_schemaz%KreaImageGenProvider.get_setup_schemas4t*,D   rKrs aspect_ratiokwargsrc |pd}t|}t|d}|st ddd|St jd}|st dd d|St\}}t|d } ||t| d } |d } t| tr| | d <|d } t| tr| r| | d <|d} t| tr| r | dd| d<|d}t|tr|r |dd| d<d|ddd}td|d} tj||| d}|n#tj$r8}|j}||jnd} ||ni}t|dt*r)|didn)|dp|dp||jddnt/|}n0#t0$r#||jddnt/|}YnwxYwt2d||t d |d!|d"d|||#cYd}~Sd}~wtj$rt d$d%d|||#cYStj$r"}t d&|d'd|||#cYd}~Sd}~wwxYw |}n/#t0$r"}t d(|d)d|||#cYd}~Sd}~wwxYw|d*}t|t.r|st d+d)d|||#Std,|}d|dd-}t:}t=jt@z}d} t=j!|tE|tFztH} tj||d/}|n#tj$rx}|j}||jnd}t2d0|||tJvst=j|kr t d1|d2|d"d|||#cYd}~SYd}~d}~wtjtjf$r^}t2&d3||t=j|kr t d4|d5|d%d|||#cYd}~SYd}~Yd}~wwxYw |}nh#t0$r[}t2&d6||t=j|krt d7|d)d|||#cYd}~SYd}~d}~wwxYwt|t*r|d8nd} t| t.r | }| tNvrnwt|t*r|d9rnLt=j|kr3t d:|d;tt@d<|pd=d>d%d|||#St|t*st d?d)d|||#S|d@krqt|dAt*r*|dApidnd}!t d:|dB|!pdCd"d|||#S|dDkrt d:|dEdDd|||#S|dA}"t|"t*st dFdGd|||#Sd}#|"dH}$t|$trF|$rD|$D]A}%t|%t.r*|%r|%}#nB|#R|"dI}&t|&t.r(|&r|&}#|#t dJdGd|||#S tQ|#dK|L}'t/|'}(n5#t0$r(}t2&dM|#||#}(Yd}~nd}~wwxYw|t| |dN})t|d9t.r |d9|)d9<tS|(|||d|)OS)PNrz1Prompt is required and must be a non-empty stringinvalid_argumentr?)error error_typeproviderryrbuKREA_API_KEY not set. Run `hermes tools` → Image Generation → Krea to configure, or get a key at https://www.krea.ai/settings/api-tokens. auth_requiredrN)rsry resolutionrNseedstylesimage_style_references moodboardszBearer zapplication/jsonz!Hermes-Agent/1.0 (krea-image-gen)) Authorizationz Content-Type User-Agentz/generate/image/krea/krea-2/r)headersjsontimeoutrr~messagedetaili,zKrea submit failed (%d): %szKrea image generation failed (z): api_error)r~rrr@rsryzKrea 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_responseurlsrtz"Krea result contained no image URLkrea_)prefixzEKrea image URL %s could not be cached (%s); falling back to bare URL.)krea_aspect_ratiorrNr)imager@rsryrextra)*rOr _ASPECT_MAPr3r rArBrJrTDEFAULT_RESOLUTIONr1intlistBASE_URLrequestspostraise_for_status HTTPErrorresponse status_coderr2textrCr4r5r~TimeoutConnectionError_POLL_INITIAL_INTERVALtime monotonic_POLL_TIMEOUT_SECONDSsleepmin _POLL_BACKOFF_POLL_MAX_INTERVAL_RETRYABLE_POLL_STATUSESwarning_TERMINAL_STATESr r)*rZrsryrzaspectkrea_arapi_keyrirjrNpayloadrrrrr submit_urlrr9resprbodyerr_msg submit_bodyrjob_url poll_headersintervaldeadline last_status poll_respjob status_strerrr image_urlrrGsingle saved_path image_refrs* r:generatezKreaImageGenProvider.generatesp ,B%%''%l33//&%00 !I-#  *..00 !?+#    ())$(L)A)ABB #,$ # # zz&!! dC  #"GFOH%% fd # # ' ' &GH !',D!E!E ,d 3 3 L8N L0Fss0KG, -ZZ -- j$ ' ' 3J 3$.rrNGL !1w00.=  !LLd6lLL . } H  % % ' ' ' '!   ##&;;%) 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-#   ( " "> +"LLNN  !:+#   (' :L(:L:LMMMJJII " " " NNW    "IIIIII  ""),$ ! !  cggn--s 3 3 8$'$7E. !     s7,G%%N5L(B3J<;L(<*K)&L((K))9L("N('NNM<6N<NN O$O;OO5+R""V 1A(T$V $V A VV V $V99 XA X XXc&& d0ddN)r+rC)r+r`)r+rd)r+rLr+r,)rsrCryrCrzrr+r,) __name__ __module__ __qualname____doc__propertyr[r_rcrmrorxr rrXrKr:rVrVsJJ XX4444          *1J J J J J J J rKrVNonecH|tdS)uGPlugin entry point — wire ``KreaImageGenProvider`` into the registry.N)register_image_gen_providerrV)ctxs r:registerr"s###$8$:$:;;;;;rKr)r+r<)rHrLr+rC)r+r)*r __future__rloggingrArtypingrrrrrragent.image_gen_providerr r r r r r getLoggerrr5rr__annotations__rDrrrQrrrr frozensetrrr;rJrTrVrrXrKr:rs7,#""""" 33333333333333  8 $ $ ! #ZL "`L &&"  544   %9%M%M%MNN777    11110    &y y y y y +y y y B <<<<<