File size: 973 Bytes
0e76632 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | """Tests for smart routing and query endpoint extensions."""
import pytest
from mac.services.llm_service import _smart_route
def test_smart_route_code_keywords():
messages = [{"role": "user", "content": "Write a Python function to sort a list and debug the algorithm"}]
result = _smart_route(messages)
assert result == "qwen2.5-coder:7b"
def test_smart_route_math_keywords():
messages = [{"role": "user", "content": "Solve this integral of x squared and calculate the derivative"}]
result = _smart_route(messages)
assert result == "deepseek-r1:14b"
def test_smart_route_general():
messages = [{"role": "user", "content": "Hello, how are you today?"}]
result = _smart_route(messages)
assert result == "qwen2.5:7b"
def test_smart_route_empty():
result = _smart_route([])
assert result is not None
def test_smart_route_none():
result = _smart_route(None)
assert result is not None
|