@just-every/ensemble 0.2.160 → 0.2.162

package/dist/cjs/model_providers/runway.cjs

@@ -0,0 +1,145 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runwayProvider = exports.RunwayProvider = void 0;
+const base_provider_js_1 = require("./base_provider.cjs");
+const cost_tracker_js_1 = require("../utils/cost_tracker.cjs");
+const llm_logger_js_1 = require("../utils/llm_logger.cjs");
+const fetch_with_timeout_js_1 = require("../utils/fetch_with_timeout.cjs");
+const RUNWAY_BASE = process.env.RUNWAY_API_BASE || 'https://api.dev.runwayml.com';
+const RUNWAY_VERSION = process.env.RUNWAY_API_VERSION || '2024-11-06';
+function mapRatio(size) {
+    const s = String(size || 'square');
+    if (s === 'landscape' || s === '1792x1024' || s === '1536x1024' || s === '1920x1080' || s === '1280x720')
+        return '1920:1080';
+    if (s === 'portrait' || s === '1024x1792' || s === '1080x1920' || s === '720x1280')
+        return '1080:1920';
+    return '1024:1024';
+}
+const shouldRetry = (error) => {
+    const message = String(error?.message || '').toLowerCase();
+    return error?.name === 'AbortError' || message.includes('timed out');
+};
+const withRetries = async (fn, attempts = 3) => {
+    let lastError;
+    for (let attempt = 1; attempt <= attempts; attempt++) {
+        try {
+            return await fn();
+        }
+        catch (error) {
+            lastError = error;
+            if (!shouldRetry(error) || attempt === attempts)
+                throw error;
+            await new Promise(r => setTimeout(r, attempt * 500));
+        }
+    }
+    throw lastError;
+};
+class RunwayProvider extends base_provider_js_1.BaseModelProvider {
+    constructor() {
+        super('runway');
+    }
+    async *createResponseStream() {
+        throw new Error('Runway provider does not support text streaming');
+    }
+    async createImage(prompt, model, agent, opts = {}) {
+        const apiKey = process.env.RUNWAY_API_KEY;
+        const requestId = (0, llm_logger_js_1.log_llm_request)(agent.agent_id || 'default', 'runway', model, { prompt, opts }, new Date());
+        let success = false;
+        try {
+            if (!apiKey)
+                throw new Error('RUNWAY_API_KEY is not set');
+            const body = {
+                promptText: prompt,
+                ratio: mapRatio(opts.size),
+                model: (model && model.toLowerCase().includes('turbo')) ? 'gen4_image_turbo' : 'gen4_image',
+            };
+            if (opts?.source_images) {
+                const srcs = Array.isArray(opts.source_images) ? opts.source_images : [opts.source_images];
+                body.referenceImages = srcs.slice(0, 3).map((v, i) => {
+                    const uri = typeof v === 'string' ? v : v?.data || v;
+                    return { uri, tag: `ref${i + 1}` };
+                });
+            }
+            const createRes = await withRetries(() => (0, fetch_with_timeout_js_1.fetchWithTimeout)(`${RUNWAY_BASE}/v1/text_to_image`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    Authorization: `Bearer ${apiKey}`,
+                    'X-Runway-Version': RUNWAY_VERSION,
+                },
+                body: JSON.stringify(body),
+            }, 20000), 3);
+            if (!createRes.ok)
+                throw new Error(`Runway create failed: ${createRes.status} ${await createRes.text()}`);
+            const created = await createRes.json();
+            const id = created?.id || created?.task?.id;
+            if (!id)
+                throw new Error('Runway: missing task id');
+            const pollUrl = `${RUNWAY_BASE}/v1/tasks/${encodeURIComponent(id)}`;
+            const started = Date.now();
+            const timeoutMs = 180000;
+            let urls = [];
+            while (true) {
+                const response = await withRetries(() => (0, fetch_with_timeout_js_1.fetchWithTimeout)(pollUrl, {
+                    headers: {
+                        Authorization: `Bearer ${apiKey}`,
+                        'X-Runway-Version': RUNWAY_VERSION,
+                    },
+                }, 15000), 2);
+                if (!response.ok)
+                    throw new Error(`Runway poll failed: ${response.status} ${await response.text()}`);
+                const info = await response.json();
+                const status = (info?.status || info?.task?.status || '').toLowerCase();
+                const extracted = [];
+                const tryPush = (value) => {
+                    if (!value)
+                        return;
+                    if (typeof value === 'string')
+                        extracted.push(value);
+                    else if (Array.isArray(value))
+                        value.forEach(tryPush);
+                    else if (typeof value === 'object') {
+                        if (typeof value.url === 'string')
+                            extracted.push(value.url);
+                        if (typeof value.uri === 'string')
+                            extracted.push(value.uri);
+                        if (Array.isArray(value.images))
+                            value.images.forEach(tryPush);
+                        if (Array.isArray(value.assets))
+                            value.assets.forEach(tryPush);
+                        if (value.image)
+                            tryPush(value.image);
+                    }
+                };
+                tryPush(info?.output);
+                tryPush(info?.assets);
+                tryPush(info?.task?.output);
+                if (status === 'succeeded' || status === 'completed' || (extracted.length > 0 && (status === 'success' || status === ''))) {
+                    urls = extracted;
+                    break;
+                }
+                if (status === 'failed' || status === 'canceled' || status === 'error') {
+                    throw new Error(`Runway task failed (status=${status || 'unknown'})`);
+                }
+                if (Date.now() - started > timeoutMs)
+                    throw new Error('Runway generation timed out');
+                await new Promise(r => setTimeout(r, 1200));
+            }
+            if (!urls.length)
+                throw new Error('Runway: no image url in response');
+            cost_tracker_js_1.costTracker.addUsage({ model, image_count: urls.length, request_id: opts?.request_id, metadata: { source: 'runway' } });
+            success = true;
+            return urls;
+        }
+        catch (err) {
+            (0, llm_logger_js_1.log_llm_error)(requestId, err);
+            throw err;
+        }
+        finally {
+            (0, llm_logger_js_1.log_llm_response)(requestId, { ok: success });
+        }
+    }
+}
+exports.RunwayProvider = RunwayProvider;
+exports.runwayProvider = new RunwayProvider();
+//# sourceMappingURL=runway.js.map

