File size: 1,691 Bytes
8e5831a |
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 42 43 |
import requests
class APIClient:
"""Handles all API interactions with error handling."""
def __init__(self, base_url):
self.base_url = base_url.rstrip('/')
def get_questions(self):
"""Fetch the full list of evaluation questions."""
try:
resp = requests.get(f"{self.base_url}/questions", timeout=10)
resp.raise_for_status()
return resp.json()
except Exception as e:
raise RuntimeError(f"Failed to fetch questions: {e}")
def get_random_question(self):
"""Fetch a single random question."""
try:
resp = requests.get(f"{self.base_url}/random-question", timeout=10)
resp.raise_for_status()
return resp.json()
except Exception as e:
raise RuntimeError(f"Failed to fetch random question: {e}")
def get_file(self, task_id):
"""Download a specific file associated with a given task ID."""
try:
resp = requests.get(f"{self.base_url}/files/{task_id}", timeout=20)
resp.raise_for_status()
return resp.content
except Exception as e:
raise RuntimeError(f"Failed to fetch file for task {task_id}: {e}")
def submit(self, username, agent_code, answers):
"""Submit agent answers and return the result."""
data = {"username": username, "agent_code": agent_code, "answers": answers}
try:
resp = requests.post(f"{self.base_url}/submit", json=data, timeout=60)
resp.raise_for_status()
return resp.json()
except Exception as e:
raise RuntimeError(f"Failed to submit answers: {e}") |