curl -X POST "$ATAI_API_URL/agents/instances/agt_01jc9q8v5nm3ry7t2bkz4dhs6f/query" \
  -H "Authorization: Bearer $ATAI_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "records": [
      {"timestamp": "2026-09-18T11:24:03Z", "vibration": 0.41, "flow": 12.6}
    ]
  }'

import os
import requests

base_url = os.environ["ATAI_API_URL"]
api_key = os.environ["ATAI_API_KEY"]

response = requests.post(
    f"{base_url}/agents/instances/agt_01jc9q8v5nm3ry7t2bkz4dhs6f/query",
    headers={
        "Authorization": f"Bearer {api_key}",
        "Content-Type": "application/json",
    },
    json={
        "records": [
            {"timestamp": "2026-09-18T11:24:03Z", "vibration": 0.41, "flow": 12.6}
        ]
    },
)

if response.status_code == 200:
    body = response.json()
    print(f"{body['status']} in {body['query_response_time_ms']} ms")
    print(body["response"])
else:
    print(f"Error: {response.json()['errors']}")

const response = await fetch(
  `${process.env.ATAI_API_URL}/agents/instances/agt_01jc9q8v5nm3ry7t2bkz4dhs6f/query`,
  {
    method: 'POST',
    headers: {
      'Authorization': `Bearer ${process.env.ATAI_API_KEY}`,
      'Content-Type': 'application/json'
    },
    body: JSON.stringify({
      records: [
        { timestamp: '2026-09-18T11:24:03Z', vibration: 0.41, flow: 12.6 }
      ]
    })
  }
);

const body = await response.json();

if (response.ok) {
  console.log(`${body.status} in ${body.query_response_time_ms} ms`);
  console.log(body.response);
} else {
  console.error('Error:', body.errors);
}

{
  "status": "completed",
  "response": {
    "predicted_state": "running",
    "p_running": 0.94,
    "p_idle": 0.06
  },
  "query_response_time_ms": 38.4
}

{
  "errors": [
    {
      "code": "<error_code>",
      "message": "No agent with this id in the caller's org.",
      "suggestion": null,
      "error_uid": "err-xxxxxxxx"
    }
  ]
}

{
  "errors": [
    {
      "code": "<error_code>",
      "message": "The agent was not run in serving mode, or is no longer running.",
      "suggestion": null,
      "error_uid": "err-xxxxxxxx"
    }
  ]
}