File size: 1,232 Bytes
e2640ca | 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
from config import API_URL
_TIMEOUT_FAST = 10
_TIMEOUT_QUERY = 60
_TIMEOUT_UPLOAD = 120
_TIMEOUT_DATASET = 600
def health_check() -> dict:
resp = requests.get(f"{API_URL}/health", timeout=_TIMEOUT_FAST)
resp.raise_for_status()
return resp.json()
def upload_pdfs(files) -> dict:
payload = [("files", (f.name, f.read(), "application/pdf")) for f in files]
resp = requests.post(f"{API_URL}/upload_pdfs/", files=payload, timeout=_TIMEOUT_UPLOAD)
resp.raise_for_status()
return resp.json()
def ask_question(question: str) -> dict:
resp = requests.post(f"{API_URL}/ask/", data={"question": question}, timeout=_TIMEOUT_QUERY)
resp.raise_for_status()
return resp.json()
def get_available_datasets() -> list[dict]:
resp = requests.get(f"{API_URL}/hf_datasets/", timeout=_TIMEOUT_FAST)
resp.raise_for_status()
return resp.json().get("available_datasets", [])
def load_datasets(dataset_names: list[str], max_samples: int = 500) -> dict:
resp = requests.post(
f"{API_URL}/hf_datasets/load/",
json={"datasets": dataset_names, "max_samples": max_samples},
timeout=_TIMEOUT_DATASET,
)
resp.raise_for_status()
return resp.json()
|