From 6266a4fc012a43bf9a0f8d35b8ab8746f234d5d6 Mon Sep 17 00:00:00 2001
From: OG T <ogt@WOOOMacMiniM4.local>
Date: Thu, 2 Apr 2026 23:39:46 +0800
Subject: [PATCH] =?UTF-8?q?fix(test):=20=E6=9B=B4=E6=96=B0=20AIProviderEnu?=
 =?UTF-8?q?m=20=E6=B8=AC=E8=A9=A6=20=E2=80=94=20NVIDIA=20=E2=86=92=20NEMOT?=
 =?UTF-8?q?RON=20(Phase=2024=20B3)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- test_nvidia_provider_in_router: 改為驗證 NEMOTRON enum
- test_tool_calling_route: 改為期望 NEMOTRON provider
- test_existing_routing_not_affected: 排除 NEMOTRON (非一般路由)

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
---
 apps/api/tests/test_nvidia_provider.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

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()