L0&jH94UdZddlmZddlZddlZddlmZmZmZm Z m Z ddl m Z m Z mZmZmZmZejeZdZddd d d d d ddd ddddd dZded<dZddddZdZdZdZd6d!Zd7d#Zd8d%Zd9d*Z d:d-Z!d;d/Z"du3OpenAI 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.5z%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.Ys111111kmm*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=fs# 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<<Bpromptr6sizerc ftdtddd|dgdgdt||dd d d gd d ddigdddS)zDBuild the Codex Responses request body for an image_generation call.Fmessageuser input_text)typetext)rLrolecontentimage_generationpngopaque)rLr3rGr output_format backgroundpartial_images allowed_toolsrequiredrL)rLmodetoolsT)r3store instructionsinputrZ tool_choicestream)_CODEX_CHAT_MODEL_CODEX_INSTRUCTIONS API_MODELrFrGrs r-_build_responses_payloadrds#+!-v>>?   '""   $123  /  r>r;rcd}t|tr|ddkr.|d}t|tr|r|}|d}t|tr|r|}|D]}t |}|r|}n-t|t r|D]}t |}|r|}|S)zBReturn the newest image b64 embedded in a Responses event payload.NrLimage_generation_callresultpartial_image_b64)r$r%r&r6values_extract_image_b64list)r;foundrgpartialchildnesteds r-rjrjs E% 99V   7 7 7YYx((F&#&& 6 ))/00 gs # #  E\\^^  E'..F   E4   E'..F  Lr>responsec#dKdgfd}|D]}t|tr|dd}t |}|dkr|}||VT|drj|dr*|t dd|d r<|t d d |}||VdSdS) a Yield JSON payloads from an SSE response without OpenAI SDK parsing. The ChatGPT/Codex backend can emit image-generation events newer than the pinned Python SDK understands. Parsing raw SSE keeps this provider tolerant of those event-shape changes. NcsddSd}}dg|r|dkrdStj|}t |t r |r d|vr||d<|S)N z[DONE]rL)joinrCjsonloadsr$r%)raweventpayload data_lines event_names r-flushz_iter_sse_json..flushs J4ii ##))++   cXoo4*S// gt $ $ $ $63H3H#GFOr>zutf-8replace)errors:zevent:zdata:) iter_linesr$bytesdecoder6 startswithlenrCappendlstrip)rpr|lineryrzr{s @@r-_iter_sse_jsonrs^!%JJ ##%%<< dE " " :;;wy;99D4yy 2::eggG"  ??3     ??8 $ $ <c(mmnn-3355JJ __W % % <   d3w<<==188:: ; ; ;eggG r>rEc ddl}ddlm}||}|dd|ddt |||}|d d d d d }d} ||| 5} | d td|5} | n_#|j $rR} | j | j j dd} td| j jd| | d} ~ wwxYwt!| D]}t#|}|r|}  dddn #1swxYwYdddn #1swxYwY| S)zHStream a Codex Responses image_generation call and return the b64 image.rN)_codex_cloudflare_headersztext/event-streamzBearer zapplication/json)Accept Authorizationz Content-Typercgr@g>@)connectreadwritepool)timeoutheadersPOSTz /responses)ruiz"Codex Responses API returned HTTP z: )httpxrBrupdaterdTimeoutClientr__CODEX_BASE_URLraise_for_statusHTTPStatusErrorrprrM RuntimeError status_coderrj)rErFrGrrrrryr image_b64httprpr,bodyrxrls r-_collect_image_b64rsBLLL@@@@@@''..G NN%*5*** 'f4QQQGmmE4e4dmSSG#I gw 7 7 &4 [[O!?!?!?g[ N N &RZ ))++++(    !!###|(#."[9Q[[UY[[  (11 & &*511& %I & & & & & & & & & & & & & & & & & & & & & & & & & & & & & & & s[7!EEB/.E/ D 9A DD  (E4 EE EE EEEcneZdZdZeddZeddZddZdd Zdd Z dd Z e fddZ dS)OpenAICodexImageGenProviderzEgpt-image-2 routed through ChatGPT/Codex OAuth instead of an API key.rr6cdS)Nr2selfs r-namez OpenAICodexImageGenProvider.names~r>cdS)NOpenAI (Codex auth)rrs r- display_namez(OpenAICodexImageGenProvider.display_name"s$$r>boolcTtsdS ddl}n#t$rYdSwxYwdS)NFrT)rAr ImportError)rrs r- is_availablez(OpenAICodexImageGenProvider.is_available&sL')) 5  LLLL   55 ts  %%List[Dict[str, Any]]cHdtDS)NcHg|]\}}||d|d|ddd S)rrrvaries)idrrrpricer).0model_idmetas r- z;OpenAICodexImageGenProvider.list_models..0sQ    $  ?g!+.!       r>)ritemsrs r- list_modelsz'OpenAICodexImageGenProvider.list_models/s-   #*--//    r>r?ctS)N)r7rs r- default_modelz)OpenAICodexImageGenProvider.default_model;sr>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_schema>s")P=   r>rF aspect_ratiokwargsrc |pd}t|}|stddd|Ststddd|S ddl}n##t $rtd d d|cYSwxYwt \}}t|td }t} | stddd||| 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||d dS)Nrz1Prompt 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_requiredrz6httpx Python package not installed (pip install httpx)missing_dependencyr)rrrr3rFrrrczCodex 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)rGr)imager3rFrrextra)rCr r rArrr=_SIZESr&rr'r(r)r rr6) rrFrraspectrtier_idrrGrEb64r, saved_paths r-generatez$OpenAICodexImageGenProvider.generateJs ,B%%''%l33 !I-'#  ()) !T+'#   LLLL   !N/'#     '(( zz&&"233(** !T+'#     $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)rFr6rr6rrrr) __name__ __module__ __qualname____doc__propertyrrrrrrr rrr>r-rrsOO X%%%X%          1f f f f f f f r>rNonecH|tdS)uDPlugin entry point — register the Codex-backed image-gen provider.N)register_image_gen_providerr)ctxs r-registerrs###$?$A$ABBBBBr>r)rr/r)rFr6rGr6rr6rr)r;rrr?)rpr) rEr6rFr6rGr6rr6rr?)rr)&r __future__rruloggingtypingrrrrragent.image_gen_providerr r r r r r getLoggerrr(rbr__annotations__r7rr`rrar.r=rArdrjrrrrrr>r-rsR&#""""" 33333333333333  8 $ $  '2 *+ (C &&*%   95    11114$:.,,,,^LU U U U U "2U U U zCCCCCCr>