@nodetool-ai/runtime 0.7.0-rc.11

package/dist/providers/rodin-provider.js

@@ -0,0 +1,236 @@
+/**
+ * Rodin AI Provider — wraps the Rodin AI (Hyperhuman) 3D generation API to
+ * expose `textTo3D` and `imageTo3D` through the standard {@link BaseProvider}
+ * interface.
+ *
+ * Rodin's primary capability is image-to-3D, but the same `/v2/rodin`
+ * endpoint accepts a text prompt for text-to-3D generation.
+ *
+ * API docs: https://hyperhuman.deemos.com/api
+ *
+ * Ported from the historical Python implementation at
+ * `nodetool-core@25e60910:src/nodetool/providers/rodin_provider.py`. The
+ * Python `image_to_3d` accidentally submitted the task twice to fetch the
+ * subscription key — fixed here by reading both `uuids` and
+ * `subscription_key` from a single submit response.
+ */
+import { BaseProvider } from "./base-provider.js";
+import { createLogger } from "@nodetool-ai/config";
+const log = createLogger("nodetool.runtime.providers.rodin");
+const RODIN_API_BASE_URL = "https://hyperhuman.deemos.com/api";
+const DEFAULT_POLL_INTERVAL_MS = 5000;
+const DEFAULT_MAX_POLL_ATTEMPTS = 120;
+const DEFAULT_OUTPUT_FORMAT = "glb";
+export const RODIN_3D_MODELS = [
+    {
+        id: "rodin-gen-1",
+        name: "Rodin Gen-1 Image-to-3D",
+        provider: "rodin",
+        supportedTasks: ["image_to_3d"],
+        outputFormats: ["glb", "fbx", "obj", "usdz"]
+    },
+    {
+        id: "rodin-gen-1-turbo",
+        name: "Rodin Gen-1 Turbo Image-to-3D",
+        provider: "rodin",
+        supportedTasks: ["image_to_3d"],
+        outputFormats: ["glb", "fbx", "obj", "usdz"]
+    },
+    {
+        id: "rodin-sketch",
+        name: "Rodin Sketch Text-to-3D",
+        provider: "rodin",
+        supportedTasks: ["text_to_3d"],
+        outputFormats: ["glb", "fbx", "obj", "usdz"]
+    }
+];
+function detectImageType(image) {
+    if (image.length >= 8 &&
+        image[0] === 0x89 &&
+        image[1] === 0x50 &&
+        image[2] === 0x4e &&
+        image[3] === 0x47 &&
+        image[4] === 0x0d &&
+        image[5] === 0x0a &&
+        image[6] === 0x1a &&
+        image[7] === 0x0a) {
+        return "png";
+    }
+    if (image.length >= 3 &&
+        image[0] === 0xff &&
+        image[1] === 0xd8 &&
+        image[2] === 0xff) {
+        return "jpeg";
+    }
+    return "png"; // Match Python default
+}
+function encodeImageForRodin(image) {
+    return {
+        type: detectImageType(image),
+        data: Buffer.from(image).toString("base64")
+    };
+}
+export class RodinProvider extends BaseProvider {
+    apiKey;
+    pollIntervalMs;
+    maxPollAttempts;
+    constructor(secrets = {}, options = {}) {
+        super("rodin");
+        this.apiKey = secrets["RODIN_API_KEY"] ?? "";
+        this.pollIntervalMs = options.pollIntervalMs ?? DEFAULT_POLL_INTERVAL_MS;
+        this.maxPollAttempts = options.maxPollAttempts ?? DEFAULT_MAX_POLL_ATTEMPTS;
+        if (!this.apiKey) {
+            log.warn("Rodin API key not configured");
+        }
+    }
+    static requiredSecrets() {
+        return ["RODIN_API_KEY"];
+    }
+    async generateMessage(_args) {
+        throw new Error("rodin does not support chat generation");
+    }
+    async *generateMessages(_args) {
+        throw new Error("rodin does not support chat generation");
+    }
+    async getAvailable3DModels() {
+        if (!this.apiKey) {
+            log.debug("No Rodin API key configured, returning empty 3D model list");
+            return [];
+        }
+        return RODIN_3D_MODELS;
+    }
+    async textTo3D(params) {
+        if (!params.prompt)
+            throw new Error("prompt must not be empty");
+        if (!this.apiKey)
+            throw new Error("Rodin API key is not configured");
+        const maxAttempts = this.computeMaxAttempts(params.timeoutSeconds);
+        const format = (params.outputFormat ?? DEFAULT_OUTPUT_FORMAT).toUpperCase();
+        const payload = {
+            prompt: params.prompt,
+            geometry_file_format: format
+        };
+        if (params.seed != null && params.seed >= 0)
+            payload.seed = params.seed;
+        const submit = await this.submitTask(payload);
+        log.debug(`Rodin text-to-3D task submitted: ${submit.uuid}`);
+        await this.pollTaskStatus(submit.subscriptionKey, maxAttempts);
+        return this.downloadResult(submit.uuid);
+    }
+    async imageTo3D(image, params) {
+        if (!image || image.length === 0)
+            throw new Error("image must not be empty");
+        if (!this.apiKey)
+            throw new Error("Rodin API key is not configured");
+        const maxAttempts = this.computeMaxAttempts(params.timeoutSeconds);
+        const format = (params.outputFormat ?? DEFAULT_OUTPUT_FORMAT).toUpperCase();
+        const encoded = encodeImageForRodin(image);
+        const payload = {
+            images: [encoded],
+            geometry_file_format: format
+        };
+        if (params.prompt)
+            payload.prompt = params.prompt;
+        if (params.seed != null && params.seed >= 0)
+            payload.seed = params.seed;
+        const submit = await this.submitTask(payload);
+        log.debug(`Rodin image-to-3D task submitted: ${submit.uuid}`);
+        await this.pollTaskStatus(submit.subscriptionKey, maxAttempts);
+        return this.downloadResult(submit.uuid);
+    }
+    // --- internals ---
+    headers() {
+        return {
+            Authorization: `Bearer ${this.apiKey}`,
+            "Content-Type": "application/json"
+        };
+    }
+    computeMaxAttempts(timeoutSeconds) {
+        if (!timeoutSeconds || timeoutSeconds <= 0) {
+            return this.maxPollAttempts;
+        }
+        return Math.max(1, Math.floor((timeoutSeconds * 1000) / this.pollIntervalMs));
+    }
+    async submitTask(payload) {
+        const url = `${RODIN_API_BASE_URL}/v2/rodin`;
+        const res = await fetch(url, {
+            method: "POST",
+            headers: this.headers(),
+            body: JSON.stringify(payload)
+        });
+        if (res.status !== 200 && res.status !== 202) {
+            const errText = await res.text();
+            throw new Error(`Rodin API error (${res.status}): ${errText}`);
+        }
+        const data = (await res.json());
+        const uuids = data.uuids;
+        const subscriptionKey = data.subscription_key;
+        const uuid = uuids?.[0];
+        if (!uuid) {
+            throw new Error(`Rodin submit returned no task uuid: ${JSON.stringify(data)}`);
+        }
+        if (!subscriptionKey) {
+            throw new Error(`Rodin submit returned no subscription_key: ${JSON.stringify(data)}`);
+        }
+        return { uuid, subscriptionKey };
+    }
+    async pollTaskStatus(subscriptionKey, maxAttempts) {
+        const url = `${RODIN_API_BASE_URL}/v2/status`;
+        for (let attempt = 0; attempt < maxAttempts; attempt++) {
+            const res = await fetch(url, {
+                method: "POST",
+                headers: this.headers(),
+                body: JSON.stringify({ subscription_key: subscriptionKey })
+            });
+            if (res.status !== 200) {
+                const errText = await res.text();
+                throw new Error(`Rodin API poll error (${res.status}): ${errText}`);
+            }
+            const data = (await res.json());
+            const jobs = data.jobs ?? [];
+            const job = jobs[0];
+            if (!job) {
+                log.debug(`Rodin task: no jobs yet (attempt ${attempt + 1}/${maxAttempts})`);
+                await new Promise((r) => setTimeout(r, this.pollIntervalMs));
+                continue;
+            }
+            const status = String(job.status ?? "").toUpperCase();
+            if (status === "DONE")
+                return job;
+            if (status === "FAILED" ||
+                status === "ERROR" ||
+                status === "CANCELLED") {
+                const errMsg = job.error ?? "Unknown error";
+                throw new Error(`Rodin task failed: ${errMsg}`);
+            }
+            log.debug(`Rodin task status: ${status} (attempt ${attempt + 1}/${maxAttempts})`);
+            await new Promise((r) => setTimeout(r, this.pollIntervalMs));
+        }
+        throw new Error(`Rodin task timed out after ${(maxAttempts * this.pollIntervalMs) / 1000}s`);
+    }
+    async downloadResult(taskUuid) {
+        const downloadInfoRes = await fetch(`${RODIN_API_BASE_URL}/v2/download`, {
+            method: "POST",
+            headers: this.headers(),
+            body: JSON.stringify({ task_uuid: taskUuid })
+        });
+        if (downloadInfoRes.status !== 200) {
+            const errText = await downloadInfoRes.text();
+            throw new Error(`Failed to get Rodin download URL (${downloadInfoRes.status}): ${errText}`);
+        }
+        const info = (await downloadInfoRes.json());
+        const downloadUrl = info.model_url ?? info.url;
+        if (!downloadUrl) {
+            throw new Error(`No download URL in Rodin response: ${JSON.stringify(info)}`);
+        }
+        const dlRes = await fetch(downloadUrl);
+        if (!dlRes.ok) {
+            const errText = await dlRes.text().catch(() => "");
+            throw new Error(`Failed to download Rodin result (${dlRes.status}): ${errText}`);
+        }
+        const bytes = new Uint8Array(await dlRes.arrayBuffer());
+        log.debug(`Downloaded ${bytes.length} bytes of 3D model data from Rodin`);
+        return bytes;
+    }
+}
+//# sourceMappingURL=rodin-provider.js.map

