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()