fix(ai): remove 188 ollama provider
This commit is contained in:
@@ -83,7 +83,7 @@ async def test_router_uses_failover_when_ollama_initial_provider():
|
||||
return_value=_make_failover_result(
|
||||
primary_provider="gemini",
|
||||
primary_model="gemini-1.5-flash",
|
||||
fallback=[("ollama_188", "qwen2.5:7b-instruct"), ("nemotron", "nvidia/nemotron-mini-4b-instruct")],
|
||||
fallback=[("ollama_local", "qwen2.5:7b-instruct"), ("nemotron", "nvidia/nemotron-mini-4b-instruct")],
|
||||
)
|
||||
)
|
||||
|
||||
@@ -109,14 +109,14 @@ async def test_router_uses_failover_when_ollama_initial_provider():
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_router_failover_fallback_chain_converted():
|
||||
"""failover_manager 回傳 fallback_chain → decision.fallback_chain 包含 OLLAMA_188"""
|
||||
"""failover_manager 回傳 fallback_chain → decision.fallback_chain 包含 OLLAMA_LOCAL"""
|
||||
mock_fm = MagicMock()
|
||||
mock_fm.select_provider = AsyncMock(
|
||||
return_value=_make_failover_result(
|
||||
primary_provider="gemini",
|
||||
primary_model="gemini-1.5-flash",
|
||||
fallback=[
|
||||
("ollama_188", "qwen2.5:7b-instruct"),
|
||||
("ollama_local", "qwen2.5:7b-instruct"),
|
||||
("nemotron", "nvidia/nemotron-mini-4b-instruct"),
|
||||
("claude", "claude-haiku-4-5-20251001"),
|
||||
],
|
||||
@@ -134,8 +134,8 @@ async def test_router_failover_fallback_chain_converted():
|
||||
decision = await router.route("test alert message")
|
||||
|
||||
fb_providers = [p for p, _ in decision.fallback_chain]
|
||||
assert AIProviderEnum.OLLAMA_188 in fb_providers, (
|
||||
f"OLLAMA_188 not in fallback_chain: {fb_providers}"
|
||||
assert AIProviderEnum.OLLAMA_LOCAL in fb_providers, (
|
||||
f"OLLAMA_LOCAL not in fallback_chain: {fb_providers}"
|
||||
)
|
||||
assert AIProviderEnum.NEMOTRON in fb_providers
|
||||
assert AIProviderEnum.CLAUDE in fb_providers
|
||||
|
||||
Reference in New Issue
Block a user