epj!8UdZddlmZddlZddlZddlmZmZmZm Z m Z ddl m Z ddl mZmZmZmZmZmZmZmZejeZdZdd d d d d dddd ddddd dZded<dZddddZd*dZd+d!Zd,d%Z Gd&d'eZ!d-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) get_secret)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/thesage/.hermes/hermes-agent/plugins/image_gen/openai/__init__.py_load_openai_configr1Us111111kmm*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)rr1r'r(str DEFAULT_MODEL) env_overrider- openai_cfg candidatevaluetops r0_resolve_modelr@bs:>>"677L3 //W\222   C&01B1BD&I&IQ"""rJ#I*d##w'' eS ! ! ew&6&6Iggg c3   C7NNI'),,, '-0 00refr9Tuple[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_safetyrNopenreadr7pathbasename) rBrQrSrespnamerXheader_b64extrNfhdatas r0_load_image_bytesrhs ))++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!r9cdS)Nr5selfs r0razOpenAIImageGenProvider.namexrAcdS)NOpenAIrlrms r0 display_namez#OpenAIImageGenProvider.display_namerorAboolcVtdsdS ddl}n#t$rYdSwxYwdS)NOPENAI_API_KEYFrT)r r5 ImportError)rnr5s r0 is_availablez#OpenAIImageGenProvider.is_availablesO*++ 5  MMMM   55 ts  &&List[Dict[str, Any]]cHdtDS)NcHg|]\}}||d|d|ddd S)rrrvaries)idrrrpricerl).0model_idmetas r0 z6OpenAIImageGenProvider.list_models..sQ    $  ?g!+.!       rA)ritemsrms r0 list_modelsz"OpenAIImageGenProvider.list_modelss-   #*--//    rA Optional[str]ctS)N)r:rms r0 default_modelz$OpenAIImageGenProvider.default_modelsrAr"cdddddddgdS) NrqpaiduNgpt-image-2 at low/medium/high quality tiers — text-to-image & image editingruzOpenAI API keyz$https://platform.openai.com/api-keys)keyprompturl)rabadgetagenv_varsrlrms r0get_setup_schemaz'OpenAIImageGenProvider.get_setup_schemas4c,.A   rAcddgddS)Ntextimage) modalitiesmax_reference_imagesrlrms r0 capabilitiesz#OpenAIImageGenProvider.capabilitiess &w/LLLrAN) image_urlreference_image_urlsr aspect_ratiorrOptional[List[str]]kwargsrc |pd}t|}|stddd|Std}|stddd|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,t/|dkr|n|d || | dd}n#t&$r>}t0ddtd|dd| ||cYd}~Sd}~wwxYwt,|| d| dd} |jjd,i|}nK#t&$r>}t0ddtd|dd| ||cYd}~Sd}~wwxYwt7|ddpg}|std d!d| ||S|d }t7|d"d}t7|d#d}t7|d$d}|rU t9|d%| &}n/#t&$r"}td'|dd| ||cYd}~Sd}~wwxYwt|}nq|rZ t;|d%| &}t|}nJ#t&$r(}t0d(|||}Yd}~nd}~wwxYwtd)d!d| ||S| | dd*}|r||d$<t?|| ||d||+S)-Nz1Prompt is required and must be a non-empty stringinvalid_argumentr5)error error_typeproviderrruuzOPENAI_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_dependencyrrrr)api_keyz)Could not load source image for editing: io_error)rrrr6rrrHr)r6rrsizernzOpenAI image edit failedT)exc_infozOpenAI image editing failed: api_error)r6rrrrzOpenAI image generation failedz OpenAI image generation failed: rgzOpenAI 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)rr6rrrmodalityextrarl) rPrr r r5rvr@_SIZESr)r'r9appendr rsrqiorhBytesIOrar*imagesedit API_MODELlenr+r,generategetattrrrwarningr) rnrrrrraspectrr5tier_idrrsourcesrBis_editrclientrfilesrgfnamebior/responsepayloadfirstrdrr saved_path image_refrs r0rzOpenAIImageGenProvider.generates^,B%%''%l33 !I-!#  -.. !&+!#      MMMM   !P/!#     '(( zz&&"233  i % % .)//*;*; . NN9??,, - - -./CDDJ  C NN3    #2#,w--%1776w// ?  III "&&C"3C"8"8KD%**T**C$CHLL%%%% &    %KcKK)%!!!'   !=--##&u::>>%%uQx! O .    7$ GGG%?#??*%!!!'  #  ? ''G 16=1<rs .#""""" 33333333333333))))))                      8 $ $  '2 *+ (C &&*%       1111:Lt t t t t -t t t x>>>>>>rA