i|UdZddlmZddlZddlZddlZddlZddlmZddl m Z m Z ej e ZiZded<iZded <d aeejjd z d z ZddZd dZd!dZd"dZd#dZd$dZdS)%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-providersprofilerreturnNonecV|t|j<|jD]}|jt|<dS)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)r aliass 7/home/piyush/.hermes/hermes-agent/providers/__init__.pyregister_providerr5s8&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_profilerAs=  T4((I == # ##rlist[ProviderProfile]ctstt}g}tD]?}t |}||vr*||||@|S)zAReturn all registered provider profiles (one per canonical name).)rrsetrvaluesidaddappend)seenresultr pids rlist_providersr'Lsz UUD$&F##%%##kk d?? HHSMMM MM' " " " Mr 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_dirr/[sj444444 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)r0r1 init_file safe_name module_namespecmoduleexcs r_import_plugin_dirrKfse ]*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)r4.r5userrr4basez providers.z.Failed to import legacy provider module %s: %s)r_BUNDLED_PLUGINS_DIRr,sortediterdirr startswithrKr/pkgutil providers iter_modules__path__r< import_module ImportErrorrBrCr-)childuser_dirrT_pkg _importermodname_ispkgrJs 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()r0rr1rr r )r r )__doc__ __future__rr<importlib.utilloggingr:pathlibrproviders.baserr getLogger__name__rBr__annotations__rr__file__resolveparentrPrrr'r/rKrrrrmsg<#""""" <<<<<<<<  8 $ $(* ****  DNN#*Y69JJ ' ' ' '$$$$    #+#+#+#+L3 3 3 3 3 3 r