mirror of
https://github.com/x1xhlol/system-prompts-and-models-of-ai-tools.git
synced 2026-06-18 15:29:36 +00:00
50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
"""Unit tests for core.config.models.smart_route."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from core.config.models import (
|
|
Provider,
|
|
Task,
|
|
_arabic_ratio,
|
|
ordered_providers,
|
|
smart_route,
|
|
)
|
|
|
|
|
|
def test_classification_uses_groq():
|
|
cfg = smart_route(Task.CLASSIFICATION)
|
|
assert cfg.provider == Provider.GROQ
|
|
|
|
|
|
def test_arabic_heavy_routes_to_glm():
|
|
arabic_text = "مرحبا بك في دياليكس، نحن نحتاج مساعدة عاجلة"
|
|
cfg = smart_route(Task.SUMMARY, text_sample=arabic_text)
|
|
assert cfg.provider == Provider.GLM
|
|
|
|
|
|
def test_critical_routes_to_anthropic():
|
|
cfg = smart_route(Task.REASONING, critical=True)
|
|
assert cfg.provider == Provider.ANTHROPIC
|
|
|
|
|
|
def test_code_routes_to_deepseek():
|
|
cfg = smart_route(Task.CODE)
|
|
assert cfg.provider == Provider.DEEPSEEK
|
|
|
|
|
|
def test_research_uses_gemini():
|
|
cfg = smart_route(Task.RESEARCH)
|
|
assert cfg.provider == Provider.GEMINI
|
|
|
|
|
|
def test_arabic_ratio():
|
|
assert _arabic_ratio("") == 0.0
|
|
assert _arabic_ratio("hello") == 0.0
|
|
assert _arabic_ratio("مرحبا") > 0.9
|
|
|
|
|
|
def test_ordered_providers_starts_with_primary():
|
|
chain = ordered_providers(Task.CLASSIFICATION)
|
|
assert chain[0] == Provider.GROQ
|
|
assert len(chain) >= 2
|