package/dist/providers/rodin-provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rodin-provider.js","sourceRoot":"","sources":["../../src/providers/rodin-provider.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AASnD,MAAM,GAAG,GAAG,YAAY,CAAC,kCAAkC,CAAC,CAAC;AAE7D,MAAM,kBAAkB,GAAG,mCAAmC,CAAC;AAE/D,MAAM,wBAAwB,GAAG,IAAI,CAAC;AACtC,MAAM,yBAAyB,GAAG,GAAG,CAAC;AAEtC,MAAM,qBAAqB,GAAG,KAAK,CAAC;AAEpC,MAAM,CAAC,MAAM,eAAe,GAAc;IACxC;QACE,EAAE,EAAE,aAAa;QACjB,IAAI,EAAE,yBAAyB;QAC/B,QAAQ,EAAE,OAAO;QACjB,cAAc,EAAE,CAAC,aAAa,CAAC;QAC/B,aAAa,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC;KAC7C;IACD;QACE,EAAE,EAAE,mBAAmB;QACvB,IAAI,EAAE,+BAA+B;QACrC,QAAQ,EAAE,OAAO;QACjB,cAAc,EAAE,CAAC,aAAa,CAAC;QAC/B,aAAa,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC;KAC7C;IACD;QACE,EAAE,EAAE,cAAc;QAClB,IAAI,EAAE,yBAAyB;QAC/B,QAAQ,EAAE,OAAO;QACjB,cAAc,EAAE,CAAC,YAAY,CAAC;QAC9B,aAAa,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC;KAC7C;CACF,CAAC;AAiBF,SAAS,eAAe,CAAC,KAAiB;IACxC,IACE,KAAK,CAAC,MAAM,IAAI,CAAC;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,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI;QACjB,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,EACjB,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IACE,KAAK,CAAC,MAAM,IAAI,CAAC;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,MAAM,CAAC;IAChB,CAAC;IACD,OAAO,KAAK,CAAC,CAAC,uBAAuB;AACvC,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAiB;IAC5C,OAAO;QACL,IAAI,EAAE,eAAe,CAAC,KAAK,CAAC;QAC5B,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC;KAC5C,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,aAAc,SAAQ,YAAY;IAC5B,MAAM,CAAS;IACf,cAAc,CAAS;IACvB,eAAe,CAAS;IAEzC,YACE,UAAmC,EAAE,EACrC,UAAgC,EAAE;QAElC,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,MAAM,GAAI,OAAO,CAAC,eAAe,CAAY,IAAI,EAAE,CAAC;QACzD,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,wBAAwB,CAAC;QACzE,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,IAAI,yBAAyB,CAAC;QAC5E,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,GAAG,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,MAAM,CAAU,eAAe;QAC7B,OAAO,CAAC,eAAe,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,KAAqD;QAErD,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,CAAC,gBAAgB,CACrB,KAAsD;QAEtD,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC5D,CAAC;IAEQ,KAAK,CAAC,oBAAoB;QACjC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,GAAG,CAAC,KAAK,CAAC,4DAA4D,CAAC,CAAC;YACxE,OAAO,EAAE,CAAC;QACZ,CAAC;QACD,OAAO,eAAe,CAAC;IACzB,CAAC;IAEQ,KAAK,CAAC,QAAQ,CAAC,MAAsB;QAC5C,IAAI,CAAC,MAAM,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAChE,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QAErE,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACnE,MAAM,MAAM,GAAG,CAAC,MAAM,CAAC,YAAY,IAAI,qBAAqB,CAAC,CAAC,WAAW,EAAE,CAAC;QAE5E,MAAM,OAAO,GAA4B;YACvC,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,oBAAoB,EAAE,MAAM;SAC7B,CAAC;QACF,IAAI,MAAM,CAAC,IAAI,IAAI,IAAI,IAAI,MAAM,CAAC,IAAI,IAAI,CAAC;YAAE,OAAO,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QAExE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC9C,GAAG,CAAC,KAAK,CAAC,oCAAoC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;QAE7D,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC;QAC/D,OAAO,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC1C,CAAC;IAEQ,KAAK,CAAC,SAAS,CACtB,KAAiB,EACjB,MAAuB;QAEvB,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAC9B,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QAC7C,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QAErE,MAAM,WAAW,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACnE,MAAM,MAAM,GAAG,CAAC,MAAM,CAAC,YAAY,IAAI,qBAAqB,CAAC,CAAC,WAAW,EAAE,CAAC;QAC5E,MAAM,OAAO,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,OAAO,GAA4B;YACvC,MAAM,EAAE,CAAC,OAAO,CAAC;YACjB,oBAAoB,EAAE,MAAM;SAC7B,CAAC;QACF,IAAI,MAAM,CAAC,MAAM;YAAE,OAAO,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAClD,IAAI,MAAM,CAAC,IAAI,IAAI,IAAI,IAAI,MAAM,CAAC,IAAI,IAAI,CAAC;YAAE,OAAO,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QAExE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC9C,GAAG,CAAC,KAAK,CAAC,qCAAqC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;QAE9D,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,eAAe,EAAE,WAAW,CAAC,CAAC;QAC/D,OAAO,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC1C,CAAC;IAED,oBAAoB;IAEZ,OAAO;QACb,OAAO;YACL,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;YACtC,cAAc,EAAE,kBAAkB;SACnC,CAAC;IACJ,CAAC;IAEO,kBAAkB,CAAC,cAAyC;QAClE,IAAI,CAAC,cAAc,IAAI,cAAc,IAAI,CAAC,EAAE,CAAC;YAC3C,OAAO,IAAI,CAAC,eAAe,CAAC;QAC9B,CAAC;QACD,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;IAChF,CAAC;IAEO,KAAK,CAAC,UAAU,CACtB,OAAgC;QAEhC,MAAM,GAAG,GAAG,GAAG,kBAAkB,WAAW,CAAC;QAC7C,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;YAC3B,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;YACvB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;SAC9B,CAAC,CAAC;QACH,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YAC7C,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YACjC,MAAM,IAAI,KAAK,CAAC,oBAAoB,GAAG,CAAC,MAAM,MAAM,OAAO,EAAE,CAAC,CAAC;QACjE,CAAC;QACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA4B,CAAC;QAC3D,MAAM,KAAK,GAAG,IAAI,CAAC,KAA6B,CAAC;QACjD,MAAM,eAAe,GAAG,IAAI,CAAC,gBAAsC,CAAC;QACpE,MAAM,IAAI,GAAG,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACxB,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CACb,uCAAuC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAC9D,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CACb,8CAA8C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CACrE,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;IACnC,CAAC;IAEO,KAAK,CAAC,cAAc,CAC1B,eAAuB,EACvB,WAAmB;QAEnB,MAAM,GAAG,GAAG,GAAG,kBAAkB,YAAY,CAAC;QAC9C,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;YACvD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAC3B,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;gBACvB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,gBAAgB,EAAE,eAAe,EAAE,CAAC;aAC5D,CAAC,CAAC;YACH,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBACvB,MAAM,OAAO,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;gBACjC,MAAM,IAAI,KAAK,CAAC,yBAAyB,GAAG,CAAC,MAAM,MAAM,OAAO,EAAE,CAAC,CAAC;YACtE,CAAC;YACD,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA4B,CAAC;YAC3D,MAAM,IAAI,GAAI,IAAI,CAAC,IAA8C,IAAI,EAAE,CAAC;YACxE,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YACpB,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,GAAG,CAAC,KAAK,CACP,oCAAoC,OAAO,GAAG,CAAC,IAAI,WAAW,GAAG,CAClE,CAAC;gBACF,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;gBAC7D,SAAS;YACX,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;YACtD,IAAI,MAAM,KAAK,MAAM;gBAAE,OAAO,GAAG,CAAC;YAClC,IACE,MAAM,KAAK,QAAQ;gBACnB,MAAM,KAAK,OAAO;gBAClB,MAAM,KAAK,WAAW,EACtB,CAAC;gBACD,MAAM,MAAM,GAAI,GAAG,CAAC,KAA4B,IAAI,eAAe,CAAC;gBACpE,MAAM,IAAI,KAAK,CAAC,sBAAsB,MAAM,EAAE,CAAC,CAAC;YAClD,CAAC;YACD,GAAG,CAAC,KAAK,CACP,sBAAsB,MAAM,aAAa,OAAO,GAAG,CAAC,IAAI,WAAW,GAAG,CACvE,CAAC;YACF,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC;QAC/D,CAAC;QACD,MAAM,IAAI,KAAK,CACb,8BAA8B,CAAC,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,IAAI,GAAG,CAC5E,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,QAAgB;QAC3C,MAAM,eAAe,GAAG,MAAM,KAAK,CAAC,GAAG,kBAAkB,cAAc,EAAE;YACvE,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE;YACvB,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC;SAC9C,CAAC,CAAC;QACH,IAAI,eAAe,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACnC,MAAM,OAAO,GAAG,MAAM,eAAe,CAAC,IAAI,EAAE,CAAC;YAC7C,MAAM,IAAI,KAAK,CACb,qCAAqC,eAAe,CAAC,MAAM,MAAM,OAAO,EAAE,CAC3E,CAAC;QACJ,CAAC;QACD,MAAM,IAAI,GAAG,CAAC,MAAM,eAAe,CAAC,IAAI,EAAE,CAA4B,CAAC;QACvE,MAAM,WAAW,GACd,IAAI,CAAC,SAAgC,IAAK,IAAI,CAAC,GAA0B,CAAC;QAC7E,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CACb,sCAAsC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAC7D,CAAC;QACJ,CAAC;QACD,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC;YACd,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YACnD,MAAM,IAAI,KAAK,CACb,oCAAoC,KAAK,CAAC,MAAM,MAAM,OAAO,EAAE,CAChE,CAAC;QACJ,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;QACxD,GAAG,CAAC,KAAK,CAAC,cAAc,KAAK,CAAC,MAAM,oCAAoC,CAAC,CAAC;QAC1E,OAAO,KAAK,CAAC;IACf,CAAC;CACF"}

package/dist/providers/scripted-provider.d.ts

@@ -0,0 +1,141 @@
+/**
+ * ScriptedProvider -- a deterministic fake provider for agent E2E tests.
+ *
+ * Instead of a single fixed response, each generateMessages call dispatches
+ * to the next script function in the sequence. Scripts can inspect the full
+ * messages and tools list to decide what to yield, making it possible to
+ * simulate a real LLM that responds to context.
+ *
+ * Usage:
+ *   const provider = new ScriptedProvider([
+ *     planScript({ title: "My Plan", steps: [...] }),   // call 0: planning
+ *     stepScript({ answer: "42" }),                     // call 1: step 1
+ *     stepScript({ summary: "done" }),                  // call 2: step 2
+ *   ]);
+ */
+import { BaseProvider } from "./base-provider.js";
+import type { Message, ProviderStreamItem, ProviderTool } from "./types.js";
+export type ScriptItem = {
+    type: "chunk";
+    content: string;
+    done?: boolean;
+} | {
+    type: "tool_call";
+    name: string;
+    args: Record<string, unknown>;
+    id?: string;
+};
+/**
+ * A script function is called with the current messages and tools for a
+ * generateMessages invocation, and returns an array of items to yield.
+ */
+export type ScriptFn = (messages: Message[], tools: ProviderTool[]) => ScriptItem[];
+export declare class ScriptedProvider extends BaseProvider {
+    private readonly scripts;
+    private callIdx;
+    /** Full log of all generateMessages calls (messages + tools per call). */
+    readonly callLog: Array<{
+        messages: Message[];
+        tools: ProviderTool[];
+    }>;
+    constructor(scripts: ScriptFn[]);
+    /** Reset call index and log (useful between test cases). */
+    reset(): void;
+    generateMessage(args: {
+        messages: Message[];
+        model: string;
+        tools?: ProviderTool[];
+        maxTokens?: number;
+    }): Promise<Message>;
+    generateMessages(args: {
+        messages: Message[];
+        model: string;
+        tools?: ProviderTool[];
+        maxTokens?: number;
+    }): AsyncGenerator<ProviderStreamItem>;
+}
+export interface TaskPlanSpec {
+    title: string;
+    steps: Array<{
+        id: string;
+        instructions: string;
+        depends_on?: string[];
+        output_schema?: string;
+        tools?: string[];
+    }>;
+}
+/**
+ * Spec for a multi-task plan where each task runs as an independent sub-agent.
+ */
+export interface MultiTaskPlanSpec {
+    title: string;
+    tasks: Array<{
+        id: string;
+        title: string;
+        depends_on?: string[];
+        steps: Array<{
+            id: string;
+            instructions: string;
+            depends_on?: string[];
+            output_schema?: string;
+            tools?: string[];
+        }>;
+    }>;
+}
+/**
+ * Script that calls the `create_task` tool with the provided plan.
+ * Use this for the planning phase of agent execution (single-task mode).
+ */
+export declare function planScript(taskSpec: TaskPlanSpec): ScriptFn;
+/**
+ * Script that calls the `create_plan` tool with a multi-task plan.
+ * Use this for the planning phase of agent execution (multi-task mode).
+ */
+export declare function multiTaskPlanScript(planSpec: MultiTaskPlanSpec): ScriptFn;
+/**
+ * Script that calls `finish_step` with the provided result.
+ * Use this for step execution phases.
+ */
+export declare function stepScript(result: unknown): ScriptFn;
+/**
+ * Script that emits a plain text chunk (no tool call).
+ * Useful for unstructured steps or testing fallback behavior.
+ */
+export declare function textScript(content: string): ScriptFn;
+/**
+ * Script that calls an arbitrary tool by name.
+ */
+export declare function toolCallScript(name: string, args: Record<string, unknown>): ScriptFn;
+/**
+ * Auto-detecting script: inspects the `tools` list for each call and decides:
+ * - If `create_plan` is available → call it with `opts.multiTaskPlan` or auto-wrap `opts.plan`
+ * - If `create_task` is available → call it with `opts.plan`
+ * - If `finish_step` is available → call it with `opts.result`
+ * - Otherwise → emit `opts.text` as a chunk
+ *
+ * This mimics a "smart" LLM that knows what tools are available.
+ */
+export declare function autoScript(opts: {
+    plan?: TaskPlanSpec;
+    multiTaskPlan?: MultiTaskPlanSpec;
+    result?: unknown;
+    text?: string;
+}): ScriptFn;
+/**
+ * Sequence of tool calls followed by a finish_step.
+ * Models a step where the LLM first uses tools, then completes.
+ *
+ * Example:
+ *   toolThenFinishScript(
+ *     [{ name: "calculator", args: { expression: "2+2" } }],
+ *     { answer: 4 }
+ *   )
+ *
+ * The provider will yield the tool call on the first iteration, and on the
+ * next iteration (after tool results are fed back) it will yield finish_step.
+ */
+export declare function toolThenFinishScript(toolCalls: Array<{
+    name: string;
+    args: Record<string, unknown>;
+}>, finalResult: unknown): ScriptFn[];
+//# sourceMappingURL=scripted-provider.d.ts.map

package/dist/providers/scripted-provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"scripted-provider.d.ts","sourceRoot":"","sources":["../../src/providers/scripted-provider.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,KAAK,EAAE,OAAO,EAAE,kBAAkB,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAM5E,MAAM,MAAM,UAAU,GAClB;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,OAAO,CAAA;CAAE,GAClD;IACE,IAAI,EAAE,WAAW,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC9B,EAAE,CAAC,EAAE,MAAM,CAAC;CACb,CAAC;AAEN;;;GAGG;AACH,MAAM,MAAM,QAAQ,GAAG,CACrB,QAAQ,EAAE,OAAO,EAAE,EACnB,KAAK,EAAE,YAAY,EAAE,KAClB,UAAU,EAAE,CAAC;AAMlB,qBAAa,gBAAiB,SAAQ,YAAY;IAChD,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAa;IACrC,OAAO,CAAC,OAAO,CAAK;IAEpB,0EAA0E;IAC1E,QAAQ,CAAC,OAAO,EAAE,KAAK,CAAC;QAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;QAAC,KAAK,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC,CAAM;gBAEjE,OAAO,EAAE,QAAQ,EAAE;IAO/B,4DAA4D;IAC5D,KAAK,IAAI,IAAI;IAKP,eAAe,CAAC,IAAI,EAAE;QAC1B,QAAQ,EAAE,OAAO,EAAE,CAAC;QACpB,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE,YAAY,EAAE,CAAC;QACvB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,GAAG,OAAO,CAAC,OAAO,CAAC;IAUb,gBAAgB,CAAC,IAAI,EAAE;QAC5B,QAAQ,EAAE,OAAO,EAAE,CAAC;QACpB,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,CAAC,EAAE,YAAY,EAAE,CAAC;QACvB,SAAS,CAAC,EAAE,MAAM,CAAC;KACpB,GAAG,cAAc,CAAC,kBAAkB,CAAC;CA4BvC;AAMD,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,KAAK,CAAC;QACX,EAAE,EAAE,MAAM,CAAC;QACX,YAAY,EAAE,MAAM,CAAC;QACrB,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;QACtB,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;KAClB,CAAC,CAAC;CACJ;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,KAAK,CAAC;QACX,EAAE,EAAE,MAAM,CAAC;QACX,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;QACtB,KAAK,EAAE,KAAK,CAAC;YACX,EAAE,EAAE,MAAM,CAAC;YACX,YAAY,EAAE,MAAM,CAAC;YACrB,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;YACtB,aAAa,CAAC,EAAE,MAAM,CAAC;YACvB,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;SAClB,CAAC,CAAC;KACJ,CAAC,CAAC;CACJ;AAED;;;GAGG;AACH,wBAAgB,UAAU,CAAC,QAAQ,EAAE,YAAY,GAAG,QAAQ,CAQ3D;AAED;;;GAGG;AACH,wBAAgB,mBAAmB,CAAC,QAAQ,EAAE,iBAAiB,GAAG,QAAQ,CAQzE;AAED;;;GAGG;AACH,wBAAgB,UAAU,CAAC,MAAM,EAAE,OAAO,GAAG,QAAQ,CAIpD;AAED;;;GAGG;AACH,wBAAgB,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,QAAQ,CAEpD;AAED;;GAEG;AACH,wBAAgB,cAAc,CAC5B,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,QAAQ,CAEV;AAED;;;;;;;;GAQG;AACH,wBAAgB,UAAU,CAAC,IAAI,EAAE;IAC/B,IAAI,CAAC,EAAE,YAAY,CAAC;IACpB,aAAa,CAAC,EAAE,iBAAiB,CAAC;IAClC,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;CACf,GAAG,QAAQ,CA4FX;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,oBAAoB,CAClC,SAAS,EAAE,KAAK,CAAC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,CAAC,EACjE,WAAW,EAAE,OAAO,GACnB,QAAQ,EAAE,CAaZ"}

package/dist/providers/scripted-provider.js

@@ -0,0 +1,246 @@
+/**
+ * ScriptedProvider -- a deterministic fake provider for agent E2E tests.
+ *
+ * Instead of a single fixed response, each generateMessages call dispatches
+ * to the next script function in the sequence. Scripts can inspect the full
+ * messages and tools list to decide what to yield, making it possible to
+ * simulate a real LLM that responds to context.
+ *
+ * Usage:
+ *   const provider = new ScriptedProvider([
+ *     planScript({ title: "My Plan", steps: [...] }),   // call 0: planning
+ *     stepScript({ answer: "42" }),                     // call 1: step 1
+ *     stepScript({ summary: "done" }),                  // call 2: step 2
+ *   ]);
+ */
+import { randomUUID } from "node:crypto";
+import { BaseProvider } from "./base-provider.js";
+// ---------------------------------------------------------------------------
+// ScriptedProvider
+// ---------------------------------------------------------------------------
+export class ScriptedProvider extends BaseProvider {
+    scripts;
+    callIdx = 0;
+    /** Full log of all generateMessages calls (messages + tools per call). */
+    callLog = [];
+    constructor(scripts) {
+        super("fake");
+        if (scripts.length === 0)
+            throw new Error("ScriptedProvider requires at least one script");
+        this.scripts = scripts;
+    }
+    /** Reset call index and log (useful between test cases). */
+    reset() {
+        this.callIdx = 0;
+        this.callLog.length = 0;
+    }
+    async generateMessage(args) {
+        let content = "";
+        for await (const item of this.generateMessages(args)) {
+            if ("type" in item && item.type === "chunk") {
+                content += item.content ?? "";
+            }
+        }
+        return { role: "assistant", content: content || "Done." };
+    }
+    async *generateMessages(args) {
+        const tools = args.tools ?? [];
+        this.callLog.push({ messages: args.messages, tools });
+        // Use the next script; if exhausted, repeat the last one
+        const idx = Math.min(this.callIdx, this.scripts.length - 1);
+        this.callIdx++;
+        const script = this.scripts[idx];
+        const items = script(args.messages, tools);
+        for (const item of items) {
+            if (item.type === "chunk") {
+                yield {
+                    type: "chunk",
+                    content: item.content,
+                    done: item.done ?? true,
+                    content_type: "text"
+                };
+            }
+            else {
+                // tool_call
+                yield {
+                    id: item.id ?? randomUUID(),
+                    name: item.name,
+                    args: item.args
+                };
+            }
+        }
+    }
+}
+/**
+ * Script that calls the `create_task` tool with the provided plan.
+ * Use this for the planning phase of agent execution (single-task mode).
+ */
+export function planScript(taskSpec) {
+    return (_messages, _tools) => [
+        {
+            type: "tool_call",
+            name: "create_task",
+            args: taskSpec
+        }
+    ];
+}
+/**
+ * Script that calls the `create_plan` tool with a multi-task plan.
+ * Use this for the planning phase of agent execution (multi-task mode).
+ */
+export function multiTaskPlanScript(planSpec) {
+    return (_messages, _tools) => [
+        {
+            type: "tool_call",
+            name: "create_plan",
+            args: planSpec
+        }
+    ];
+}
+/**
+ * Script that calls `finish_step` with the provided result.
+ * Use this for step execution phases.
+ */
+export function stepScript(result) {
+    return (_messages, _tools) => [
+        { type: "tool_call", name: "finish_step", args: { result } }
+    ];
+}
+/**
+ * Script that emits a plain text chunk (no tool call).
+ * Useful for unstructured steps or testing fallback behavior.
+ */
+export function textScript(content) {
+    return (_messages, _tools) => [{ type: "chunk", content, done: true }];
+}
+/**
+ * Script that calls an arbitrary tool by name.
+ */
+export function toolCallScript(name, args) {
+    return (_messages, _tools) => [{ type: "tool_call", name, args }];
+}
+/**
+ * Auto-detecting script: inspects the `tools` list for each call and decides:
+ * - If `create_plan` is available → call it with `opts.multiTaskPlan` or auto-wrap `opts.plan`
+ * - If `create_task` is available → call it with `opts.plan`
+ * - If `finish_step` is available → call it with `opts.result`
+ * - Otherwise → emit `opts.text` as a chunk
+ *
+ * This mimics a "smart" LLM that knows what tools are available.
+ */
+export function autoScript(opts) {
+    return (messages, tools) => {
+        const toolNames = new Set(tools.map((t) => t.name));
+        // Incremental planner: add_task + finish_plan
+        if (toolNames.has("add_task") &&
+            toolNames.has("finish_plan") &&
+            (opts.multiTaskPlan || opts.plan)) {
+            const planSpec = opts.multiTaskPlan ?? {
+                title: opts.plan.title,
+                tasks: [
+                    {
+                        id: "task_1",
+                        title: opts.plan.title,
+                        depends_on: [],
+                        steps: opts.plan.steps
+                    }
+                ]
+            };
+            // Count add_task tool results already in history to decide next step.
+            let addedCount = 0;
+            for (const m of messages) {
+                if (m.role !== "tool")
+                    continue;
+                const content = typeof m.content === "string" ? m.content : "";
+                if (content.includes('"status":"task_added"'))
+                    addedCount++;
+            }
+            if (addedCount < planSpec.tasks.length) {
+                const task = planSpec.tasks[addedCount];
+                return [
+                    {
+                        type: "tool_call",
+                        name: "add_task",
+                        args: task
+                    }
+                ];
+            }
+            return [
+                {
+                    type: "tool_call",
+                    name: "finish_plan",
+                    args: { title: planSpec.title }
+                }
+            ];
+        }
+        if (toolNames.has("create_plan") && (opts.multiTaskPlan || opts.plan)) {
+            // Legacy one-shot multi-task plan tool.
+            const planArgs = opts.multiTaskPlan ?? {
+                title: opts.plan.title,
+                tasks: [
+                    {
+                        id: "task_1",
+                        title: opts.plan.title,
+                        depends_on: [],
+                        steps: opts.plan.steps
+                    }
+                ]
+            };
+            return [
+                {
+                    type: "tool_call",
+                    name: "create_plan",
+                    args: planArgs
+                }
+            ];
+        }
+        if (toolNames.has("create_task") && opts.plan) {
+            return [
+                {
+                    type: "tool_call",
+                    name: "create_task",
+                    args: opts.plan
+                }
+            ];
+        }
+        if (toolNames.has("finish_step")) {
+            return [
+                {
+                    type: "tool_call",
+                    name: "finish_step",
+                    args: { result: opts.result ?? {} }
+                }
+            ];
+        }
+        return [
+            { type: "chunk", content: opts.text ?? "Task completed.", done: true }
+        ];
+    };
+}
+/**
+ * Sequence of tool calls followed by a finish_step.
+ * Models a step where the LLM first uses tools, then completes.
+ *
+ * Example:
+ *   toolThenFinishScript(
+ *     [{ name: "calculator", args: { expression: "2+2" } }],
+ *     { answer: 4 }
+ *   )
+ *
+ * The provider will yield the tool call on the first iteration, and on the
+ * next iteration (after tool results are fed back) it will yield finish_step.
+ */
+export function toolThenFinishScript(toolCalls, finalResult) {
+    const scripts = [
+        // First call: yield the tool calls
+        (_messages, _tools) => toolCalls.map((tc) => ({
+            type: "tool_call",
+            name: tc.name,
+            args: tc.args
+        })),
+        // Second call (after tool results): yield finish_step
+        stepScript(finalResult)
+    ];
+    return scripts;
+}
+//# sourceMappingURL=scripted-provider.js.map

