epjztUdZddlmZddlZddlZddlZddlZddlmZm Z m Z m Z m Z ddl Z ddlmZddlmZmZmZmZmZmZmZejeZdZdd d d d d dddddd dddddd dZded<dZddddZd Z d!Z!hd"Z"d#Z#d$Z$d%Z%d&Z&e'hd'Z(hd(Z)d:d+Z*d;dd5Z.Gd6d7eZ/d?d9Z0dS)@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) get_secret)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.configr0 isinstancedictget Exceptionloggerdebug)r0cfgsectionexcs E/home/thesage/.hermes/hermes-agent/plugins/image_gen/krea/__init__.py_load_krea_configr=xs111111kmm*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) r3strstriprosenvironr5r=r4 DEFAULT_MODEL)r> env_overrider9krea_cfg candidatevaluetops r<_resolve_modelrOs`(C  ;X^^%5%5%@%@~~)9)9!::::>>"455L3 //W\222   C",SWWV__d"C"CKswwvH#I(D!! W%% eS ! ! ew&6&6Iggg c3   C7NNI'),,, '-0 00c: ddlm}ddlm}n3#t$r&}t d|Yd}~dSd}~wwxYwtdr |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_KEYr1rCz*Managed Krea gateway resolution failed: %s)tools.managed_tool_gatewayrRtools.tool_backend_helpersrSr6r7r8r )rRrSr;s r<_resolve_managed_krea_gatewayrWsKKKKKK>>>>>>>  FLLLttttt.!!//+*F*Ft++F333  A3GGGttttts(  ?:? A** B4BBboolc ddlm}n#t$rYdSwxYw t|dS#t$rYdSwxYw)zz4KreaImageGenProvider.list_models..sU    $  ?g!+.g       rP)ritemsrhs r< list_modelsz KreaImageGenProvider.list_modelss-   #*--//    rPr?ctSN)rIrhs r< default_modelz"KreaImageGenProvider.default_modelsrPr.cdddddddgdS) NrmpaiduKrea 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.rTz Krea API keyz'https://www.krea.ai/settings/api-tokens)keyprompturl)rjbadgetagenv_varsrgrhs r<get_setup_schemaz%KreaImageGenProvider.get_setup_schemas7Q*,D   rPcddgddS)Ntextimage ) modalitiesmax_reference_imagesrgrhs r< capabilitiesz!KreaImageGenProvider.capabilities s &w/LLLrPN) 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#d }|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 t/j|d<|d |d!} t3j|||d"#}|n#t2j$r~}|j}||jnd$} ||ni} t | d%t@r)| d%id&n)| d&p| d'p||j!dd(nt |}!n0#tD$r#||j!dd(nt |}!YnwxYwtF$d)||!|Dd*|cxkrd+kr7nn4|d,krd-nd.|d/}"td0|d1|d2|!d3|"d4d |||cYd}~Std5|d2|!d4d |||cYd}~Sd}~wt2j%$rtd6d7d |||cYSt2j&$r"}td8|d9d |||cYd}~Sd}~wwxYw |}#n/#tD$r"}td:|d;d |||cYd}~Sd}~wwxYw|#d<}$t |$t r|$std=d;d |||S|d>|$}%d|dd?}&tN}'tQj)tTz}(d}) tQj+|'tY|'tZzt\}' t3j|%|&d"A}*|*n#t2j$rx}|j}||jnd$}tF$dB||$|t^vstQj)|(kr tdC|dD|$d4d |||cYd}~SYd}~d}~wt2j%t2j&f$r^}tF0dE|$|tQj)|(kr tdF|$dG|d7d |||cYd}~SYd}~Yd}~wwxYw |*}+nh#tD$r[}tF0dH|$|tQj)|(krtdI|d;d |||cYd}~SYd}~d}~wwxYwt |+t@r|+dJnd},t |,t r |,})|,tbvrnwt |+t@r|+dKrnLtQj)|(kr3tdL|$dMt+tTdN|)pdOdPd7d |||St |+t@stdQd;d |||S|)dRkrqt |+dSt@r*|+dSpid%nd}-tdL|$dT|-pdUd4d |||S|)dVkrtdL|$dWdVd |||S|+dS}.t |.t@stdXdYd |||Sd}/|.dZ}0t |0trF|0rD|0D]A}1t |1t r*|1r|1}/nB|/R|.d[}2t |2t r(|2r|2}/|/td\dYd |||S te|/d]|^}3t |3}4n5#tD$r(}tF0d_|/||/}4Yd}~nd}~wwxYw|t(||$d`}5t |+dKt r |+dK|5dK<tg|4|||d ||5aS)bNrimage_style_referencesrrrz1Prompt is required and must be a non-empty stringinvalid_argumentrC)error error_typeproviderr/rTuKREA_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_requiredrDr^styleszManaged Krea (Nous Subscription) does not support trained styles (LoRAs). Set KREA_API_KEY to use Krea directly, or omit `styles`.unsupported_argument)rrrrDrr 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)rrDrrrmodalityextra)4rFr _ASPECT_MAPr5r3rEappendr listsetaddr rWgateway_originrstripnous_user_tokenBASE_URLr rOrcDEFAULT_RESOLUTIONint!_DEFAULT_STYLE_REFERENCE_STRENGTHuuiduuid4requestspostraise_for_status HTTPErrorresponse status_coderr4rr6r7rTimeoutConnectionError_POLL_INITIAL_INTERVALtime monotonic_POLL_TIMEOUT_SECONDSsleepmin _POLL_BACKOFF_POLL_MAX_INTERVAL_RETRYABLE_POLL_STATUSESwarning_TERMINAL_STATESrr)6rirrrrraspectkrea_ar style_refsref legacy_refsseendedupedrrmanagedbase_url auth_tokenrvrwr^payloadrrnormalized_refsrr submit_urlrr;resprbodyerr_msghint submit_bodyrjob_url poll_headersintervaldeadline last_status poll_respjob status_strerrrresult_image_urlrrLsingle saved_path image_refrs6 r<generatezKreaImageGenProvider.generates5,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 0JJH#N33J %, /#!'    ( 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,RY"X #B3UX *VX VAX #Y")X Y" 'Y"2Y"YY"Y"&Y;; Z'Z"Z'"Z'+]>>a< A(`5a<a<A a7+a<7a<b c:A c5)c:5c:o o4 o//o4)r-rEr-rX)r-rq)r-r?r-r.) rrErrErr?rrrrr-r.)__name__ __module__ __qualname____doc__propertyrjrnrprzr}rrr rrgrPr<reresJJ XXQQQQ          MMMM1I $(48 I I I I I I I I rPreNonecH|tdS)uGPlugin entry point — wire ``KreaImageGenProvider`` into the registry.N)register_image_gen_providerre)ctxs r<registerrs###$8$:$:;;;;;rPrr|)r>r?r-r@r)rMr?r-rE)r-r)1r __future__rloggingrGrrtypingrrrrrragent.secret_scoper agent.image_gen_providerr r r r rrr getLoggerrr7rr__annotations__rIrrrr`rrrr frozensetrrr=rOrWr\rcrerrgrPr<rs,#""""" 33333333333333))))))  8 $ $ ! #ZL "`L )Q7 &&0   %(!544   %9%M%M%MNN777    11111@4        &@ @ @ @ @ +@ @ @ P<<<<<