mirror of
https://github.com/x1xhlol/system-prompts-and-models-of-ai-tools.git
synced 2026-06-17 23:09:35 +00:00
27 lines
768 B
Python
27 lines
768 B
Python
"""Model task routing."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from auto_client_acquisition.ai.model_router import ModelTask, get_model_route, requires_guardrail
|
|
|
|
|
|
def test_model_routes_exist():
|
|
r = get_model_route(ModelTask.ARABIC_WRITING)
|
|
assert r.task == ModelTask.ARABIC_WRITING
|
|
assert r.eval_metric
|
|
|
|
|
|
def test_high_risk_tasks_require_guardrail():
|
|
assert requires_guardrail(ModelTask.COMPLIANCE_GUARDRAIL) is True
|
|
assert requires_guardrail(ModelTask.STRATEGIC_REASONING) is True
|
|
|
|
|
|
def test_arabic_writing_route():
|
|
r = get_model_route(ModelTask.ARABIC_WRITING)
|
|
assert r.guardrail_required is True
|
|
|
|
|
|
def test_project_understanding_route():
|
|
r = get_model_route(ModelTask.PROJECT_CODE_UNDERSTANDING)
|
|
assert r.quality_tier == "high"
|