yj,UdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z ej e ZiZded<iZded <dad ed <d aeejjd z dz Zd!dZd"dZd#dZd$dZd%dZd&d ZdS)'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]_ALIASESzlist[ProviderProfile] | None_PROVIDER_LIST_CACHEFpluginsmodel-providersprofilerreturnNonecZ|t|j<|jD]}|jt|<dadS)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)rnamealiasesrr )r aliass 8/home/ragecks/.hermes/hermes-agent/providers/__init__.pyregister_providerr6s:&Igl''!,rstrProviderProfile | Nonectstt||}t|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_profilerDs=  T4((I == # ##rlist[ProviderProfile]cVtsttttSt }g}t D]?}t|}||vr*||| |@|at|S)zAReturn all registered provider profiles (one per canonical name).) rrr listsetrvaluesidaddappend)seenresultr pids rlist_providersr)Os '()))UUD$&F##%%##kk d?? HHSMMM MM' " " "! <<r Path | Nonec ddlm}|dz dz }|r|ndS#t$rYdSwxYw)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_dirr1bsj444444 O   ),= =HHJJ(qqD( tts -0 >> plugin_dirrsourcecp|dz }|sdS|jdd}|dkrd|}nd|}|tjvrdS t j||t|g}||j dSt j |}|tj|<|j |dS#t$rM}td ||j|tj|dYd}~dSd}~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)existsrreplacesysmodules importlibutilspec_from_file_locationrloadermodule_from_spec exec_moduler/loggerwarningpop)r2r3 init_file safe_name module_namespecmoduleexcs r_import_plugin_dirrMmse ]*I      ''S11I <<< :y:: ck!! +~55 J?P6   <4;. F0066#) K  ''''' +++ 6 QT     T********* +s 9CAC D5(AD00D5cHtrdSdatrhttD]A}|r|jdr1t|dBt}|ct|D]A}|r|jdr1t|dB ddl }ddl }| |j D]m\}}}|ds|dkr" tjd |;#t$r&}t d ||Yd}~fd}~wwxYwdS#t$$rYdSwxYw) 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)r6.r7userrr6basez providers.z.Failed to import legacy provider module %s: %s)r_BUNDLED_PLUGINS_DIRr.sortediterdirr startswithrMr1pkgutil providers iter_modules__path__r> import_module ImportErrorrDrEr/)childuser_dirrV_pkg _importermodname_ispkgrLs rrrsK""$$1088::;; 1 1E<<>> UZ%:%::%F%F  ui 0 0 0 0 !""HH,,..// . .E<<>> UZ%:%::%F%F  uf - - - -      *1*>*>t}*M*M   &Iw!!#&& 'V*;*; '(>W(>(>????   Dgs          s=AFEF F(F F FF F! F!)r rr r)rrr r)r r)r r*)r2rr3rr r)r r)__doc__ __future__rr>importlib.utilloggingr<pathlibrproviders.baserr getLogger__name__rDr__annotations__rr r__file__resolveparentrRrrr)r1rMrrrrosy<#""""" <<<<<<<<  8 $ $(* ****599999  DNN#*Y69JJ     $$$$&#+#+#+#+L3 3 3 3 3 3 r