i&UdZddlmZddlZddlZddlmZmZmZm Z m Z ddl m Z m Z mZmZmZmZejeZdZddd d d d d ddd ddddd dZded<dZddddZd%dZd&d ZGd!d"e Zd'd$ZdS)(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_responseresolve_aspect_ratiosave_b64_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 F/home/ubuntu/.hermes/hermes-agent/plugins/image_gen/openai/__init__.py_load_openai_configr.Rs111111kmm*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&rr.r$r%str DEFAULT_MODEL) env_overrider* openai_cfg candidatevaluetops r-_resolve_modelr=_s:>>"677L3 //W\222   C&01B1BD&I&IQ"""rJ#I*d##w'' eS ! ! ew&6&6Iggg c3   C7NNI'),,, '-0 00cneZdZdZeddZeddZddZdd Zdd Z dd Z e fddZ dS)OpenAIImageGenProvideruFOpenAI ``images.generate`` backend — gpt-image-2 at low/medium/high.rr6cdS)Nr2selfs r-namezOpenAIImageGenProvider.namexr>cdS)NOpenAIrBrCs r- display_namez#OpenAIImageGenProvider.display_namerFr>boolcvtjdsdS ddl}n#t$rYdSwxYwdS)NOPENAI_API_KEYFrT)r4r5r&r2 ImportError)rDr2s r- is_availablez#OpenAIImageGenProvider.is_availablesUz~~.// 5  MMMM   55 ts ( 66List[Dict[str, Any]]cHdtDS)NcHg|]\}}||d|d|ddd S)rrrvaries)idrrrpricerB).0model_idmetas r- z6OpenAIImageGenProvider.list_models..sQ    $  ?g!+.!       r>)ritemsrCs r- list_modelsz"OpenAIImageGenProvider.list_modelss-   #*--//    r> Optional[str]ctS)N)r7rCs r- default_modelz$OpenAIImageGenProvider.default_modelsr>rcdddddddgdS) NrHpaidz,gpt-image-2 at low/medium/high quality tiersrLzOpenAI API keyz$https://platform.openai.com/api-keys)keyprompturl)rEbadgetagenv_varsrBrCs r-get_setup_schemaz'OpenAIImageGenProvider.get_setup_schemas4A,.A   r>ra aspect_ratiokwargsrc |pd}t|}|stddd|Stjdstddd|S d dl}n##t$rtd d d|cYSwxYwt\}}t|td }t||d |dd} | } | j j d#i| } nK#t$r>} tddtd| dd|||cYd} ~ Sd} ~ wwxYwt#| ddpg} | stddd|||S| d }t#|dd}t#|dd}t#|dd}|rU t%|d|}n/#t$r"} td| dd|||cYd} ~ Sd} ~ wwxYwt'|}n|r|}ntd dd|||S||dd!}|r||d<t)||||d|"S)$Nz1Prompt is required and must be a non-empty stringinvalid_argumentr2)error error_typeproviderrgrLuzOPENAI_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_dependencyrr)r3rasizenrzOpenAI image generation failedT)exc_infoz OpenAI image generation failed: api_error)rlrmrnr3rargdatazOpenAI returned no image dataempty_responseb64_jsonrbrevised_promptopenai_)prefixzCould not save image to cache: io_errorz2OpenAI response contained neither b64_json nor URL)rrr)imager3rargrnextrarB)stripr r r4r5r&r2rMr=_SIZES API_MODELrHimagesgenerater'r(r)getattrr r6r)rDrargrhaspectr2tier_idrWrrpayloadclientresponser,rvfirstb64rbry saved_path image_refr~s r-rzOpenAIImageGenProvider.generates ,B%%''%l33 !I-!#  z~~.// !&+!#      MMMM   !P/!#     '(( zz&&"233 I # #  ]]__F-v}-8888HH    LL9DL I I I!>>>&!#        x..4" !5+!#  QeZ..eUD)) (8$??   +C8K'8K8KLLL    %ACAA)%!!!'  JII  II!J+!#  *.$y/ J J  5&4E" #     sN.A33BB!&D E3E E E;G G<G71G<7G<N)rr6)rrJ)rrO)rr[rr)rar6rgr6rhrrr) __name__ __module__ __qualname____doc__propertyrErIrNrZr]rfr rrBr>r-r@r@|sPP XX          "1x x x x x x x r>r@NonecH|tdS)uIPlugin entry point — wire ``OpenAIImageGenProvider`` into the registry.N)register_image_gen_providerr@)ctxs r-registerr-s###$:$<$<=====r>r)rr/)rr)r __future__rloggingr4typingrrrrragent.image_gen_providerr r r r r r getLoggerrr(rr__annotations__r7rr.r=r@rrBr>r-rs.#""""" 33333333333333  8 $ $  '2 *+ (C &&*%       1111:i i i i i -i i i b>>>>>>r>