mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-06-28 07:35:27 -04:00
Alexandre Teixeira
83 lines
4.0 KiB
Python
83 lines
4.0 KiB
Python
"""Provider detection tests — build_chat_url / build_models_url routing (re: #768).
|
|
|
|
These import the *real* helpers from ``src.llm_core`` (not local copies) so a
|
|
regression in hostname matching is actually caught. The point of the change
|
|
under test is that provider detection keys off the URL's *hostname*, not a
|
|
substring of the whole URL — so a domain appearing in a path/query, or a
|
|
look-alike host, must not be misclassified.
|
|
"""
|
|
import pytest
|
|
|
|
from src import llm_core
|
|
from src import endpoint_resolver
|
|
from src.endpoint_resolver import build_chat_url, build_models_url
|
|
|
|
|
|
class TestBuildersRejectLookalikeHosts:
|
|
"""build_chat_url / build_models_url must route look-alike and
|
|
domain-in-path hosts to the OpenAI-compatible default, not the
|
|
anthropic/ollama branches. Before #815's follow-up these builders still
|
|
fell back to ``host.endswith("anthropic.com")`` style checks, so
|
|
``notanthropic.com`` was misrouted to the Anthropic messages endpoint.
|
|
"""
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _stub_dns(self, monkeypatch):
|
|
# build_* call resolve_url(), which does real DNS + tailscale lookups.
|
|
# Provider routing is independent of name resolution, so stub it out to
|
|
# keep these deterministic and offline.
|
|
monkeypatch.setattr(endpoint_resolver, "resolve_url", lambda u: u)
|
|
|
|
def test_real_anthropic_chat(self):
|
|
assert build_chat_url("https://api.anthropic.com") == "https://api.anthropic.com/v1/messages"
|
|
|
|
def test_chatgpt_subscription_chat_uses_responses(self):
|
|
assert build_chat_url("https://chatgpt.com/backend-api/codex") == "https://chatgpt.com/backend-api/codex/responses"
|
|
|
|
def test_chatgpt_subscription_models_uses_no_live_probe(self):
|
|
assert build_models_url("https://chatgpt.com/backend-api/codex") is None
|
|
|
|
def test_lookalike_anthropic_chat_is_openai(self):
|
|
assert build_chat_url("https://notanthropic.com") == "https://notanthropic.com/chat/completions"
|
|
|
|
def test_lookalike_anthropic_models_is_openai(self):
|
|
assert llm_core._detect_provider("https://anthropic.com.evil.com") == "openai"
|
|
assert build_models_url("https://anthropic.com.evil.com") == "https://anthropic.com.evil.com/models"
|
|
|
|
def test_anthropic_domain_in_path_is_openai(self):
|
|
assert build_chat_url("https://myproxy.internal/anthropic.com/v1") == "https://myproxy.internal/anthropic.com/v1/chat/completions"
|
|
|
|
def test_real_ollama_chat(self):
|
|
assert build_chat_url("https://ollama.com") == "https://ollama.com/api/chat"
|
|
|
|
def test_lookalike_ollama_chat_is_openai(self):
|
|
assert build_chat_url("https://notollama.com") == "https://notollama.com/chat/completions"
|
|
|
|
def test_lookalike_ollama_models_is_openai(self):
|
|
assert llm_core._detect_provider("https://notollama.com") == "openai"
|
|
assert build_models_url("https://notollama.com") == "https://notollama.com/models"
|
|
|
|
|
|
class TestBuildersLocalAndDockerEndpoints:
|
|
"""Local and docker endpoints must keep working after the hostname change:
|
|
a local ``/v1`` base stays OpenAI-compatible, and a native Ollama ``/api``
|
|
path is still detected by path even on a non-ollama.com host such as
|
|
host.docker.internal.
|
|
"""
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _stub_dns(self, monkeypatch):
|
|
monkeypatch.setattr(endpoint_resolver, "resolve_url", lambda u: u)
|
|
|
|
def test_local_v1_chat_is_openai_compatible(self):
|
|
assert build_chat_url("http://localhost:8000/v1") == "http://localhost:8000/v1/chat/completions"
|
|
|
|
def test_local_v1_models_is_openai_compatible(self):
|
|
assert build_models_url("http://127.0.0.1:1234/v1") == "http://127.0.0.1:1234/v1/models"
|
|
|
|
def test_docker_internal_ollama_api_path_is_native_chat(self):
|
|
assert build_chat_url("http://host.docker.internal:11434/api") == "http://host.docker.internal:11434/api/chat"
|
|
|
|
def test_docker_internal_ollama_api_path_is_native_models(self):
|
|
assert build_models_url("http://host.docker.internal:11434/api") == "http://host.docker.internal:11434/api/tags"
|