@kubb/studio 5.3.3 → 5.3.4

package/dist/index.d.ts

@@ -98,7 +98,9 @@ export declare function createJob({ studioUrl, token, type, agentId, name, versi
   config?: Record<string, unknown>;
 }): Promise<StudioJob>;
 /**
- * Polls `GET /api/jobs/{id}` until the job reaches `success` or `failed`.
+ * Polls `GET /api/jobs/{id}` until the job reaches `success` or `failed`, waiting
+ * {@link INITIAL_POLL_DELAY_MS} first and doubling up to {@link MAX_POLL_INTERVAL_MS} so a long
+ * job stays inside the API key's rate limit.
  *
  * A `failed` job resolves normally. Check `job.status` and `job.error`. Throws only when the
  * deadline passes before Studio finishes.

package/dist/index.js

@@ -609,18 +609,41 @@ async function createJob({ studioUrl, token, type, agentId, name, version, confi
 	return job;
 }
 /**
-* Polls `GET /api/jobs/{id}` until the job reaches `success` or `failed`.
+* A job runs a generation and packs a tarball, so it is never done the instant it is queued.
+*/
+const INITIAL_POLL_DELAY_MS = 2e3;
+/**
+* Slowest the poll backs off to. Requests per run are roughly `timeoutMs` divided by this, and
+* every concurrent run on the same organization key draws on one budget.
+*/
+const MAX_POLL_INTERVAL_MS = 3e4;
+/**
+* Polls `GET /api/jobs/{id}` until the job reaches `success` or `failed`, waiting
+* {@link INITIAL_POLL_DELAY_MS} first and doubling up to {@link MAX_POLL_INTERVAL_MS} so a long
+* job stays inside the API key's rate limit.
 *
 * A `failed` job resolves normally. Check `job.status` and `job.error`. Throws only when the
 * deadline passes before Studio finishes.
 */
 async function waitForJob({ studioUrl, token, id, timeoutMs = 6e4 }) {
 	const deadline = Date.now() + timeoutMs;
+	let interval = INITIAL_POLL_DELAY_MS;
 	for (;;) {
-		const { job } = await ofetch(`${studioUrl}/api/jobs/${id}`, { headers: { "x-api-key": token } });
-		if (job.status === "success" || job.status === "failed") return job;
+		await new Promise((resolve) => setTimeout(resolve, Math.max(Math.min(interval, deadline - Date.now()), 0)));
 		if (Date.now() >= deadline) throw new Error("Timed out waiting for the Studio job");
-		await new Promise((resolve) => setTimeout(resolve, 1e3));
+		interval = Math.min(interval * 2, MAX_POLL_INTERVAL_MS);
+		try {
+			const { job } = await ofetch(`${studioUrl}/api/jobs/${id}`, {
+				headers: { "x-api-key": token },
+				retry: false
+			});
+			if (job.status === "success" || job.status === "failed") return job;
+		} catch (error) {
+			const response = error.response;
+			if (response?.status !== 429) throw error;
+			const retryAfter = response._data?.data?.tryAgainIn;
+			interval = Math.max(interval, typeof retryAfter === "number" && Number.isFinite(retryAfter) && retryAfter > 0 ? retryAfter : MAX_POLL_INTERVAL_MS);
+		}
 	}
 }
 /**
@@ -650,7 +673,7 @@ async function createAgent({ studioUrl, token, name, machineToken }) {
 }
 //#endregion
 //#region package.json
-var version = "5.3.3";
+var version = "5.3.4";
 //#endregion
 //#region src/hooks.ts
 /**