package/dist/cjs/model_providers/runway.d.ts

@@ -0,0 +1,9 @@
+import { BaseModelProvider } from "./base_provider.cjs";
+import type { AgentDefinition, ImageGenerationOpts, ProviderStreamEvent } from "../types/types.cjs";
+export declare class RunwayProvider extends BaseModelProvider {
+    constructor();
+    createResponseStream(): AsyncGenerator<ProviderStreamEvent>;
+    createImage(prompt: string, model: string, agent: AgentDefinition, opts?: ImageGenerationOpts): Promise<string[]>;
+}
+export declare const runwayProvider: RunwayProvider;
+//# sourceMappingURL=runway.d.ts.map

package/dist/cjs/model_providers/runway.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"runway.d.ts","sourceRoot":"","sources":["../../../model_providers/runway.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,eAAe,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAsCnG,qBAAa,cAAe,SAAQ,iBAAiB;;IAK1C,oBAAoB,IAAI,cAAc,CAAC,mBAAmB,CAAC;IAI5D,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,IAAI,GAAE,mBAAwB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;CA2F9H;AAED,eAAO,MAAM,cAAc,gBAAuB,CAAC"}

package/dist/cjs/model_providers/runway.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runway.js","sourceRoot":"","sources":["../../../model_providers/runway.ts"],"names":[],"mappings":";;;AAAA,yDAAuD;AAEvD,8DAAuD;AACvD,0DAA0F;AAC1F,0EAAkE;AAMlE,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,8BAA8B,CAAC;AAClF,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,YAAY,CAAC;AAEtE,SAAS,QAAQ,CAAC,IAAkC;IAChD,MAAM,CAAC,GAAG,MAAM,CAAC,IAAI,IAAI,QAAQ,CAAC,CAAC;IACnC,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,UAAU;QAAE,OAAO,WAAW,CAAC;IAC7H,IAAI,CAAC,KAAK,UAAU,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,UAAU;QAAE,OAAO,WAAW,CAAC;IACvG,OAAO,WAAW,CAAC;AACvB,CAAC;AAED,MAAM,WAAW,GAAG,CAAC,KAAU,EAAE,EAAE;IAC/B,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;IAC3D,OAAO,KAAK,EAAE,IAAI,KAAK,YAAY,IAAI,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;AACzE,CAAC,CAAC;AAEF,MAAM,WAAW,GAAG,KAAK,EAAK,EAAoB,EAAE,QAAQ,GAAG,CAAC,EAAc,EAAE;IAC5E,IAAI,SAAc,CAAC;IACnB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,QAAQ,EAAE,OAAO,EAAE,EAAE,CAAC;QACnD,IAAI,CAAC;YACD,OAAO,MAAM,EAAE,EAAE,CAAC;QACtB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,SAAS,GAAG,KAAK,CAAC;YAClB,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,OAAO,KAAK,QAAQ;gBAAE,MAAM,KAAK,CAAC;YAC7D,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,OAAO,GAAG,GAAG,CAAC,CAAC,CAAC;QACzD,CAAC;IACL,CAAC;IACD,MAAM,SAAS,CAAC;AACpB,CAAC,CAAC;AAEF,MAAa,cAAe,SAAQ,oCAAiB;IACjD;QACI,KAAK,CAAC,QAAe,CAAC,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,CAAC,oBAAoB;QACvB,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;IACvE,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,KAAa,EAAE,KAAsB,EAAE,OAA4B,EAAE;QACnG,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAC1C,MAAM,SAAS,GAAG,IAAA,+BAAe,EAAC,KAAK,CAAC,QAAQ,IAAI,SAAS,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;QAC9G,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC;YACD,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YAG1D,MAAM,IAAI,GAAQ;gBACd,UAAU,EAAE,MAAM;gBAClB,KAAK,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC1B,KAAK,EAAE,CAAC,KAAK,IAAI,KAAK,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,YAAY;aAC9F,CAAC;YACF,IAAI,IAAI,EAAE,aAAa,EAAE,CAAC;gBACtB,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBAC3F,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,CAAS,EAAE,EAAE;oBAC9D,MAAM,GAAG,GAAG,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,CAAC,CAAC;oBACrD,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACvC,CAAC,CAAC,CAAC;YACP,CAAC;YAED,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,GAAG,EAAE,CAAC,IAAA,wCAAgB,EAAC,GAAG,WAAW,mBAAmB,EAAE;gBAC1F,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACL,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,MAAM,EAAE;oBACjC,kBAAkB,EAAE,cAAc;iBACrC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC7B,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;YACd,IAAI,CAAC,SAAS,CAAC,EAAE;gBAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,SAAS,CAAC,MAAM,IAAI,MAAM,SAAS,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAC1G,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,CAAC;YACvC,MAAM,EAAE,GAAuB,OAAO,EAAE,EAAE,IAAI,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC;YAChE,IAAI,CAAC,EAAE;gBAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;YAGpD,MAAM,OAAO,GAAG,GAAG,WAAW,aAAa,kBAAkB,CAAC,EAAE,CAAC,EAAE,CAAC;YACpE,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC3B,MAAM,SAAS,GAAG,MAAM,CAAC;YACzB,IAAI,IAAI,GAAa,EAAE,CAAC;YACxB,OAAO,IAAI,EAAE,CAAC;gBACV,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,GAAG,EAAE,CAAC,IAAA,wCAAgB,EAAC,OAAO,EAAE;oBAC/D,OAAO,EAAE;wBACL,aAAa,EAAE,UAAU,MAAM,EAAE;wBACjC,kBAAkB,EAAE,cAAc;qBACrC;iBACJ,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;gBAEd,IAAI,CAAC,QAAQ,CAAC,EAAE;oBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,QAAQ,CAAC,MAAM,IAAI,MAAM,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACrG,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACnC,MAAM,MAAM,GAAG,CAAC,IAAI,EAAE,MAAM,IAAI,IAAI,EAAE,IAAI,EAAE,MAAM,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;gBAExE,MAAM,SAAS,GAAa,EAAE,CAAC;gBAC/B,MAAM,OAAO,GAAG,CAAC,KAAU,EAAE,EAAE;oBAC3B,IAAI,CAAC,KAAK;wBAAE,OAAO;oBACnB,IAAI,OAAO,KAAK,KAAK,QAAQ;wBAAE,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;yBAChD,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;wBAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;yBACjD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;wBACjC,IAAI,OAAO,KAAK,CAAC,GAAG,KAAK,QAAQ;4BAAE,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;wBAC7D,IAAI,OAAO,KAAK,CAAC,GAAG,KAAK,QAAQ;4BAAE,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;wBAC7D,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC;4BAAE,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;wBAC/D,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC;4BAAE,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;wBAC/D,IAAI,KAAK,CAAC,KAAK;4BAAE,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;oBAC1C,CAAC;gBACL,CAAC,CAAC;gBACF,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;gBACtB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;gBACtB,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;gBAE5B,IAAI,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,WAAW,IAAI,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,EAAE,CAAC,CAAC,EAAE,CAAC;oBACxH,IAAI,GAAG,SAAS,CAAC;oBACjB,MAAM;gBACV,CAAC;gBACD,IAAI,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,UAAU,IAAI,MAAM,KAAK,OAAO,EAAE,CAAC;oBACrE,MAAM,IAAI,KAAK,CAAC,8BAA8B,MAAM,IAAI,SAAS,GAAG,CAAC,CAAC;gBAC1E,CAAC;gBACD,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,GAAG,SAAS;oBAAE,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;gBACrF,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;YAChD,CAAC;YAED,IAAI,CAAC,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACtE,6BAAW,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,EAAE,CAAC,CAAC;YACxH,OAAO,GAAG,IAAI,CAAC;YACf,OAAO,IAAI,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,IAAA,6BAAa,EAAC,SAAS,EAAE,GAAG,CAAC,CAAC;YAC9B,MAAM,GAAG,CAAC;QACd,CAAC;gBAAS,CAAC;YACP,IAAA,gCAAgB,EAAC,SAAS,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QACjD,CAAC;IACL,CAAC;CACJ;AApGD,wCAoGC;AAEY,QAAA,cAAc,GAAG,IAAI,cAAc,EAAE,CAAC"}

