i=1UdZddlmZddlZddlmZmZmZmZm Z ddl m Z m Z m Z mZmZmZejeZdZddd d d d d ddd ddddd dZded<dZddddZdZdZdZd2d!Zd3d#Zd4d%Zd&Zd5d-Z Gd.d/e Z!d6d1Z"dS)7u3OpenAI image generation backend — ChatGPT/Codex OAuth variant. Identical model catalog and tier semantics to the ``openai`` image-gen plugin (``gpt-image-2`` at low/medium/high quality), but routes the request through the Codex Responses API ``image_generation`` tool instead of the ``images.generate`` REST endpoint. This lets users who are already authenticated with Codex/ChatGPT generate images without configuring a separate ``OPENAI_API_KEY``. Selection precedence for the tier (first hit wins): 1. ``OPENAI_IMAGE_MODEL`` env var (escape hatch for scripts / tests) 2. ``image_gen.openai-codex.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`` Output is saved as PNG under ``$HERMES_HOME/cache/images/``. ) 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) landscapesquareportraitzgpt-5.4z%https://chatgpt.com/backend-api/codexzrYou are an assistant that must fulfill image generation requests by using the image_generation tool when provided.returnDict[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 L/home/ubuntu/.hermes/hermes-agent/plugins/image_gen/openai-codex/__init__.py_load_image_gen_configr.Xs111111kmm*4S$*?*?I#''+&&&T$Wd33;ww;  :C@@@ sAA B!B=BBTuple[str, Dict[str, Any]]cfddl}|jd}|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)``.rNOPENAI_IMAGE_MODEL openai-codexmodel) osenvironr&rr.r$r%str DEFAULT_MODEL)r4 env_overrider*sub candidatevaluetops r-_resolve_modelr=es# III:>>"677L3 //W\222 " "C%/0G0G%N%N V#''. ! ! !TVC#I#t   eS ! ! ew&6&6Iggg c3   C7NNI'),,, '-0 00 Optional[str]c ddlm}|}t|tr(|r|SdS#t $r&}t d|Yd}~dSd}~wwxYw)zReturn a usable Codex OAuth token, or None. Delegates to the canonical reader in ``agent.auxiliary_client`` so token expiry, credential pool selection, and JWT decoding stay in one place. r)_read_codex_access_tokenNz(Could not resolve Codex access token: %s)agent.auxiliary_clientrAr$r6stripr'r(r))_readertokenr,s r-rArAs NNNNNN  eS ! ! !ekkmm !;;== t  ?EEEtttttsA A BA<<Bct}|sdS ddl}ddlm}||t ||S#t $r&}td|Yd}~dSd}~wwxYw)zFReturn an OpenAI client pointed at the ChatGPT/Codex backend, or None.Nr)_codex_cloudflare_headers)api_keybase_urldefault_headersz&Could not build Codex image client: %s) rAopenairBrGOpenAI_CODEX_BASE_URLr'r(r))rErKrGr,s r-_build_codex_clientrNs $ & &E t  DDDDDD}}$55e<<    =sCCCttttts/A A4A//A4clientrpromptr6sizercd}|jtdtddd|dgdgdt||d d d d gd dddigd5}|D]}t |dd}|dkrQt |dd}t |dddkr*t |dd} t | tr| r| }j|dkr*t |dd} t | tr| r| }|} dddn #1swxYwYt | ddpgD]A}t |dddkr*t |dd} t | tr| r| }B|S)zHStream a Codex Responses image_generation call and return the b64 image.NFmessageuser input_text)typetext)rVrolecontentimage_generationpngopaque)rVr3rQr output_format backgroundpartial_images allowed_toolsrequiredrV)rVmodetools)r3store instructionsinputrd tool_choicezresponse.output_item.doneitemimage_generation_callresultz,response.image_generation_call.partial_imagepartial_image_b64output) responsesstream_CODEX_CHAT_MODEL_CODEX_INSTRUCTIONS API_MODELgetattrr$r6get_final_response) rOrPrQr image_b64rpevent event_typerjrlpartialfinals r-_collect_image_b64r{s/#I   (!-v>>?   '""   $123  % !  $,.  ( (E 33J888ufd334..2III$T8T::F!&#..+6+$* MMM!%)cdS)NOpenAI (Codex auth)rrs r- display_namez(OpenAICodexImageGenProvider.display_names$$r>boolcTtsdS ddl}n#t$rYdSwxYwdS)NFrT)rArK ImportError)rrKs r- is_availablez(OpenAICodexImageGenProvider.is_availablesL')) 5  MMMM   55 ts  %%List[Dict[str, Any]]cHdtDS)NcHg|]\}}||d|d|ddd S)rrrvaries)idrrrpricer).0model_idmetas r- z;OpenAICodexImageGenProvider.list_models..sQ    $  ?g!+.!       r>)ritemsrs r- list_modelsz'OpenAICodexImageGenProvider.list_modelss-   #*--//    r>r?ctS)N)r7rs r- default_modelz)OpenAICodexImageGenProvider.default_modelsr>rcdddgddS)Nrfreeu;gpt-image-2 via ChatGPT/Codex OAuth — no API key requireduiSign in with `hermes auth codex` (or `hermes setup` → Codex) if you haven't already. No API key needed.)rbadgetagenv_varspost_setup_hintrrs r-get_setup_schemaz,OpenAICodexImageGenProvider.get_setup_schemas")P=   r>rP aspect_ratiokwargsrc |pd}t|}|stddd|Ststddd|S ddl}n##t $rtd d d|cYSwxYwt \}}t|td }t} | td dd||| S t| |||d} nK#t$r>} t ddtd| dd||| cYd} ~ Sd} ~ wwxYw| stddd||| S t| d|} n/#t$r"} td| dd||| cYd} ~ Sd} ~ wwxYwtt!| |||d||ddS)Nriz1Prompt is required and must be a non-empty stringinvalid_argumentr2)error error_typeproviderruoNo Codex/ChatGPT OAuth credentials available. Run `hermes auth codex` (or `hermes setup` → Codex) to sign in. auth_requiredrz8openai Python package not installed (pip install openai)missing_dependencyrz'Could not initialize Codex image client)rrrr3rPrr)rPrQrzCodex image generation failedT)exc_infoz/OpenAI image generation via Codex auth failed: api_errorz8Codex response contained no image_generation_call resultempty_response openai_codex_)prefixzCould not save image to cache: io_error)rQr)imager3rPrrextra)rCr r rArKrr=_SIZESr&rNr{r'r(r)r rr6) rrPrraspectrKtier_idrrQrOb64r, saved_paths r-generatez$OpenAICodexImageGenProvider.generate s ,B%%''%l33 !I-'#  ()) !T+'#   MMMM   !P/'#     '(( zz&&"233$&& >!?*'#   $Y CC     LL84L H H H!MMM&'#         !P+'#   '4MG4M4MNNNJJ   !===%'#         j//#DO<<     sNA""BB"C<< E3D?9E?EE44 F >FF F N)rr6)rr)rrrr?rr)rPr6rr6rrrr) __name__ __module__ __qualname____doc__propertyrrrrrrr rrr>r-r}r}sOO X%%%X%          1c c c c c c c r>r}NonecH|tdS)uDPlugin entry point — register the Codex-backed image-gen provider.N)register_image_gen_providerr})ctxs r-registerrxs###$?$A$ABBBBBr>r)rr/r) rOrrPr6rQr6rr6rr?)rr)#r __future__rloggingtypingrrrrragent.image_gen_providerr r r r r r getLoggerrr(rsr__annotations__r7rrqrMrrr.r=rArNr{r}rrr>r-rs&#"""""33333333333333  8 $ $  '2 *+ (C &&*%   95    11114$&2222tR R R R R "2R R R tCCCCCCr>