߲j3UdZddlmZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZddlmZmZmZmZmZej*eZddd d Zd ed <d ZdZddddddddddddddddZded<dZd Ze d!e eeffZ d?d"Z!d@dAd#Z"dd$dd%d& dBd'Z# d@ dCd(Z$ dDdd) dEd*Z% dFd+Z&Gd,d-eZ'd.d.d/dGd0Z(dHd1Z)dId2Z*dJd3Z+edd4 dKd5Z,edd6 dLd7Z-dMd8Z.eGd9d:Z/dd;dNd<Z0e.d%d= dOd>Z1y)PzShared helpers for the bundled ``image_gen`` provider plugins. Providers are loaded by path (``hermes_plugins.image_gen__``) and resolve this via the repo root on ``sys.path``; not a plugin itself (the scanner only looks at directories).) annotationsN) dataclass)AnyCallableDictIterableListOptionalTuple)ImageGenProvidererror_responsenormalize_reference_imagessave_b64_imagesave_url_image 1536x1024 1024x1024 1024x1536) landscapesquareportraitDict[str, str] OPENAI_SIZESz gpt-image-2gpt-image-2-mediumzGPT Image 2 (Low)z~15szFast iteration, lowest costlow)displayspeed strengthsqualityzGPT Image 2 (Medium)z~40suBalanced — defaultmediumzGPT Image 2 (High)z~2minz,Highest fidelity, strongest prompt adherencehigh)zgpt-image-2-lowrzgpt-image-2-highDict[str, Dict[str, Any]]GPT_IMAGE_2_TIERSz1Prompt is required and must be a non-empty stringz8openai Python package not installed (pip install openai).c<tj|tdS)zCOpenAI ``size`` string for a semantic aspect (square when unknown).r)rget)aspects >/home/ubuntu/.hermes/hermes-agent/plugins/image_gen/_common.pysize_forr'2s   FL$: ;;cF|dnd|} ddlm}|}t|tr|j dnd}|#t|tr|j |nd}t|tr|SiS#t $r#}t jd||icYd}~Sd}~wwxYw)zTRead ``image_gen`` (or ``image_gen.``) from config.yaml; ``{}`` on any failure.N image_genz image_gen.r) load_configzCould not load %s config: %s)hermes_cli.configr+ isinstancedictr$ Exceptionloggerdebug)sublabelr+cfgsectionexcs r&load_image_gen_configr77s;Kj,>E 1m*4S$*?#''+&T ?*4Wd*Cgkk#&G$Wd3w;;  3UC@ s$A&A42A44 B =BB B TF)explicitinclude_top_levelconfig passthroughcNt|tr5|j|vr#|j||jfStjj |}|r ||vr|||fS| t n|} | j |} t| tr| j dndg} |r | j| j d| D] } t| ts| |vs| || fcS|r"td|| dfDd} | r| | | ddfS|||fS)uB``(model_id, meta)`` from a fixed catalog; first *known* id wins (unknown ids fall through): explicit → ``env_var`` → ``image_gen..model`` → ``image_gen.model`` → ``default``. ``passthrough``: an unknown id from ``env_var`` or the provider-scoped ``image_gen..model`` is sent verbatim as the API model with no ``quality`` (OpenAI-compatible gateways serve their own image model names and reject unknown enum values, #97928). The shared top-level ``image_gen.model`` never passes through — it may hold another provider's id.Nmodelc3K|]6}t|ts|js%|j8ywN)r-strstrip).0cs r& z'resolve_static_model..as-lQ:VWY\K]bcbibibkqwwyls >>>r)r api_modelr) r-r@rAosenvironr$r7r.appendnext)modelsdefaultenv_var config_keyr8r9r:r; env_overrider4scoped candidates candidatecustoms r&resolve_static_modelrTGs.(C X^^%5%?~~(8!999::>>'*L .VL111%+^  !C WWZ F)3FD)A&**W%tLJ#'''*+0 i %)v*=fY// /0l<A*Glnpq vFtTT T F7O ##r(cg}t|tr/|jr|j|j|j t |xsg||d|S|S)zNPrimary ``image_url`` first, then normalized references, clamped to ``limit``.N)r-r@rArIextendr) image_urlreference_image_urlslimitsourcess r&collect_source_imagesr[gs]G)S!ioo&7y() NN-.BCIrJ#/76E?j r(ceZdZUdZded<ded<iZded<dZded <dZded <d Zd ed <iZ ded<e ddZ e ddZ ddZ ddZddZy)StaticImageGenProvideraIdentity + picker surface from class attributes: ``provider_id``/``label``; fixed catalog via ``models`` (+ ``default_model_id``, ``price``, ``catalog_fields``); single-env-var auth via ``setup`` (kwargs for :func:`api_key_setup_schema`). Dynamic providers override methods.r@ provider_idr3r!rKN Optional[str]default_model_idr]rrrr]zTuple[str, ...]catalog_fieldsDict[str, Any]setupc|jSr?)rtselfs r&rlzStaticImageGenProvider.namesr(c|jSr?)r3r|s r& display_namez#StaticImageGenProvider.display_names zzr(cZt|j|j|jS)Nr\)rgrKrxr]r|s r& list_modelsz"StaticImageGenProvider.list_modelssDKK)<)   P%2r(rsrEr=rjc dfd }|S)zPReturn ``fail(error, error_type, **override)`` pre-bound to this call's context.c`t}|j|td||d|S)N)providerr=rj aspect_ratio)error error_typerp)r.updater )rroverridekwargsr%r=rjrs r&failzerror_factory..fails5xuVRXY hKEjKFKKr()rr@rr@rrrryrp)rr%r=rjrs```` r& error_factoryrsLL Kr(c0t||tdS)Ninvalid_argument)rPROMPT_REQUIRED)rr%s r&prompt_required_errorrs *=6 *?#f566    CL KK]_dfikn o s A  AA A c |jjdijd|jddS#t$r!||jddcYSt |cYSwxYw)zJ``error.message`` from an HTTP error body, else the first 300 chars of it.rmessageNi,)jsonr$textr/r@)responser6s r&requests_error_messagersiI}}""7B/33Ix}}Tc?RSS I&.&:x}}Tc"HCHIsO=PPRSU^`` `  # #g[%8KC5/QSefff  $ $[& [ug5Fse,LkZZZ[  [ ug%=cUCEWadehai    se%AD8D5$?B)#D5)0D5D5)D<D5D5D0*D50D58 E*E%E*%E*)r%r@rr@r?)r2rurry)rKr!rLr@rMr@rNr@r8rur9boolr:zOptional[Dict[str, Any]]r;rrzTuple[str, Dict[str, Any]])rWrurXzOptional[List[str]]rYz Optional[int]rz List[str])rw)rKr!raz Iterable[str]r]rurr)rlr@rmr@rnr@rir@rjr@rkr@rry) rr@r%r@r=r@rjr@rErrorFn)rr@r%r@rry)rr)rr@r%r@rz$Tuple[Any, Optional[Dict[str, Any]]])rrurkrurr@r3r@rr@r=r@rjr@r%r@rlogging.Loggerr%Optional[Callable[[Exception], None]]rz.Tuple[Optional[str], Optional[Dict[str, Any]]]) rkr@rr@r3r@rrrrrr@)rrr6r/rr@) rrr=r@rr@rrurNone)rkr@rrrryrrr3r@rzCallable[[Any, Exception], str]rrrz+Tuple[Optional[Any], Optional[HttpFailure]])2r __future__rloggingrG dataclassesrtypingrrrrr r r agent.image_gen_providerr r rrr getLoggerrr0rrGPT_IMAGE_2_API_MODELGPT_IMAGE_2_DEFAULTr"rrr@rr'r7rTr[rgrqrsrrrrrrrrrrrpr(r&rs<@# !GGGbb   8 $.9K]hi ni&*'2 *+ (C 0,*FK 3S#X& '< $#d'+$ %$03$AD$RU$$7;$ %$;?$ $Bae==4G=P]==G  %   "  #-0:=DG2-2>?APRP05;9=` `*`7:`CF`RU`^a` ``#1`7`4 `&AG59    %( /=  2   I     \`m*6L$)# # (# 3A# LO# X[# 2# "# 1 # r(