File size: 1,409 Bytes
1902271 | 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 34 35 36 37 38 39 40 41 | import sys
import os
# Add the backend directory to sys.path so we can import app modules
sys.path.append(os.path.join(os.getcwd(), 'backend'))
from app.config.agents import get_agents_config
def test_dispatch():
config = get_agents_config()
print(f"Loaded {len(config.assistants)} assistants.")
test_cases = [
# (Language, Age Group, Category, Grading Mode, Expected Name Pattern)
("zh-TW", "國中", "一般文章", "全文批改", ["九大複句教練", "會考作文", "AI 寫作引路人"]),
("zh-TW", "國小", "自然科普", "全文批改", ["科普寫作教練"]),
("zh-TW", "大學", "一般文章", "全文批改", ["作文練習整體回饋"]),
("zh-TW", "高中", "一般文章", "結構建議", ["段落批改機器人"]),
]
print(f"\n{'='*60}")
print(f"{'TEST CASE':<40} | {'SELECTED AGENT':<20} | {'ID'}")
print(f"{'='*60}")
for lang, age, cat, mode, expected_keywords in test_cases:
agent = config.find_assistant(
language=lang,
age_group=age,
category=cat,
grading_mode=mode
)
case_desc = f"{age}/{cat}/{mode}"
if agent:
print(f"{case_desc:<40} | {agent.name:<20} | {agent.assistant_id}")
else:
print(f"{case_desc:<40} | {'None':<20} | -")
if __name__ == "__main__":
test_dispatch()
|