fix(test): 更新 AIProviderEnum 測試 — NVIDIA → NEMOTRON (Phase 24 B3)
All checks were successful
CD Pipeline / build-and-deploy (push) Successful in 7m6s
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:
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user