mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-05-02 02:01:47 +00:00
84d1673e2f
feat: provider modules — ProviderProfile ABC, 29 providers, fetch_models, transport single-path Introduces providers/ as the single source of truth for every inference provider. All 29 providers declared with correct data cross-checked against auth.py, runtime_provider.py and auxiliary_client.py. Rebased onto main (30307a980). Incorporates post-salvage fixes from56724147e(gmi aux model google/gemini-3.1-flash-lite-preview, already set in providers/gmi.py).
76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
"""Provider module registry.
|
|
|
|
Auto-discovers ProviderProfile instances from providers/*.py modules.
|
|
Each module should define a module-level PROVIDER or PROVIDERS list.
|
|
|
|
Usage:
|
|
from providers import get_provider_profile
|
|
profile = get_provider_profile("nvidia") # returns ProviderProfile or None
|
|
profile = get_provider_profile("kimi") # checks name + aliases
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from providers.base import OMIT_TEMPERATURE, ProviderProfile # noqa: F401
|
|
|
|
_REGISTRY: dict[str, ProviderProfile] = {}
|
|
_ALIASES: dict[str, str] = {}
|
|
_discovered = False
|
|
|
|
|
|
def register_provider(profile: ProviderProfile) -> None:
|
|
"""Register a provider profile by name and aliases."""
|
|
_REGISTRY[profile.name] = profile
|
|
for alias in profile.aliases:
|
|
_ALIASES[alias] = profile.name
|
|
|
|
|
|
def get_provider_profile(name: str) -> ProviderProfile | None:
|
|
"""Look up a provider profile by name or alias.
|
|
|
|
Returns None if the provider has no profile (falls back to generic).
|
|
"""
|
|
if not _discovered:
|
|
_discover_providers()
|
|
canonical = _ALIASES.get(name, name)
|
|
return _REGISTRY.get(canonical)
|
|
|
|
|
|
def list_providers() -> list[ProviderProfile]:
|
|
"""Return all registered provider profiles (one per canonical name)."""
|
|
if not _discovered:
|
|
_discover_providers()
|
|
# Deduplicate: _REGISTRY has canonical names; _ALIASES points to same objects
|
|
seen: set[int] = set()
|
|
result: list[ProviderProfile] = []
|
|
for profile in _REGISTRY.values():
|
|
pid = id(profile)
|
|
if pid not in seen:
|
|
seen.add(pid)
|
|
result.append(profile)
|
|
return result
|
|
|
|
|
|
def _discover_providers() -> None:
|
|
"""Import all provider modules to trigger registration."""
|
|
global _discovered
|
|
if _discovered:
|
|
return
|
|
_discovered = True
|
|
|
|
import importlib
|
|
import pkgutil
|
|
|
|
import providers as _pkg
|
|
|
|
for _importer, modname, _ispkg in pkgutil.iter_modules(_pkg.__path__):
|
|
if modname.startswith("_") or modname == "base":
|
|
continue
|
|
try:
|
|
importlib.import_module(f"providers.{modname}")
|
|
except ImportError as e:
|
|
import logging
|
|
|
|
logging.getLogger(__name__).warning(
|
|
"Failed to import provider module %s: %s", modname, e
|
|
)
|