Jj7UdZddlmZddlZddlZddlmZmZmZm Z m Z ddl m Z m Z mZmZmZmZmZmZejeZdZddd d d d d ddd ddddd dZded<dZddddZd)dZd*d Zd+d$ZGd%d&e Zd,d(Z dS)-uOpenAI image generation backend. Exposes OpenAI's ``gpt-image-2`` model at three quality tiers as an :class:`ImageGenProvider` implementation. The tiers are implemented as three virtual model IDs so the ``hermes tools`` model picker and the ``image_gen.model`` config key behave like any other multi-model backend: gpt-image-2-low ~15s fastest, good for iteration gpt-image-2-medium ~40s default — balanced gpt-image-2-high ~2min slowest, highest fidelity All three hit the same underlying API model (``gpt-image-2``) with a different ``quality`` parameter. Output is base64 JSON → saved under ``$HERMES_HOME/cache/images/``. Selection precedence (first hit wins): 1. ``OPENAI_IMAGE_MODEL`` env var (escape hatch for scripts / tests) 2. ``image_gen.openai.model`` in ``config.yaml`` 3. ``image_gen.model`` in ``config.yaml`` (when it's one of our tier IDs) 4. :data:`DEFAULT_MODEL` — ``gpt-image-2-medium`` ) annotationsN)AnyDictListOptionalTuple)DEFAULT_ASPECT_RATIOImageGenProvidererror_responsenormalize_reference_imagesresolve_aspect_ratiosave_b64_imagesave_url_imagesuccess_responsez gpt-image-2zGPT 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-lowgpt-image-2-mediumzgpt-image-2-highzDict[str, Dict[str, Any]]_MODELSr 1536x1024 1024x1024 1024x1536) landscapesquareportraitreturnDict[str, Any]c ddlm}|}t|tr|dnd}t|tr|niS#t $r'}t d|icYd}~Sd}~wwxYw)z@Read ``image_gen`` from config.yaml (returns {} on any failure).r) load_config image_genNz#Could not load image_gen config: %s)hermes_cli.configr# isinstancedictget Exceptionloggerdebug)r#cfgsectionexcs G/home/rurouni/.hermes/hermes-agent/plugins/image_gen/openai/__init__.py_load_openai_configr0Ts111111kmm*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)z9Decide which tier to use and return ``(model_id, meta)``.OPENAI_IMAGE_MODELopenaiNmodel) osenvironr(rr0r&r'str DEFAULT_MODEL) env_overrider, openai_cfg candidatevaluetops r/_resolve_modelr?as:>>"677L3 //W\222   C&01B1BD&I&IQ"""rJ#I*d##w'' eS ! ! ew&6&6Iggg c3   C7NNI'),,, '-0 00refr8Tuple[bytes, str]cT|}|}|drpddl}||d}||dddddd pd }|j|fS|d ruddl }| d \}}}d } d|vr8|ddddddpd } | |d| fSddl m } | |t|d5} | } dddn #1swxYwYt j|pd }| |fS)zLoad image bytes from a URL or local file path. Returns ``(data, filename)``. Raises on any network / IO error so the caller can surface a clean error_response. )zhttp://zhttps://rN<)timeout?/z image.pngzdata:,pngzimage/;zimage.)raise_if_read_blockedrb)striplower startswithrequestsr(raise_for_statussplitrsplitcontentbase64 partition b64decodeagent.file_safetyrMopenreadr6pathbasename) rArPrRrespnamerWheader_b64extrMfhdatas r/_load_image_bytesrg~s ))++C IIKKE /00"||C|,, yya  #**32226E+|T!!   5 s++3 v  ,,x++A.44S!<E>c~eZdZdZeddZeddZddZdd Zdd Z dd Z ddZ e fdddddZ dS)OpenAIImageGenProvideruEOpenAI ``images.generate`` / ``images.edit`` backend — gpt-image-2.r r8cdS)Nr4selfs r/r`zOpenAIImageGenProvider.namexr@cdS)NOpenAIrkrls r/ display_namez#OpenAIImageGenProvider.display_namernr@boolcvtjdsdS ddl}n#t$rYdSwxYwdS)NOPENAI_API_KEYFrT)r6r7r(r4 ImportError)rmr4s r/ is_availablez#OpenAIImageGenProvider.is_availablesUz~~.// 5  MMMM   55 ts ( 66List[Dict[str, Any]]cHdtDS)NcHg|]\}}||d|d|ddd S)rrrvaries)idrrrpricerk).0model_idmetas r/ z6OpenAIImageGenProvider.list_models..sQ    $  ?g!+.!       r@)ritemsrls r/ list_modelsz"OpenAIImageGenProvider.list_modelss-   #*--//    r@ Optional[str]ctS)N)r9rls r/ default_modelz$OpenAIImageGenProvider.default_modelsr@r!cdddddddgdS) NrppaiduNgpt-image-2 at low/medium/high quality tiers — text-to-image & image editingrtzOpenAI API keyz$https://platform.openai.com/api-keys)keyprompturl)r`badgetagenv_varsrkrls r/get_setup_schemaz'OpenAIImageGenProvider.get_setup_schemas4c,.A   r@cddgddS)Ntextimage) modalitiesmax_reference_imagesrkrls r/ capabilitiesz#OpenAIImageGenProvider.capabilitiess &w/LLLr@N) image_urlreference_image_urlsr aspect_ratiorrOptional[List[str]]kwargsrc |pd}t|}|stddd|Stjdstddd|S d dl}n##t$rtd d d|cYSwxYwt\}} t|td } g} t|tr;|r'| |t|pgD]} | | | dd } t| } | rdnd}|}| rd dl} g}| D]E} t#| \}}||}||_| |Fn/#t($r"}td|dd|||cYd}~Sd}~wwxYw |jt.t1|dkr|n|d || | dd}n#t($r>}t2ddtd|dd|||cYd}~Sd}~wwxYwt.|| d| dd} |jjd+i|}nK#t($r>}t2ddtd|dd|||cYd}~Sd}~wwxYwt9|ddpg}|stdd d|||S|d }t9|d!d}t9|d"d}t9|d#d}|rU t;|d$|%}n/#t($r"}td&|dd|||cYd}~Sd}~wwxYwt|}nq|rZ t=|d$|%}t|}nJ#t($r(}t2d'|||}Yd}~nd}~wwxYwtd(d d|||S| | dd)}|r||d#<tA||||d||*S),Nz1Prompt is required and must be a non-empty stringinvalid_argumentr4)error error_typeproviderrrtuzOPENAI_API_KEY not set. Run `hermes tools` → Image Generation → OpenAI to configure, or `hermes setup` to add the key. auth_requiredrz8openai Python package not installed (pip install openai)missing_dependencyrrrrz)Could not load source image for editing: io_error)rrrr5rrrGr)r5rrsizernzOpenAI image edit failedT)exc_infozOpenAI image editing failed: api_error)r5rrrrzOpenAI image generation failedz OpenAI image generation failed: rfzOpenAI returned no image dataempty_responseb64_jsonrrevised_promptopenai_)prefixzCould not save image to cache: zGOpenAI image URL %s could not be cached (%s); falling back to bare URL.z2OpenAI response contained neither b64_json nor URL)rr)rr5rrrmodalityextrark)!rOr r r6r7r(r4rur?_SIZESr&r8appendr rrrpiorgBytesIOr`r)imagesedit API_MODELlenr*r+generategetattrrrwarningr)rmrrrrraspectr4tier_idrrsourcesrAis_editrclientrfilesrffnamebior.responsepayloadfirstrcrr saved_path image_refrs r/rzOpenAIImageGenProvider.generatesY,B%%''%l33 !I-!#  z~~.// !&+!#      MMMM   !P/!#     '(( zz&&"233  i % % .)//*;*; . NN9??,, - - -./CDDJ  C NN3    #2#,w--%1776 ?  III "&&C"3C"8"8KD%**T**C$CHLL%%%% &    %KcKK)%!!!'   !=--##&u::>>%%uQx! O .    7$ GGG%?#??*%!!!'  #  ? ''G 16=1<AI J 3JJ J $J77 K?3K:4K?:K?*M?? N+ N& N+&N+O%% P/PP)r r8)r rr)r rw)r rr r!) rr8rr8rrrrrrr r!)__name__ __module__ __qualname____doc__propertyr`rqrvrrrrr rrkr@r/ririsOO XX          MMMM1} $(48 } } } } } } } } r@riNonecH|tdS)uIPlugin entry point — wire ``OpenAIImageGenProvider`` into the registry.N)register_image_gen_providerri)ctxs r/registerrs###$:$<$<=====r@r)r r1)rAr8r rB)r r)!r __future__rloggingr6typingrrrrragent.image_gen_providerr r r r r rrr getLoggerrr*rr__annotations__r9rr0r?rgrirrkr@r/rs.#""""" 33333333333333                      8 $ $  '2 *+ (C &&*%       1111:Ls s s s s -s s s v>>>>>>r@