ritvik360/nl2sql-bench

1import requests
2
3space_url = "https://ritvik360-nl2sql-bench.hf.space"
4
5print("1. Resetting environment...")
6try:
7    res_reset = requests.post(f"{space_url}/reset", json={"task_name": "simple-filter"})
8    print(f"Reset Status: {res_reset.status_code}")
9except Exception as e:
10    print(f"Network error on reset: {e}")
11
12print("\n2. Sending test SQL step...")
13try:
14    # We send a perfectly valid query that should score > 0
15    payload = {"query": "SELECT id, name, email, country FROM customers WHERE tier = 'gold'"}
16    res_step = requests.post(f"{space_url}/step", json=payload)
17    
18    print(f"Step Status: {res_step.status_code}")
19    print(f"Step Response:\n{res_step.text}")
20except Exception as e:
21    print(f"Network error on step: {e}")