sjq UdZddlZddlZddlZddlZddlmZddlmZddl m Z m Z m Z m Z mZddlmZddlZejeZdZdZd Ziae ee fed <daeed <daeed <ejZejZ d a!eGddZ"eGddZ#iddddddddddddddddddddddddd dd!d!d"d"d#d#d$d#id%d&d'd(d)d*d+d+d,d-d.d/d0d0d1d2d2d2d3d3d4d3d5d5d6d6d7d8d8d8d9d9d:d:d;dd?Z$e eefed@<da%e e ee efedA<dBedCe efdDZ&dCefdEZ'dCefdFZ(dCefdGZ)dHedCdfdIZ*ddJZ+dCefdKZ,dLe dCe-fdMZ.dCe ee ffdNZ/dOedCdfdPZ0dCe efdQZ1ddLe ee fdHedCdfdSZ2GdTdUe3Z4d dVdWe-dCee ee feffdXZ5ddYZ6dRdZdLe ee fdHed[edCdfd\Z7d[edCdfd]Z8d^e3d[edCdfd_Z9dd`Z:ddaZ; ddbdcdde-dee-dCe ee ffdfZdgedCe e=fdjZ?dke ee fdCe e=fdlZ@eGdmdnZAeBZCeBedo<dCe ee ffdpZDdgedCe e ee ffdqZEdgedhedCe e ee ffdrZFdgedCe e ee ffdsZGdgedhedte-dCe e ee ffduZHdve ee fdwedCe e=fdxZIdgedhedCe e=fdyZJdve ee fdCee ee fe e-ffdzZKd{e ee fdve ee fdCe ee ffd|ZLd dcdgedee-dCe e ee ffd}ZMd~e ee fdhedCe e ee ffdZNd dcdgedhedee-dCe eAfdZOdbdcdgedee-dCe efdZPddlQZQeQjRdeQjSZTeQjUed<eVhdZWdgededCe-fdZXdbdcdgedee-dCe efdZYded{e ee fdedCe"fdZZded{e ee fdCe#fdZ[dbdcdedee-dCe e#fdZ\d dcdededee-dCe e"fdZ]dS)uModels.dev registry integration — primary database for providers and models. Fetches from https://models.dev/api.json — a community-maintained database of 4000+ models across 109+ providers. Provides: - **Provider metadata**: name, base URL, env vars, documentation link - **Model metadata**: context window, max output, cost/M tokens, capabilities (reasoning, tools, vision, PDF, audio), modalities, knowledge cutoff, open-weights flag, family grouping, deprecation status Data resolution order: 1. In-memory cache (fresh, or stale served immediately while a single background daemon thread refreshes) 2. Disk cache (~/.hermes/models_dev_cache.json — any age; stale data is served rather than blocking callers on the network) 3. Network fetch (https://models.dev/api.json) — only when no cache exists at all; failed refreshes back off for 5 minutes process-wide Network hardening: - **ETag conditional GET**: network refreshes send ``If-None-Match`` with the last-known ETag whenever a servable registry is held (memory, hydrated from disk on cold force-refresh). A 304 Not Modified response is a no-op — the existing cache is re-confirmed fresh without re-downloading the full registry (≈2 MB). The ETag is persisted atomically alongside the cache file. - **No-network-on-hot-paths invariant**: resolution, picker, and resume paths NEVER perform network I/O. ``allow_network=False`` is threaded through every query function, and hot-path callers (vision routing, image routing, cost guard, context-length lookup) pass it explicitly. - **Corrupt-cache rejection**: a disk cache that fails to parse, is not a dict, or is empty is ignored with a warning rather than served as ``{}`` and silently breaking provider/model resolution. - **Mirror URL override**: ``models_dev.url`` in config.yaml lets deployments point at a mirror (e.g. a self-hosted copy) without code changes. Other modules should import the dataclasses and query functions from here rather than parsing the raw JSON themselves. N) dataclass)Path)AnyDictListOptionalTuple)atomic_json_writezhttps://models.dev/api.jsoni@8i,_models_dev_cache_models_dev_cache_time_models_dev_retry_afterFceZdZUdZeed<eed<eed<eed<dZeed<dZeed<dZ eed <dZ eed <dZ eed <dZ eed <d Z eedfed<d Zeedfed<dZeed<dZeed<dZeeed<dZeed<dZeed<dZeeed<dZeeed<dZeed<dZeed<dZeed<dZeed<d efd!Zd efd"Z d efd#Z!d efd$Z"d efd%Z#d efd&Z$dS)' ModelInfoz1Full metadata for a single model from models.dev.idnamefamily provider_idF reasoning tool_call attachment temperaturestructured_output open_weights.input_modalitiesoutput_modalitiesrcontext_window max_outputN max_inputg cost_input cost_outputcost_cache_readcost_cache_writeknowledge_cutoff release_datestatus interleavedreturnc.|jdkp |jdkS)Nr)r r!selfs 4/home/agent/.hermes/hermes-agent/agent/models_dev.py has_cost_datazModelInfo.has_cost_dataos":d&6&::c"|jpd|jvS)Nimage)rrr+s r-supports_visionzModelInfo.supports_visionrsB'T-B"BBr/cd|jvS)Npdfrr+s r- supports_pdfzModelInfo.supports_pdfus---r/cd|jvS)Naudior5r+s r-supports_audio_inputzModelInfo.supports_audio_inputxs$///r/c|sdSd|jddd|jddg}|j|d|jddd |S) zrr2r6r9rrr?)r,capss r-format_capabilitieszModelInfo.format_capabilitiess > % KK $ $ $ > ! KK    ! ! " KK ! ! !       KK     $ $ & & ! KK  ! - KK+ , , ,   ( KK ' ' '"&3tyyG3r/)%__name__ __module__ __qualname____doc__str__annotations__rboolrrrrrrr rrintrrrr floatr!r"r#r%r&r'r(rr.r2r6r9rArHrr/r-rrIsE;; 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) rIrJrKrLrMrNr rVrWrPrr/r-rSrSsd77 GGG III sCx HHHCMMMKr/rS openrouternovitaz novita-ai anthropicopenaiz openai-codexzaikimizkimi-for-codingz kimi-codingmoonshotstepfunzkimi-coding-cnminimaxz minimax-oauthz minimax-cndeepseekalibabaz qwen-oauthcopilotzgithub-copilotz ai-gatewayvercelz opencode-zenopencodez opencode-gokilocodekilo fireworksz fireworks-ai huggingfacegeminigooglexaiz xai-oauthxiaominvidiazmeta-aimetagroqmistral togetherai perplexitycohere ollama-cloud)rrrsrtruPROVIDER_TO_MODELS_DEV_MODELS_DEV_TO_PROVIDERmdev_idr)ctLi}tD].\}}||g|/|at|gS)zAReturn the Hermes provider ids that map to *mdev_id* (may be []).)rwrvitems setdefaultr>get)rxreverse hermes_idmappeds r-_models_dev_to_hermes_idsrsr&(*!7!=!=!?!? = = Iv   vr * * 1 1) < < < <") " & &w 3 33r/c(ddlm}|dz S)zReturn path to disk cache file.rget_hermes_homezmodels_dev_cache.jsonhermes_constantsrrs r-_get_cache_pathr(000000 ?  6 66r/c(ddlm}|dz S)z9Return path to the ETag sidecar file for conditional GET.rrzmodels_dev_cache.etagrrs r-_get_etag_pathrrr/c t}|r(|dSn2#t$r%}t d|Yd}~nd}~wwxYwdS)z?Load the last-known ETag from disk, or empty string if missing.utf-8encodingz"Failed to load models.dev ETag: %sNr$)rexists read_textstrip Exceptionloggerdebug) etag_pathes r- _load_etagrs>"$$      A&&&88>>@@ @ A >>> 91========> 2sA A A<A77A<etagc ddlm}t}|jdd|||dS#t $r&}t d|Yd}~dSd}~wwxYw)z/Persist an ETag to the sidecar file atomically.r)atomic_write_textT)parentsexist_okz"Failed to save models.dev ETag: %sN)utilsrrparentmkdirrrr)rrrrs r- _save_etagrs>++++++"$$ td;;;)T***** >>> 91=========>s>>>>>>>>?s"& AAAc ddlm}m}|}||ddd}t|tr(|r|Sn#t $rYnwxYwtS)aResolve the models.dev API URL, honoring a config.yaml override. The ``models_dev.url`` config key lets deployments point at a mirror (e.g. a self-hosted copy behind a corporate proxy) without code changes. Falls back to the default public URL when unset or empty. rcfg_getload_config_readonly models_devurlr$default)hermes_cli.configrr isinstancerMrrMODELS_DEV_URL)rrcfgrs r-_get_models_dev_urlrs CCCCCCCC""$$gc<;;; c3   CIIKK 99;;        sAA!! A.-A.datacRt|tot|dkS)z?Return True if *data* is a non-empty dict suitable for serving.r)rdictlen)rs r-_validate_registryr.s! dD ! ! 3c$ii!m3r/c t}|ryt|d5}tj|}dddn #1swxYwYt |s+t dt|iS|Sn_#t$rR}t d| ttn#t$rYnwxYwYd}~nd}~wwxYwiS)zLoad models.dev data from disk cache. A corrupt cache (invalid JSON, not a dict, or empty) is rejected with a warning so it doesn't silently masquerade as ``{}`` and break provider/model resolution for every caller. rrNzSmodels.dev disk cache is corrupt or empty; quarantining (will refetch from network)z6Failed to load models.dev disk cache; quarantining: %s) rropenjsonloadrrwarning_quarantine_corrupt_cacher) cache_pathfrrs r-_load_disk_cacher3sf$&&      j7333 $qy|| $ $ $ $ $ $ $ $ $ $ $ $ $ $ $%d++ ?**555 K   Da     %o&7&7 8 8 8 8    D   Isd3BA BABAz#_commit_registry..sG S Sz!T?R?R SCh## $ $ S S S S S Sr/N) rr rr r rrrsumvalues)rrrs r-_commit_registryrs}T4   !Y[[ LLK  D  S Sdkkmm S S SSS r/ctsFttjtzat d|dStjadat d|dS)uURe-confirm the existing cache as fresh after a 304 Not Modified. Callers must hold ``_models_dev_fetch_lock``. Clears the backoff and resets the in-memory cache timestamp so the next caller hits the fast path. The disk cache itself is not rewritten — its contents are unchanged, only its freshness marker is advanced. zomodels.dev returned 304 but no cached registry is held (%s); cleared ETag sidecar, will refetch unconditionallyNrzNmodels.dev registry unchanged (304 Not Modified, %s); cache re-confirmed fresh) r rrrr rrr rrs r-_confirm_cache_not_modifiedrs   "&)++0G"G A    !Y[[ LL # r/excctjtzatd|t|dS)zRecord a failed refresh: arm the process-wide 5-minute backoff. Callers must hold ``_models_dev_fetch_lock`` (see ``_commit_registry``). zC*B; CB CB CC> CC$C6 CC C C C C>CC>!C11C58C5>D!D D!D D!D D!ctjtkrdSt5tr ddddSdadddn #1swxYwYt jt dd} |dS#t$rG}t5dadddn #1swxYwYt d|Yd}~dSd}~wwxYw)zStart one daemon refresh worker if none is already running. Honors the process-wide failure backoff: after a failed refresh, no new background worker is spawned until ``_models_dev_retry_after``. NTzmodels-dev-refresh)targetrdaemonFz-Failed to start models.dev refresh thread: %s) rr rr threadingThreadrstartrrr)threadrs r-$_start_background_refresh_models_devr(s y{{,,, !-- (  --------)-%--------------- - !F I  III& 2 2,1 ) 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 DaHHHHHHHHH IsR A A  AA2B CCB( C(B, ,C/B, 0CCT allow_network force_refreshrc.|sOtrtSt}|r*|at}|tj|z ndatS|s2tr+tjtz t krtS|sDtr=t tt dtS|st}|t}|r|a|t kr@tj|z at dt||n7t tt d|tS|s#tjtkrtSt5tj}|s8trtcdddS|tkrtcdddS|rtst}|r|ada ttt\}}t||d|cdddS#t $r&t#d tcYcdddSt$$r}t'|d Yd}~nd}~wwxYwtsDtadatr-t d tttcdddS#1swxYwYdS) u\Fetch models.dev registry. Cache hierarchy: in-mem → disk → network. Returns the full registry dict keyed by provider ID, or empty dict on failure. Network requests use ETag conditional GET when a cached ETag exists AND a servable registry is held (on a cold ``force_refresh`` the memory cache is hydrated from disk first). A 304 Not Modified response re-confirms the existing cache's freshness without re-downloading the full (~2 MB) registry. Cache hierarchy (when ``force_refresh=False``): 1. Fresh in-memory cache → return immediately. 2. Stale in-memory cache → return immediately and refresh in a single background daemon thread. Callers never block on the network while any cache exists; ``models.dev`` only changes when providers add new models, so stale data is preferable to a foreground timeout. 3. Disk cache file (any age) → load, populate in-mem, return immediately. Stale disk caches trigger the same background refresh. A corrupt or empty disk cache is rejected with a warning. 4. No cache at all → singleflight foreground network fetch. On success, save to disk + in-mem and return. 5. Any failed refresh (foreground or background) suppresses further automatic refreshes for 5 minutes process-wide. When ``force_refresh=True`` (used by ``hermes config refresh``, the "refresh model catalog" code path), cache fast paths and the failure backoff are bypassed; the function hits the network and only falls back to cached data if the call fails. When ``allow_network=False``, any memory or disk cache is returned regardless of age and no request is made — used by latency-sensitive paths (gateway route-identity checks, vision routing, context-length lookup) that must never wait on the network. Nrz@Using stale in-memory models.dev cache; refreshing in backgroundzALoaded models.dev from fresh disk cache (%d providers, age=%.0fs)zGUsing stale models.dev disk cache (age=%.0fs); refreshing in backgroundr foregroundrrz1Loaded stale models.dev disk cache (%d providers))r rrrr rrrrrrr rrrOrrrrr) rr disk_datadisk_agenowdiskrrrs r-fetch_models_devr Ds1L  !  %$ $$&&   ) .00H*2*> h&&A #! ! !Y[[1 15J J J   !.!!!!,... N   !  )*,,  (**I )$-!333.2Y[[8-C*LL458^^X ,---8:::LL3  )( !TY[[+BBB  -!-!ikk )  )( -!-!-!-!-!-!-!-! ,,,( -!-!-!-!-!-!-!-!  +!2 +#%%D +$(!)*& 97 !233JD$ TL A A A A1-!-!-!-!-!-!-!-!2 % % % (l ; ; ; ;$ $ $;-!-!-!-!-!-!-!-!< 9 9 9 !!< 8 8 8 8 8 8 8 8 8 9 !  0 2 2 %& "   G)** ![-!-!-!-!-!-!-!-!-!-!-!-!-!-!-!-!-!-!sO%#L L 3L 8I!J)7L  J)J$L $J))AL  LLprovidermodelct||}||St|}|st|S|rt nt d}||}t |t st|S|di}t |t st|S||}|rt|} | r| S|} | D]2\} } | | krt| } | r| cS3dD]} || z}||}|rt|} | r| cS| | z}| D]4\} } | |krt| } | r| ccS5t|S)uLook 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. An EXPLICIT ``model_overrides`` config entry for this provider+model wins over the catalog value; ``_default`` entries fill the gap only when the catalog has no answer — the supported self-unblock path for models with wrong or missing context in models.dev (#84482). ``allow_network`` defaults to False — context-length lookup is a hot path (called during every conversation turn) and must never block on the network. Pass True only from explicit refresh flows. NFrrz:cloudz-cloud) _override_context_windowrvr|_default_override_contextr rr_extract_contextlowerrz)r rr override_ctxmdev_provider_idr provider_datarentryctx model_lowermidmdatasuffix suffixed_keysuffixed_lowers r-lookup_models_dev_contextr s-$,He<rrr)r  overrides provider_key candidatesrr~keysections r-_provider_override_sectionr:s&''I tN))++L tJ # ' ' 5 5F "&L((&!!!.|<<))  $ $   i ( ( (--$$ gt $ $ NNN  4r/c|pd}|sdSt|}|dS||}t|tr|S|}|D]=\}}|dkr ||krt|tr|cS>dS)zReturn the explicit per-provider+model override dict, or None. Model ids match exactly first, then case-insensitively (skipping the ``_default`` sentinel), mirroring catalog lookup behavior. r$N_default)rr:r|rrrrz)r r model_keyr9rrrrs r-_explicit_model_overrider>s "##%%I t(22Gt KK " "E% //##Kmmoo U *    99;;+ % %*UD*A*A %LLL 4r/ct|}|,|d}t|tr|St }|d}t|tr|SdS)uReturn the fill-gap ``_default`` override for *provider*, or None. Checks the per-provider ``_default`` first, then the global one. Only consulted for models the catalog does not know — see the block comment. Nr<)r:r|rrr3)r r9rr5global_defaults r-r"r"sx )22G++j)) gt $ $ N%''I]]:..N.$'' 4r/ catalog_hitcPt||}||S|rdSt|S)zSelect the override dict for a lookup, honoring fill-gap semantics. Explicit per-provider+model overrides always apply. ``_default`` entries apply only when the catalog has no entry for the model. N)r>r")r rrAexplicits r- _override_forrDs9(%88Ht "8 , ,,r/overrider8c8||}|dS t|}|dkr|Sn#ttf$rYnwxYw|t |f}|t vr6t |td||dS)zDCoerce an override field to a positive int, warning once on garbage.NrzKmodel_overrides: ignoring invalid %s value %r (expected a positive integer)) r|rP TypeErrorrreprr/addrr)rEr8r2valuewarn_keys r-r#r#s ,,s  C {t C 199L  z "     T#YYH,,,!!(+++ ,-0#    4s3AAcJt||}|dSt|dS)aReturn the EXPLICITLY overridden context_window, or None. Explicit-only on purpose: this runs early in the resolution chain (agent/model_metadata.py step 0b, before custom_providers and live probes), where a ``_default`` must not preempt more specific sources. Fill-gap defaults are applied later by ``lookup_models_dev_context`` once the catalog has actually missed. Nr)r>r#)r rovs r-rrs. "(E 2 2B zt - . ..r/chi}i}t|d}|||d<t|d}|||d<|r||d<d|vrt|d|d<d |vrt|d |d <d}d |vrt|d }||d <d |vrt|d pd|d<||fS)uRTranslate canonical override keys into a models.dev-shaped patch. ``get_model_info``/``_parse_model_info`` consume the raw catalog shape (``limit.context``, ``tool_call``, ...). All override consumers accept ONE canonical schema (the documented ``context_window``/``supports_*`` keys), so this boundary translates rather than forcing users to know the internal catalog shape. Returns ``(patch, vision)`` — vision is returned out-of-band (not as a key in the patch) because it maps onto the catalog's ``modalities.input`` list rather than a scalar field. rNr&r-outputr%r)rr*rr2rr.r$r)r#rOrM)rEpatchr%routrDs r-_override_to_catalog_shaperRsEE "2 3 3C i "5 6 6C h g8##!(+;"<==kx''!(+?"@AAk!FH$$h0122$l!!h~6<"==h &=r/r2ct|\}}t|}|dd}|rU|d}t |trt|ni}||||d<||d}t |trt|ni}|d}t |t rt |ng}|rd|vr|dn|sd|vr|d||d<||d<|||S)uPatch a catalog entry with a canonical-schema override. Sub-dicts (``limit``, ``modalities``) are merged, not clobbered — an override setting only ``context_window`` must not wipe the catalog's ``limit.output``. r%N modalitiesinputr1) rRrpopr|rupdatelistr>remove) r2rEshapedvision_overridemerged limit_patch base_limit base_mods input_modss r-"_merge_catalog_entry_with_overridera)sl9BBFO #YYF**Wd++K%WWW%% )3J)E)EMT*%%%2 +&&&$w"GGL)) '1)T'B'BJDOOO ]]7++ )3J)E)EMT*%%%2  'wj88   g & & & &  'W %:%:   g & & &' '(| MM& Mr/c4t|}|sdS|rtntd}||}t|tsdS|di}t|tsdS|S)u-Resolve 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. ``allow_network`` defaults to False — this is called from hot paths (vision routing, image routing, capability checks) and must never block. NFrr)rvr|r rr)r rrrrrs r-_get_provider_modelsrcIs.11(;; t  3 E 2 2 2 HH-..M mT * *t   x , ,F fd # #t Mr/rc,||}t|tr|S|}|D]6\}}||krt|tr|cS7dD]}|||z}t|tr|cS||z}|D]8\}}||krt|tr|ccS9dS)uFind a model entry: exact, case-insensitive, then suffix fallback. The ``:cloud``/``-cloud`` suffix fallback mirrors ``lookup_models_dev_context`` so "is this model in the catalog" means the same thing to every consumer — important for ``model_overrides`` fill-gap ``_default`` semantics, where a suffix-keyed catalog model (e.g. ``kimi-k2.6:cloud``) must count as KNOWN and keep its catalog metadata rather than being displaced by a ``_default``. rN)r|rrrrz)rrrrrrrrs r-_find_model_entryrejs$ JJu  E% ++--Kllnn U 99;;+ % %*UD*A*A %LLL' 56>** eT " " LLL$v- ,,..  JCyy{{n,,E41H1H,   4r/c t||}|t||nd}t|||du}||dS|t|dd}|di}t |t r|d}nd}t |trd|v}n#t|d d}t|d d} |d i} t | t si} | d } t | ttfr| d krt| nd} | d} t | ttfr| d krt| nd}|ddpd}n d}d}d} d} d}d}|d|vrt|d}d|vrt|d}d|vrt|d} t|d}||} t|d}||}d|vrt|dpd}t||| | ||S)u|Look 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. EXPLICIT ``model_overrides`` entries (per-provider+model) win over catalog values for the fields they set. ``_default`` entries fill the gap only for models the catalog does not know — the supported self-unblock path for custom/local models (#8731) and for models with wrong metadata in models.dev (#84482). An override may set any subset of fields; unspecified fields fall through to the catalog value (or sensible defaults when the model is absent from the catalog). ``allow_network`` defaults to False — capability lookup is a hot path (vision routing, image routing) and must never block on the network. 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 rNrArFrTrUr1rrr%r&rr+rOr,rr$Tr)r2r*rr-r.)r)r2r*rr-r.) rcrerDrOr|rrrXrPrQr#rMr()r rrrrrEr)r`r2r*r%rrrQr-r.ctx_ovout_ovs r-get_model_capabilitiesrjs6"(- H H HF060B fe , , ,EXu%t:KLLLH })t eii U;;<<YY|R00 j$ ' ' #00JJJ j$ ' ' C%3OO"599\5#A#ABBO!%))K"?"?@@ '2&&%&& Eii ""%/c5\%B%BZsQwwSTZii!!(23e (E(E[#PQ''CHHHW[yy2..4" "   x ' '!(+;"<==N  ( ("8,=#>??O 8 + +!%h/C&D!E!E x)9::  #Nx)<==   &  X % %x7=2>>L %'-%+!    r/cddlm}|pt|}|gSfd|DS)uJReturn all model IDs for a provider from models.dev. Returns an empty list if the provider is unknown or has no data. ``allow_network`` defaults to True — this is called from the model picker (``hermes model``), which is an interactive user-facing flow where a fresh catalog is worth a short network wait. r)normalize_providerrNc4g|]}t||Sr)"_should_hide_from_provider_catalog)rrr s r- z(list_provider_models..s9   1(C@@     r/)hermes_cli.modelsrlrckeys)r rrlrs` r-list_provider_modelsrrs{544444!!(++7xH !(- H H HF ~     {{}}   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$>rjrkTF)rr_GOOGLE_HIDDEN_MODELS)r rprovider_lowerrs r-rnrn3sen"++--3355N>r((**0022K---+AV2V2Vt 5r/cDt||}|gSg}|D]s\}}t|tst ||r,|ddsCt |r^||t|S)utReturn 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. ``allow_network`` defaults to True — like ``list_provider_models``, this is called from interactive model selection flows. rNrF) rcrzrrrnr|rssearchr>)r rrresultrrs r-list_agentic_modelsr;s"(- H H HF ~ Fllnn   U%&&   -h < <  yye,,    ! !# & &   c Mr/rc |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.r%costrTrUrOr&rNrrr$rrrFrrrrrrrrrrrr r!r" cache_readr# cache_writer% knowledger&r'r() r|rrrPrQrrOrXtuple)rr2rr%rrTr` output_modsrctx_intrQout_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.rTrrr$rrUrVr)rrrTrUrVrW)r|rSrrXrrr)rr2rTrs r-_parse_provider_infors ''%.. BC WWX   $"F   WWVR /K$S$// 7E#JJJR GGE2   $" GGE2   $"#-fd#;#;BCKKK    r/ct||}|rtntd}||}t|tsdSt ||S)uGet 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. ``allow_network`` defaults to True — the primary caller is ``resolve_provider_full`` during interactive setup, where a fresh catalog is worth a short network wait. Hot-path callers should pass ``allow_network=False``. FrN)rvr|r rrr)rrrxrr2s r-get_provider_infors|%((kBBG  3 E 2 2 2 ((7  C c4 t  - --r/c4 t dttf fd }|rt nt d}| }t |t s |S|di}t |t s |Sdtdtttfdtf fd }|}t |t r ||S } | D]@\} } | | kr#t | t r|| | cSA|S) uGet 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. ``model_overrides`` entries use the SAME canonical schema as every other consumer (``context_window``, ``max_output_tokens``, ``supports_*``, ``model_family``) — they are translated into the catalog shape at this boundary, and sub-dicts (``limit``, ``modalities``) are merged rather than clobbered. EXPLICIT entries patch known catalog models; ``_default`` entries fill the gap only for models the catalog does not know (#8731, #84482). ``allow_network`` defaults to False — model info lookup is a hot path (cost guard, inventory) and must never block on the network. r)ctd}|dSddddd}t||}t|S)NFrgr+r,)r&rOT)r%rrDrar)rEbaserZrxrrs r-_from_override_alonez,get_model_info.._from_override_alonesa hEJJJ  4 "(488  4D(CC 67;;;r/Frrrr2ctd}|!t||}t||St||S)NTrgr)rr2rEr\rxrrs r-_with_overridez&get_model_info.._with_overridesP hDIII  7XFFF$S&':: : c7333r/) rvr|rrr rrrMrrrrz) rrrrrpdatarrr2rrrrxs `` @r-get_model_infors&%((kBBG <(9"5 < < < < < < < <(  3 E 2 2 2 HHW  E eT " "&##%%% YYx $ $F fd # #&##%%%4C4d38n444444444 **X  C#t-~h,,,..""Kllnn.. U 99;;+ % %*UD*A*A %!>#u-- - - -  ! !!r/)r)N)r$)F)^rLrloggingrr dataclassesrpathlibrtypingrrrrr rr r getLoggerrIrrrrr rMrNr rQr LockrrrrrSrvrwrrrrrrrrOrrrrrrrrrrrrrrr rPr rrr(setr/r3r:r>r"rDr#rrRrarcrerjrrrecompile IGNORECASErsPattern frozensetrrnrrrrrrr/r-rs '''R  !!!!!!33333333333333######  8 $ $. %'4S>&&& !!!!!""""')))9>++ % K4K4K4K4K4K4K4 K4\   .*,.* k.*.* h .* H .*  5 .* .*$.*!.*y.*'.*y.*Y.*,.* .* y!.*")#.*.*$%.*&('.*(J).**=+.*,-.*./.*0=1.*2 h3.*4 h5.*6 57.*<=.*> h?.*@ hA.*LvM.*N FO.*P FQ.*RyS.*.*T"[.*.*.*S#X...d;?$sDI~"67>>>4s4tCy444477777 77777 C >S >T > > > > ? ? ? ?S(4S4T4444 $sCx.>$4"%4  4S>  d    NNNNN9NNN !*2*2*2*2 4S>3 *2*2*2*2Z , , , ,;=4S>CST*#$B y C D    22222IIII: X!:>X!X!X!X!37X! #s(^X!X!X!X!x9>P/P/P/P/P/15P/ c]P/P/P/P/f44 4444 DcN x}    *  B!SUUs"""tCH~"$sCx.1I<s38DcN;S4chtCH~.F$ - - -/3 - d38n - - - - DcN#* /s /3 /8C= / / / /#38n# 4S>8D> )*####L c3h#'S> #s(^B-2%) d38nB d38n S Xd3PS8n=U    H9>eeeee15e  eeeeR-1%) #Y2 (bj0M" ###4-1%) #YF00$sCx.0s0y0000f c S#X <    &04....(,. l....H?DH"H"H"H" #H"7;H" iH"H"H"H"H"H"r/