yj'UdZddlmZddlZddlZddlZddlmZddlm Z m Z ddl Z ej e ZgdZded<gd Zd ed <ddZddZddZddZddZdddZdS) z8Codex model discovery from API, local cache, and config.) annotationsN)Path)ListOptional) gpt-5.6-solgpt-5.6-sol-pro gpt-5.6-terragpt-5.6-terra-pro gpt-5.6-lunagpt-5.6-luna-progpt-5.5 gpt-5.4-minigpt-5.4 gpt-5.3-codexgpt-5.3-codex-spark List[str]DEFAULT_CODEX_MODELS) )rr r)rr)r r)r r)r r)r r)r )rrr)rr)rr)rrz!List[tuple[str, tuple[str, ...]]]_FORWARD_COMPAT_TEMPLATE_MODELS model_idsreturnc<g}t|D]0}|vr*|||1tD]O\}}|vr t fd|Dr*|||P|S)a Add Clawdbot-style synthetic forward-compat Codex models. If a newer Codex slug isn't returned by live discovery, surface it when an older compatible template model is present. This mirrors Clawdbot's synthetic catalog / forward-compat behavior for GPT-5 Codex variants. c3 K|]}|vV dSN).0templateseens =/home/ragecks/.hermes/hermes-agent/hermes_cli/codex_models.py z-_add_forward_compat_models..Zs(@@Hx4@@@@@@)setappendaddrany)rorderedmodel_idsynthetic_modeltemplate_modelsrs @r _add_forward_compat_modelsr+IsGUUD 4   NN8 $ $ $ HHX   ,K&&( d " "  @@@@@@@ @ @ & NN? + + + HH_ % % % Nr" access_tokenstr Optional[str]c |d}t|dkrdS|ddt|d dzzz}tjt j|}t |tr)|didnd}t |tr|r|ndS#t$rYdSwxYw) uBest-effort extraction of ``chatgpt_account_id`` from the OAuth JWT. The Codex backend requires the ``ChatGPT-Account-Id`` header for the per-account catalog. Without it, ``GET /backend-api/codex/models`` returns ``{"models":[]}`` (HTTP 200) — which masquerades as "no models available" and silently degrades the picker to the curated fallback list. The request-side path in ``auxiliary_client.py`` already extracts the same claim; this mirrors that logic here so the probe sees the same catalog the request path will actually use. Returns ``None`` on any parse error — the probe then degrades gracefully to the unauthenticated fallback list instead of crashing. .N=zhttps://api.openai.com/authchatgpt_account_id) splitlenjsonloadsbase64urlsafe_b64decode isinstancedictgetr- Exception)r,parts payload_b64claimsacct_ids r _extract_chatgpt_account_idrDas ""3'' u::>>4AhU1X(:!;; F4[AABB&$'' FJJ4b 9 9 = =>R S S S  %Wc22HwHwwDH tts(CB%C C C c ddl}dd|i}t|}|r||d<|d|d}|jd krgS|}t |t r|d gng}n4#t$r'}t d |gcYd}~Sd}~wwxYwg}|D]} t | t s| d } t | tr| sX| } | d d} t | tr)| dvr| d} t | ttfrt| nd} || | f|dt#d|DS)zUFetch available models from the Codex API. Returns visible models sorted by priority.rN AuthorizationzBearer zChatGPT-Account-IdzAhttps://chatgpt.com/backend-api/codex/models?client_version=1.0.0 )headerstimeoutmodelsz)Failed to fetch Codex models from API: %sslug visibility>hidehiddenpriority'c"|d|dfSNrr2r)xs r z(_fetch_models_from_api..s1qt r"keycg|]\}}|Srr)r_rLs r z*_fetch_models_from_api..s&D&D&D4t&D&D&Dr")httpxrDr> status_coder8r<r=r?loggerdebugr-striplowerintfloatr$sortr+)r,r\rHrCrespdataentriesexcsortableitemrLrMrQranks r _fetch_models_from_apirls- "$r-r`)rm config_pathrtpayloadrrs r _read_default_modelrxs},K     t tt-- 5 5w 5 G GHH tt$.w$=$= GGKK 4E%%++--{{}} 4s" 00)A A,+A,c|dz }|sgS tj|d}n#t$rgcYSwxYwt |t r|dnd}g}t |tr|D]}t |t s|d}t |tr| sX| }|d}t |tr)| dvr|d}t |ttfrt|nd } || |f|d g} |D]\} }|| vr| || S) Nzmodels_cache.jsonrorprKrLrM>rOrPrQrRc"|d|dfSrTr)rjs r rVz$_read_cache_models..sDGT!W#5r"rW)rsr8r9rur?r<r=r>listr-r`rarbrcr$rd) rm cache_pathrawrgrirjrLrMrQrkdedupedrZs r _read_cache_modelsrs11J      j--w-??@@  $.c4#8#8BcgghdGH'4  * * *DdD)) 88F##DdC((   ::< config.toml default > local cache > hardcoded defaults. CODEX_HOMErNz.codex) osgetenvr`r-rhome expanduserrlr+rxr$rr)r,codex_home_strrmr' api_models default_modelr(s r get_codex_model_idsrs Y|R006688WC h@Vrs]>>>"""""" !!!!!!!!  8 $ $&#&#&#&&&&PFFF$0<)F)F)F)FX$""""J///////r"