diff --git a/apps/api/tests/test_nvidia_provider.py b/apps/api/tests/test_nvidia_provider.py index 57d821dc5..4df03a7b8 100644 --- a/apps/api/tests/test_nvidia_provider.py +++ b/apps/api/tests/test_nvidia_provider.py @@ -476,15 +476,15 @@ class TestAIRouterNvidiaIntegration: assert fallback_providers == [ AIProviderEnum.OLLAMA_GCP_B, AIProviderEnum.OLLAMA_LOCAL, + AIProviderEnum.CLAUDE, AIProviderEnum.GEMINI, ] assert AIProviderEnum.NEMOTRON not in fallback_providers - assert AIProviderEnum.CLAUDE not in fallback_providers reset_ai_router() def test_existing_routing_not_affected(self): - """測試現有路由規則不受影響""" + """同步路由也遵守唯一 production provider 順序。""" from src.services.ai_router import ( AIProviderEnum, get_ai_router, @@ -501,6 +501,7 @@ class TestAIRouterNvidiaIntegration: assert [provider for provider, _ in decision.fallback_chain] == [ AIProviderEnum.OLLAMA_GCP_B, AIProviderEnum.OLLAMA_LOCAL, + AIProviderEnum.CLAUDE, AIProviderEnum.GEMINI, ]