Rj UdZddlmZddlZddlZddlmZddlmZm 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-codexgpt-5.3-codex-spark List[str]DEFAULT_CODEX_MODELS))r)r rr )rr )r r)r rz!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..Ks(@@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]}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.rNzAhttps://chatgpt.com/backend-api/codex/models?client_version=1.0.0 AuthorizationzBearer )headerstimeoutmodelsz)Failed to fetch Codex models from API: %sslug visibility>hidehiddenpriority'c"|d|dfSNrr)xs rz(_fetch_models_from_api..vs1qt rkeycg|]\}}|Srr)r_r.s r z*_fetch_models_from_api..ws&D&D&D4t&D&D&Dr)httpxget status_codejson isinstancedict Exceptionloggerdebugr&striplowerintfloatrsortr$) r%r?respdataentriesexcsortableitemr.r/r3ranks r_fetch_models_from_apirTRs  yy O$&> &>&>?    s " "Iyy{{,6tT,B,BJ$((8R(((  @#FFF H&&$%%  xx$$$ DJJLL  zz|| 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) existstomllibrEloads read_textrCrDr@r&rH)rU config_pathr]payloadr[s r_read_default_modelrbzs},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.jsonrXrYr-r.r/>r1r2r3r4c"|d|dfSr6r)rRs rr9z$_read_cache_models..sDGT!W#5rr:)r\rBr^r_rErCrDr@listr&rHrIrJrKrrL) rU cache_pathrawrOrQrRr.r/r3rSdedupedr=s r_read_cache_modelsris11J      j--w-??@@  $.c4#8#8BcgghdGH'4  * * *DdD)) 88F##DdC((   ::< config.toml default > local cache > hardcoded defaults. CODEX_HOMEr0z.codex) osgetenvrHr&rhome expanduserrTr$rbrrir )r%codex_home_strrUr api_models default_modelr!s rget_codex_model_idsrss Y|R006688WC h@Vr|s@>>>"""""" !!!!!!!!  8 $ $###@ F F F    0%F%F%F%FP$""""J///////r