Bj|&UdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z eje ZiZded<iZded <d aeej)j*j*d z d z Zdd ZddZddZddZddZddZy)uProvider module registry. Provider profiles can live in two places: 1. Bundled plugins: ``plugins/model-providers//`` (shipped with hermes-agent) 2. User plugins: ``$HERMES_HOME/plugins/model-providers//`` Each plugin directory contains: - ``__init__.py`` — calls ``register_provider(profile)`` at import - ``plugin.yaml`` — manifest (name, kind: model-provider, version, description) Discovery is lazy: the first call to ``get_provider_profile()`` or ``list_providers()`` scans both locations and imports every plugin. User plugins override bundled plugins on name collision (last-writer-wins), so third parties can monkey-patch or replace any built-in profile without editing the repo. For backward compatibility, ``providers/*.py`` files (other than ``base.py`` and ``__init__.py``) are still discovered via ``pkgutil.iter_modules``. This lets out-of-tree users drop a single-file profile into an editable install without the plugin dir structure. New profiles should prefer the plugin layout. Usage:: from providers import get_provider_profile profile = get_provider_profile("nvidia") # ProviderProfile or None profile = get_provider_profile("kimi") # checks name + aliases ) annotationsN)Path)OMIT_TEMPERATUREProviderProfilezdict[str, ProviderProfile] _REGISTRYzdict[str, str]_ALIASESFpluginsmodel-providerscr|t|j<|jD]}|jt|<y)uRegister a provider profile by name and aliases. Later registrations with the same name replace earlier ones — so user plugins under ``$HERMES_HOME/plugins/model-providers/`` can override bundled profiles without editing repo code. N)rnamealiasesr)profilealiass 7/home/ubuntu/.hermes/hermes-agent/providers/__init__.pyregister_providerr5s2&Igll'!,,'cxts ttj||}tj|S)z{Look up a provider profile by name or alias. Returns None if the provider has no profile (falls back to generic). ) _discovered_discover_providersrgetr)r canonicals rget_provider_profilerAs,  T4(I == ##rcts tt}g}tj D]4}t |}||vs|j ||j|6|S)zAReturn all registered provider profiles (one per canonical name).)rrsetrvaluesidaddappend)seenresultrpids rlist_providersr"Ls] UD$&F##%#k d? HHSM MM' " # Mrcp ddlm}|dz dz }|jr|SdS#t$rYywxYw)z>Return ``$HERMES_HOME/plugins/model-providers/`` if it exists.r)get_hermes_homer r N)hermes_constantsr$is_dir Exception)r$ds r_user_plugins_dirr)[sB4   ),= =HHJq(D( s$)) 55cp|dz }|jsy|jjdd}|dk(rd|}nd|}|tjvry t j j||t|g}| |jyt j j|}|tj|<|jj|y#t$rL}tjd ||j|tjj|dYd}~yd}~wwxYw) zImport a single plugin directory so it self-registers. ``source`` is "bundled" or "user", used only for log messages. z __init__.pyN-_bundledzplugins.model_providers._hermes_user_provider_)submodule_search_locationsz(Failed to load %s provider plugin %s: %s)existsr replacesysmodules importlibutilspec_from_file_locationstrloadermodule_from_spec exec_moduler'loggerwarningpop) plugin_dirsource init_file safe_name module_namespecmoduleexcs r_import_plugin_dirrFfs ]*I     ''S1I 0 < .yk: ckk! +~~55 J?P6  <4;;. 006#) K  ' + 6 QT   T** +s :C A C D5)AD00D5ctrydatjrZttj D]:}|jr|j j dr/t|d<t}|Vt|j D]:}|jr|j j dr/t|d< ddl }ddl }|j|jD]6\}}}|j ds|dk(r tjd |8y#t$r!}t j#d ||Yd}~_d}~wwxYw#t$$rYywxYw) aPopulate the registry by importing every provider plugin. Order: 1. Bundled plugins at ``/plugins/model-providers//`` 2. User plugins at ``$HERMES_HOME/plugins/model-providers//`` 3. Legacy per-file modules at ``providers/.py`` (back-compat) Each step imports its plugins, which call ``register_provider()`` at module-level. Later steps win on name collision. NT)r,.r-userrr,basez providers.z.Failed to import legacy provider module %s: %s)r_BUNDLED_PLUGINS_DIRr&sortediterdirr startswithrFr)pkgutil providers iter_modules__path__r4 import_module ImportErrorr;r<r')childuser_dirrO_pkg _importermodname_ispkgrEs rrrsUK""$088:; 1E<<>UZZ%:%::%F ui 0 1!"HH,,./ .E<<>UZZ%:%::%F uf - .  *1*>*>t}}*M  &Iw!!#&'V*; ''*WI(>?    Dgs     s=AE%D85E%8 E"EE%E""E%% E10E1)rrreturnNone)r r7r[zProviderProfile | None)r[zlist[ProviderProfile])r[z Path | None)r>rr?r7r[r\)r[r\)__doc__ __future__rr4importlib.utilloggingr2pathlibrproviders.baserr getLogger__name__r;r__annotations__rr__file__resolveparentrKrrr"r)rFrrrrjs<# <   8 $(* %*.  N##**Y69JJ '$ #+L3 r