mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-04-30 01:41:43 +00:00
a6a6cf047d
Registers tencent-tokenhub (https://tokenhub.tencentmaas.com/v1) as a new API-key provider with model tencent/hy3-preview (256K context). - PROVIDER_REGISTRY entry + TOKENHUB_API_KEY / TOKENHUB_BASE_URL env vars - Aliases: tencent, tokenhub, tencent-cloud, tencentmaas - openai_chat transport with is_tokenhub branch for top-level reasoning_effort (Hy3 is a reasoning model) - tencent/hy3-preview:free added to OpenRouter curated list - 60+ tests (provider registry, aliases, runtime resolution, credentials, model catalog, URL mapping, context length) - Docs: integrations/providers.md, environment-variables.md, model-catalog.json Author: simonweng <simonweng@tencent.com> Salvaged from PR #16860 onto current main (resolved conflicts with #16935 Azure Anthropic env-var hint tests and the --provider choices= list removal in chat_parser).
494 lines
20 KiB
Python
494 lines
20 KiB
Python
"""Tests for Tencent TokenHub provider support (Hy3 Preview)."""
|
|
|
|
import json
|
|
import os
|
|
|
|
import pytest
|
|
|
|
from hermes_cli.auth import (
|
|
PROVIDER_REGISTRY,
|
|
resolve_provider,
|
|
get_api_key_provider_status,
|
|
resolve_api_key_provider_credentials,
|
|
AuthError,
|
|
)
|
|
|
|
|
|
# Other provider env vars to clear during auto-detection tests
|
|
_OTHER_PROVIDER_KEYS = (
|
|
"OPENAI_API_KEY", "ANTHROPIC_API_KEY", "DEEPSEEK_API_KEY",
|
|
"GOOGLE_API_KEY", "GEMINI_API_KEY", "DASHSCOPE_API_KEY",
|
|
"XAI_API_KEY", "KIMI_API_KEY", "KIMI_CN_API_KEY",
|
|
"MINIMAX_API_KEY", "MINIMAX_CN_API_KEY", "AI_GATEWAY_API_KEY",
|
|
"KILOCODE_API_KEY", "HF_TOKEN", "GLM_API_KEY", "ZAI_API_KEY",
|
|
"XIAOMI_API_KEY", "OPENROUTER_API_KEY", "COPILOT_GITHUB_TOKEN",
|
|
"GH_TOKEN", "GITHUB_TOKEN", "ARCEEAI_API_KEY",
|
|
)
|
|
|
|
|
|
# =============================================================================
|
|
# Provider Registry
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubProviderRegistry:
|
|
"""Verify tencent-tokenhub is registered correctly in the PROVIDER_REGISTRY."""
|
|
|
|
def test_registered(self):
|
|
assert "tencent-tokenhub" in PROVIDER_REGISTRY
|
|
|
|
def test_name(self):
|
|
assert PROVIDER_REGISTRY["tencent-tokenhub"].name == "Tencent TokenHub"
|
|
|
|
def test_auth_type(self):
|
|
assert PROVIDER_REGISTRY["tencent-tokenhub"].auth_type == "api_key"
|
|
|
|
def test_inference_base_url(self):
|
|
assert PROVIDER_REGISTRY["tencent-tokenhub"].inference_base_url == "https://tokenhub.tencentmaas.com/v1"
|
|
|
|
def test_api_key_env_vars(self):
|
|
assert PROVIDER_REGISTRY["tencent-tokenhub"].api_key_env_vars == ("TOKENHUB_API_KEY",)
|
|
|
|
def test_base_url_env_var(self):
|
|
assert PROVIDER_REGISTRY["tencent-tokenhub"].base_url_env_var == "TOKENHUB_BASE_URL"
|
|
|
|
|
|
# =============================================================================
|
|
# Aliases
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubAliases:
|
|
"""All aliases should resolve to 'tencent-tokenhub'."""
|
|
|
|
@pytest.mark.parametrize("alias", [
|
|
"tencent-tokenhub", "tencent", "tokenhub", "tencent-cloud", "tencentmaas",
|
|
])
|
|
def test_alias_resolves(self, alias, monkeypatch):
|
|
for key in _OTHER_PROVIDER_KEYS:
|
|
monkeypatch.delenv(key, raising=False)
|
|
monkeypatch.setenv("TOKENHUB_API_KEY", "sk-test-key-12345678")
|
|
assert resolve_provider(alias) == "tencent-tokenhub"
|
|
|
|
def test_normalize_provider_models_py(self):
|
|
from hermes_cli.models import normalize_provider
|
|
assert normalize_provider("tencent") == "tencent-tokenhub"
|
|
assert normalize_provider("tokenhub") == "tencent-tokenhub"
|
|
assert normalize_provider("tencent-cloud") == "tencent-tokenhub"
|
|
assert normalize_provider("tencentmaas") == "tencent-tokenhub"
|
|
|
|
def test_normalize_provider_providers_py(self):
|
|
from hermes_cli.providers import normalize_provider
|
|
assert normalize_provider("tencent") == "tencent-tokenhub"
|
|
assert normalize_provider("tokenhub") == "tencent-tokenhub"
|
|
assert normalize_provider("tencent-cloud") == "tencent-tokenhub"
|
|
assert normalize_provider("tencentmaas") == "tencent-tokenhub"
|
|
|
|
|
|
# =============================================================================
|
|
# Auto-detection
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubAutoDetection:
|
|
"""Setting TOKENHUB_API_KEY should auto-detect the provider."""
|
|
|
|
def test_auto_detect(self, monkeypatch):
|
|
for var in _OTHER_PROVIDER_KEYS:
|
|
monkeypatch.delenv(var, raising=False)
|
|
monkeypatch.setenv("TOKENHUB_API_KEY", "sk-tokenhub-test-12345678")
|
|
provider = resolve_provider("auto")
|
|
assert provider == "tencent-tokenhub"
|
|
|
|
|
|
# =============================================================================
|
|
# Credentials
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubCredentials:
|
|
"""Test credential resolution for the tencent-tokenhub provider."""
|
|
|
|
def test_status_configured(self, monkeypatch):
|
|
monkeypatch.setenv("TOKENHUB_API_KEY", "sk-test-12345678")
|
|
status = get_api_key_provider_status("tencent-tokenhub")
|
|
assert status["configured"]
|
|
|
|
def test_status_not_configured(self, monkeypatch):
|
|
monkeypatch.delenv("TOKENHUB_API_KEY", raising=False)
|
|
status = get_api_key_provider_status("tencent-tokenhub")
|
|
assert not status["configured"]
|
|
|
|
def test_resolve_credentials(self, monkeypatch):
|
|
monkeypatch.setenv("TOKENHUB_API_KEY", "sk-test-12345678")
|
|
monkeypatch.delenv("TOKENHUB_BASE_URL", raising=False)
|
|
creds = resolve_api_key_provider_credentials("tencent-tokenhub")
|
|
assert creds["api_key"] == "sk-test-12345678"
|
|
assert creds["base_url"] == "https://tokenhub.tencentmaas.com/v1"
|
|
|
|
def test_openrouter_key_does_not_make_tokenhub_configured(self, monkeypatch):
|
|
"""OpenRouter users should NOT see tencent-tokenhub as configured."""
|
|
monkeypatch.delenv("TOKENHUB_API_KEY", raising=False)
|
|
monkeypatch.setenv("OPENROUTER_API_KEY", "sk-or-test")
|
|
status = get_api_key_provider_status("tencent-tokenhub")
|
|
assert not status["configured"]
|
|
|
|
def test_custom_base_url_override(self, monkeypatch):
|
|
monkeypatch.setenv("TOKENHUB_API_KEY", "sk-test-12345678")
|
|
monkeypatch.setenv("TOKENHUB_BASE_URL", "https://custom.tokenhub.example/v1")
|
|
creds = resolve_api_key_provider_credentials("tencent-tokenhub")
|
|
assert creds["base_url"] == "https://custom.tokenhub.example/v1"
|
|
|
|
|
|
# =============================================================================
|
|
# Model catalog
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubModelCatalog:
|
|
"""Tencent TokenHub static model list."""
|
|
|
|
def test_static_model_list_exists(self):
|
|
from hermes_cli.models import _PROVIDER_MODELS
|
|
assert "tencent-tokenhub" in _PROVIDER_MODELS
|
|
assert len(_PROVIDER_MODELS["tencent-tokenhub"]) >= 1
|
|
|
|
def test_hy3_preview_in_model_list(self):
|
|
from hermes_cli.models import _PROVIDER_MODELS
|
|
assert "hy3-preview" in _PROVIDER_MODELS["tencent-tokenhub"]
|
|
|
|
def test_default_model(self):
|
|
from hermes_cli.models import get_default_model_for_provider
|
|
assert get_default_model_for_provider("tencent-tokenhub") == "hy3-preview"
|
|
|
|
|
|
# =============================================================================
|
|
# CANONICAL_PROVIDERS (hermes model picker)
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubCanonicalProvider:
|
|
"""Tencent TokenHub appears in the interactive model picker."""
|
|
|
|
def test_in_canonical_providers(self):
|
|
from hermes_cli.models import CANONICAL_PROVIDERS
|
|
slugs = [p.slug for p in CANONICAL_PROVIDERS]
|
|
assert "tencent-tokenhub" in slugs
|
|
|
|
def test_label(self):
|
|
from hermes_cli.models import CANONICAL_PROVIDERS
|
|
entry = next(p for p in CANONICAL_PROVIDERS if p.slug == "tencent-tokenhub")
|
|
assert entry.label == "Tencent TokenHub"
|
|
|
|
def test_description_contains_hy3(self):
|
|
from hermes_cli.models import CANONICAL_PROVIDERS
|
|
entry = next(p for p in CANONICAL_PROVIDERS if p.slug == "tencent-tokenhub")
|
|
assert "Hy3 Preview" in entry.tui_desc
|
|
|
|
|
|
# =============================================================================
|
|
# OpenRouter / Nous Portal curated lists
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentInOpenRouterAndNous:
|
|
"""tencent/hy3-preview:free should appear in OpenRouter and Nous curated lists."""
|
|
|
|
def test_in_openrouter_fallback(self):
|
|
from hermes_cli.models import OPENROUTER_MODELS
|
|
ids = [mid for mid, _ in OPENROUTER_MODELS]
|
|
assert "tencent/hy3-preview:free" in ids
|
|
|
|
def test_in_nous_provider_models(self):
|
|
from hermes_cli.models import _PROVIDER_MODELS
|
|
assert "tencent/hy3-preview" in _PROVIDER_MODELS["nous"]
|
|
|
|
|
|
# =============================================================================
|
|
# Model normalization
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubNormalization:
|
|
"""Model name normalization — Tencent TokenHub is a direct provider
|
|
not in _MATCHING_PREFIX_STRIP_PROVIDERS, so names pass through as-is.
|
|
"""
|
|
|
|
def test_bare_name_passthrough(self):
|
|
"""hy3-preview should remain unchanged when targeting tencent-tokenhub."""
|
|
from hermes_cli.model_normalize import normalize_model_for_provider
|
|
result = normalize_model_for_provider("hy3-preview", "tencent-tokenhub")
|
|
assert result == "hy3-preview"
|
|
|
|
def test_vendor_prefixed_passthrough(self):
|
|
"""tencent/hy3-preview is not stripped since tencent-tokenhub is not in
|
|
_MATCHING_PREFIX_STRIP_PROVIDERS — the slash survives."""
|
|
from hermes_cli.model_normalize import normalize_model_for_provider
|
|
result = normalize_model_for_provider("tencent/hy3-preview", "tencent-tokenhub")
|
|
# Direct providers not in any special set → passthrough
|
|
assert result == "tencent/hy3-preview"
|
|
|
|
def test_not_in_matching_prefix_strip_set(self):
|
|
"""tencent-tokenhub does NOT need prefix stripping — it only has
|
|
one model (hy3-preview) and users won't copy vendor/ form."""
|
|
from hermes_cli.model_normalize import _MATCHING_PREFIX_STRIP_PROVIDERS
|
|
assert "tencent-tokenhub" not in _MATCHING_PREFIX_STRIP_PROVIDERS
|
|
|
|
def test_not_in_lowercase_providers(self):
|
|
"""tencent-tokenhub does not require lowercase normalization."""
|
|
from hermes_cli.model_normalize import _LOWERCASE_MODEL_PROVIDERS
|
|
assert "tencent-tokenhub" not in _LOWERCASE_MODEL_PROVIDERS
|
|
|
|
@pytest.mark.parametrize("empty_input", ["", None, " "])
|
|
def test_normalize_empty_and_none(self, empty_input):
|
|
"""None, empty, and whitespace-only inputs return empty string."""
|
|
from hermes_cli.model_normalize import normalize_model_for_provider
|
|
result = normalize_model_for_provider(empty_input, "tencent-tokenhub")
|
|
assert result == "" or result.strip() == ""
|
|
|
|
|
|
# =============================================================================
|
|
# Provider label
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubProviderLabel:
|
|
"""Test provider_label() from models.py for tencent-tokenhub."""
|
|
|
|
def test_label_from_provider_labels_dict(self):
|
|
from hermes_cli.models import _PROVIDER_LABELS
|
|
assert _PROVIDER_LABELS["tencent-tokenhub"] == "Tencent TokenHub"
|
|
|
|
def test_provider_label_function(self):
|
|
from hermes_cli.models import provider_label
|
|
assert provider_label("tencent-tokenhub") == "Tencent TokenHub"
|
|
|
|
def test_provider_label_via_alias(self):
|
|
from hermes_cli.models import provider_label
|
|
assert provider_label("tencent") == "Tencent TokenHub"
|
|
assert provider_label("tokenhub") == "Tencent TokenHub"
|
|
|
|
|
|
# =============================================================================
|
|
# URL mapping
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubURLMapping:
|
|
"""Test URL → provider inference for Tencent TokenHub endpoints."""
|
|
|
|
def test_url_to_provider(self):
|
|
from agent.model_metadata import _URL_TO_PROVIDER
|
|
assert _URL_TO_PROVIDER.get("tokenhub.tencentmaas.com") == "tencent-tokenhub"
|
|
|
|
def test_provider_prefixes(self):
|
|
from agent.model_metadata import _PROVIDER_PREFIXES
|
|
assert "tencent-tokenhub" in _PROVIDER_PREFIXES
|
|
assert "tencent" in _PROVIDER_PREFIXES
|
|
assert "tokenhub" in _PROVIDER_PREFIXES
|
|
|
|
def test_infer_from_url(self):
|
|
from agent.model_metadata import _infer_provider_from_url
|
|
assert _infer_provider_from_url("https://tokenhub.tencentmaas.com/v1") == "tencent-tokenhub"
|
|
|
|
|
|
# =============================================================================
|
|
# Context length
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubContextLength:
|
|
"""hy3-preview context length is registered."""
|
|
|
|
def test_hy3_preview_context_length(self):
|
|
from agent.model_metadata import get_model_context_length
|
|
ctx = get_model_context_length("hy3-preview")
|
|
assert ctx == 256000
|
|
|
|
|
|
# =============================================================================
|
|
# providers.py (unified provider module)
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubProvidersModule:
|
|
"""Test Tencent TokenHub in the unified providers module."""
|
|
|
|
def test_overlay_exists(self):
|
|
from hermes_cli.providers import HERMES_OVERLAYS
|
|
assert "tencent-tokenhub" in HERMES_OVERLAYS
|
|
overlay = HERMES_OVERLAYS["tencent-tokenhub"]
|
|
assert overlay.transport == "openai_chat"
|
|
assert overlay.base_url_env_var == "TOKENHUB_BASE_URL"
|
|
assert not overlay.is_aggregator
|
|
|
|
def test_alias_resolves(self):
|
|
from hermes_cli.providers import normalize_provider
|
|
assert normalize_provider("tencent") == "tencent-tokenhub"
|
|
assert normalize_provider("tokenhub") == "tencent-tokenhub"
|
|
|
|
def test_label(self):
|
|
from hermes_cli.providers import get_label
|
|
assert get_label("tencent-tokenhub") == "Tencent TokenHub"
|
|
|
|
def test_get_provider(self):
|
|
pdef = None
|
|
try:
|
|
from hermes_cli.providers import get_provider
|
|
pdef = get_provider("tencent-tokenhub")
|
|
except Exception:
|
|
pass
|
|
if pdef is not None:
|
|
assert pdef.id == "tencent-tokenhub"
|
|
assert pdef.transport == "openai_chat"
|
|
|
|
|
|
# =============================================================================
|
|
# Auxiliary client
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubAuxiliary:
|
|
"""Tencent TokenHub auxiliary model routing."""
|
|
|
|
def test_aux_model_registered(self):
|
|
from agent.auxiliary_client import _API_KEY_PROVIDER_AUX_MODELS
|
|
assert "tencent-tokenhub" in _API_KEY_PROVIDER_AUX_MODELS
|
|
assert _API_KEY_PROVIDER_AUX_MODELS["tencent-tokenhub"] == "hy3-preview"
|
|
|
|
def test_aux_aliases(self):
|
|
from agent.auxiliary_client import _PROVIDER_ALIASES
|
|
assert _PROVIDER_ALIASES.get("tencent") == "tencent-tokenhub"
|
|
assert _PROVIDER_ALIASES.get("tokenhub") == "tencent-tokenhub"
|
|
|
|
|
|
# =============================================================================
|
|
# Doctor
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubDoctor:
|
|
"""Verify hermes doctor recognizes Tencent TokenHub env vars."""
|
|
|
|
def test_provider_env_hints(self):
|
|
from hermes_cli.doctor import _PROVIDER_ENV_HINTS
|
|
assert "TOKENHUB_API_KEY" in _PROVIDER_ENV_HINTS
|
|
|
|
|
|
# =============================================================================
|
|
# Agent init (no SyntaxError, correct api_mode)
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubAgentInit:
|
|
"""Verify the agent can be constructed with tencent-tokenhub provider without errors."""
|
|
|
|
def test_no_syntax_errors(self):
|
|
"""Importing run_agent with tencent-tokenhub should not raise."""
|
|
import importlib
|
|
importlib.import_module("run_agent")
|
|
|
|
def test_api_mode_is_chat_completions(self):
|
|
from hermes_cli.providers import HERMES_OVERLAYS, TRANSPORT_TO_API_MODE
|
|
overlay = HERMES_OVERLAYS["tencent-tokenhub"]
|
|
api_mode = TRANSPORT_TO_API_MODE[overlay.transport]
|
|
assert api_mode == "chat_completions"
|
|
|
|
|
|
# =============================================================================
|
|
# CLI model flow dispatch (main.py)
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubCLIDispatch:
|
|
"""Verify tencent-tokenhub is routed through _model_flow_api_key_provider."""
|
|
|
|
def test_in_api_key_provider_tuple(self):
|
|
"""tencent-tokenhub must appear in the elif tuple in _model_flow dispatch
|
|
so ``hermes model`` routes it through the generic api_key_provider flow.
|
|
"""
|
|
import inspect
|
|
from hermes_cli import main as main_mod
|
|
source = inspect.getsource(main_mod)
|
|
# The source should contain tencent-tokenhub in the dispatch block
|
|
assert '"tencent-tokenhub"' in source or "'tencent-tokenhub'" in source
|
|
|
|
|
|
# =============================================================================
|
|
# Remote model catalog (model-catalog.json)
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubModelCatalogJSON:
|
|
"""Verify tencent/hy3-preview:free is present in the website model-catalog.json."""
|
|
|
|
def test_in_model_catalog_json(self):
|
|
catalog_path = os.path.join(
|
|
os.path.dirname(__file__),
|
|
"..", "..",
|
|
"website", "static", "api", "model-catalog.json",
|
|
)
|
|
if not os.path.isfile(catalog_path):
|
|
pytest.skip("model-catalog.json not found in workspace")
|
|
with open(catalog_path) as f:
|
|
data = json.load(f)
|
|
# Collect all model IDs across all provider lists.
|
|
# providers is a dict keyed by provider name, each value has a "models" list.
|
|
all_ids = set()
|
|
providers = data.get("providers", {})
|
|
if isinstance(providers, dict):
|
|
for provider_entry in providers.values():
|
|
for model in provider_entry.get("models", []):
|
|
all_ids.add(model.get("id", ""))
|
|
else:
|
|
for provider_entry in providers:
|
|
for model in provider_entry.get("models", []):
|
|
all_ids.add(model.get("id", ""))
|
|
assert "tencent/hy3-preview:free" in all_ids
|
|
|
|
|
|
# =============================================================================
|
|
# determine_api_mode (providers.py)
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubApiMode:
|
|
"""Verify determine_api_mode routes tencent-tokenhub correctly."""
|
|
|
|
def test_determine_api_mode_direct(self):
|
|
from hermes_cli.providers import determine_api_mode
|
|
mode = determine_api_mode("tencent-tokenhub")
|
|
assert mode == "chat_completions"
|
|
|
|
def test_determine_api_mode_with_base_url(self):
|
|
from hermes_cli.providers import determine_api_mode
|
|
mode = determine_api_mode("tencent-tokenhub", "https://tokenhub.tencentmaas.com/v1")
|
|
assert mode == "chat_completions"
|
|
|
|
def test_determine_api_mode_via_alias(self):
|
|
from hermes_cli.providers import determine_api_mode
|
|
mode = determine_api_mode("tencent")
|
|
assert mode == "chat_completions"
|
|
|
|
|
|
# =============================================================================
|
|
# _KNOWN_PROVIDER_NAMES (models.py)
|
|
# =============================================================================
|
|
|
|
|
|
class TestTencentTokenhubKnownProviderNames:
|
|
"""Verify tencent-tokenhub and its aliases are recognized as valid
|
|
provider names for the ``provider:model`` syntax.
|
|
"""
|
|
|
|
def test_canonical_id_known(self):
|
|
from hermes_cli.models import _KNOWN_PROVIDER_NAMES
|
|
assert "tencent-tokenhub" in _KNOWN_PROVIDER_NAMES
|
|
|
|
@pytest.mark.parametrize("alias", [
|
|
"tencent", "tokenhub", "tencent-cloud", "tencentmaas",
|
|
])
|
|
def test_alias_known(self, alias):
|
|
from hermes_cli.models import _KNOWN_PROVIDER_NAMES
|
|
assert alias in _KNOWN_PROVIDER_NAMES
|
|
|