diff --git a/tests/test_provider_detection.py b/tests/test_provider_detection.py index 372a3950d..997f2c572 100644 --- a/tests/test_provider_detection.py +++ b/tests/test_provider_detection.py @@ -107,7 +107,10 @@ class TestBuildersRejectLookalikeHosts: assert build_chat_url("https://notanthropic.com") == "https://notanthropic.com/chat/completions" def test_lookalike_anthropic_models_is_openai(self): - assert build_models_url("https://anthropic.com.evil.com") == "https://anthropic.com.evil.com/models" + # Must hit the generic OpenAI branch, not Anthropic — assert the + # provider directly since both branches now end in /v1/models. + 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/v1/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" @@ -119,7 +122,9 @@ class TestBuildersRejectLookalikeHosts: assert build_chat_url("https://notollama.com") == "https://notollama.com/chat/completions" def test_lookalike_ollama_models_is_openai(self): - assert build_models_url("https://notollama.com") == "https://notollama.com/models" + # Must hit the generic OpenAI branch, not Ollama. + assert llm_core._detect_provider("https://notollama.com") == "openai" + assert build_models_url("https://notollama.com") == "https://notollama.com/v1/models" class TestBuildersLocalAndDockerEndpoints: