@nodetool-ai/runtime 0.7.0-rc.23 → 0.8.2

package/dist/providers/atlascloud-webhook-registry.js

@@ -0,0 +1,104 @@
+/**
+ * In-memory registry for AtlasCloud webhook callbacks.
+ *
+ * AtlasCloud calls back once a prediction reaches a terminal state
+ * (https://atlascloud.ai/docs/en/webhooks). The provider registers the
+ * prediction id it got from submit, the signed ingress route
+ * (`packages/websocket/src/routes/atlascloud-webhook.ts`) resolves it with the
+ * callback's `payload`, and the run continues without polling.
+ *
+ * Delivery is at-least-once and best effort, so this registry is never the
+ * only way a run finishes: `atlasAwaitResult` races it against a slow
+ * reconciliation poll of the prediction endpoint.
+ */
+import { createLogger } from "@nodetool-ai/config";
+const log = createLogger("nodetool.runtime.providers.atlascloud-webhook");
+/** The wait's own deadline passed with no callback. */
+export class AtlasWebhookWaitTimeout extends Error {
+    constructor(predictionId, timeoutMs) {
+        super(`AtlasCloud webhook not received within ` +
+            `${Math.round(timeoutMs / 1000)}s (predictionId: ${predictionId})`);
+        this.name = "AtlasWebhookWaitTimeout";
+    }
+}
+const pending = new Map();
+/**
+ * Wait for the callback AtlasCloud sends for `predictionId`. Rejects on the
+ * provider's own failure callback, on timeout, and on abort.
+ */
+export function registerAtlasWebhookWait(predictionId, timeoutMs, signal) {
+    return new Promise((resolve, reject) => {
+        if (signal?.aborted) {
+            reject(signal.reason instanceof Error ? signal.reason : new Error("Aborted"));
+            return;
+        }
+        // Declared before the handlers that call it so every exit from the wait —
+        // callback, timeout, abort — drops the timer, the listener and the entry.
+        let settle = () => undefined;
+        const timer = setTimeout(() => {
+            settle();
+            reject(new AtlasWebhookWaitTimeout(predictionId, timeoutMs));
+        }, timeoutMs);
+        const onAbort = () => {
+            settle();
+            reject(signal?.reason instanceof Error ? signal.reason : new Error("Aborted"));
+        };
+        signal?.addEventListener("abort", onAbort, { once: true });
+        settle = () => {
+            clearTimeout(timer);
+            signal?.removeEventListener("abort", onAbort);
+            pending.delete(predictionId);
+        };
+        pending.set(predictionId, {
+            resolve: (result) => {
+                settle();
+                resolve(result);
+            },
+            reject: (error) => {
+                settle();
+                reject(error);
+            }
+        });
+        log.debug("Registered AtlasCloud webhook wait", {
+            predictionId,
+            timeoutMs
+        });
+    });
+}
+/**
+ * Called by the ingress route for a success callback. Returns false when this
+ * process has no waiter, which is the normal case for a duplicate delivery or
+ * for a replica that did not submit the prediction.
+ */
+export function resolveAtlasWebhook(predictionId, result) {
+    const entry = pending.get(predictionId);
+    if (!entry) {
+        log.debug("No pending AtlasCloud prediction for callback", {
+            predictionId
+        });
+        return false;
+    }
+    log.info("AtlasCloud webhook callback received", { predictionId });
+    entry.resolve(result);
+    return true;
+}
+/** Called by the ingress route for a failure callback. */
+export function rejectAtlasWebhook(predictionId, reason) {
+    const entry = pending.get(predictionId);
+    if (!entry)
+        return false;
+    log.info("AtlasCloud webhook failure callback received", {
+        predictionId,
+        reason
+    });
+    entry.reject(new Error(`AtlasCloud job failed (webhook): ${reason} (predictionId: ${predictionId})`));
+    return true;
+}
+export function hasPendingAtlasWebhook(predictionId) {
+    return pending.has(predictionId);
+}
+/** Visible for testing. */
+export function atlasWebhookPendingCount() {
+    return pending.size;
+}
+//# sourceMappingURL=atlascloud-webhook-registry.js.map

package/dist/providers/atlascloud-webhook-registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"atlascloud-webhook-registry.js","sourceRoot":"","sources":["../../src/providers/atlascloud-webhook-registry.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAGnD,MAAM,GAAG,GAAG,YAAY,CAAC,+CAA+C,CAAC,CAAC;AAE1E,uDAAuD;AACvD,MAAM,OAAO,uBAAwB,SAAQ,KAAK;IAChD,YAAY,YAAoB,EAAE,SAAiB;QACjD,KAAK,CACH,yCAAyC;YACvC,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,oBAAoB,YAAY,GAAG,CACrE,CAAC;QACF,IAAI,CAAC,IAAI,GAAG,yBAAyB,CAAC;IACxC,CAAC;CACF;AAOD,MAAM,OAAO,GAAG,IAAI,GAAG,EAA6B,CAAC;AAErD;;;GAGG;AACH,MAAM,UAAU,wBAAwB,CACtC,YAAoB,EACpB,SAAiB,EACjB,MAAoB;IAEpB,OAAO,IAAI,OAAO,CAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACtD,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,MAAM,CACJ,MAAM,CAAC,MAAM,YAAY,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CACtE,CAAC;YACF,OAAO;QACT,CAAC;QAED,0EAA0E;QAC1E,0EAA0E;QAC1E,IAAI,MAAM,GAAG,GAAS,EAAE,CAAC,SAAS,CAAC;QAEnC,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC5B,MAAM,EAAE,CAAC;YACT,MAAM,CAAC,IAAI,uBAAuB,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC,CAAC;QAC/D,CAAC,EAAE,SAAS,CAAC,CAAC;QAEd,MAAM,OAAO,GAAG,GAAS,EAAE;YACzB,MAAM,EAAE,CAAC;YACT,MAAM,CACJ,MAAM,EAAE,MAAM,YAAY,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CACvE,CAAC;QACJ,CAAC,CAAC;QACF,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAE3D,MAAM,GAAG,GAAS,EAAE;YAClB,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC9C,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAC/B,CAAC,CAAC;QAEF,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE;YACxB,OAAO,EAAE,CAAC,MAAM,EAAE,EAAE;gBAClB,MAAM,EAAE,CAAC;gBACT,OAAO,CAAC,MAAM,CAAC,CAAC;YAClB,CAAC;YACD,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;gBAChB,MAAM,EAAE,CAAC;gBACT,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC;SACF,CAAC,CAAC;QAEH,GAAG,CAAC,KAAK,CAAC,oCAAoC,EAAE;YAC9C,YAAY;YACZ,SAAS;SACV,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,mBAAmB,CACjC,YAAoB,EACpB,MAAuB;IAEvB,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IACxC,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,GAAG,CAAC,KAAK,CAAC,+CAA+C,EAAE;YACzD,YAAY;SACb,CAAC,CAAC;QACH,OAAO,KAAK,CAAC;IACf,CAAC;IACD,GAAG,CAAC,IAAI,CAAC,sCAAsC,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;IACnE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACtB,OAAO,IAAI,CAAC;AACd,CAAC;AAED,0DAA0D;AAC1D,MAAM,UAAU,kBAAkB,CAChC,YAAoB,EACpB,MAAc;IAEd,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IACxC,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAC;IACzB,GAAG,CAAC,IAAI,CAAC,8CAA8C,EAAE;QACvD,YAAY;QACZ,MAAM;KACP,CAAC,CAAC;IACH,KAAK,CAAC,MAAM,CACV,IAAI,KAAK,CACP,oCAAoC,MAAM,mBAAmB,YAAY,GAAG,CAC7E,CACF,CAAC;IACF,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,YAAoB;IACzD,OAAO,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;AACnC,CAAC;AAED,2BAA2B;AAC3B,MAAM,UAAU,wBAAwB;IACtC,OAAO,OAAO,CAAC,IAAI,CAAC;AACtB,CAAC"}

package/dist/providers/audio-mime.d.ts

@@ -0,0 +1,11 @@
+/**
+ * Detect an audio container from its leading magic bytes.
+ *
+ * Used by providers that return an audio file (mp3/wav/…) without a reliable
+ * Content-Type — e.g. FAL/KIE result URLs and ElevenLabs responses.
+ */
+/** Identify an audio container; `null` when nothing matches. */
+export declare function sniffAudioMimeOrNull(bytes: Uint8Array): string | null;
+/** As {@link sniffAudioMimeOrNull}, defaulting to the common TTS encoding. */
+export declare function sniffAudioMime(bytes: Uint8Array): string;
+//# sourceMappingURL=audio-mime.d.ts.map

package/dist/providers/audio-mime.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"audio-mime.d.ts","sourceRoot":"","sources":["../../src/providers/audio-mime.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,gEAAgE;AAChE,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,UAAU,GAAG,MAAM,GAAG,IAAI,CAyCrE;AAED,8EAA8E;AAC9E,wBAAgB,cAAc,CAAC,KAAK,EAAE,UAAU,GAAG,MAAM,CAExD"}

package/dist/providers/audio-mime.js

@@ -0,0 +1,46 @@
+/**
+ * Detect an audio container from its leading magic bytes.
+ *
+ * Used by providers that return an audio file (mp3/wav/…) without a reliable
+ * Content-Type — e.g. FAL/KIE result URLs and ElevenLabs responses.
+ */
+/** Identify an audio container; `null` when nothing matches. */
+export function sniffAudioMimeOrNull(bytes) {
+    // RIFF is a generic container (WAVE, WEBP, AVI, …). Only "RIFF....WAVE" is
+    // audio — claiming every RIFF is a WAV mislabels a WebP image as audio/wav.
+    if (bytes[0] === 0x52 &&
+        bytes[1] === 0x49 &&
+        bytes[2] === 0x46 &&
+        bytes[3] === 0x46 &&
+        bytes[8] === 0x57 &&
+        bytes[9] === 0x41 &&
+        bytes[10] === 0x56 &&
+        bytes[11] === 0x45) {
+        return "audio/wav";
+    }
+    // "OggS"
+    if (bytes[0] === 0x4f &&
+        bytes[1] === 0x67 &&
+        bytes[2] === 0x67 &&
+        bytes[3] === 0x53) {
+        return "audio/ogg";
+    }
+    // "fLaC"
+    if (bytes[0] === 0x66 &&
+        bytes[1] === 0x4c &&
+        bytes[2] === 0x61 &&
+        bytes[3] === 0x43) {
+        return "audio/flac";
+    }
+    // ID3-tagged MP3 ("ID3") or a raw MPEG frame sync (0xFFEx).
+    if ((bytes[0] === 0x49 && bytes[1] === 0x44 && bytes[2] === 0x33) ||
+        (bytes[0] === 0xff && (bytes[1] & 0xe0) === 0xe0)) {
+        return "audio/mpeg";
+    }
+    return null;
+}
+/** As {@link sniffAudioMimeOrNull}, defaulting to the common TTS encoding. */
+export function sniffAudioMime(bytes) {
+    return sniffAudioMimeOrNull(bytes) ?? "audio/mpeg";
+}
+//# sourceMappingURL=audio-mime.js.map

package/dist/providers/audio-mime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"audio-mime.js","sourceRoot":"","sources":["../../src/providers/audio-mime.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,gEAAgE;AAChE,MAAM,UAAU,oBAAoB,CAAC,KAAiB;IACpD,2EAA2E;IAC3E,4EAA4E;IAC5E,IACE,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,EAAE,CAAC,KAAK,IAAI;QAClB,KAAK,CAAC,EAAE,CAAC,KAAK,IAAI,EAClB,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IACD,SAAS;IACT,IACE,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,EACjB,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IACD,SAAS;IACT,IACE,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,EACjB,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC;IACD,4DAA4D;IAC5D,IACE,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC;QAC7D,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,IAAI,CAAC,EACjD,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,8EAA8E;AAC9E,MAAM,UAAU,cAAc,CAAC,KAAiB;IAC9C,OAAO,oBAAoB,CAAC,KAAK,CAAC,IAAI,YAAY,CAAC;AACrD,CAAC"}