diff --git a/apps/api/tests/test_nvidia_provider.py b/apps/api/tests/test_nvidia_provider.py index 2027c4a6..e2469e70 100644 --- a/apps/api/tests/test_nvidia_provider.py +++ b/apps/api/tests/test_nvidia_provider.py @@ -398,11 +398,11 @@ class TestAIRouterNvidiaIntegration: """測試 AIRouter NVIDIA 整合""" def test_nvidia_provider_in_router(self): - """測試 AIProviderEnum 包含 NVIDIA""" + """測試 AIProviderEnum 包含 NEMOTRON (Phase 24 B3: NVIDIA → NEMOTRON)""" from src.services.ai_router import AIProviderEnum - assert hasattr(AIProviderEnum, "NVIDIA") - assert AIProviderEnum.NVIDIA.value == "nvidia" + assert hasattr(AIProviderEnum, "NEMOTRON") + assert AIProviderEnum.NEMOTRON.value == "nemotron" def test_tool_calling_route(self): """測試 Tool Calling 路由""" @@ -413,7 +413,7 @@ class TestAIRouterNvidiaIntegration: provider, model, fallback_chain = router.route_tool_calling() - assert provider == AIProviderEnum.NVIDIA + assert provider == AIProviderEnum.NEMOTRON assert "nvidia" in model.lower() or "nemotron" in model.lower() # Fallback 應該包含 Gemini 和 Claude fallback_providers = [p for p, _ in fallback_chain] @@ -429,7 +429,7 @@ class TestAIRouterNvidiaIntegration: reset_ai_router() router = get_ai_router() - # 測試同步路由 (不涉及 NVIDIA) + # 測試同步路由 (不涉及 NEMOTRON) decision = router.route_sync("重啟 api pod") # 應該還是使用 Ollama (低複雜度) @@ -438,8 +438,8 @@ class TestAIRouterNvidiaIntegration: AIProviderEnum.GEMINI, AIProviderEnum.CLAUDE, ] - # NVIDIA 不應該出現在一般路由中 - assert decision.selected_provider != AIProviderEnum.NVIDIA + # NEMOTRON 不應該出現在一般路由中 + assert decision.selected_provider != AIProviderEnum.NEMOTRON reset_ai_router()