#!/usr/bin/env python3
"""End-to-end test of the Symbiont router."""

import sys
import json

sys.path.insert(0, "/data/symbiont")
from symbiont.router import route_task

print("=" * 60)
print("SYMBIONT ROUTER - END TO END TEST")
print("=" * 60)

# Test 1: Simple task (should route to Haiku)
print()
print("--- Test 1: Simple extraction task ---")
result = route_task("Extract email addresses from: Contact hello@example.com or support@test.org")
print(json.dumps(result, indent=2))

# Test 2: Medium task (should route to Sonnet)
print()
print("--- Test 2: Content writing task ---")
result = route_task("Write a 3-sentence product description for an AI task router that saves money by using cheaper models")
print(json.dumps(result, indent=2))

print()
print("=" * 60)
print("TESTS COMPLETE")
print("=" * 60)