#!/usr/bin/env bash
set -euo pipefail

: "${COLABHIVE_API_KEY:?set COLABHIVE_API_KEY}"
: "${COLABHIVE_ACCOUNT_ID:?set COLABHIVE_ACCOUNT_ID}"
: "${COLABHIVE_ENDPOINT_ID:?set COLABHIVE_ENDPOINT_ID}"

base="${COLABHIVE_API_URL:-https://api.colabhive.com}/api/builder/v1"
headers=(-H "X-API-Key: ${COLABHIVE_API_KEY}" -H "X-Account-ID: ${COLABHIVE_ACCOUNT_ID}")

curl --fail-with-body "${headers[@]}" \
  "${base}/cohorts/capabilities?endpoint_id=${COLABHIVE_ENDPOINT_ID}"

curl --fail-with-body "${headers[@]}" -H 'Content-Type: application/json' \
  -H "Idempotency-Key: cohort-$(date +%s)" \
  -d "{\"endpoint_id\":\"${COLABHIVE_ENDPOINT_ID}\",\"input\":{\"text\":\"hello\"},\"execution\":{\"mode\":\"cohort\",\"fallback\":\"single\",\"trace\":\"summary\"}}" \
  "${base}/cohorts:run"