Async tasks and polling

import os, time, requests

BASE = "https://agents.nanonets.com"
HEADERS = {"Authorization": f"Bearer {os.environ['NANONETS_API_KEY']}"}
TERMINAL = {"completed", "failed", "stopped"}

def wait_for_task(task_id: str, timeout_s: int = 600) -> dict:
    deadline = time.time() + timeout_s
    delay = 1.0
    while time.time() < deadline:
        r = requests.get(f"{BASE}/api/v1/tasks/{task_id}", headers=HEADERS, timeout=10)
        r.raise_for_status()
        body = r.json()
        if body["status"] in TERMINAL:
            return body
        if body["status"] == "waiting_for_input":
            raise RuntimeError(f"Task {task_id} needs a reply; see /interactive-tasks")
        time.sleep(delay)
        delay = min(delay * 1.5, 10.0)  # gentle exponential backoff, cap at 10s
    raise TimeoutError(f"Task {task_id} did not finish within {timeout_s}s")

final = wait_for_task("f1c2a3b4-...")
if final["status"] == "completed":
    summary = requests.get(
        f"{BASE}/api/v1/tasks/{final['task_id']}/summary", headers=HEADERS
    ).json()
    print(summary)

const BASE = "https://agents.nanonets.com";
const headers = { Authorization: `Bearer ${process.env.NANONETS_API_KEY}` };
const TERMINAL = new Set(["completed", "failed", "stopped"]);

async function waitForTask(taskId, timeoutMs = 600_000) {
  const deadline = Date.now() + timeoutMs;
  let delay = 1000;
  while (Date.now() < deadline) {
    const res = await fetch(`${BASE}/api/v1/tasks/${taskId}`, { headers });
    if (!res.ok) throw new Error(`status ${res.status}`);
    const body = await res.json();
    if (TERMINAL.has(body.status)) return body;
    if (body.status === "waiting_for_input") {
      throw new Error(`Task ${taskId} needs a reply; see /interactive-tasks`);
    }
    await new Promise((r) => setTimeout(r, delay));
    delay = Math.min(delay * 1.5, 10_000);
  }
  throw new Error(`Task ${taskId} timed out`);
}

TASK_ID="f1c2a3b4-..."
while :; do
  STATUS=$(curl -s "https://agents.nanonets.com/api/v1/tasks/$TASK_ID" \
    -H "Authorization: Bearer $NANONETS_API_KEY" | jq -r .status)
  echo "$STATUS"
  case "$STATUS" in
    completed|failed|stopped) break ;;
    waiting_for_input) echo "Needs a reply"; exit 1 ;;
  esac
  sleep 2
done
curl -s "https://agents.nanonets.com/api/v1/tasks/$TASK_ID/summary" \
  -H "Authorization: Bearer $NANONETS_API_KEY"