fix(test): 更新 AIProviderEnum 測試 — NVIDIA → NEMOTRON (Phase 24 B3)
All checks were successful
CD Pipeline / build-and-deploy (push) Successful in 7m6s

- 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>
This commit is contained in:
OG T
2026-04-02 23:39:46 +08:00
parent e9a1ac6276
commit 6266a4fc01

View File

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