Agent API

set -euo pipefail
: "${KREA_API_KEY:?Set your Krea API key}"

START=$(curl --fail-with-body --silent --show-error --max-time 60 \
  https://api.krea.ai/agent/send-message \
  -H "Authorization: Bearer $KREA_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "prompt": "Create three square product images for a ceramic coffee mug on a neutral studio background. Use no text or logos.",
    "name": "Coffee mug campaign"
  }')

SESSION_ID=$(printf '%s' "$START" | jq -er '.sessionId')
RUN_ID=$(printf '%s' "$START" | jq -er '.runId')
printf '%s' "$START" | jq -r '.url'

DEADLINE=$((SECONDS + 300))

while (( SECONDS < DEADLINE - 10 )); do
  WAIT_SECONDS=$((DEADLINE - SECONDS - 10))
  if (( WAIT_SECONDS > 60 )); then WAIT_SECONDS=60; fi
  WAIT_BODY=$(jq -n --arg sessionId "$SESSION_ID" --arg runId "$RUN_ID" \
    --argjson timeoutSeconds "$WAIT_SECONDS" \
    '{sessionId: $sessionId, runId: $runId, timeoutSeconds: $timeoutSeconds}')
  RESULT=$(curl --fail-with-body --silent --show-error --max-time "$((WAIT_SECONDS + 10))" \
    https://api.krea.ai/agent/wait-for-session \
    -H "Authorization: Bearer $KREA_API_KEY" \
    -H "Content-Type: application/json" \
    -d "$WAIT_BODY")
  printf '%s' "$RESULT" | jq '{status, finished, timedOut, text, media, error, url}'
  if printf '%s' "$RESULT" | jq -e '.finished' > /dev/null; then
    if printf '%s' "$RESULT" | jq -e '.status == "completed"' > /dev/null; then
      exit 0
    fi
    printf '%s' "$RESULT" | jq -r '"Turn ended: \(.status). \(.error // "No error details")"' >&2
    exit 1
  fi
done

printf 'Wait budget exhausted; the turn continues. Resume waiting with sessionId=%s runId=%s\n' \
  "$SESSION_ID" "$RUN_ID" >&2
exit 1

import json
import os
import time
import requests

api = "https://api.krea.ai"
headers = {"Authorization": f"Bearer {os.environ['KREA_API_KEY']}"}

response = requests.post(
    f"{api}/agent/send-message",
    headers=headers,
    json={
        "prompt": "Create three square product images for a ceramic coffee mug on a neutral studio background. Use no text or logos.",
        "name": "Coffee mug campaign",
    },
    timeout=60,
)
response.raise_for_status()
start = response.json()
session_id, run_id = start["sessionId"], start["runId"]
print(start["url"])

deadline = time.monotonic() + 300
while True:
    wait_seconds = min(60, int(deadline - time.monotonic()) - 10)
    if wait_seconds < 1:
        raise TimeoutError(
            "Wait budget exhausted; the turn continues. Resume waiting with "
            f"sessionId={session_id} runId={run_id}"
        )
    response = requests.post(
        f"{api}/agent/wait-for-session",
        headers=headers,
        json={
            "sessionId": session_id,
            "runId": run_id,
            "timeoutSeconds": wait_seconds,
        },
        timeout=wait_seconds + 10,
    )
    response.raise_for_status()
    result = response.json()
    print(json.dumps(result, indent=2))
    if result["finished"]:
        if result["status"] != "completed":
            raise RuntimeError(
                f"Turn ended: {result['status']}. {result.get('error', 'No error details')}"
            )
        break

import { performance } from "node:perf_hooks";

const token = process.env.KREA_API_KEY;
if (!token) throw new Error("Set your Krea API key");
const api = "https://api.krea.ai";

async function post(path, body, timeoutMs) {
  const response = await fetch(`${api}${path}`, {
    method: "POST",
    headers: {
      Authorization: `Bearer ${token}`,
      "Content-Type": "application/json",
    },
    body: JSON.stringify(body),
    signal: AbortSignal.timeout(timeoutMs),
  });
  if (!response.ok) {
    throw new Error(`HTTP ${response.status}: ${await response.text()}`);
  }
  return response.json();
}

const start = await post("/agent/send-message", {
  prompt: "Create three square product images for a ceramic coffee mug on a neutral studio background. Use no text or logos.",
  name: "Coffee mug campaign",
}, 60_000);
const { sessionId, runId } = start;
console.log(start.url);

const deadline = performance.now() + 300_000;
while (true) {
  const timeoutSeconds = Math.min(60, Math.floor((deadline - performance.now()) / 1000) - 10);
  if (timeoutSeconds < 1) {
    throw new Error(
      `Wait budget exhausted; the turn continues. Resume waiting with sessionId=${sessionId} runId=${runId}`,
    );
  }
  const result = await post("/agent/wait-for-session", {
    sessionId,
    runId,
    timeoutSeconds,
  }, (timeoutSeconds + 10) * 1000);
  console.log(JSON.stringify(result, null, 2));
  if (result.finished) {
    if (result.status !== "completed") {
      throw new Error(`Turn ended: ${result.status}. ${result.error ?? "No error details"}`);
    }
    break;
  }
}

import { performance } from "node:perf_hooks";
import { Krea } from "@krea-ai/sdk";

const token = process.env.KREA_API_KEY;
if (!token) throw new Error("Set your Krea API key");
const krea = new Krea({ apiKey: token });

const start = await krea.requestRaw("/agent/send-message", {
  method: "POST",
  retry: false,
  signal: AbortSignal.timeout(60_000),
  body: {
    prompt: "Create three square product images for a ceramic coffee mug on a neutral studio background. Use no text or logos.",
    name: "Coffee mug campaign",
  },
});
const { sessionId, runId } = start;
console.log(start.url);

const deadline = performance.now() + 300_000;
while (true) {
  const timeoutSeconds = Math.min(60, Math.floor((deadline - performance.now()) / 1000) - 15);
  if (timeoutSeconds < 1) {
    throw new Error(
      `Wait budget exhausted; the turn continues. Resume waiting with sessionId=${sessionId} runId=${runId}`,
    );
  }
  const result = await krea.requestRaw("/agent/wait-for-session", {
    method: "POST",
    retry: false,
    signal: AbortSignal.timeout((timeoutSeconds + 15) * 1000),
    body: { sessionId, runId, timeoutSeconds },
  });
  console.log(JSON.stringify(result, null, 2));
  if (result.finished) {
    if (result.status !== "completed") {
      throw new Error(`Turn ended: ${result.status}. ${result.error ?? "No error details"}`);
    }
    break;
  }
}