eiRUdZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlZejeZdZdZiae ee fed <daeed <eGd d ZeGd dZidddddddddddddddddddddddddddddd d!d"d#d#d$d%d&d'd'd(d)d*d+d,d-d.d/d0d1Ze eefed2<dZe e eefed3<d4efd5Zd4e ee ffd6Z d7e ee fd4dfd8Z!dTd:e"d4e ee ffd;Z#dZ%d?e ee fd4e e$fd@Z&eGdAdBZ'd)r+capss r,format_capabilitieszModelInfo.format_capabilitiesis > % 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,rr.sE;; GGG III KKKItItJK#t###L$)+eCHo***)+uS#X+++NCJ#Ix}###JK'+OXe_+++(,huo,,,cL#FCK;t;;;;CCCCC.d....0d0000 S    4S444444r.rcheZdZUdZeed<eed<eedfed<eed<dZeed<d Ze ed <d S) ProviderInfoz-Full metadata for a provider from models.dev.rr.envapir#docr model_countN) rHrIrJrKrLrMr rUrVrOrr.r,rRrR}sd77 GGG III sCx HHHCMMMKr.rR openrouter anthropicopenaiz openai-codexzaiz kimi-codingzkimi-for-codingstepfunzkimi-coding-cnminimaxz minimax-cndeepseekalibabaz qwen-oauthcopilotzgithub-copilotz ai-gatewayvercelz opencode-zenopencodez opencode-gokiloz fireworks-ai huggingfacegooglexaixiaominvidiagroqmistral togetherai perplexitycohere ollama-cloud)kilocode fireworksrcgeminirdrerfrgrhrirjrkrlrmPROVIDER_TO_MODELS_DEV_MODELS_DEV_TO_PROVIDERr(c(ddlm}|dz S)zReturn path to disk cache file.rget_hermes_homezmodels_dev_cache.json)hermes_constantsrurts r,_get_cache_pathrws(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)rwexistsopenjsonload 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)rwr r~rr)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).0ps 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_statusr|rrrrrrsumvaluesr~r)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)rqrrrr_extract_contextloweritems) 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,rr6szEEND!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)rqrrrr)rrrrrs r,_get_provider_modelsrBs .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)rrrrr)rrrrrrs r,_find_model_entryrWs JJu  E% ++--Kllnn U 99;;+ % %*UD*A*A %LLL 4r.ct|}|dSt||}|dSt|dd}|di}t |t r|dg}ng}t|ddpd|v}t|dd}|d i}t |t si}|d } t | t tfr| d krt | nd } |d } t | t tfr| d krt | nd} |ddpd} 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 NrF modalitiesinputrr0rrrrroutputrrr#)rr1rrrr) rrrNrrrrOrPr)rrrrr input_modsr1rrrroutrrs r,get_model_capabilitiesrgs"( + +F ~t fe , ,E }t%))K7788N<,,J*d##^^GR00  599\599::Sg>SOeii U;;<< IIgr " "E eT " " ))I  C!+C#u!>!>V377SXXXPVN ))H  C$.sS%L$A$AWcAggCSW99Xr**0bL %'-%+!    r.cddlm}|pt}|gSfd|DS)zReturn all model IDs for a provider from models.dev. Returns an empty list if the provider is unknown or has no data. r)normalize_providerNc4g|]}t||Sr)"_should_hide_from_provider_catalog)rrrs r, z(list_provider_models..s9   1(C@@     r.)hermes_cli.modelsrrkeys)rrrs` r,list_provider_modelsrsv 544444!!(++7xH !( + +F ~     {{}}   r.zZ-tts\b|embedding|live-|-(preview|exp)-\d{2,4}[-_]|-image\b|-image-preview\b|-customtools\b_NOISE_PATTERNS> gemma-3-1b gemma-3-2b gemma-3-4b gemma-3-12b gemma-3-27b gemma-3-1b-it gemma-3-2b-it gemma-3-4b-itgemini-1.5-progemma-3-12b-itgemma-3-27b-itgemma-4-26b-itgemma-4-31b-itgemini-1.5-flashgemini-2.0-flashgemma-4-26b-a4b-itgemini-1.5-flash-8bgemini-2.0-flash-litemodel_idc|pd}|pd}|dvr |tvrdSdS)Nr#>rprdTF)stripr_GOOGLE_HIDDEN_MODELS)rrprovider_lowerrs r,rrsen"++--3355N>r((**0022K---+AV2V2Vt 5r.c@t|}|gSg}|D]s\}}t|tst ||r,|ddsCt |r^||t|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) rrrrrrrsearchr=)rrresultrrs r,list_agentic_modelsrs"( + +F ~ Fllnn   U%&&   -h < <  yye,,    ! !# & &   c Mr.rawrc |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.rcostrrrrrNrrr#rrrFrrrrrrrrrrrrr r! cache_readr" cache_writer$ knowledger%r&r') rrrrOrPrrNlisttuple)rrrrrrr output_modsrctx_introut_intinpinp_ints r,_parse_model_infors@ 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_infor1s ''%.. 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)rqrrrrr)rmdev_idrrs r,get_provider_inforCs\%((kBBG   D ((7  C c4 t  - --r.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)rqrrrrrrr) rrrrpdatarrrrrs r,get_model_inforXs%((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.)F)9rKr|loggingr dataclassesrpathlibrtypingrrrrr utilsr r getLoggerrHrrrr rLrMr rPrrRrqrrrwrrrNrrOrrrrrrrrecompile IGNORECASErPattern frozensetrrrrrrrrr.r,r s(  !!!!!!33333333333333######  8 $ $.%'4S>&&& !!!! K4K4K4K4K4K4K4 K4\   *, * * h *H *  5 * $ *y *' *y *, *  *y *) * *( * J! *"=# *$   "? * * *S#X   F59$sCx.188877777 $sCx.    D4S>DdDDDD))D)T#s(^))))X""C"HSM""""J DcN x}    *  38DcN+C* d38n S Xd3PS8n=U     6S66BS9T6666r349& (bj0M" ###4#$s)<00$sCx.0s0y0000f c S#X <    $.3.8L+A....* # ir.