package/dist/cjs/model_providers/stability.cjs

@@ -0,0 +1,130 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.stabilityProvider = exports.StabilityProvider = void 0;
+const base_provider_js_1 = require("./base_provider.cjs");
+const cost_tracker_js_1 = require("../utils/cost_tracker.cjs");
+const llm_logger_js_1 = require("../utils/llm_logger.cjs");
+const STABILITY_BASE = 'https://api.stability.ai/v2beta';
+function mapAspect(size) {
+    if (!size)
+        return undefined;
+    const s = String(size);
+    if (s === 'square' || s === '1024x1024' || s === '512x512' || s === '256x256')
+        return '1:1';
+    if (s === 'landscape' || s === '1792x1024' || s === '1536x1024')
+        return '16:9';
+    if (s === 'portrait' || s === '1024x1792' || s === '1024x1536')
+        return '9:16';
+    return undefined;
+}
+class StabilityProvider extends base_provider_js_1.BaseModelProvider {
+    constructor() {
+        super('stability');
+    }
+    async *createResponseStream() {
+        throw new Error('Stability provider does not support text streaming');
+    }
+    endpointFor(model) {
+        const m = model.toLowerCase();
+        if (m.includes('ultra'))
+            return `${STABILITY_BASE}/stable-image/generate/ultra`;
+        if (m.includes('core'))
+            return `${STABILITY_BASE}/stable-image/generate/core`;
+        return `${STABILITY_BASE}/stable-image/generate/sd3`;
+    }
+    async createImage(prompt, model, agent, opts = {}) {
+        const apiKey = process.env.STABILITY_API_KEY;
+        const requestId = (0, llm_logger_js_1.log_llm_request)(agent.agent_id || 'default', 'stability', model, { prompt, opts }, new Date());
+        try {
+            if (!apiKey)
+                throw new Error('STABILITY_API_KEY is not set');
+            const endpoint = this.endpointFor(model);
+            const form = new FormData();
+            form.set('prompt', prompt);
+            if (opts?.response_format === 'url')
+                form.set('output_format', 'png');
+            const ml = model.toLowerCase();
+            if (ml.includes('sd3.5-large-turbo'))
+                form.set('model', 'sd3.5-large-turbo');
+            else if (ml.includes('sd3.5-large'))
+                form.set('model', 'sd3.5-large');
+            else if (ml.includes('sd3.5-medium'))
+                form.set('model', 'sd3.5-medium');
+            else if (ml.includes('sd3.5-flash'))
+                form.set('model', 'sd3.5-flash');
+            let isI2I = false;
+            if (opts?.source_images) {
+                const arr = Array.isArray(opts.source_images) ? opts.source_images : [opts.source_images];
+                const first = arr[0];
+                if (typeof first === 'string' && first.startsWith('data:image/')) {
+                    const [, mime, b64] = first.match(/^data:(image\/[^;]+);base64,(.+)$/) || [];
+                    if (!b64)
+                        throw new Error('Invalid base64 source image');
+                    const bin = Buffer.from(b64, 'base64');
+                    form.set('image', new Blob([bin], { type: mime || 'image/png' }), 'image.png');
+                    form.set('mode', 'image-to-image');
+                    form.set('strength', '0.75');
+                    isI2I = true;
+                }
+                else if (typeof first === 'string') {
+                    const r = await fetch(first);
+                    const ct = r.headers.get('content-type') || 'image/png';
+                    const buf = new Uint8Array(await r.arrayBuffer());
+                    form.set('image', new Blob([buf], { type: ct }), 'image');
+                    form.set('mode', 'image-to-image');
+                    form.set('strength', '0.75');
+                    isI2I = true;
+                }
+            }
+            if (!isI2I) {
+                const aspect = mapAspect(opts.size);
+                if (aspect)
+                    form.set('aspect_ratio', aspect);
+            }
+            const res = await fetch(endpoint, {
+                method: 'POST',
+                headers: {
+                    Authorization: `Bearer ${apiKey}`,
+                    Accept: 'application/json',
+                },
+                body: form,
+            });
+            if (!res.ok)
+                throw new Error(`Stability create failed: ${res.status} ${await res.text()}`);
+            const contentType = res.headers.get('content-type') || '';
+            if (contentType.startsWith('image/')) {
+                const buf = new Uint8Array(await res.arrayBuffer());
+                const b64 = Buffer.from(buf).toString('base64');
+                const mime = contentType.split(';')[0];
+                const dataUrl = `data:${mime};base64,${b64}`;
+                cost_tracker_js_1.costTracker.addUsage({ model, image_count: 1, request_id: opts?.request_id, metadata: { source: 'stability' } });
+                return [dataUrl];
+            }
+            const json = await res.json();
+            const images = [];
+            if (Array.isArray(json?.artifacts)) {
+                for (const a of json.artifacts)
+                    if (a?.base64)
+                        images.push(`data:image/png;base64,${a.base64}`);
+            }
+            if (json?.image)
+                images.push(`data:image/png;base64,${json.image}`);
+            if (json?.images?.[0]?.base64)
+                images.push(`data:image/png;base64,${json.images[0].base64}`);
+            if (!images.length)
+                throw new Error('Stability: no image in response');
+            cost_tracker_js_1.costTracker.addUsage({ model, image_count: images.length, request_id: opts?.request_id, metadata: { source: 'stability' } });
+            return images;
+        }
+        catch (err) {
+            (0, llm_logger_js_1.log_llm_error)(requestId, err);
+            throw err;
+        }
+        finally {
+            (0, llm_logger_js_1.log_llm_response)(requestId, { ok: true });
+        }
+    }
+}
+exports.StabilityProvider = StabilityProvider;
+exports.stabilityProvider = new StabilityProvider();
+//# sourceMappingURL=stability.js.map