package/dist/providers/scripted-provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"scripted-provider.js","sourceRoot":"","sources":["../../src/providers/scripted-provider.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAyBlD,8EAA8E;AAC9E,mBAAmB;AACnB,8EAA8E;AAE9E,MAAM,OAAO,gBAAiB,SAAQ,YAAY;IAC/B,OAAO,CAAa;IAC7B,OAAO,GAAG,CAAC,CAAC;IAEpB,0EAA0E;IACjE,OAAO,GAA0D,EAAE,CAAC;IAE7E,YAAY,OAAmB;QAC7B,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;QACnE,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,4DAA4D;IAC5D,KAAK;QACH,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC;QACjB,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,IAKrB;QACC,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;YACrD,IAAI,MAAM,IAAI,IAAI,IAAK,IAAyB,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBAClE,OAAO,IAAK,IAA6B,CAAC,OAAO,IAAI,EAAE,CAAC;YAC1D,CAAC;QACH,CAAC;QACD,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,IAAI,OAAO,EAAE,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,CAAC,gBAAgB,CAAC,IAKvB;QACC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;QAC/B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEtD,yDAAyD;QACzD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC5D,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACjC,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAE3C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBAC1B,MAAM;oBACJ,IAAI,EAAE,OAAO;oBACb,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,IAAI,EAAE,IAAI,CAAC,IAAI,IAAI,IAAI;oBACvB,YAAY,EAAE,MAAM;iBACZ,CAAC;YACb,CAAC;iBAAM,CAAC;gBACN,YAAY;gBACZ,MAAM;oBACJ,EAAE,EAAE,IAAI,CAAC,EAAE,IAAI,UAAU,EAAE;oBAC3B,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,IAAI,EAAE,IAAI,CAAC,IAAI;iBAChB,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAoCD;;;GAGG;AACH,MAAM,UAAU,UAAU,CAAC,QAAsB;IAC/C,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC;QAC5B;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE,QAA8C;SACrD;KACF,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,mBAAmB,CAAC,QAA2B;IAC7D,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC;QAC5B;YACE,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE,QAA8C;SACrD;KACF,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,UAAU,CAAC,MAAe;IACxC,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC;QAC5B,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,MAAM,EAAE,EAAE;KAC7D,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,UAAU,CAAC,OAAe;IACxC,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;AACzE,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAC5B,IAAY,EACZ,IAA6B;IAE7B,OAAO,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;AACpE,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,UAAU,CAAC,IAK1B;IACC,OAAO,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE;QACzB,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAEpD,8CAA8C;QAC9C,IACE,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC;YACzB,SAAS,CAAC,GAAG,CAAC,aAAa,CAAC;YAC5B,CAAC,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,IAAI,CAAC,EACjC,CAAC;YACD,MAAM,QAAQ,GAAsB,IAAI,CAAC,aAAa,IAAI;gBACxD,KAAK,EAAE,IAAI,CAAC,IAAK,CAAC,KAAK;gBACvB,KAAK,EAAE;oBACL;wBACE,EAAE,EAAE,QAAQ;wBACZ,KAAK,EAAE,IAAI,CAAC,IAAK,CAAC,KAAK;wBACvB,UAAU,EAAE,EAAE;wBACd,KAAK,EAAE,IAAI,CAAC,IAAK,CAAC,KAAK;qBACxB;iBACF;aACF,CAAC;YAEF,sEAAsE;YACtE,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;gBACzB,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM;oBAAE,SAAS;gBAChC,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC/D,IAAI,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAAC;oBAAE,UAAU,EAAE,CAAC;YAC9D,CAAC;YAED,IAAI,UAAU,GAAG,QAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;gBACvC,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBACxC,OAAO;oBACL;wBACE,IAAI,EAAE,WAAW;wBACjB,IAAI,EAAE,UAAU;wBAChB,IAAI,EAAE,IAA0C;qBACjD;iBACF,CAAC;YACJ,CAAC;YACD,OAAO;gBACL;oBACE,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,aAAa;oBACnB,IAAI,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE;iBAChC;aACF,CAAC;QACJ,CAAC;QAED,IAAI,SAAS,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACtE,wCAAwC;YACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,IAAI;gBACrC,KAAK,EAAE,IAAI,CAAC,IAAK,CAAC,KAAK;gBACvB,KAAK,EAAE;oBACL;wBACE,EAAE,EAAE,QAAQ;wBACZ,KAAK,EAAE,IAAI,CAAC,IAAK,CAAC,KAAK;wBACvB,UAAU,EAAE,EAAE;wBACd,KAAK,EAAE,IAAI,CAAC,IAAK,CAAC,KAAK;qBACxB;iBACF;aACF,CAAC;YACF,OAAO;gBACL;oBACE,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,aAAa;oBACnB,IAAI,EAAE,QAA8C;iBACrD;aACF,CAAC;QACJ,CAAC;QACD,IAAI,SAAS,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC9C,OAAO;gBACL;oBACE,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,aAAa;oBACnB,IAAI,EAAE,IAAI,CAAC,IAA0C;iBACtD;aACF,CAAC;QACJ,CAAC;QACD,IAAI,SAAS,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;YACjC,OAAO;gBACL;oBACE,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,aAAa;oBACnB,IAAI,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,IAAI,EAAE,EAAE;iBACpC;aACF,CAAC;QACJ,CAAC;QACD,OAAO;YACL,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,CAAC,IAAI,IAAI,iBAAiB,EAAE,IAAI,EAAE,IAAI,EAAE;SACvE,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,oBAAoB,CAClC,SAAiE,EACjE,WAAoB;IAEpB,MAAM,OAAO,GAAe;QAC1B,mCAAmC;QACnC,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,CACpB,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;YACrB,IAAI,EAAE,WAAoB;YAC1B,IAAI,EAAE,EAAE,CAAC,IAAI;YACb,IAAI,EAAE,EAAE,CAAC,IAAI;SACd,CAAC,CAAC;QACL,sDAAsD;QACtD,UAAU,CAAC,WAAW,CAAC;KACxB,CAAC;IACF,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/providers/together-provider.d.ts

@@ -0,0 +1,20 @@
+import OpenAI from "openai";
+import { OpenAIProvider } from "./openai-provider.js";
+import type { LanguageModel } from "./types.js";
+interface TogetherProviderOptions {
+    client?: OpenAI;
+    clientFactory?: (apiKey: string) => OpenAI;
+    fetchFn?: typeof fetch;
+}
+export declare class TogetherProvider extends OpenAIProvider {
+    static requiredSecrets(): string[];
+    private _togetherFetch;
+    constructor(secrets: {
+        TOGETHER_API_KEY?: string;
+    }, options?: TogetherProviderOptions);
+    getContainerEnv(): Record<string, string>;
+    hasToolSupport(_model: string): Promise<boolean>;
+    getAvailableLanguageModels(): Promise<LanguageModel[]>;
+}
+export {};
+//# sourceMappingURL=together-provider.d.ts.map