iUdZddlmZddlZddlZddlmZddlmZm Z ddl Z ej e Z gdZded<gd Zd ed <ddZddZddZddZdddZdS)z8Codex model discovery from API, local cache, and config.) annotationsN)Path)ListOptional)gpt-5.5 gpt-5.4-minigpt-5.4 gpt-5.3-codex gpt-5.2-codexzgpt-5.1-codex-maxzgpt-5.1-codex-mini List[str]DEFAULT_CODEX_MODELS))r)r rr )rr r )r r)r )r z!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 z-_add_forward_compat_models..1s(@@Hx4@@@@@@)setappendaddrany)rorderedmodel_idsynthetic_modeltemplate_modelsrs @r_add_forward_compat_modelsr$ sGUUD 4   NN8 $ $ $ HHX   ,K&&( d " "  @@@@@@@ @ @ & NN? + + + HH_ % % % Nr access_tokenstrc ddl}|ddd|id}|jdkrgS|}t |t r|d gng}n4#t $r'}td |gcYd}~Sd}~wwxYwg}|D]1}t |t s|d }t |tr| sX| }|d d ur|dd} t | tr)| dvr|d} t | ttfrt| nd} || |f3|dt!d|DS)zUFetch available models from the Codex API. Returns visible models sorted by priority.rNzAhttps://chatgpt.com/backend-api/codex/models?client_version=1.0.0 AuthorizationzBearer )headerstimeoutmodelsz)Failed to fetch Codex models from API: %sslugsupported_in_apiF visibilityhidehiddenpriority'c"|d|dfSNrr)xs rz(_fetch_models_from_api..Zs1qt rkeycg|]\}}|Srr)r_r.s r z*_fetch_models_from_api..[s&D&D&D4t&D&D&Dr)httpxget status_codejson isinstancedict Exceptionloggerdebugr&striplowerintfloatrsortr$) r%rArespdataentriesexcsortableitemr.r0r5ranks r_fetch_models_from_apirV8s!  yy O$&> &>&>?    s " "Iyy{{,6tT,B,BJ$((8R(((  @#FFF H&&$%%  xx$$$ DJJLL  zz|| 88& ' '5 0 0 XXlB// j# & & :+;+;+=+=+C+C+E+EI[+[+[ 88J'' *8c5\ B BNs8}}}t %%%% MM,,M--- %&D&D8&D&D&D E EEs#-A2AA22 B#<BB#B# codex_homer Optional[str]c|dz }|sdS ddl}n#t$rYdSwxYw ||d}n#t$rYdSwxYwt |t r|dnd}t |tr(| r| SdS)Nz config.tomlrutf-8encodingmodel) existstomllibrGloads read_textrErFrBr&rJ)rW config_pathr_payloadr]s r_read_default_modelrd^s},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 |tr4|D]0}t |t s|d}t |tr| sX| }|ddur|d}t |tr)| d vr|d }t |ttfrt|nd } || |f2|d g} |D]\} }|| vr| || S)Nzmodels_cache.jsonrZr[r-r.r/Fr0r2r5r6c"|d|dfSr8r)rTs rr;z$_read_cache_models..sDGT!W#5rr<)r^rDr`rarGrErFrBlistr&rJrKrLrMrrN) rW cache_pathrawrQrSrTr.r0r5rUdedupedr?s r_read_cache_modelsrkps11J      j--w-??@@  $.c4#8#8BcgghdGH'4  * * *DdD)) 88F##DdC((   ::< config.toml default > local cache > hardcoded defaults. CODEX_HOMEr1z.codex) osgetenvrJr&rhome expanduserrVr$rdrrkr )r%codex_home_strrWr api_models default_modelr!s rget_codex_model_idsrus Y|R006688WC h@Vr~s?>>>"""""" !!!!!!!!  8 $ $###FFF0#F#F#F#FL$!!!!H///////r