Follow-up tasks

# 1. start
TASK_ID=$(curl -sS -X POST "https://api.autumn.ai/task" \
  -H "X-API-Key: $AUTUMN_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{"prompt": "Find 20 AI infrastructure startups hiring founding engineers", "clarify": false}' \
  | sed -n 's/.*"task_id": *"\([^"]*\)".*/\1/p')

# 2. send more instructions to the same task (queued with 202 if a turn is running)
curl -sS -X POST "https://api.autumn.ai/task/$TASK_ID/continue" \
  -H "X-API-Key: $AUTUMN_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{"message": "Add each company'\''s founder LinkedIn URL, and find 10 more rows"}'

# 3. wait for a status change until the turn is done
headers=$(mktemp)
trap 'rm -f "$headers"' EXIT
while :; do
  response=$(curl -sS -D "$headers" -w '\n%{http_code}' \
    "https://api.autumn.ai/task/$TASK_ID/state?wait=50" \
    -H "X-API-Key: $AUTUMN_API_KEY")
  status=${response##*$'\n'}
  task=${response%$'\n'*}
  if [ "$status" = 429 ] || [ "$status" = 503 ]; then
    retry_after=$(awk 'tolower($1) == "retry-after:" { gsub("\r", "", $2); delay=$2 } END { print delay }' "$headers")
    sleep "${retry_after:-5}"
    continue
  fi
  if [ "$status" -lt 200 ] || [ "$status" -ge 300 ]; then
    printf 'Status request failed (HTTP %s): %s\n' "$status" "$task" >&2
    exit 1
  fi
  printf '%s\n' "$task" | grep -q '"done": *true' && break
done

state=$(printf '%s\n' "$task" | grep -o '"state": *"[^"]*"' | head -n 1 | cut -d'"' -f4)
case $state in
  completed)
    curl -sS "https://api.autumn.ai/task/$TASK_ID/output" \
      -H "X-API-Key: $AUTUMN_API_KEY"
    ;;
  input-required)
    printf '%s\n' "$task" | grep -o -E '"message": *"([^"\\]|\\.)*"' | head -n 1 | sed -E 's/^"message": *"//; s/"$//; s/\\"/"/g; s/\\\\/\\/g'
    ;;
  failed|canceled)
    printf 'Task %s: %s\n' "$state" "$task" >&2
    exit 1
    ;;
  *)
    printf 'Task %s: %s\n' "$state" "$task" >&2
    exit 1
    ;;
esac

import os, requests, time

AUTUMN = "https://api.autumn.ai"
HEADERS = {"X-API-Key": os.environ["AUTUMN_API_KEY"]}

# start a task
task = requests.post(f"{AUTUMN}/task", headers=HEADERS, json={
    "prompt": "Find 20 AI infrastructure startups hiring founding engineers",
    "clarify": False,
}).json()
task_id = task["task_id"]

def get_status(task_id):
    while True:
        r = requests.get(
            f"{AUTUMN}/task/{task_id}",
            headers=HEADERS,
            params=dict(wait=50),
        )
        if r.status_code in (429, 503):
            time.sleep(float(r.headers.get("Retry-After", 5)))
            continue
        if not 200 <= r.status_code < 300:
            raise RuntimeError(r.json().get("message") or r.text)
        return r.json()

# send more instructions to the same task
task = requests.post(f"{AUTUMN}/task/{task_id}/continue", headers=HEADERS, json={
    "message": "Add each company's founder LinkedIn URL, and find 10 more rows",
}).json()

# wait for a status change until the turn is done
task = get_status(task_id)
while not task["done"]:
    task = get_status(task_id)

if task["state"] == "completed":
    rows = requests.get(f"{AUTUMN}/task/{task_id}/output", headers=HEADERS).json()["rows"]
elif task["state"] == "input-required":
    print(task["message"])
elif task["state"] in {"failed", "canceled"}:
    error = task.get("error") or {}
    raise RuntimeError(
        error.get("message") or error.get("code") or task.get("message") or task["state"]
    )

const AUTUMN = "https://api.autumn.ai";
const HEADERS = {
  "X-API-Key": process.env.AUTUMN_API_KEY,
  "Content-Type": "application/json",
};

const { task_id } = await fetch(`${AUTUMN}/task`, {
  method: "POST",
  headers: HEADERS,
  body: JSON.stringify({
    prompt: "Find 20 AI infrastructure startups hiring founding engineers",
    clarify: false,
  }),
}).then((r) => r.json());

async function getStatus(taskId) {
  while (true) {
    const r = await fetch(`${AUTUMN}/task/${taskId}?wait=50`, { headers: HEADERS });
    if (r.status === 429 || r.status === 503) {
      const seconds = Number(r.headers.get("Retry-After") || 5);
      await new Promise((resolve) => setTimeout(resolve, seconds * 1000));
      continue;
    }
    const body = await r.json();
    if (!r.ok) throw new Error(body.message || "Status request failed");
    return body;
  }
}

let task = await fetch(`${AUTUMN}/task/${task_id}/continue`, {
  method: "POST",
  headers: HEADERS,
  body: JSON.stringify({
    message: "Add each company's founder LinkedIn URL, and find 10 more rows",
  }),
}).then((r) => r.json());

// wait for a status change until the turn is done
task = await getStatus(task_id);
while (!task.done) {
  task = await getStatus(task_id);
}

if (task.state === "completed") {
  const { rows } = await fetch(`${AUTUMN}/task/${task_id}/output`, { headers: HEADERS })
    .then((r) => r.json());
} else if (task.state === "input-required") {
  console.log(task.message);
} else if (task.state === "failed" || task.state === "canceled") {
  const error = task.error || {};
  throw new Error(error.message || error.code || task.message || task.state);
}