import asyncio
import os
import json
from client import CustomerSupportEnv
from models import CustomerSupportAction

async def main():
    os.environ["TASK_NAME"] = "task1"
    env = CustomerSupportEnv(base_url="http://localhost:8001")
    res = await env.reset()
    print("RESET RESULT:", res)
    
    action = CustomerSupportAction(action_type="assign", department="TechSupport", priority="High")
    res = await env.step(action)
    print("STEP RESULT:", res)

asyncio.run(main())