fix(test): AIProvider → AIProviderEnum (Phase 24 C1 rename fix)
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:
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user