Ӄie UdZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z mZmZddlmZddlZejeZdZdZiae ee fed <daeed <eGd d ZeGd dZidddddddddddddddddddddddddd d!d"d#d#d$d%d%d%d&d'd(d)d*d+d,d-Ze eefed.<da ee eefed/<d0e eeffd1Z!d0e fd2Z"d0e ee ffd3Z#d4e ee fd0dfd5Z$dXd7e%d0e ee ffd8Z&d9ed:ed0ee'fd;Z(dd?Z*d9ed0ee ee ffd@Z+dAe ee fd:ed0ee ee ffdBZ,d9ed:ed0ee*fdCZ-d9ed0e efdDZ.ddl/Z/e/j0dEe/j1Z2e/j3edF<d9ed0e efdGZ4 dYdIed9edJe'd0e e ee ffdKZ5dLedMe ee fdNed0efdOZ6dNedMe ee fd0efdPZ7dNed0eefdQZ8d0e eeffdRZ9dSed0e efdTZ:dNedLed0eefdUZ;dLed0eefdVZ)r+capss r,format_capabilitieszModelInfo.format_capabilitiesks > % KK $ $ $ > ! KK    ! ! " KK ! ! !       KK     $ $ & & ! KK  ! - KK+ , , ,   ( KK ' ' '"&3tyyG3r.)%__name__ __module__ __qualname____doc__str__annotations__rboolrrrrrrr rrintrrrrfloatr r!r"r$r%r&r'rr-r1r5r8r@rGrr.r,rr0sE;; GGG III KKKItItJK#t###L$)+eCHo***)+uS#X+++NCJ#Ix}###JK'+OXe_+++(,huo,,,cL#FCK;t;;;;CCCCC.d....0d0000 S    4S444444r.rcteZdZUdZeed<eed<eedfed<eed<dZeed<d Ze ed <d e fd Z d S) ProviderInfoz-Full metadata for a provider from models.dev.rr.envapir#docr model_countr(c*t|jS)N)rNrTr*s r, has_api_urlzProviderInfo.has_api_urlsDH~~r.N) rHrIrJrKrLrMr rUrVrOrNrXrr.r,rRrRs77 GGG III sCx HHHCMMMKTr.rR openrouter anthropiczaiz kimi-codingzkimi-for-codingminimaxz minimax-cndeepseekalibabacopilotzgithub-copilotz ai-gatewayvercelz opencode-zenopencodez opencode-gokilocodekilo fireworksz fireworks-ai huggingfacegeminigooglexainvidiagroqmistral togetherai perplexitycohere)rhrirjrkrlrmrnPROVIDER_TO_MODELS_DEV_MODELS_DEV_TO_PROVIDERr(cdt#dtDatS)u4Return models.dev ID → Hermes provider ID mapping.Nci|]\}}|| Srr).0kvs r, z(_get_reverse_mapping..s"S"S"SDAq1a"S"S"Sr.)rproitemsrr.r,_get_reverse_mappingrxs1&"S"S4J4P4P4R4R"S"S"S ""r.c(ddlm}|dz S)zReturn path to disk cache file.rget_hermes_homezmodels_dev_cache.json)hermes_constantsr{rzs r,_get_cache_pathr}s(000000 ?  6 66r.c, t}|r=t|d5}tj|cdddS#1swxYwYn2#t $r%}t d|Yd}~nd}~wwxYwiS)z%Load models.dev data from disk cache.zutf-8)encodingNz(Failed to load models.dev disk cache: %s)r}existsopenjsonload Exceptionloggerdebug) cache_pathfes r,_load_disk_cachersD$&&      $j7333 $qy|| $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ $ DDD ?CCCCCCCCD Is:3A"A A"AA"AA"" B,B  Bdatac t}t||dddS#t$r&}td|Yd}~dSd}~wwxYw)z.Save models.dev data to disk cache atomically.N),:)indent separatorsz(Failed to save models.dev disk cache: %s)r}r rrr)rrrs r,_save_disk_cacherszD$&& *d4JOOOOOO DDD ?CCCCCCCCCDs!% AAAF force_refreshc R|s2tr+tjtz tkrtS t jt d}||}t|trz|rx|atjat|t dt|td|D|Sn2#t"$r%}t d|Yd}~nd}~wwxYwts`t%atrKtjtz dzat dtttS) zFetch models.dev registry. In-memory cache (1hr) + disk fallback. Returns the full registry dict keyed by provider ID, or empty dict on failure. )timeoutz:Fetched models.dev registry: %d providers, %d total modelsc3K|]<}t|tt|diV=dS)modelsN) isinstancedictlenget)rsps r, z#fetch_models_dev..sI[[zRSUYGZGZ[Ch++,,[[[[[[r.zFailed to fetch models.dev: %sNi,z0Loaded models.dev from disk cache (%d providers))r timer _MODELS_DEV_CACHE_TTLrequestsrMODELS_DEV_URLraise_for_statusrrrrrrrsumvaluesrr)rresponserrs r,fetch_models_devrs ! !Y[[1 15J J J  :<;;;!!###}} dD ! ! d  $ %)Y[[ " T " " " LLLD [[dkkmm[[[[[    K ::: 5q99999999: e,..  e%)Y[[3H%H3%N " LLKSQbMcMc d d d sCD D8D33D8providermodelct|}|sdSt}||}t|tsdS|di}t|tsdS||}|rt |}|r|S|}|D]2\} } | |krt | }|r|cS3dS)zLook up context_length for a provider+model combo in models.dev. Returns the context window in tokens, or None if not found. Handles case-insensitive matching and filters out context=0 entries. Nr)rorrrr_extract_contextlowerrw) rrmdev_provider_idr provider_datarentryctx model_lowermidmdatas r,lookup_models_dev_contextrs .11(;; t   DHH-..M mT * *t   x , ,F fd # #t JJu  E u%%  J++--Kllnn U 99;;+ % %"5))C  4r.rct|tsdS|d}t|tsdS|d}t|ttfr|dkrt|SdS)zExtract context_length from a models.dev model entry. Returns None for invalid/zero values (some audio/image models have context=0). Nlimitcontextr)rrrrOrP)rrrs r,rr&s eT " "t IIg  E eT " "t ))I  C#U|$$q3xx 4r.cheZdZUdZdZeed<dZeed<dZeed<dZ e ed<d Z e ed <d Z e ed <d S)ModelCapabilitiesz;Structured capability metadata for a model from models.dev.Tsupports_toolsFr1supports_reasoning@ r max_output_tokensr# model_familyN)rHrIrJrKrrNrMr1rrrOrrrLrr.r,rr;szEEND!OT!!!$$$$ NC   !s!!!L#r.rct|}|sdSt}||}t|tsdS|di}t|tsdS|S)zResolve a Hermes provider ID to its models dict from models.dev. Returns the models dict or None if the provider is unknown or has no data. Nr)rorrrr)rrrrrs r,_get_provider_modelsrGs .11(;; t   DHH-..M mT * *t   x , ,F fd # #t Mr.rc||}t|tr|S|}|D]6\}}||krt|tr|cS7dS)zBFind a model entry by exact match, then case-insensitive fallback.N)rrrrrw)rrrrrrs r,_find_model_entryr\s JJu  E% ++--Kllnn U 99;;+ % %*UD*A*A %LLL 4r.ct|}|dSt||}|dSt|dd}t|dd}t|dd}|di}t |t si}|d}t |t tfr|dkrt |nd } |d } t | t tfr| dkrt | nd } |d d pd } t|||| | | S)uLook up full capability metadata from models.dev cache. Uses the existing fetch_models_dev() and PROVIDER_TO_MODELS_DEV mapping. Returns None if model not found. Extracts from model entry fields: - reasoning (bool) → supports_reasoning - tool_call (bool) → supports_tools - attachment (bool) → supports_vision - limit.context (int) → context_window - limit.output (int) → max_output_tokens - family (str) → model_family NrFrrrrrroutputrrr#)rr1rrrr) rrrNrrrrOrPr) rrrrrr1rrrroutrrs r,get_model_capabilitiesrlsn"( + +F ~t fe , ,E }t%))K7788N599\599::Oeii U;;<< IIgr " "E eT " " ))I  C!+C#u!>!>V377SXXXPVN ))H  C$.sS%L$A$AWcAggCSW99Xr**0bL %'-%+!    r.cjt|}|gSt|S)zReturn all model IDs for a provider from models.dev. Returns an empty list if the provider is unknown or has no data. )rlistkeys)rrs r,list_provider_modelsrs1 "( + +F ~   r.zZ-tts\b|embedding|live-|-(preview|exp)-\d{2,4}[-_]|-image\b|-image-preview\b|-customtools\b_NOISE_PATTERNSct|}|gSg}|D]b\}}t|ts|dds2t |rM||c|S)zReturn model IDs suitable for agentic use from models.dev. Filters for tool_call=True and excludes noise (TTS, embedding, dated preview snapshots, live/streaming, image-only models). Returns an empty list on any failure. NrF)rrwrrrrsearchr=)rrresultrrs r,list_agentic_modelsrs"( + +F ~ Fllnn U%&&  yye,,    ! !# & &   c Mr.queryrc~t}|sgSg}|t|}|sgS||i}t|tr]|di}t|tr2|D]\}} |||| fntD]\} } || i}t|tr]|di}t|tr2|D]\}} || || f|sgSd|D} |} g}|D]5\}}} | |vr|||| d6tj | | |dzd}t}g}|D]W}|d|d f}||vrA| |||t||kr|cSX|D]r}|D]m\}}} ||krO||f}||vrG| ||||| dt||kr|ccSns|S) aFuzzy search across models.dev catalog. Returns matching model entries. Args: query: Search string to match against model IDs. provider: Optional Hermes provider ID to restrict search scope. If None, searches across all providers in PROVIDER_TO_MODELS_DEV. limit: Maximum number of results to return. Returns: List of dicts, each containing 'provider', 'model_id', and the full model 'entry' from models.dev. NrcBg|]}|dS))r)rscs r, z%search_models_dev..s$888qtzz||888r.)rmodel_idrg?)ncutoffrr) rrorrrrwr=rdifflibget_close_matchessetaddr)rrrr candidatesrrrrr hermes_prov mdev_provmodel_ids_lower query_lowersubstring_matchesprov fuzzy_idsseen_idsresultsmatchkeyfids r,search_models_devrsN   D  !J155h?? I!1266 mT * * >"&&x44F&$'' >"(,,..>>JC%%xe&<===='=&B&B&D&D E E "K HHY33M-.. E&**8R88fd++E&,llnnEE U"));U*CDDDD  98Z888O++--K&ZZc5 #))++ % %  $ $$CRW%X%X Y Y Y)_ #IEEH$&G#Z % "34 h   LL    NN5 ! ! !7||u$$ ' ' * ' ' D#uyy{{c!!Skh&&LL%%%NN#PU#V#VWWW7||u,,& ' Nr.rrawrc |dpi}t|tsi}|dpi}t|tsi}|dpi}t|tsi}|dpg}|dpg}|d}t|ttfr|dkrt|nd} |d} t| ttfr| dkrt| nd} |d} t| ttfr| dkrt| nd} t did |d |d d p|d |d d pd d |dt |dddt |dddt |dddt |dddt |dddt |dddt|trt|nddt|trt|ndd| d| d| dt |ddpddt |ddpddd|vr|dt |dnddd |vr|d t |d ndd!|d"d pd d#|d#d pd d$|d$d pd d%|d%dS)&zEConvert a raw models.dev model entry dict into a ModelInfo dataclass.rcost modalitiesinputrrrNrrr#rrrFrrrrrrrrrrrrr r! cache_readr" cache_writer$ knowledger%r&r') rrrrOrPrrNrtuple)rrrrrr input_mods output_modsrctx_introut_intinpinp_ints r,_parse_model_infor&s@ GGG   "E eT " " 776?? bD dD ! !&&,"J j$ ' ' ((.BJ..**0bK ))I  C$S3,77JC!GGc#hhhG ))H  C$S3,77JC!GGc#hhhG ))G  C$S3,77MC!GGc#hhhG     8  WWVR ,H wwx$$*  K   sww{E22333   sww{E22333   e44555 66777 sww':EBBCCC #''.%88999 /9T.J.JRz***PR 1;;0M0MU% ,,,SU w 7 '  '1--2333! "$((8Q//41555# $6BT5I5IdS_N`Nld<0111rv% &8E7L7LQUVcQdQptM2333vz' (b117R) *WW^R006B+ ,wwx$$*- .GGM5111/ r.c |dpg}|dpi}t||ddp|t|trt |nd|ddpd|ddpdt|t rt |nd S) zAConvert a raw models.dev provider entry dict into a ProviderInfo.rSrrr#rrTrUr)rrrSrTrUrV)rrRrrrrr)rrrSrs r,_parse_provider_inforYs ''%.. BC WWX   $"F   WWVR /K$S$// 7E#JJJR GGE2   $" GGE2   $"#-fd#;#;BCKKK    r.ct||}t}||}t|tsdSt ||S)zGet full provider metadata from models.dev. Accepts either a Hermes provider ID (e.g. "kilocode") or a models.dev ID (e.g. "kilo"). Returns None if the provider is not in the catalog. N)rorrrrr)rmdev_idrrs r,get_provider_inforks\%((kBBG   D ((7  C c4 t  - --r.ct}i}|D]/\}}t|trt ||}|||<0|S)uReturn all providers from models.dev as {provider_id: ProviderInfo}. Returns the full catalog — 109+ providers. For providers that have a Hermes alias, both the models.dev ID and the Hermes ID are included. )rrwrrr)rrpidpdatainfos r,list_all_providersr|s_   D&(Fjjll U eT " " 'U33DF3K Mr.env_varc t}g}|D]^\}}t|trD|dg}t|t r||vr||_|S)zReverse lookup: find all providers that use a given env var. Useful for auto-detection: "user has ANTHROPIC_API_KEY set, which providers does that enable?" Returns list of models.dev provider IDs. rS)rrwrrrrr=)rrmatchesrrrSs r,get_providers_for_env_varr s   DGjjll$$ U eT " " $))E2&&C#t$$ $Cs### Nr.c\t||}t}||}t|tsdS|di}t|tsdS||}t|trt |||S|}|D]E\}} ||kr(t| trt || |cSFdS)zGet full model metadata from models.dev. Accepts Hermes or models.dev provider ID. Tries exact match then case-insensitive fallback. Returns None if not found. Nr)rorrrrrrrw) rrrrrrrrrrs r,get_model_infor s%((kBBG   D HHW  E eT " "t YYx $ $F fd # #t **X  C#t9 3888..""Kllnn:: U 99;;+ % %*UD*A*A %$S%99 9 9 9 4r.ct}tD] \}}||}t |t s1|di}t |t s]||}t |t rt |||cS|}|D]G\}} ||kr*t | t rt || |ccSH |D]\} }| tvrt |t s,|di}t |t sX||}t |t rt ||| cSdS)zSearch all providers for a model by ID. Useful when you have a full slug like "anthropic/claude-sonnet-4.6" or a bare name and want to find it anywhere. Checks Hermes-mapped providers first, then falls back to all models.dev providers. rN) rrorwrrrrrrx) rr hermes_idrrrrrrrrs r,get_model_info_any_providerrs   D5::<<>> 7!!%&&  8R((&$''  jj"" c4  =$XsG<< < < <nn&&  ,,.. > >JCyy{{k))j.E.E)(eW======= > jjll 9 9 U &(( ( ( %&&  8R((&$''  jj"" c4  9$XsC88 8 8 8 9 4r.ct||}t}||}t|tsgS|di}t|tsgSg}|D]\\}}t|ts|dd}|dkr8|t|||]|S)zjReturn all models for a provider as ModelInfo objects. Filters out deprecated models by default. rr&r# deprecated)rorrrrrwr=r) rrrrrrrrr&s r,list_provider_model_infosrs %((kBBG   D HHW  E eT " " YYx $ $F fd # #  Fllnn>> U%&&  8R(( \ ! !  'UG<<==== Mr.)F)Nr)>rKrrloggingosr dataclassesrpathlibrtypingrrrrr utilsr r getLoggerrHrrrr rLrMr rPrrRrorprxr}rrrNrrOrrrrrrrrecompile IGNORECASErPatternrrrrrrr r rrrr.r,rs(  !!!!!!33333333333333######  8 $ $.%'4S>&&& !!!! K4K4K4K4K4K4K4 K4\          &*,** 5*$ * y * , * *y**(*J*=***=*  h!*" h#*$  1***S#X859$sCx.1888#d38n####77777 $sCx.    D4S>DdDDDD))D)T#s(^))))X""C"HSM""""J DcN x}    *  38DcN+C* d38n S Xd3PS8n=U     /S//BS9T////d349 (bj0M#$s)045UU UU-0U $sCx.UUUUx00$sCx.0s0y0000f c S#X <    $.3.8L+A...."Dl!23"stCy0 # iB*#*(92E****Z34 ?r.