fix(test): AIProvider → AIProviderEnum (Phase 24 C1 rename fix)
All checks were successful
CD Pipeline / build-and-deploy (push) Successful in 7m11s
E2E Health Check / e2e-health (push) Successful in 16s

C1 修復 (3ad7b60) 重命名 AIProvider Enum 為 AIProviderEnum
test_nvidia_provider.py 未同步更新,導致 CD 測試失敗。

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
OG T
2026-04-02 19:38:04 +08:00
parent 9afb518ea6
commit 5a7919f55c

View File

@@ -398,33 +398,33 @@ class TestAIRouterNvidiaIntegration:
"""測試 AIRouter NVIDIA 整合"""
def test_nvidia_provider_in_router(self):
"""測試 AIProvider 包含 NVIDIA"""
from src.services.ai_router import AIProvider
"""測試 AIProviderEnum 包含 NVIDIA"""
from src.services.ai_router import AIProviderEnum
assert hasattr(AIProvider, "NVIDIA")
assert AIProvider.NVIDIA.value == "nvidia"
assert hasattr(AIProviderEnum, "NVIDIA")
assert AIProviderEnum.NVIDIA.value == "nvidia"
def test_tool_calling_route(self):
"""測試 Tool Calling 路由"""
from src.services.ai_router import AIProvider, get_ai_router, reset_ai_router
from src.services.ai_router import AIProviderEnum, get_ai_router, reset_ai_router
reset_ai_router()
router = get_ai_router()
provider, model, fallback_chain = router.route_tool_calling()
assert provider == AIProvider.NVIDIA
assert provider == AIProviderEnum.NVIDIA
assert "nvidia" in model.lower() or "nemotron" in model.lower()
# Fallback 應該包含 Gemini 和 Claude
fallback_providers = [p for p, _ in fallback_chain]
assert AIProvider.GEMINI in fallback_providers
assert AIProvider.CLAUDE in fallback_providers
assert AIProviderEnum.GEMINI in fallback_providers
assert AIProviderEnum.CLAUDE in fallback_providers
reset_ai_router()
def test_existing_routing_not_affected(self):
"""測試現有路由規則不受影響"""
from src.services.ai_router import AIProvider, get_ai_router, reset_ai_router
from src.services.ai_router import AIProviderEnum, get_ai_router, reset_ai_router
reset_ai_router()
router = get_ai_router()
@@ -434,12 +434,12 @@ class TestAIRouterNvidiaIntegration:
# 應該還是使用 Ollama (低複雜度)
assert decision.selected_provider in [
AIProvider.OLLAMA,
AIProvider.GEMINI,
AIProvider.CLAUDE,
AIProviderEnum.OLLAMA,
AIProviderEnum.GEMINI,
AIProviderEnum.CLAUDE,
]
# NVIDIA 不應該出現在一般路由中
assert decision.selected_provider != AIProvider.NVIDIA
assert decision.selected_provider != AIProviderEnum.NVIDIA
reset_ai_router()