mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-05-21 03:39:54 +00:00
yannsunn
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
"""Upstream adapter registry for the local proxy server.
|
|
|
|
Each adapter wraps a provider's OAuth state and exposes a uniform interface
|
|
the proxy server can use to forward requests with a freshly-minted bearer
|
|
token. See :class:`UpstreamAdapter` for the contract.
|
|
"""
|
|
|
|
from typing import Dict, Type
|
|
|
|
from hermes_cli.proxy.adapters.base import UpstreamAdapter
|
|
from hermes_cli.proxy.adapters.nous_portal import NousPortalAdapter
|
|
from hermes_cli.proxy.adapters.xai import XAIGrokAdapter
|
|
|
|
# Registry of available adapter classes keyed by provider name as used on
|
|
# the ``hermes proxy start --provider <name>`` CLI flag.
|
|
ADAPTERS: Dict[str, Type[UpstreamAdapter]] = {
|
|
"nous": NousPortalAdapter,
|
|
"xai": XAIGrokAdapter,
|
|
}
|
|
|
|
|
|
def get_adapter(name: str) -> UpstreamAdapter:
|
|
"""Instantiate an adapter by provider name.
|
|
|
|
Raises:
|
|
ValueError: if ``name`` is not a registered adapter.
|
|
"""
|
|
key = (name or "").strip().lower()
|
|
if key not in ADAPTERS:
|
|
available = ", ".join(sorted(ADAPTERS)) or "(none)"
|
|
raise ValueError(
|
|
f"Unknown proxy upstream provider: {name!r}. Available: {available}"
|
|
)
|
|
return ADAPTERS[key]()
|
|
|
|
|
|
__all__ = ["UpstreamAdapter", "ADAPTERS", "get_adapter"]
|