Run a workflow from code

AGENT_ID="YOUR_AGENT_ID"

curl -X POST "https://app.sushibelt.ai/api/v1/agents/$AGENT_ID/runs" \
  -H "Authorization: Bearer $SUSHIBELT_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{}'

// sushibelt.mts: a small client for the Sushibelt API. Node 18 or later.
const API = 'https://app.sushibelt.ai/api/v1'
const KEY = process.env.SUSHIBELT_API_KEY

export interface Run {
  jobId: string
  status: string
  progress: number
  steps: { node_id: string; step_type: string; status: string; error?: string }[]
  output: { node_id: string; step_type: string; result: unknown; licence?: unknown }[]
}

export class SushibeltError extends Error {
  constructor(readonly status: number, readonly body: Record<string, unknown>) {
    super(`Sushibelt answered ${status}: ${JSON.stringify(body)}`)
  }
}

const sleep = (ms: number) => new Promise((r) => setTimeout(r, ms))

/** One request. On 429 and 503 it waits for Retry-After and tries again, up to 5 times. */
async function call(method: 'GET' | 'POST', path: string, body?: unknown): Promise<Record<string, unknown>> {
  for (let attempt = 1; ; attempt++) {
    const res = await fetch(`${API}${path}`, {
      method,
      headers: { Authorization: `Bearer ${KEY}`, 'Content-Type': 'application/json' },
      body: body === undefined ? undefined : JSON.stringify(body),
    })
    const json = (await res.json().catch(() => ({}))) as Record<string, unknown>
    if (res.ok) return json
    if ((res.status === 429 || res.status === 503) && attempt < 5) {
      await sleep(Number(res.headers.get('Retry-After') ?? 30) * 1000)
      continue
    }
    throw new SushibeltError(res.status, json)
  }
}

/**
 * Start a run and return its id. If the workflow already has a run in
 * progress, returns that run's id instead: it may have been started with
 * other inputs, so only do this when any run of the workflow will do.
 */
export async function startRun(workflowId: string, body: Record<string, unknown> = {}): Promise<string> {
  try {
    return (await call('POST', `/workflows/${workflowId}/runs`, body)).jobId as string
  } catch (err) {
    if (err instanceof SushibeltError && err.status === 409) return err.body.activeJobId as string
    throw err
  }
}

/** Read a run until it completes or fails. Gives up after timeoutMs. */
export async function waitForRun(jobId: string, timeoutMs = 15 * 60_000): Promise<Run> {
  const deadline = Date.now() + timeoutMs
  let delay = 2_000
  while (Date.now() < deadline) {
    const run = (await call('GET', `/runs/${jobId}`)) as unknown as Run
    if (run.status === 'completed' || run.status === 'failed') return run
    await sleep(delay)
    delay = Math.min(delay * 1.5, 10_000)
  }
  throw new Error(`Run ${jobId} did not finish within ${timeoutMs / 1000} seconds`)
}

// Use it:
const run = await waitForRun(await startRun('YOUR_WORKFLOW_ID'))
if (run.status === 'failed') {
  for (const s of run.steps.filter((s) => s.status === 'failed')) console.error(`${s.step_type} failed: ${s.error}`)
} else {
  console.log(JSON.stringify(run.output, null, 2))
}

WORKFLOW_ID="YOUR_WORKFLOW_ID"

curl -X POST "https://app.sushibelt.ai/api/v1/workflows/$WORKFLOW_ID/runs" \
  -H "Authorization: Bearer $SUSHIBELT_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "mode": "draft"
  }'

WORKFLOW_ID="YOUR_WORKFLOW_ID"

curl -X POST "https://app.sushibelt.ai/api/v1/workflows/$WORKFLOW_ID/runs" \
  -H "Authorization: Bearer $SUSHIBELT_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "defer": true,
    "maxMultiplier": 0.7
  }'

DEFERRED-RUN_ID="YOUR_DEFERRED-RUN_ID"

curl "https://app.sushibelt.ai/api/v1/deferred-runs/$DEFERRED-RUN_ID" \
  -H "Authorization: Bearer $SUSHIBELT_API_KEY"