curl "$ATAI_API_URL/agents/optimizations/opt_01jcb1m3t7v5xq8nr2h6kdzs4w" \
  -H "Authorization: Bearer $ATAI_API_KEY"

import os
import time

import requests

base_url = os.environ["ATAI_API_URL"]
api_key = os.environ["ATAI_API_KEY"]
headers = {"Authorization": f"Bearer {api_key}"}

optimization_id = "opt_01jcb1m3t7v5xq8nr2h6kdzs4w"

while True:
    response = requests.get(f"{base_url}/agents/optimizations/{optimization_id}", headers=headers)
    run = response.json()
    progress = run["progress"]
    done = progress["completed"] + progress["failed"] + progress["cancelled"]
    print(f"{run['status']}: {done}/{run['budget']['max_trials']} trials, {progress['infeasible']} infeasible")

    if run["status"] in ("completed", "failed", "cancelled"):
        break
    time.sleep(15)

if run["best_trial_id"]:
    print(f"Winner: {run['best_trial_id']}")

const response = await fetch(
  `${process.env.ATAI_API_URL}/agents/optimizations/opt_01jcb1m3t7v5xq8nr2h6kdzs4w`,
  {
    headers: {
      'Authorization': `Bearer ${process.env.ATAI_API_KEY}`
    }
  }
);

const body = await response.json();

if (response.ok) {
  const { completed, failed, cancelled, infeasible } = body.progress;
  const done = completed + failed + cancelled;
  console.log(`${body.status}: ${done}/${body.budget.max_trials} trials, ${infeasible} infeasible`);
  if (body.best_trial_id) {
    console.log(`Winner: ${body.best_trial_id}`);
  }
} else {
  console.error('Error:', body.errors);
}

{
  "id": "opt_01jcb1m3t7v5xq8nr2h6kdzs4w",
  "name": "Window size sweep",
  "org_id": "org_01jc8m5r2vq9xt4bn7h3kdzs6w",
  "blueprint_id": "blp_01jc9n7k3xf8mbq2v5t0ary6de",
  "objective": "macro_f1",
  "search_space": {
    "parameters": {
      "window_size": {
        "kind": "value",
        "spec": {"type": "categorical", "values": [512, 1024, 2048]}
      },
      "n_neighbors": {
        "kind": "fitting",
        "spec": {"type": "int_range", "min": 3, "max": 25}
      }
    }
  },
  "budget": {"max_trials": 24},
  "constraints": {
    "min_metric": [
      {"metric": "recall", "class": "fault_bearing", "min_value": 0.8}
    ]
  },
  "training_examples": [
    {
      "name": "file_abc123",
      "ordinal": 1,
      "inputs": [
        {"type": "file", "id": "file_abc123", "format": "csv", "crc32c": "AAAAAA=="}
      ]
    }
  ],
  "calibration_examples": null,
  "validation_examples": [
    {
      "name": "file_def456",
      "ordinal": 1,
      "inputs": [
        {"type": "file", "id": "file_def456", "format": "csv", "crc32c": "AAAAAA=="}
      ]
    }
  ],
  "status": "completed",
  "progress": {
    "pending": 0,
    "running": 0,
    "completed": 22,
    "failed": 2,
    "cancelled": 0,
    "infeasible": 5
  },
  "best_trial_id": "otr_01jcb2v9h4x7mq3nt8k5rdzy6w",
  "created_by": "usr_01jc8m4p3rt6vx9qn2h5kdzb7y",
  "created_at": "2026-09-18T12:02:44Z",
  "started_at": "2026-09-18T12:02:51Z",
  "completed_at": "2026-09-18T13:47:05Z",
  "error": null
}

{
  "errors": [
    {
      "code": "<error_code>",
      "message": "Optimization not found.",
      "suggestion": null,
      "error_uid": "err-xxxxxxxx"
    }
  ]
}