package/dist/cjs/model_providers/stability.d.ts

@@ -0,0 +1,10 @@
+import { BaseModelProvider } from "./base_provider.cjs";
+import type { AgentDefinition, ImageGenerationOpts, ProviderStreamEvent } from "../types/types.cjs";
+export declare class StabilityProvider extends BaseModelProvider {
+    constructor();
+    createResponseStream(): AsyncGenerator<ProviderStreamEvent>;
+    private endpointFor;
+    createImage(prompt: string, model: string, agent: AgentDefinition, opts?: ImageGenerationOpts): Promise<string[]>;
+}
+export declare const stabilityProvider: StabilityProvider;
+//# sourceMappingURL=stability.d.ts.map

package/dist/cjs/model_providers/stability.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stability.d.ts","sourceRoot":"","sources":["../../../model_providers/stability.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,eAAe,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAkBnG,qBAAa,iBAAkB,SAAQ,iBAAiB;;IAK7C,oBAAoB,IAAI,cAAc,CAAC,mBAAmB,CAAC;IAIlE,OAAO,CAAC,WAAW;IAQb,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,IAAI,GAAE,mBAAwB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;CAuF9H;AAED,eAAO,MAAM,iBAAiB,mBAA0B,CAAC"}

package/dist/cjs/model_providers/stability.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stability.js","sourceRoot":"","sources":["../../../model_providers/stability.ts"],"names":[],"mappings":";;;AAAA,yDAAuD;AAEvD,8DAAuD;AACvD,0DAA0F;AAK1F,MAAM,cAAc,GAAG,iCAAiC,CAAC;AAEzD,SAAS,SAAS,CAAC,IAAkC;IACjD,IAAI,CAAC,IAAI;QAAE,OAAO,SAAS,CAAC;IAC5B,MAAM,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,CAAC,KAAK,QAAQ,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,SAAS,IAAI,CAAC,KAAK,SAAS;QAAE,OAAO,KAAK,CAAC;IAC5F,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,WAAW;QAAE,OAAO,MAAM,CAAC;IAC/E,IAAI,CAAC,KAAK,UAAU,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,WAAW;QAAE,OAAO,MAAM,CAAC;IAC9E,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,MAAa,iBAAkB,SAAQ,oCAAiB;IACpD;QACI,KAAK,CAAC,WAAkB,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,CAAC,oBAAoB;QACvB,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;IAC1E,CAAC;IAEO,WAAW,CAAC,KAAa;QAC7B,MAAM,CAAC,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;QAC9B,IAAI,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,OAAO,GAAG,cAAc,8BAA8B,CAAC;QAChF,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,GAAG,cAAc,6BAA6B,CAAC;QAE9E,OAAO,GAAG,cAAc,4BAA4B,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,KAAa,EAAE,KAAsB,EAAE,OAA4B,EAAE;QACnG,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QAC7C,MAAM,SAAS,GAAG,IAAA,+BAAe,EAAC,KAAK,CAAC,QAAQ,IAAI,SAAS,EAAE,WAAW,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;QACjH,IAAI,CAAC;YACD,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;YAE7D,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACzC,MAAM,IAAI,GAAG,IAAI,QAAQ,EAAE,CAAC;YAC5B,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAC3B,IAAI,IAAI,EAAE,eAAe,KAAK,KAAK;gBAAE,IAAI,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;YAGtE,MAAM,EAAE,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;YAC/B,IAAI,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;gBAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,mBAAmB,CAAC,CAAC;iBACxE,IAAI,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC;gBAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;iBACjE,IAAI,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC;gBAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC;iBACnE,IAAI,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC;gBAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YAGtE,IAAI,KAAK,GAAG,KAAK,CAAC;YAClB,IAAI,IAAI,EAAE,aAAa,EAAE,CAAC;gBACtB,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;gBAC1F,MAAM,KAAK,GAAG,GAAG,CAAC,CAAC,CAAQ,CAAC;gBAC5B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;oBAC/D,MAAM,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,mCAAmC,CAAC,IAAI,EAAE,CAAC;oBAC7E,IAAI,CAAC,GAAG;wBAAE,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;oBACzD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;oBACvC,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,IAAI,WAAW,EAAE,CAAQ,EAAE,WAAW,CAAC,CAAC;oBACtF,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;oBACnC,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;oBAC7B,KAAK,GAAG,IAAI,CAAC;gBACjB,CAAC;qBAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;oBAEnC,MAAM,CAAC,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,CAAC;oBAC7B,MAAM,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,WAAW,CAAC;oBACxD,MAAM,GAAG,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;oBAClD,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAQ,EAAE,OAAO,CAAC,CAAC;oBACjE,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;oBACnC,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;oBAC7B,KAAK,GAAG,IAAI,CAAC;gBACjB,CAAC;YACL,CAAC;YAGD,IAAI,CAAC,KAAK,EAAE,CAAC;gBACT,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACpC,IAAI,MAAM;oBAAE,IAAI,CAAC,GAAG,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;YACjD,CAAC;YAED,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE;gBAC9B,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACL,aAAa,EAAE,UAAU,MAAM,EAAE;oBACjC,MAAM,EAAE,kBAAkB;iBAC7B;gBACD,IAAI,EAAE,IAAW;aACpB,CAAC,CAAC;YAEH,IAAI,CAAC,GAAG,CAAC,EAAE;gBAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,GAAG,CAAC,MAAM,IAAI,MAAM,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAE3F,MAAM,WAAW,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;YAC1D,IAAI,WAAW,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACnC,MAAM,GAAG,GAAG,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC;gBACpD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;gBAChD,MAAM,IAAI,GAAG,WAAW,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBACvC,MAAM,OAAO,GAAG,QAAQ,IAAI,WAAW,GAAG,EAAE,CAAC;gBAC7C,6BAAW,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC;gBACjH,OAAO,CAAC,OAAO,CAAC,CAAC;YACrB,CAAC;YACD,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YAE9B,MAAM,MAAM,GAAa,EAAE,CAAC;YAC5B,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,SAAS,CAAC,EAAE,CAAC;gBACjC,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,SAAS;oBAAE,IAAI,CAAC,EAAE,MAAM;wBAAE,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACpG,CAAC;YACD,IAAI,IAAI,EAAE,KAAK;gBAAE,MAAM,CAAC,IAAI,CAAC,yBAAyB,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;YACpE,IAAI,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,MAAM;gBAAE,MAAM,CAAC,IAAI,CAAC,yBAAyB,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YAC7F,IAAI,CAAC,MAAM,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;YACvE,6BAAW,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC;YAC7H,OAAO,MAAM,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,IAAA,6BAAa,EAAC,SAAS,EAAE,GAAG,CAAC,CAAC;YAC9B,MAAM,GAAG,CAAC;QACd,CAAC;gBAAS,CAAC;YACP,IAAA,gCAAgB,EAAC,SAAS,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9C,CAAC;IACL,CAAC;CACJ;AAxGD,8CAwGC;AAEY,QAAA,iBAAiB,GAAG,IAAI,iBAAiB,EAAE,CAAC"}