@just-every/ensemble 0.2.159 → 0.2.161

package/dist/cjs/model_providers/bytedance.cjs

@@ -0,0 +1,202 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.bytedanceProvider = exports.ByteDanceProvider = void 0;
+const base_provider_js_1 = require("./base_provider.cjs");
+const fetch_with_timeout_js_1 = require("../utils/fetch_with_timeout.cjs");
+const cost_tracker_js_1 = require("../utils/cost_tracker.cjs");
+const llm_logger_js_1 = require("../utils/llm_logger.cjs");
+const ARK_BASE = 'https://ark.ap-southeast.bytepluses.com/api/v3';
+function mapSize(size) {
+    if (!size)
+        return undefined;
+    const s = String(size);
+    if (s === 'square')
+        return '1024x1024';
+    if (s === 'landscape')
+        return '1536x1024';
+    if (s === 'portrait')
+        return '1024x1536';
+    if (s.toUpperCase() === '2K' || s.toUpperCase() === '4K' || s.toUpperCase() === '720P' || s.toUpperCase() === '1080P')
+        return s;
+    if (/^\d+x\d+$/i.test(s))
+        return s;
+    return undefined;
+}
+function normalizeModelId(model) {
+    if (model === 'seedream-4' || model === 'seedream-4.0')
+        return 'seedream-4-0-250828';
+    return model.replace(/^bytedance[\/:-]/, '');
+}
+class ByteDanceProvider extends base_provider_js_1.BaseModelProvider {
+    constructor() {
+        super('bytedance');
+    }
+    async *createResponseStream() {
+        throw new Error('Bytedance provider does not support text streaming');
+    }
+    async createImage(prompt, model, agent, opts = {}) {
+        const apiKey = process.env.ARK_API_KEY ||
+            process.env.BYTEPLUS_API_KEY ||
+            process.env.BYTEDANCE_API_KEY;
+        if (!apiKey)
+            throw new Error('Bytedance provider: set ARK_API_KEY (or BYTEPLUS_API_KEY/BYTEDANCE_API_KEY)');
+        const requestId = (0, llm_logger_js_1.log_llm_request)(agent.agent_id || 'default', 'bytedance', model, { prompt, opts }, new Date());
+        try {
+            const n = Math.max(1, Math.min(10, opts.n || 1));
+            const size = mapSize(opts.size) || '1024x1024';
+            const response_format = opts.response_format === 'b64_json' ? 'b64_json' : 'url';
+            const body = {
+                model: normalizeModelId(model),
+                prompt,
+                n,
+                size,
+                response_format,
+                sequential_image_generation: opts.sequential_image_generation || 'disabled',
+                sequential_image_generation_options: opts.sequential_image_generation_options,
+                stream: Boolean(opts.stream),
+                watermark: opts.watermark !== undefined ? !!opts.watermark : false,
+                seed: typeof opts.seed === 'number' ? opts.seed : undefined,
+                guidance_scale: typeof opts.guidance_scale === 'number' ? opts.guidance_scale : undefined,
+            };
+            if (opts?.source_images) {
+                const arr = Array.isArray(opts.source_images) ? opts.source_images : [opts.source_images];
+                const images = [];
+                for (const s of arr.slice(0, 10)) {
+                    const v = typeof s === 'string' ? s : s?.data;
+                    if (typeof v === 'string' && v)
+                        images.push(v);
+                }
+                if (images.length === 1) {
+                    body.image = images[0];
+                    body.reference_image = images[0];
+                }
+                else if (images.length > 1) {
+                    body.image = images;
+                }
+            }
+            const res = await (0, fetch_with_timeout_js_1.fetchWithTimeout)(`${ARK_BASE}/images/generations`, {
+                method: 'POST',
+                headers: {
+                    Authorization: `Bearer ${apiKey}`,
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify(body),
+            }, 60000);
+            if (!res.ok) {
+                const text = await res.text();
+                throw new Error(`Bytedance images.generation failed: ${res.status} ${text}`);
+            }
+            const contentType = res.headers.get('content-type') || '';
+            const out = [];
+            let observedUsage = null;
+            if (contentType.includes('text/event-stream')) {
+                if (!res.body)
+                    throw new Error('Bytedance: missing response body for stream');
+                const reader = res.body.getReader();
+                const decoder = new TextDecoder();
+                let buffer = '';
+                let dataBlock = '';
+                const flush = () => {
+                    const trimmed = dataBlock.trim();
+                    dataBlock = '';
+                    if (!trimmed || trimmed === '[DONE]')
+                        return;
+                    try {
+                        const payload = JSON.parse(trimmed);
+                        const collect = (value) => {
+                            if (!value)
+                                return;
+                            if (typeof value === 'string') {
+                                if (/^https?:\/\//i.test(value) || value.startsWith('data:'))
+                                    out.push(value);
+                            }
+                            else if (value.url) {
+                                out.push(String(value.url));
+                            }
+                            else if (value.b64_json) {
+                                out.push(`data:image/png;base64,${value.b64_json}`);
+                            }
+                        };
+                        if (Array.isArray(payload?.data)) {
+                            for (const item of payload.data)
+                                collect(item);
+                        }
+                        if (payload?.url)
+                            collect(payload.url);
+                        if (Array.isArray(payload?.images)) {
+                            for (const item of payload.images)
+                                collect(item);
+                        }
+                        if (payload?.image_base64)
+                            collect(`data:image/png;base64,${payload.image_base64}`);
+                        if (payload?.result?.images) {
+                            for (const item of payload.result.images)
+                                collect(item);
+                        }
+                        if (payload?.usage)
+                            observedUsage = payload.usage;
+                    }
+                    catch (err) {
+                        console.warn('[bytedance] Failed to parse SSE payload', err);
+                    }
+                };
+                while (true) {
+                    const { value, done } = await reader.read();
+                    if (done)
+                        break;
+                    buffer += decoder.decode(value, { stream: true });
+                    let idx;
+                    while ((idx = buffer.indexOf('\n')) >= 0) {
+                        const line = buffer.slice(0, idx);
+                        buffer = buffer.slice(idx + 1);
+                        const trimmed = line.trim();
+                        if (!trimmed) {
+                            flush();
+                            continue;
+                        }
+                        if (trimmed.startsWith('data:')) {
+                            const payloadPart = trimmed.slice(trimmed.indexOf(':') + 1).trim();
+                            dataBlock += payloadPart;
+                            dataBlock += '\n';
+                        }
+                    }
+                }
+                if (dataBlock.trim())
+                    flush();
+            }
+            else {
+                const json = await res.json();
+                if (Array.isArray(json?.data)) {
+                    for (const d of json.data) {
+                        if (d?.url)
+                            out.push(String(d.url));
+                        else if (d?.b64_json)
+                            out.push(`data:image/png;base64,${d.b64_json}`);
+                    }
+                }
+                if (!out.length && json?.result?.images) {
+                    for (const im of json.result.images)
+                        if (im?.url)
+                            out.push(String(im.url));
+                }
+                if (!out.length && typeof json?.image_base64 === 'string') {
+                    out.push(`data:image/png;base64,${json.image_base64}`);
+                }
+                if (json?.usage)
+                    observedUsage = json.usage;
+            }
+            if (!out.length)
+                throw new Error('Bytedance: no image result in response');
+            cost_tracker_js_1.costTracker.addUsage({ model, image_count: out.length, request_id: opts?.request_id, metadata: { source: 'bytedance', usage: observedUsage } });
+            (0, llm_logger_js_1.log_llm_response)(requestId, { ok: true, image_count: out.length });
+            return out;
+        }
+        catch (err) {
+            (0, llm_logger_js_1.log_llm_error)(requestId, err);
+            throw err;
+        }
+    }
+}
+exports.ByteDanceProvider = ByteDanceProvider;
+exports.bytedanceProvider = new ByteDanceProvider();
+//# sourceMappingURL=bytedance.js.map

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

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"bytedance.d.ts","sourceRoot":"","sources":["../../../model_providers/bytedance.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,eAAe,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAiCnG,qBAAa,iBAAkB,SAAQ,iBAAiB;;IAM7C,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;CAyJ9H;AAED,eAAO,MAAM,iBAAiB,mBAA0B,CAAC"}

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

@@ -0,0 +1 @@
+{"version":3,"file":"bytedance.js","sourceRoot":"","sources":["../../../model_providers/bytedance.ts"],"names":[],"mappings":";;;AAAA,yDAAuD;AAEvD,0EAAkE;AAClE,8DAAuD;AACvD,0DAA0F;AAQ1F,MAAM,QAAQ,GAAG,gDAAgD,CAAC;AAElE,SAAS,OAAO,CAAC,IAAkC;IAC/C,IAAI,CAAC,IAAI;QAAE,OAAO,SAAS,CAAC;IAC5B,MAAM,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,CAAC,KAAK,QAAQ;QAAE,OAAO,WAAW,CAAC;IACvC,IAAI,CAAC,KAAK,WAAW;QAAE,OAAO,WAAW,CAAC;IAC1C,IAAI,CAAC,KAAK,UAAU;QAAE,OAAO,WAAW,CAAC;IAEzC,IAAI,CAAC,CAAC,WAAW,EAAE,KAAK,IAAI,IAAI,CAAC,CAAC,WAAW,EAAE,KAAK,IAAI,IAAI,CAAC,CAAC,WAAW,EAAE,KAAK,MAAM,IAAI,CAAC,CAAC,WAAW,EAAE,KAAK,OAAO;QAAE,OAAO,CAAC,CAAC;IAEhI,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC;QAAE,OAAO,CAAC,CAAC;IACnC,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAa;IAEnC,IAAI,KAAK,KAAK,YAAY,IAAI,KAAK,KAAK,cAAc;QAAE,OAAO,qBAAqB,CAAC;IAErF,OAAO,KAAK,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;AACjD,CAAC;AAED,MAAa,iBAAkB,SAAQ,oCAAiB;IACpD;QACI,KAAK,CAAC,WAAkB,CAAC,CAAC;IAC9B,CAAC;IAGD,KAAK,CAAC,CAAC,oBAAoB;QACvB,MAAM,IAAI,KAAK,CAAC,oDAAoD,CAAC,CAAC;IAC1E,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,KAAa,EAAE,KAAsB,EAAE,OAA4B,EAAE;QACnG,MAAM,MAAM,GACR,OAAO,CAAC,GAAG,CAAC,WAAW;YACvB,OAAO,CAAC,GAAG,CAAC,gBAAgB;YAC5B,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;QAClC,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,6EAA6E,CAAC,CAAC;QAE5G,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,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACjD,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,WAAW,CAAC;YAC/C,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,KAAK,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC;YAEjF,MAAM,IAAI,GAAQ;gBACd,KAAK,EAAE,gBAAgB,CAAC,KAAK,CAAC;gBAC9B,MAAM;gBACN,CAAC;gBACD,IAAI;gBACJ,eAAe;gBAEf,2BAA2B,EAAE,IAAI,CAAC,2BAA2B,IAAI,UAAU;gBAC3E,mCAAmC,EAAE,IAAI,CAAC,mCAAmC;gBAC7E,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC;gBAE5B,SAAS,EAAE,IAAI,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK;gBAClE,IAAI,EAAE,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;gBAC3D,cAAc,EAAE,OAAO,IAAI,CAAC,cAAc,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS;aAC5F,CAAC;YAGF,IAAI,IAAI,EAAE,aAAa,EAAE,CAAC;gBACtB,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAE,IAAI,CAAC,aAAuB,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,aAAoB,CAAC,CAAC;gBAC5G,MAAM,MAAM,GAAa,EAAE,CAAC;gBAC5B,KAAK,MAAM,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;oBAC/B,MAAM,CAAC,GAAG,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;oBAC9C,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC;wBAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBACnD,CAAC;gBACD,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACrB,IAAY,CAAC,KAAK,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;oBAC/B,IAAY,CAAC,eAAe,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;gBAC9C,CAAC;qBAAM,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC1B,IAAY,CAAC,KAAK,GAAG,MAAM,CAAC;gBACjC,CAAC;YACL,CAAC;YAED,MAAM,GAAG,GAAG,MAAM,IAAA,wCAAgB,EAAC,GAAG,QAAQ,qBAAqB,EAAE;gBACjE,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACL,aAAa,EAAE,UAAU,MAAM,EAAE;oBACjC,cAAc,EAAE,kBAAkB;iBACrC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aAC7B,EAAE,KAAK,CAAC,CAAC;YAEV,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACV,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;gBAC9B,MAAM,IAAI,KAAK,CAAC,uCAAuC,GAAG,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC,CAAC;YACjF,CAAC;YAED,MAAM,WAAW,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;YAC1D,MAAM,GAAG,GAAa,EAAE,CAAC;YACzB,IAAI,aAAa,GAAQ,IAAI,CAAC;YAE9B,IAAI,WAAW,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;gBAC5C,IAAI,CAAC,GAAG,CAAC,IAAI;oBAAE,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;gBAC9E,MAAM,MAAM,GAAG,GAAG,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACpC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;gBAClC,IAAI,MAAM,GAAG,EAAE,CAAC;gBAChB,IAAI,SAAS,GAAG,EAAE,CAAC;gBAEnB,MAAM,KAAK,GAAG,GAAG,EAAE;oBACf,MAAM,OAAO,GAAG,SAAS,CAAC,IAAI,EAAE,CAAC;oBACjC,SAAS,GAAG,EAAE,CAAC;oBACf,IAAI,CAAC,OAAO,IAAI,OAAO,KAAK,QAAQ;wBAAE,OAAO;oBAC7C,IAAI,CAAC;wBACD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;wBACpC,MAAM,OAAO,GAAG,CAAC,KAAU,EAAE,EAAE;4BAC3B,IAAI,CAAC,KAAK;gCAAE,OAAO;4BACnB,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gCAC5B,IAAI,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC;oCAAE,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;4BAClF,CAAC;iCAAM,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;gCACnB,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;4BAChC,CAAC;iCAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;gCACxB,GAAG,CAAC,IAAI,CAAC,yBAAyB,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;4BACxD,CAAC;wBACL,CAAC,CAAC;wBAEF,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,EAAE,CAAC;4BAC/B,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,IAAI;gCAAE,OAAO,CAAC,IAAI,CAAC,CAAC;wBACnD,CAAC;wBACD,IAAI,OAAO,EAAE,GAAG;4BAAE,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;wBACvC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,MAAM,CAAC,EAAE,CAAC;4BACjC,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,MAAM;gCAAE,OAAO,CAAC,IAAI,CAAC,CAAC;wBACrD,CAAC;wBACD,IAAI,OAAO,EAAE,YAAY;4BAAE,OAAO,CAAC,yBAAyB,OAAO,CAAC,YAAY,EAAE,CAAC,CAAC;wBACpF,IAAI,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;4BAC1B,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,MAAM,CAAC,MAAM;gCAAE,OAAO,CAAC,IAAI,CAAC,CAAC;wBAC5D,CAAC;wBACD,IAAI,OAAO,EAAE,KAAK;4BAAE,aAAa,GAAG,OAAO,CAAC,KAAK,CAAC;oBACtD,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACX,OAAO,CAAC,IAAI,CAAC,yCAAyC,EAAE,GAAG,CAAC,CAAC;oBACjE,CAAC;gBACL,CAAC,CAAC;gBAEF,OAAO,IAAI,EAAE,CAAC;oBACV,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;oBAC5C,IAAI,IAAI;wBAAE,MAAM;oBAChB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;oBAClD,IAAI,GAAG,CAAC;oBACR,OAAO,CAAC,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;wBACvC,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;wBAClC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC;wBAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;wBAC5B,IAAI,CAAC,OAAO,EAAE,CAAC;4BACX,KAAK,EAAE,CAAC;4BACR,SAAS;wBACb,CAAC;wBACD,IAAI,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;4BAC9B,MAAM,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;4BACnE,SAAS,IAAI,WAAW,CAAC;4BACzB,SAAS,IAAI,IAAI,CAAC;wBACtB,CAAC;oBACL,CAAC;gBACL,CAAC;gBACD,IAAI,SAAS,CAAC,IAAI,EAAE;oBAAE,KAAK,EAAE,CAAC;YAClC,CAAC;iBAAM,CAAC;gBACJ,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;gBAE9B,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC;oBAC5B,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;wBACxB,IAAI,CAAC,EAAE,GAAG;4BAAE,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;6BAC/B,IAAI,CAAC,EAAE,QAAQ;4BAAE,GAAG,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;oBAC1E,CAAC;gBACL,CAAC;gBACD,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;oBACtC,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM;wBAAE,IAAI,EAAE,EAAE,GAAG;4BAAE,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;gBAC/E,CAAC;gBACD,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,OAAO,IAAI,EAAE,YAAY,KAAK,QAAQ,EAAE,CAAC;oBACxD,GAAG,CAAC,IAAI,CAAC,yBAAyB,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;gBAC3D,CAAC;gBACD,IAAI,IAAI,EAAE,KAAK;oBAAE,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC;YAChD,CAAC;YAED,IAAI,CAAC,GAAG,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;YAE3E,6BAAW,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,CAAC,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,CAAC,CAAC;YAChJ,IAAA,gCAAgB,EAAC,SAAS,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;YACnE,OAAO,GAAG,CAAC;QACf,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,IAAA,6BAAa,EAAC,SAAS,EAAE,GAAG,CAAC,CAAC;YAC9B,MAAM,GAAG,CAAC;QACd,CAAC;IACL,CAAC;CACJ;AAnKD,8CAmKC;AAEY,QAAA,iBAAiB,GAAG,IAAI,iBAAiB,EAAE,CAAC"}

package/dist/cjs/model_providers/fal.cjs

@@ -0,0 +1,97 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.falProvider = exports.FALProvider = 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");
+function mapImageSize(size) {
+    if (!size)
+        return undefined;
+    const s = String(size);
+    if (s === 'square')
+        return 'square_hd';
+    if (s === '1024x1024')
+        return 'square_hd';
+    if (s === 'landscape' || s === '1792x1024' || s === '1536x1024')
+        return 'landscape_16_9';
+    if (s === 'portrait' || s === '1024x1792' || s === '1024x1536')
+        return 'portrait_16_9';
+    return undefined;
+}
+class FALProvider extends base_provider_js_1.BaseModelProvider {
+    constructor() {
+        super('fal');
+    }
+    async *createResponseStream() {
+        throw new Error('FAL provider does not support text streaming');
+    }
+    endpointFor(model) {
+        const m = model.toLowerCase();
+        if (m.startsWith('recraft'))
+            return { path: 'fal-ai/recraft/v3/text-to-image', bodyMode: 'top' };
+        if (m.includes('runway') || m.includes('gen4'))
+            return { path: 'runwayml/gen4-image', bodyMode: 'input' };
+        if (m.includes('schnell'))
+            return { path: 'fal-ai/flux/schnell', bodyMode: 'top' };
+        if (m.includes('dev'))
+            return { path: 'fal-ai/flux/dev', bodyMode: 'top' };
+        if (m.includes('kontext') || m.includes('pro'))
+            return { path: 'fal-ai/flux-pro/kontext', bodyMode: 'top' };
+        return { path: 'fal-ai/flux/schnell', bodyMode: 'top' };
+    }
+    async createImage(prompt, model, agent, opts = {}) {
+        const falKey = process.env.FAL_KEY;
+        const requestId = (0, llm_logger_js_1.log_llm_request)(agent.agent_id || 'default', 'fal', model, { prompt, opts }, new Date());
+        try {
+            if (!falKey)
+                throw new Error('FAL_KEY is not set');
+            const { path, bodyMode } = this.endpointFor(model);
+            const size = mapImageSize(opts.size);
+            const bodyInput = bodyMode === 'top' ? { prompt } : { input: { prompt } };
+            if (size) {
+                if (bodyMode === 'top')
+                    bodyInput.image_size = size;
+                else
+                    bodyInput.input.image_size = size;
+            }
+            if (opts?.response_format === 'b64_json') {
+                if (bodyMode === 'top')
+                    bodyInput.sync_mode = true;
+                else
+                    bodyInput.input.sync_mode = true;
+            }
+            const res = await fetch(`https://fal.run/${path}`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    Authorization: `Key ${falKey}`,
+                },
+                body: JSON.stringify(bodyInput),
+            });
+            if (!res.ok)
+                throw new Error(`FAL request failed: ${res.status} ${await res.text()}`);
+            const data = await res.json();
+            const images = [];
+            const arr = data?.images || data?.output?.images || [];
+            for (const im of arr)
+                if (im?.url)
+                    images.push(im.url);
+            if (!images.length && data?.url)
+                images.push(data.url);
+            if (!images.length)
+                throw new Error('FAL: no image url in response');
+            cost_tracker_js_1.costTracker.addUsage({ model, image_count: images.length, request_id: opts?.request_id, metadata: { source: 'fal' } });
+            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.FALProvider = FALProvider;
+exports.falProvider = new FALProvider();
+//# sourceMappingURL=fal.js.map

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

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"fal.d.ts","sourceRoot":"","sources":["../../../model_providers/fal.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,eAAe,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAiBnG,qBAAa,WAAY,SAAQ,iBAAiB;;IAKvC,oBAAoB,IAAI,cAAc,CAAC,mBAAmB,CAAC;IAIlE,OAAO,CAAC,WAAW;IAWb,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,IAAI,GAAE,mBAAwB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;CAyC9H;AAED,eAAO,MAAM,WAAW,aAAoB,CAAC"}

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

@@ -0,0 +1 @@
+{"version":3,"file":"fal.js","sourceRoot":"","sources":["../../../model_providers/fal.ts"],"names":[],"mappings":";;;AAAA,yDAAuD;AAEvD,8DAAuD;AACvD,0DAA0F;AAK1F,SAAS,YAAY,CAAC,IAAkC;IACpD,IAAI,CAAC,IAAI;QAAE,OAAO,SAAS,CAAC;IAC5B,MAAM,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;IACvB,IAAI,CAAC,KAAK,QAAQ;QAAE,OAAO,WAAW,CAAC;IACvC,IAAI,CAAC,KAAK,WAAW;QAAE,OAAO,WAAW,CAAC;IAC1C,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,WAAW;QAAE,OAAO,gBAAgB,CAAC;IACzF,IAAI,CAAC,KAAK,UAAU,IAAI,CAAC,KAAK,WAAW,IAAI,CAAC,KAAK,WAAW;QAAE,OAAO,eAAe,CAAC;IACvF,OAAO,SAAS,CAAC;AACrB,CAAC;AAED,MAAa,WAAY,SAAQ,oCAAiB;IAC9C;QACI,KAAK,CAAC,KAAY,CAAC,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,CAAC,oBAAoB;QACvB,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;IACpE,CAAC;IAEO,WAAW,CAAC,KAAa;QAC7B,MAAM,CAAC,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;QAC9B,IAAI,CAAC,CAAC,UAAU,CAAC,SAAS,CAAC;YAAE,OAAO,EAAE,IAAI,EAAE,iCAAiC,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QACjG,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,EAAE,IAAI,EAAE,qBAAqB,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;QAE1G,IAAI,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,OAAO,EAAE,IAAI,EAAE,qBAAqB,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QACnF,IAAI,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,EAAE,IAAI,EAAE,iBAAiB,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QAC3E,IAAI,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,EAAE,IAAI,EAAE,yBAAyB,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;QAC5G,OAAO,EAAE,IAAI,EAAE,qBAAqB,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,MAAc,EAAE,KAAa,EAAE,KAAsB,EAAE,OAA4B,EAAE;QACnG,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC;QACnC,MAAM,SAAS,GAAG,IAAA,+BAAe,EAAC,KAAK,CAAC,QAAQ,IAAI,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,CAAC;QAC3G,IAAI,CAAC;YACD,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;YACnD,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACnD,MAAM,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrC,MAAM,SAAS,GAAQ,QAAQ,KAAK,KAAK,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC;YAC/E,IAAI,IAAI,EAAE,CAAC;gBACP,IAAI,QAAQ,KAAK,KAAK;oBAAE,SAAS,CAAC,UAAU,GAAG,IAAI,CAAC;;oBAC/C,SAAS,CAAC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC;YAC3C,CAAC;YACD,IAAI,IAAI,EAAE,eAAe,KAAK,UAAU,EAAE,CAAC;gBACvC,IAAI,QAAQ,KAAK,KAAK;oBAAE,SAAS,CAAC,SAAS,GAAG,IAAI,CAAC;;oBAC9C,SAAS,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC;YAC1C,CAAC;YAED,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,mBAAmB,IAAI,EAAE,EAAE;gBAC/C,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACL,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,OAAO,MAAM,EAAE;iBACjC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC;aAClC,CAAC,CAAC;YACH,IAAI,CAAC,GAAG,CAAC,EAAE;gBAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,GAAG,CAAC,MAAM,IAAI,MAAM,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACtF,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;YAC9B,MAAM,MAAM,GAAa,EAAE,CAAC;YAC5B,MAAM,GAAG,GAAG,IAAI,EAAE,MAAM,IAAI,IAAI,EAAE,MAAM,EAAE,MAAM,IAAI,EAAE,CAAC;YACvD,KAAK,MAAM,EAAE,IAAI,GAAG;gBAAE,IAAI,EAAE,EAAE,GAAG;oBAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC;YACvD,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,EAAE,GAAG;gBAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvD,IAAI,CAAC,MAAM,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;YACrE,6BAAW,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;YACvH,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;AA7DD,kCA6DC;AAEY,QAAA,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC"}

package/dist/cjs/model_providers/fireworks.cjs

@@ -0,0 +1,193 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fireworksProvider = exports.FireworksProvider = void 0;
+const base_provider_js_1 = require("./base_provider.cjs");
+const cost_tracker_js_1 = require("../utils/cost_tracker.cjs");
+const fetch_with_timeout_js_1 = require("../utils/fetch_with_timeout.cjs");
+const llm_logger_js_1 = require("../utils/llm_logger.cjs");
+const FW_BASE = 'https://api.fireworks.ai';
+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 FireworksProvider extends base_provider_js_1.BaseModelProvider {
+    constructor() {
+        super('fireworks');
+    }
+    async *createResponseStream() {
+        throw new Error('Fireworks provider does not support text streaming');
+    }
+    isKontext(model) {
+        return model.includes('kontext');
+    }
+    fireworksModelId(model) {
+        const m = model.toLowerCase();
+        if (m.includes('kontext') && m.includes('max'))
+            return 'flux-kontext-max';
+        if (m.includes('kontext'))
+            return 'flux-kontext-pro';
+        if (m.includes('pro'))
+            return 'flux-pro-1.1';
+        if (m.includes('schnell'))
+            return 'flux-schnell';
+        return model;
+    }
+    async createImage(prompt, model, agent, opts = {}) {
+        const apiKey = process.env.FIREWORKS_API_KEY;
+        const falKey = process.env.FAL_KEY;
+        const requestId = (0, llm_logger_js_1.log_llm_request)(agent.agent_id || 'default', 'fireworks', model, { prompt, opts }, new Date());
+        try {
+            if (!apiKey)
+                throw new Error('FIREWORKS_API_KEY is not set');
+            const aspect_ratio = mapAspect(opts.size);
+            const modelId = this.fireworksModelId(model);
+            if (this.isKontext(modelId)) {
+                const createUrl = `${FW_BASE}/inference/v1/workflows/accounts/fireworks/models/${modelId}`;
+                let input_image;
+                if (opts?.source_images) {
+                    const s = Array.isArray(opts.source_images) ? opts.source_images[0] : opts.source_images;
+                    const v = typeof s === 'string' ? s : s?.data || s;
+                    if (typeof v === 'string')
+                        input_image = v;
+                }
+                const res = await (0, fetch_with_timeout_js_1.fetchWithTimeout)(createUrl, {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json',
+                        Authorization: `Bearer ${apiKey}`,
+                    },
+                    body: JSON.stringify({ prompt, aspect_ratio: aspect_ratio, output_format: 'png', ...(input_image ? { input_image } : {}) }),
+                }, 60000);
+                if (!res.ok) {
+                    if ((res.status === 401 || res.status === 403) && falKey) {
+                        return this.fallbackToFAL(prompt, modelId, opts);
+                    }
+                    throw new Error(`Fireworks Kontext create failed: ${res.status} ${await res.text()}`);
+                }
+                const data = await res.json();
+                const id = data.request_id || data.id;
+                if (!id)
+                    throw new Error('Fireworks Kontext: missing request id');
+                const pollUrl = `${FW_BASE}/inference/v1/workflows/accounts/fireworks/models/${modelId}/get_result`;
+                const started = Date.now();
+                const timeoutMs = 240000;
+                while (true) {
+                    const r = await (0, fetch_with_timeout_js_1.fetchWithTimeout)(pollUrl, {
+                        method: 'POST',
+                        headers: {
+                            'Content-Type': 'application/json',
+                            Authorization: `Bearer ${apiKey}`,
+                        },
+                        body: JSON.stringify({ id }),
+                    }, 60000);
+                    if (!r.ok)
+                        throw new Error(`Fireworks Kontext poll failed: ${r.status} ${await r.text()}`);
+                    const out = await r.json();
+                    const status = String(out.status || '').toLowerCase();
+                    if (status.includes('ready') || status === 'succeeded' || status === 'completed') {
+                        const urls = [];
+                        if (Array.isArray(out.result?.images)) {
+                            for (const im of out.result.images)
+                                if (im?.url)
+                                    urls.push(im.url);
+                        }
+                        if (out.result?.image?.url)
+                            urls.push(out.result.image.url);
+                        if (out.result?.url)
+                            urls.push(out.result.url);
+                        if (out.result?.sample)
+                            urls.push(out.result.sample);
+                        if (!urls.length && out.result?.image_base64) {
+                            urls.push(`data:image/png;base64,${out.result.image_base64}`);
+                        }
+                        if (!urls.length && typeof out.result === 'string' && /^https?:\/\//.test(out.result)) {
+                            urls.push(out.result);
+                        }
+                        if (!urls.length)
+                            throw new Error('Fireworks Kontext: no image result found');
+                        cost_tracker_js_1.costTracker.addUsage({ model, image_count: urls.length, request_id: opts?.request_id, metadata: { source: 'fireworks', model: modelId } });
+                        return urls;
+                    }
+                    if (status.includes('error') || status.includes('failed')) {
+                        throw new Error(`Fireworks Kontext failed: ${status}`);
+                    }
+                    if (Date.now() - started > timeoutMs)
+                        throw new Error('Fireworks Kontext timed out');
+                    await new Promise(r2 => setTimeout(r2, 1500));
+                }
+            }
+            try {
+                return await this.createImage(prompt, 'flux-kontext-pro', agent, opts);
+            }
+            catch (e) {
+                if (falKey)
+                    return this.fallbackToFAL(prompt, model, opts);
+                throw e;
+            }
+        }
+        catch (err) {
+            (0, llm_logger_js_1.log_llm_error)(requestId, err);
+            if (process.env.FAL_KEY) {
+                try {
+                    const urls = await this.fallbackToFAL(prompt, model, opts);
+                    (0, llm_logger_js_1.log_llm_response)(requestId, { ok: true, fallback: 'fal' });
+                    return urls;
+                }
+                catch (_) {
+                }
+            }
+            throw err;
+        }
+        finally {
+            (0, llm_logger_js_1.log_llm_response)(requestId, { ok: true });
+        }
+    }
+    async fallbackToFAL(prompt, model, opts = {}) {
+        const lower = model.toLowerCase();
+        let endpoint = '';
+        if (lower.includes('schnell'))
+            endpoint = 'fal-ai/flux/schnell';
+        else if (lower.includes('dev'))
+            endpoint = 'fal-ai/flux/dev';
+        else if (lower.includes('pro') || lower.includes('kontext'))
+            endpoint = 'fal-ai/flux-pro/kontext';
+        else
+            endpoint = 'fal-ai/flux/schnell';
+        const res = await fetch(`https://fal.run/${endpoint}`, {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                Authorization: `Key ${process.env.FAL_KEY}`,
+            },
+            body: JSON.stringify({
+                prompt,
+                input: { prompt },
+            }),
+        });
+        if (!res.ok)
+            throw new Error(`FAL fallback failed: ${res.status} ${await res.text()}`);
+        const data = await res.json();
+        const images = [];
+        const arr = data?.images || data?.output?.images || [];
+        for (const im of arr)
+            if (im?.url)
+                images.push(im.url);
+        if (!images.length && data?.url)
+            images.push(data.url);
+        if (!images.length)
+            throw new Error('FAL fallback: no image url');
+        cost_tracker_js_1.costTracker.addUsage({ model, image_count: images.length, request_id: opts?.request_id, metadata: { source: 'fal-fallback' } });
+        return images;
+    }
+}
+exports.FireworksProvider = FireworksProvider;
+exports.fireworksProvider = new FireworksProvider();
+//# sourceMappingURL=fireworks.js.map

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

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"fireworks.d.ts","sourceRoot":"","sources":["../../../model_providers/fireworks.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,eAAe,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AAoBnG,qBAAa,iBAAkB,SAAQ,iBAAiB;;IAK7C,oBAAoB,IAAI,cAAc,CAAC,mBAAmB,CAAC;IAIlE,OAAO,CAAC,SAAS;IAIjB,OAAO,CAAC,gBAAgB;IAUlB,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,IAAI,GAAE,mBAAwB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;YAgH7G,aAAa;CA8B9B;AAED,eAAO,MAAM,iBAAiB,mBAA0B,CAAC"}

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

@@ -0,0 +1 @@
+{"version":3,"file":"fireworks.js","sourceRoot":"","sources":["../../../model_providers/fireworks.ts"],"names":[],"mappings":";;;AAAA,yDAAuD;AAEvD,8DAAuD;AACvD,0EAAkE;AAClE,0DAA0F;AAM1F,MAAM,OAAO,GAAG,0BAA0B,CAAC;AAE3C,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,SAAS,CAAC,KAAa;QAC3B,OAAO,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IACrC,CAAC;IAEO,gBAAgB,CAAC,KAAa;QAElC,MAAM,CAAC,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;QAC9B,IAAI,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,kBAAkB,CAAC;QAC1E,IAAI,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,OAAO,kBAAkB,CAAC;QACrD,IAAI,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,OAAO,cAAc,CAAC;QAC7C,IAAI,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,OAAO,cAAc,CAAC;QACjD,OAAO,KAAK,CAAC;IACjB,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,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC;QACnC,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;QAEjH,IAAI,CAAC;YACD,IAAI,CAAC,MAAM;gBAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;YAE7D,MAAM,YAAY,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;YAE7C,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;gBAE1B,MAAM,SAAS,GAAG,GAAG,OAAO,qDAAqD,OAAO,EAAE,CAAC;gBAE3F,IAAI,WAA+B,CAAC;gBACpC,IAAI,IAAI,EAAE,aAAa,EAAE,CAAC;oBACtB,MAAM,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAE,IAAI,CAAC,aAAqB,CAAC;oBAClG,MAAM,CAAC,GAAG,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,CAAC,CAAC;oBACnD,IAAI,OAAO,CAAC,KAAK,QAAQ;wBAAE,WAAW,GAAG,CAAC,CAAC;gBAC/C,CAAC;gBAED,MAAM,GAAG,GAAG,MAAM,IAAA,wCAAgB,EAAC,SAAS,EAAE;oBAC1C,MAAM,EAAE,MAAM;oBACd,OAAO,EAAE;wBACL,cAAc,EAAE,kBAAkB;wBAClC,aAAa,EAAE,UAAU,MAAM,EAAE;qBACpC;oBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,YAAY,EAAE,aAAa,EAAE,KAAK,EAAE,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;iBAC9H,EAAE,KAAK,CAAC,CAAC;gBAEV,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;oBAEV,IAAI,CAAC,GAAG,CAAC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,IAAI,MAAM,EAAE,CAAC;wBACvD,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;oBACrD,CAAC;oBACD,MAAM,IAAI,KAAK,CAAC,oCAAoC,GAAG,CAAC,MAAM,IAAI,MAAM,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAC1F,CAAC;gBACD,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;gBAC9B,MAAM,EAAE,GAAG,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,EAAE,CAAC;gBACtC,IAAI,CAAC,EAAE;oBAAE,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;gBAGlE,MAAM,OAAO,GAAG,GAAG,OAAO,qDAAqD,OAAO,aAAa,CAAC;gBACpG,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAC3B,MAAM,SAAS,GAAG,MAAM,CAAC;gBACzB,OAAO,IAAI,EAAE,CAAC;oBACV,MAAM,CAAC,GAAG,MAAM,IAAA,wCAAgB,EAAC,OAAO,EAAE;wBACtC,MAAM,EAAE,MAAM;wBACd,OAAO,EAAE;4BACL,cAAc,EAAE,kBAAkB;4BAClC,aAAa,EAAE,UAAU,MAAM,EAAE;yBACpC;wBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,EAAE,CAAC;qBAC/B,EAAE,KAAK,CAAC,CAAC;oBACV,IAAI,CAAC,CAAC,CAAC,EAAE;wBAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC,MAAM,IAAI,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBAC3F,MAAM,GAAG,GAAG,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;oBAC3B,MAAM,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;oBACtD,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,WAAW,EAAE,CAAC;wBAE/E,MAAM,IAAI,GAAa,EAAE,CAAC;wBAC1B,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC;4BACpC,KAAK,MAAM,EAAE,IAAI,GAAG,CAAC,MAAM,CAAC,MAAM;gCAAE,IAAI,EAAE,EAAE,GAAG;oCAAE,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC;wBACvE,CAAC;wBACD,IAAI,GAAG,CAAC,MAAM,EAAE,KAAK,EAAE,GAAG;4BAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;wBAC5D,IAAI,GAAG,CAAC,MAAM,EAAE,GAAG;4BAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;wBAC/C,IAAI,GAAG,CAAC,MAAM,EAAE,MAAM;4BAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;wBAErD,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM,EAAE,YAAY,EAAE,CAAC;4BAC3C,IAAI,CAAC,IAAI,CAAC,yBAAyB,GAAG,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;wBAClE,CAAC;wBACD,IAAI,CAAC,IAAI,CAAC,MAAM,IAAI,OAAO,GAAG,CAAC,MAAM,KAAK,QAAQ,IAAI,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;4BACpF,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;wBAC1B,CAAC;wBACD,IAAI,CAAC,IAAI,CAAC,MAAM;4BAAE,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;wBAE9E,6BAAW,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;wBAC3I,OAAO,IAAI,CAAC;oBAChB,CAAC;oBACD,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;wBACxD,MAAM,IAAI,KAAK,CAAC,6BAA6B,MAAM,EAAE,CAAC,CAAC;oBAC3D,CAAC;oBACD,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,GAAG,SAAS;wBAAE,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;oBACrF,MAAM,IAAI,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC,UAAU,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC;gBAClD,CAAC;YACL,CAAC;YAGD,IAAI,CAAC;gBACD,OAAO,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,kBAAkB,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;YAC3E,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACT,IAAI,MAAM;oBAAE,OAAO,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;gBAC3D,MAAM,CAAC,CAAC;YACZ,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,IAAA,6BAAa,EAAC,SAAS,EAAE,GAAG,CAAC,CAAC;YAE9B,IAAI,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;gBACtB,IAAI,CAAC;oBACD,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;oBAC3D,IAAA,gCAAgB,EAAC,SAAS,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;oBAC3D,OAAO,IAAI,CAAC;gBAChB,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;gBAEb,CAAC;YACL,CAAC;YACD,MAAM,GAAG,CAAC;QACd,CAAC;gBAAS,CAAC;YACP,IAAA,gCAAgB,EAAC,SAAS,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QAC9C,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,MAAc,EAAE,KAAa,EAAE,OAA4B,EAAE;QAErF,MAAM,KAAK,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC;QAClC,IAAI,QAAQ,GAAG,EAAE,CAAC;QAClB,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,QAAQ,GAAG,qBAAqB,CAAC;aAC3D,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,QAAQ,GAAG,iBAAiB,CAAC;aACxD,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,QAAQ,GAAG,yBAAyB,CAAC;;YAC7F,QAAQ,GAAG,qBAAqB,CAAC;QAEtC,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,mBAAmB,QAAQ,EAAE,EAAE;YACnD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACL,cAAc,EAAE,kBAAkB;gBAClC,aAAa,EAAE,OAAO,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE;aAC9C;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACjB,MAAM;gBACN,KAAK,EAAE,EAAE,MAAM,EAAE;aACpB,CAAC;SACL,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CAAC,EAAE;YAAE,MAAM,IAAI,KAAK,CAAC,wBAAwB,GAAG,CAAC,MAAM,IAAI,MAAM,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACvF,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAC;QAC9B,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,MAAM,GAAG,GAAG,IAAI,EAAE,MAAM,IAAI,IAAI,EAAE,MAAM,EAAE,MAAM,IAAI,EAAE,CAAC;QACvD,KAAK,MAAM,EAAE,IAAI,GAAG;YAAE,IAAI,EAAE,EAAE,GAAG;gBAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC;QACvD,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,EAAE,GAAG;YAAE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvD,IAAI,CAAC,MAAM,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAClE,6BAAW,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE,EAAE,CAAC,CAAC;QAChI,OAAO,MAAM,CAAC;IAClB,CAAC;CACJ;AArKD,8CAqKC;AAEY,QAAA,iBAAiB,GAAG,IAAI,iBAAiB,EAAE,CAAC"}