mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-04-28 15:01:34 +08:00
44 lines
2.1 KiB
Python
44 lines
2.1 KiB
Python
|
"""Regression tests for ``determine_api_mode`` hostname handling.
|
||
|
|
|
||
|
|
Companion to tests/hermes_cli/test_detect_api_mode_for_url.py — the same
|
||
|
|
false-positive class (custom URLs containing ``api.openai.com`` /
|
||
|
|
``api.anthropic.com`` as a path segment or host suffix) must be rejected
|
||
|
|
by ``determine_api_mode`` as well, since it's the code path used by
|
||
|
|
custom/unknown providers in ``resolve_custom_provider``.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from hermes_cli.providers import determine_api_mode
|
||
|
|
|
||
|
|
|
||
|
|
class TestOpenAIHostHardening:
|
||
|
|
def test_native_openai_url_is_codex_responses(self):
|
||
|
|
assert determine_api_mode("", "https://api.openai.com/v1") == "codex_responses"
|
||
|
|
|
||
|
|
def test_openai_host_suffix_is_not_codex(self):
|
||
|
|
assert determine_api_mode("", "https://api.openai.com.example/v1") == "chat_completions"
|
||
|
|
|
||
|
|
def test_openai_path_segment_is_not_codex(self):
|
||
|
|
assert determine_api_mode("", "https://proxy.example.test/api.openai.com/v1") == "chat_completions"
|
||
|
|
|
||
|
|
|
||
|
|
class TestAnthropicHostHardening:
|
||
|
|
def test_native_anthropic_url_is_anthropic_messages(self):
|
||
|
|
assert determine_api_mode("", "https://api.anthropic.com") == "anthropic_messages"
|
||
|
|
|
||
|
|
def test_anthropic_host_suffix_is_not_anthropic(self):
|
||
|
|
assert determine_api_mode("", "https://api.anthropic.com.example/v1") == "chat_completions"
|
||
|
|
|
||
|
|
def test_anthropic_path_segment_is_not_anthropic(self):
|
||
|
|
# A proxy whose path contains ``api.anthropic.com`` must not be misrouted.
|
||
|
|
# Note: the ``/anthropic`` convention for third-party gateways still wins
|
||
|
|
# via explicit path-suffix check — see test_anthropic_path_suffix_still_wins.
|
||
|
|
assert determine_api_mode("", "https://proxy.example.test/api.anthropic.com/v1") == "chat_completions"
|
||
|
|
|
||
|
|
def test_anthropic_path_suffix_still_wins(self):
|
||
|
|
# Third-party Anthropic-compatible gateways (MiniMax, Zhipu GLM, LiteLLM
|
||
|
|
# proxies) expose the Anthropic protocol under a ``/anthropic`` suffix.
|
||
|
|
# That convention must still resolve to anthropic_messages.
|
||
|
|
assert determine_api_mode("", "https://api.minimax.io/anthropic") == "anthropic_messages"
|