@just-every/ensemble 0.1.0

package/dist/utils/image_utils.d.ts

@@ -0,0 +1,18 @@
+import { Buffer } from 'buffer';
+export declare const MAX_IMAGE_HEIGHT = 2000;
+export declare const DEFAULT_QUALITY = 80;
+export declare const OPENAI_MAX_WIDTH = 1024;
+export declare const OPENAI_MAX_HEIGHT = 768;
+export declare const CLAUDE_MAX_WIDTH = 1024;
+export declare const CLAUDE_MAX_HEIGHT = 1120;
+export declare const GEMINI_MAX_WIDTH = 1024;
+export declare const GEMINI_MAX_HEIGHT = 1536;
+import { ExtractBase64ImageResult } from '../types.js';
+export type { ExtractBase64ImageResult };
+export declare function extractBase64Image(content: string): ExtractBase64ImageResult;
+export declare function createImageFromBase64(base64Data: string): Promise<Buffer>;
+export declare function createBase64FromImage(imageBuffer: Buffer): string;
+export declare function resizeAndSplitForOpenAI(imageData: string): Promise<string[]>;
+export declare function resizeAndTruncateForClaude(imageData: string): Promise<string>;
+export declare function resizeAndTruncateForGemini(imageData: string): Promise<string>;
+//# sourceMappingURL=image_utils.d.ts.map

package/dist/utils/image_utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"image_utils.d.ts","sourceRoot":"","sources":["../../utils/image_utils.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAKhC,eAAO,MAAM,gBAAgB,OAAO,CAAC;AACrC,eAAO,MAAM,eAAe,KAAK,CAAC;AAClC,eAAO,MAAM,gBAAgB,OAAO,CAAC;AACrC,eAAO,MAAM,iBAAiB,MAAM,CAAC;AACrC,eAAO,MAAM,gBAAgB,OAAO,CAAC;AACrC,eAAO,MAAM,iBAAiB,OAAO,CAAC;AACtC,eAAO,MAAM,gBAAgB,OAAO,CAAC;AACrC,eAAO,MAAM,iBAAiB,OAAO,CAAC;AAEtC,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AAGvD,YAAY,EAAE,wBAAwB,EAAE,CAAC;AASzC,wBAAgB,kBAAkB,CAAC,OAAO,EAAE,MAAM,GAAG,wBAAwB,CA6C5E;AAQD,wBAAsB,qBAAqB,CACvC,UAAU,EAAE,MAAM,GACnB,OAAO,CAAC,MAAM,CAAC,CAMjB;AAQD,wBAAgB,qBAAqB,CAAC,WAAW,EAAE,MAAM,GAAG,MAAM,CAMjE;AAUD,wBAAsB,uBAAuB,CACzC,SAAS,EAAE,MAAM,GAClB,OAAO,CAAC,MAAM,EAAE,CAAC,CAsDnB;AAwCD,wBAAsB,0BAA0B,CAC5C,SAAS,EAAE,MAAM,GAClB,OAAO,CAAC,MAAM,CAAC,CAiBjB;AAKD,wBAAsB,0BAA0B,CAC5C,SAAS,EAAE,MAAM,GAClB,OAAO,CAAC,MAAM,CAAC,CAiBjB"}

package/dist/utils/image_utils.js

@@ -0,0 +1,132 @@
+import { Buffer } from 'buffer';
+import sharp from 'sharp';
+import { randomUUID } from 'node:crypto';
+export const MAX_IMAGE_HEIGHT = 2000;
+export const DEFAULT_QUALITY = 80;
+export const OPENAI_MAX_WIDTH = 1024;
+export const OPENAI_MAX_HEIGHT = 768;
+export const CLAUDE_MAX_WIDTH = 1024;
+export const CLAUDE_MAX_HEIGHT = 1120;
+export const GEMINI_MAX_WIDTH = 1024;
+export const GEMINI_MAX_HEIGHT = 1536;
+export function extractBase64Image(content) {
+    const result = {
+        found: false,
+        originalContent: content,
+        replaceContent: content,
+        image_id: null,
+        images: {},
+    };
+    if (typeof content !== 'string')
+        return result;
+    if (!content.includes('data:image/'))
+        return result;
+    const imgRegex = /data:image\/[a-zA-Z0-9.+-]+;base64,[A-Za-z0-9+/=\s]+/g;
+    const images = {};
+    const replaceContent = content.replace(imgRegex, match => {
+        const id = randomUUID();
+        images[id] = match.replace(/\s+/g, '');
+        return `[image #${id}]`;
+    });
+    if (Object.keys(images).length === 0) {
+        return result;
+    }
+    const firstImageId = Object.keys(images)[0];
+    return {
+        found: true,
+        originalContent: content,
+        replaceContent: replaceContent,
+        image_id: firstImageId,
+        images: images,
+    };
+}
+export async function createImageFromBase64(base64Data) {
+    const base64Image = base64Data.replace(/^data:image\/\w+;base64,/, '');
+    return Buffer.from(base64Image, 'base64');
+}
+export function createBase64FromImage(imageBuffer) {
+    const base64Image = imageBuffer.toString('base64');
+    return `data:image/png;base64,${base64Image}`;
+}
+export async function resizeAndSplitForOpenAI(imageData) {
+    const MAX_WIDTH = 1024;
+    const MAX_HEIGHT = 768;
+    const base64Image = imageData.replace(/^data:image\/\w+;base64,/, '');
+    const imageFormat = imageData.match(/data:image\/(\w+);/)?.[1] || 'png';
+    const imageBuffer = Buffer.from(base64Image, 'base64');
+    const { width: origW = 0, height: origH = 0 } = await sharp(imageBuffer).metadata();
+    if (origW <= MAX_WIDTH && origH <= MAX_HEIGHT) {
+        return [imageData];
+    }
+    const newWidth = Math.min(origW, MAX_WIDTH);
+    const resizedBuffer = await sharp(imageBuffer)
+        .resize({ width: newWidth })
+        .flatten({ background: '#fff' })
+        .toFormat(imageFormat)
+        .toBuffer();
+    const { height: resizedH = 0 } = await sharp(resizedBuffer).metadata();
+    const result = [];
+    if (resizedH > MAX_HEIGHT) {
+        const segments = Math.ceil(resizedH / MAX_HEIGHT);
+        for (let i = 0; i < segments; i++) {
+            const top = i * MAX_HEIGHT;
+            const height = Math.min(MAX_HEIGHT, resizedH - top);
+            if (height <= 0)
+                continue;
+            const segmentBuf = await sharp(resizedBuffer)
+                .extract({ left: 0, top, width: newWidth, height })
+                .toFormat(imageFormat)
+                .toBuffer();
+            const segmentDataUrl = `data:image/${imageFormat};base64,${segmentBuf.toString('base64')}`;
+            result.push(segmentDataUrl);
+        }
+    }
+    else {
+        const singleUrl = `data:image/${imageFormat};base64,${resizedBuffer.toString('base64')}`;
+        result.push(singleUrl);
+    }
+    return result;
+}
+function stripDataUrl(dataUrl) {
+    const match = dataUrl.match(/^data:image\/([^;]+);base64,(.+)$/);
+    if (!match)
+        throw new Error('Invalid data-URL');
+    return { format: match[1], base64: match[2] };
+}
+async function processAndTruncate(imageBuffer, format, maxW, maxH) {
+    const resized = await sharp(imageBuffer)
+        .rotate()
+        .resize({ width: maxW, withoutEnlargement: true })
+        .flatten({ background: '#fff' })
+        .toFormat(format)
+        .toBuffer();
+    const { width, height } = await sharp(resized).metadata();
+    if (height > maxH) {
+        return await sharp(resized)
+            .extract({ left: 0, top: 0, width: width, height: maxH })
+            .toFormat(format)
+            .toBuffer();
+    }
+    return resized;
+}
+export async function resizeAndTruncateForClaude(imageData) {
+    const { format, base64 } = stripDataUrl(imageData);
+    const buf = Buffer.from(base64, 'base64');
+    const meta = await sharp(buf).metadata();
+    if (meta.width <= CLAUDE_MAX_WIDTH && meta.height <= CLAUDE_MAX_HEIGHT) {
+        return imageData;
+    }
+    const outBuf = await processAndTruncate(buf, format, CLAUDE_MAX_WIDTH, CLAUDE_MAX_HEIGHT);
+    return `data:image/${format};base64,${outBuf.toString('base64')}`;
+}
+export async function resizeAndTruncateForGemini(imageData) {
+    const { format, base64 } = stripDataUrl(imageData);
+    const buf = Buffer.from(base64, 'base64');
+    const meta = await sharp(buf).metadata();
+    if (meta.width <= GEMINI_MAX_WIDTH && meta.height <= GEMINI_MAX_HEIGHT) {
+        return imageData;
+    }
+    const outBuf = await processAndTruncate(buf, format, GEMINI_MAX_WIDTH, GEMINI_MAX_HEIGHT);
+    return `data:image/${format};base64,${outBuf.toString('base64')}`;
+}
+//# sourceMappingURL=image_utils.js.map

package/dist/utils/image_utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"image_utils.js","sourceRoot":"","sources":["../../utils/image_utils.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC;AACrC,MAAM,CAAC,MAAM,eAAe,GAAG,EAAE,CAAC;AAClC,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC;AACrC,MAAM,CAAC,MAAM,iBAAiB,GAAG,GAAG,CAAC;AACrC,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC;AACrC,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,CAAC;AACtC,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC;AACrC,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,CAAC;AActC,MAAM,UAAU,kBAAkB,CAAC,OAAe;IAE9C,MAAM,MAAM,GAA6B;QACrC,KAAK,EAAE,KAAK;QACZ,eAAe,EAAE,OAAO;QACxB,cAAc,EAAE,OAAO;QACvB,QAAQ,EAAE,IAAI;QACd,MAAM,EAAE,EAAE;KACb,CAAC;IAEF,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,MAAM,CAAC;IAG/C,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC;QAAE,OAAO,MAAM,CAAC;IAIpD,MAAM,QAAQ,GAAG,uDAAuD,CAAC;IAGzE,MAAM,MAAM,GAA2B,EAAE,CAAC;IAG1C,MAAM,cAAc,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE;QACrD,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;QAExB,MAAM,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;QACvC,OAAO,WAAW,EAAE,GAAG,CAAC;IAC5B,CAAC,CAAC,CAAC;IAGH,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACnC,OAAO,MAAM,CAAC;IAClB,CAAC;IAGD,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAE5C,OAAO;QACH,KAAK,EAAE,IAAI;QACX,eAAe,EAAE,OAAO;QACxB,cAAc,EAAE,cAAc;QAC9B,QAAQ,EAAE,YAAY;QACtB,MAAM,EAAE,MAAM;KACjB,CAAC;AACN,CAAC;AAQD,MAAM,CAAC,KAAK,UAAU,qBAAqB,CACvC,UAAkB;IAGlB,MAAM,WAAW,GAAG,UAAU,CAAC,OAAO,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;IAGvE,OAAO,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;AAC9C,CAAC;AAQD,MAAM,UAAU,qBAAqB,CAAC,WAAmB;IAErD,MAAM,WAAW,GAAG,WAAW,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAGnD,OAAO,yBAAyB,WAAW,EAAE,CAAC;AAClD,CAAC;AAUD,MAAM,CAAC,KAAK,UAAU,uBAAuB,CACzC,SAAiB;IAEjB,MAAM,SAAS,GAAG,IAAI,CAAC;IACvB,MAAM,UAAU,GAAG,GAAG,CAAC;IAGvB,MAAM,WAAW,GAAG,SAAS,CAAC,OAAO,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;IACtE,MAAM,WAAW,GAAG,SAAS,CAAC,KAAK,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC;IAGxE,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;IAGvD,MAAM,EAAE,KAAK,EAAE,KAAK,GAAG,CAAC,EAAE,MAAM,EAAE,KAAK,GAAG,CAAC,EAAE,GACzC,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,CAAC;IACxC,IAAI,KAAK,IAAI,SAAS,IAAI,KAAK,IAAI,UAAU,EAAE,CAAC;QAC5C,OAAO,CAAC,SAAS,CAAC,CAAC;IACvB,CAAC;IAGD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IAC5C,MAAM,aAAa,GAAG,MAAM,KAAK,CAAC,WAAW,CAAC;SACzC,MAAM,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;SAC3B,OAAO,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;SAC/B,QAAQ,CAAC,WAAqC,CAAC;SAC/C,QAAQ,EAAE,CAAC;IAGhB,MAAM,EAAE,MAAM,EAAE,QAAQ,GAAG,CAAC,EAAE,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC,CAAC,QAAQ,EAAE,CAAC;IAEvE,MAAM,MAAM,GAAa,EAAE,CAAC;IAG5B,IAAI,QAAQ,GAAG,UAAU,EAAE,CAAC;QACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,GAAG,UAAU,CAAC,CAAC;QAClD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,MAAM,GAAG,GAAG,CAAC,GAAG,UAAU,CAAC;YAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,GAAG,GAAG,CAAC,CAAC;YACpD,IAAI,MAAM,IAAI,CAAC;gBAAE,SAAS;YAE1B,MAAM,UAAU,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC;iBACxC,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;iBAClD,QAAQ,CAAC,WAAqC,CAAC;iBAC/C,QAAQ,EAAE,CAAC;YAEhB,MAAM,cAAc,GAAG,cAAc,WAAW,WAAW,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3F,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAChC,CAAC;IACL,CAAC;SAAM,CAAC;QAEJ,MAAM,SAAS,GAAG,cAAc,WAAW,WAAW,aAAa,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QACzF,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC3B,CAAC;IAED,OAAO,MAAM,CAAC;AAClB,CAAC;AAGD,SAAS,YAAY,CAAC,OAAe;IACjC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACjE,IAAI,CAAC,KAAK;QAAE,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;IAChD,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;AAClD,CAAC;AAED,KAAK,UAAU,kBAAkB,CAC7B,WAAmB,EACnB,MAAc,EACd,IAAY,EACZ,IAAY;IAGZ,MAAM,OAAO,GAAG,MAAM,KAAK,CAAC,WAAW,CAAC;SACnC,MAAM,EAAE;SACR,MAAM,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC;SACjD,OAAO,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;SAC/B,QAAQ,CAAC,MAAgC,CAAC;SAC1C,QAAQ,EAAE,CAAC;IAGhB,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;IAG1D,IAAI,MAAO,GAAG,IAAI,EAAE,CAAC;QACjB,OAAO,MAAM,KAAK,CAAC,OAAO,CAAC;aACtB,OAAO,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,KAAK,EAAE,KAAM,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;aACzD,QAAQ,CAAC,MAAgC,CAAC;aAC1C,QAAQ,EAAE,CAAC;IACpB,CAAC;IAED,OAAO,OAAO,CAAC;AACnB,CAAC;AAKD,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC5C,SAAiB;IAEjB,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IAG1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;IACzC,IAAI,IAAI,CAAC,KAAM,IAAI,gBAAgB,IAAI,IAAI,CAAC,MAAO,IAAI,iBAAiB,EAAE,CAAC;QACvE,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACnC,GAAG,EACH,MAAM,EACN,gBAAgB,EAChB,iBAAiB,CACpB,CAAC;IACF,OAAO,cAAc,MAAM,WAAW,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;AACtE,CAAC;AAKD,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC5C,SAAiB;IAEjB,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IAG1C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC;IACzC,IAAI,IAAI,CAAC,KAAM,IAAI,gBAAgB,IAAI,IAAI,CAAC,MAAO,IAAI,iBAAiB,EAAE,CAAC;QACvE,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,kBAAkB,CACnC,GAAG,EACH,MAAM,EACN,gBAAgB,EAChB,iBAAiB,CACpB,CAAC;IACF,OAAO,cAAc,MAAM,WAAW,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;AACtE,CAAC"}

package/dist/utils/llm_logger.d.ts

@@ -0,0 +1,8 @@
+import { EnsembleLogger } from '../types.js';
+export type { EnsembleLogger };
+export declare function setEnsembleLogger(logger: EnsembleLogger | null): void;
+export declare function getEnsembleLogger(): EnsembleLogger | null;
+export declare function log_llm_request(agentId: string, providerName: string, model: string, requestData: unknown, timestamp?: Date): string;
+export declare function log_llm_response(requestId: string | undefined, responseData: unknown, timestamp?: Date): void;
+export declare function log_llm_error(requestId: string | undefined, errorData: unknown, timestamp?: Date): void;
+//# sourceMappingURL=llm_logger.d.ts.map

package/dist/utils/llm_logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm_logger.d.ts","sourceRoot":"","sources":["../../utils/llm_logger.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAG7C,YAAY,EAAE,cAAc,EAAE,CAAC;AAI/B,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,cAAc,GAAG,IAAI,GAAG,IAAI,CAErE;AAED,wBAAgB,iBAAiB,IAAI,cAAc,GAAG,IAAI,CAEzD;AAED,wBAAgB,eAAe,CAC3B,OAAO,EAAE,MAAM,EACf,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,OAAO,EACpB,SAAS,CAAC,EAAE,IAAI,GACjB,MAAM,CAKR;AAED,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,EAAE,YAAY,EAAE,OAAO,EAAE,SAAS,CAAC,EAAE,IAAI,GAAG,IAAI,CAI7G;AAED,wBAAgB,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,CAAC,EAAE,IAAI,GAAG,IAAI,CAIvG"}

package/dist/utils/llm_logger.js

@@ -0,0 +1,24 @@
+let globalLogger = null;
+export function setEnsembleLogger(logger) {
+    globalLogger = logger;
+}
+export function getEnsembleLogger() {
+    return globalLogger;
+}
+export function log_llm_request(agentId, providerName, model, requestData, timestamp) {
+    if (globalLogger) {
+        return globalLogger.log_llm_request(agentId, providerName, model, requestData, timestamp);
+    }
+    return '';
+}
+export function log_llm_response(requestId, responseData, timestamp) {
+    if (globalLogger) {
+        globalLogger.log_llm_response(requestId, responseData, timestamp);
+    }
+}
+export function log_llm_error(requestId, errorData, timestamp) {
+    if (globalLogger) {
+        globalLogger.log_llm_error(requestId, errorData, timestamp);
+    }
+}
+//# sourceMappingURL=llm_logger.js.map

package/dist/utils/llm_logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm_logger.js","sourceRoot":"","sources":["../../utils/llm_logger.ts"],"names":[],"mappings":"AAKA,IAAI,YAAY,GAA0B,IAAI,CAAC;AAE/C,MAAM,UAAU,iBAAiB,CAAC,MAA6B;IAC3D,YAAY,GAAG,MAAM,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,iBAAiB;IAC7B,OAAO,YAAY,CAAC;AACxB,CAAC;AAED,MAAM,UAAU,eAAe,CAC3B,OAAe,EACf,YAAoB,EACpB,KAAa,EACb,WAAoB,EACpB,SAAgB;IAEhB,IAAI,YAAY,EAAE,CAAC;QACf,OAAO,YAAY,CAAC,eAAe,CAAC,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;IAC9F,CAAC;IACD,OAAO,EAAE,CAAC;AACd,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,SAA6B,EAAE,YAAqB,EAAE,SAAgB;IACnG,IAAI,YAAY,EAAE,CAAC;QACf,YAAY,CAAC,gBAAgB,CAAC,SAAS,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;IACtE,CAAC;AACL,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,SAA6B,EAAE,SAAkB,EAAE,SAAgB;IAC7F,IAAI,YAAY,EAAE,CAAC;QACf,YAAY,CAAC,aAAa,CAAC,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;IAChE,CAAC;AACL,CAAC"}

package/dist/utils/quota_tracker.d.ts

@@ -0,0 +1,22 @@
+import { ModelProviderID, ModelSpecificQuota, ProviderQuota } from '../types.js';
+export type { ModelSpecificQuota, ProviderQuota };
+export type QuotaUpdateCallback = (quotaSummary: Record<string, any>) => void;
+export declare class QuotaTracker {
+    private quotas;
+    private updateCallback?;
+    private openAIFreeQuota;
+    constructor(updateCallback?: QuotaUpdateCallback);
+    setUpdateCallback(callback: QuotaUpdateCallback | undefined): void;
+    private initializeProviderQuotas;
+    getProviderQuota(provider: ModelProviderID): ProviderQuota;
+    getModelQuota(provider: ModelProviderID, model: string): ModelSpecificQuota | null;
+    trackUsage(provider: ModelProviderID, model: string, inputTokens: number, outputTokens: number): boolean;
+    hasQuota(provider: ModelProviderID, model: string): boolean;
+    hasOpenAIFreeQuota(model: string): boolean;
+    trackCreditUsage(provider: ModelProviderID, creditAmount: number): void;
+    getCreditBalance(provider: ModelProviderID): number;
+    getSummary(): Record<string, any>;
+    sendQuotaUpdate(): void;
+}
+export declare const quotaTracker: QuotaTracker;
+//# sourceMappingURL=quota_tracker.d.ts.map

package/dist/utils/quota_tracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota_tracker.d.ts","sourceRoot":"","sources":["../../utils/quota_tracker.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,eAAe,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAGjF,YAAY,EAAE,kBAAkB,EAAE,aAAa,EAAE,CAAC;AA4BlD,MAAM,MAAM,mBAAmB,GAAG,CAAC,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,IAAI,CAAC;AAK9E,qBAAa,YAAY;IAErB,OAAO,CAAC,MAAM,CAAqC;IAGnD,OAAO,CAAC,cAAc,CAAC,CAAsB;IAG7C,OAAO,CAAC,eAAe,CAoBrB;gBAEU,cAAc,CAAC,EAAE,mBAAmB;IAQhD,iBAAiB,CAAC,QAAQ,EAAE,mBAAmB,GAAG,SAAS,GAAG,IAAI;IAOlE,OAAO,CAAC,wBAAwB;IA4FhC,gBAAgB,CAAC,QAAQ,EAAE,eAAe,GAAG,aAAa;IAa1D,aAAa,CACT,QAAQ,EAAE,eAAe,EACzB,KAAK,EAAE,MAAM,GACd,kBAAkB,GAAG,IAAI;IAgB5B,UAAU,CACN,QAAQ,EAAE,eAAe,EACzB,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,EACnB,YAAY,EAAE,MAAM,GACrB,OAAO;IA0MV,QAAQ,CAAC,QAAQ,EAAE,eAAe,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO;IAyB3D,kBAAkB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO;IAsB1C,gBAAgB,CAAC,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,GAAG,IAAI;IAiBvE,gBAAgB,CAAC,QAAQ,EAAE,eAAe,GAAG,MAAM;IAOnD,UAAU,IAAI,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;IAwEjC,eAAe,IAAI,IAAI;CAU1B;AAGD,eAAO,MAAM,YAAY,cAAqB,CAAC"}

package/dist/utils/quota_tracker.js

@@ -0,0 +1,338 @@
+export class QuotaTracker {
+    quotas = {};
+    updateCallback;
+    openAIFreeQuota = {
+        gpt4Family: {
+            limit: 1000000,
+            used: 0,
+        },
+        gptMiniFamily: {
+            limit: 10000000,
+            used: 0,
+        },
+        gpt4Models: ['gpt-4.5-preview', 'gpt-4.1', 'gpt-4o', 'o1', 'o3'],
+        gptMiniModels: [
+            'gpt-4.1-mini',
+            'gpt-4.1-nano',
+            'gpt-4o-mini',
+            'o1-mini',
+            'o3-mini',
+            'o4-mini',
+        ],
+    };
+    constructor(updateCallback) {
+        this.updateCallback = updateCallback;
+        this.initializeProviderQuotas();
+    }
+    setUpdateCallback(callback) {
+        this.updateCallback = callback;
+    }
+    initializeProviderQuotas() {
+        this.quotas['google'] = {
+            provider: 'google',
+            creditBalance: 0,
+            creditLimit: 0,
+            lastResetDate: new Date(),
+            models: {
+                'gemini-2.5-pro-exp-03-25': {
+                    model: 'gemini-2.5-pro-exp-03-25',
+                    dailyTokenLimit: 1000000,
+                    dailyTokensUsed: 0,
+                    dailyRequestLimit: 25,
+                    dailyRequestsUsed: 0,
+                    rateLimit: {
+                        requestsPerMinute: 5,
+                        tokensPerMinute: 250000,
+                    },
+                    lastResetDate: new Date(),
+                },
+                'gemini-2.5-flash-preview-04-17': {
+                    model: 'gemini-2.5-flash-preview-04-17',
+                    dailyTokenLimit: 0,
+                    dailyTokensUsed: 0,
+                    dailyRequestLimit: 500,
+                    dailyRequestsUsed: 0,
+                    rateLimit: {
+                        requestsPerMinute: 10,
+                        tokensPerMinute: 250000,
+                    },
+                    lastResetDate: new Date(),
+                },
+                'gemini-2.0-flash': {
+                    model: 'gemini-2.0-flash',
+                    dailyTokenLimit: 0,
+                    dailyTokensUsed: 0,
+                    dailyRequestLimit: 1500,
+                    dailyRequestsUsed: 0,
+                    rateLimit: {
+                        requestsPerMinute: 15,
+                        tokensPerMinute: 1000000,
+                    },
+                    lastResetDate: new Date(),
+                },
+                'gemini-2.0-flash-lite': {
+                    model: 'gemini-2.0-flash-lite',
+                    dailyTokenLimit: 0,
+                    dailyTokensUsed: 0,
+                    dailyRequestLimit: 1500,
+                    dailyRequestsUsed: 0,
+                    rateLimit: {
+                        requestsPerMinute: 30,
+                        tokensPerMinute: 1000000,
+                    },
+                    lastResetDate: new Date(),
+                },
+            },
+        };
+        this.quotas['openai'] = {
+            provider: 'openai',
+            creditBalance: 0,
+            creditLimit: 0,
+            lastResetDate: new Date(),
+            info: {
+                freeQuota: this.openAIFreeQuota,
+            },
+            models: {},
+        };
+        this.quotas['xai'] = {
+            provider: 'xai',
+            creditBalance: 150,
+            creditLimit: 150,
+            lastResetDate: new Date(),
+            models: {},
+        };
+    }
+    getProviderQuota(provider) {
+        return (this.quotas[provider] || {
+            provider,
+            models: {},
+        });
+    }
+    getModelQuota(provider, model) {
+        const providerQuota = this.getProviderQuota(provider);
+        if (providerQuota.models[model]) {
+            return providerQuota.models[model];
+        }
+        return null;
+    }
+    trackUsage(provider, model, inputTokens, outputTokens) {
+        if (!this.quotas[provider]) {
+            return true;
+        }
+        const providerQuota = this.quotas[provider];
+        let significantChange = false;
+        const totalTokens = inputTokens + outputTokens;
+        const today = new Date();
+        if (providerQuota.lastResetDate &&
+            (today.getUTCDate() !== providerQuota.lastResetDate.getUTCDate() ||
+                today.getUTCMonth() !==
+                    providerQuota.lastResetDate.getUTCMonth() ||
+                today.getUTCFullYear() !==
+                    providerQuota.lastResetDate.getUTCFullYear())) {
+            for (const modelKey in providerQuota.models) {
+                const modelQuota = providerQuota.models[modelKey];
+                modelQuota.dailyTokensUsed = 0;
+                modelQuota.dailyRequestsUsed = 0;
+                modelQuota.lastResetDate = today;
+            }
+            if (provider === 'openai' && providerQuota.info?.freeQuota) {
+                const freeQuota = providerQuota.info
+                    .freeQuota;
+                freeQuota.gpt4Family.used = 0;
+                freeQuota.gptMiniFamily.used = 0;
+            }
+            providerQuota.lastResetDate = today;
+            significantChange = true;
+        }
+        if (provider === 'openai' && providerQuota.info?.freeQuota) {
+            const freeQuota = providerQuota.info.freeQuota;
+            if (freeQuota.gpt4Models.includes(model)) {
+                const prevUsed = freeQuota.gpt4Family.used;
+                freeQuota.gpt4Family.used += totalTokens;
+                console.log(`[QuotaTracker] OpenAI GPT-4 family usage: ${freeQuota.gpt4Family.used}/${freeQuota.gpt4Family.limit}`);
+                const prevPercent = Math.floor((prevUsed / freeQuota.gpt4Family.limit) * 10);
+                const currPercent = Math.floor((freeQuota.gpt4Family.used / freeQuota.gpt4Family.limit) *
+                    10);
+                if (prevPercent !== currPercent) {
+                    significantChange = true;
+                }
+                if (freeQuota.gpt4Family.used >= freeQuota.gpt4Family.limit) {
+                    console.log(`[QuotaTracker] OpenAI GPT-4 family daily limit reached: ${freeQuota.gpt4Family.used} > ${freeQuota.gpt4Family.limit}`);
+                    significantChange = true;
+                    this.sendQuotaUpdate();
+                    return false;
+                }
+            }
+            if (freeQuota.gptMiniModels.includes(model)) {
+                const prevUsed = freeQuota.gptMiniFamily.used;
+                freeQuota.gptMiniFamily.used += totalTokens;
+                console.log(`[QuotaTracker] OpenAI GPT-Mini family usage: ${freeQuota.gptMiniFamily.used}/${freeQuota.gptMiniFamily.limit}`);
+                const prevPercent = Math.floor((prevUsed / freeQuota.gptMiniFamily.limit) * 10);
+                const currPercent = Math.floor((freeQuota.gptMiniFamily.used /
+                    freeQuota.gptMiniFamily.limit) *
+                    10);
+                if (prevPercent !== currPercent) {
+                    significantChange = true;
+                }
+                if (freeQuota.gptMiniFamily.used >=
+                    freeQuota.gptMiniFamily.limit) {
+                    console.log(`[QuotaTracker] OpenAI GPT-Mini family daily limit reached: ${freeQuota.gptMiniFamily.used} > ${freeQuota.gptMiniFamily.limit}`);
+                    significantChange = true;
+                    this.sendQuotaUpdate();
+                    return false;
+                }
+            }
+        }
+        const modelQuota = this.getModelQuota(provider, model);
+        if (!modelQuota && provider !== 'openai') {
+            return true;
+        }
+        if (modelQuota) {
+            const previousDailyTokensUsed = modelQuota.dailyTokensUsed;
+            modelQuota.dailyTokensUsed += totalTokens;
+            const previousDailyRequestsUsed = modelQuota.dailyRequestsUsed;
+            modelQuota.dailyRequestsUsed += 1;
+            if (modelQuota.dailyTokenLimit > 0) {
+                const previousPercent = Math.floor((previousDailyTokensUsed / modelQuota.dailyTokenLimit) * 10);
+                const currentPercent = Math.floor((modelQuota.dailyTokensUsed / modelQuota.dailyTokenLimit) *
+                    10);
+                if (previousPercent !== currentPercent) {
+                    significantChange = true;
+                }
+            }
+            if (modelQuota.dailyRequestLimit > 0) {
+                const previousPercent = Math.floor((previousDailyRequestsUsed / modelQuota.dailyRequestLimit) *
+                    10);
+                const currentPercent = Math.floor((modelQuota.dailyRequestsUsed /
+                    modelQuota.dailyRequestLimit) *
+                    10);
+                if (previousPercent !== currentPercent) {
+                    significantChange = true;
+                }
+            }
+            if (modelQuota.dailyTokenLimit > 0 &&
+                modelQuota.dailyTokensUsed >= modelQuota.dailyTokenLimit) {
+                console.log(`[QuotaTracker] ${provider} model ${model} daily token limit reached: ${modelQuota.dailyTokensUsed} > ${modelQuota.dailyTokenLimit}`);
+                significantChange = true;
+                this.sendQuotaUpdate();
+                return false;
+            }
+            if (modelQuota.dailyRequestLimit > 0 &&
+                modelQuota.dailyRequestsUsed >= modelQuota.dailyRequestLimit) {
+                console.log(`[QuotaTracker] ${provider} model ${model} daily request limit reached: ${modelQuota.dailyRequestsUsed} > ${modelQuota.dailyRequestLimit}`);
+                significantChange = true;
+                this.sendQuotaUpdate();
+                return false;
+            }
+        }
+        if (significantChange) {
+            this.sendQuotaUpdate();
+        }
+        return true;
+    }
+    hasQuota(provider, model) {
+        if (provider === 'openai') {
+            return this.hasOpenAIFreeQuota(model);
+        }
+        const modelQuota = this.getModelQuota(provider, model);
+        if (modelQuota) {
+            return ((modelQuota.dailyTokenLimit === 0 ||
+                modelQuota.dailyTokensUsed < modelQuota.dailyTokenLimit) &&
+                (modelQuota.dailyRequestLimit === 0 ||
+                    modelQuota.dailyRequestsUsed < modelQuota.dailyRequestLimit));
+        }
+        return true;
+    }
+    hasOpenAIFreeQuota(model) {
+        const providerQuota = this.getProviderQuota('openai');
+        if (!providerQuota || !providerQuota.info?.freeQuota)
+            return true;
+        const freeQuota = providerQuota.info.freeQuota;
+        if (freeQuota.gpt4Models.includes(model)) {
+            return freeQuota.gpt4Family.used < freeQuota.gpt4Family.limit;
+        }
+        if (freeQuota.gptMiniModels.includes(model)) {
+            return freeQuota.gptMiniFamily.used < freeQuota.gptMiniFamily.limit;
+        }
+        return true;
+    }
+    trackCreditUsage(provider, creditAmount) {
+        const providerQuota = this.getProviderQuota(provider);
+        if (!providerQuota || providerQuota.creditBalance === undefined) {
+            return;
+        }
+        if (providerQuota.creditBalance !== undefined) {
+            providerQuota.creditBalance = Math.max(0, providerQuota.creditBalance - creditAmount);
+        }
+    }
+    getCreditBalance(provider) {
+        return this.getProviderQuota(provider)?.creditBalance || 0;
+    }
+    getSummary() {
+        const summary = {};
+        for (const [provider, providerQuota] of Object.entries(this.quotas)) {
+            summary[provider] = {
+                creditBalance: providerQuota.creditBalance,
+                creditLimit: providerQuota.creditLimit,
+                models: {},
+            };
+            for (const [modelName, modelQuota] of Object.entries(providerQuota.models)) {
+                summary[provider].models[modelName] = {
+                    tokens: {
+                        used: modelQuota.dailyTokensUsed,
+                        limit: modelQuota.dailyTokenLimit,
+                        percent: modelQuota.dailyTokenLimit > 0
+                            ? (modelQuota.dailyTokensUsed /
+                                modelQuota.dailyTokenLimit) *
+                                100
+                            : 0,
+                    },
+                    requests: {
+                        used: modelQuota.dailyRequestsUsed,
+                        limit: modelQuota.dailyRequestLimit,
+                        percent: modelQuota.dailyRequestLimit > 0
+                            ? (modelQuota.dailyRequestsUsed /
+                                modelQuota.dailyRequestLimit) *
+                                100
+                            : 0,
+                    },
+                    lastReset: modelQuota.lastResetDate,
+                };
+            }
+            if (provider === 'openai' && providerQuota.info?.freeQuota) {
+                const freeQuota = providerQuota.info
+                    .freeQuota;
+                summary[provider].freeTier = {
+                    gpt4Family: {
+                        used: freeQuota.gpt4Family.used,
+                        limit: freeQuota.gpt4Family.limit,
+                        percent: (freeQuota.gpt4Family.used /
+                            freeQuota.gpt4Family.limit) *
+                            100,
+                    },
+                    gptMiniFamily: {
+                        used: freeQuota.gptMiniFamily.used,
+                        limit: freeQuota.gptMiniFamily.limit,
+                        percent: (freeQuota.gptMiniFamily.used /
+                            freeQuota.gptMiniFamily.limit) *
+                            100,
+                    },
+                };
+            }
+        }
+        return summary;
+    }
+    sendQuotaUpdate() {
+        try {
+            if (this.updateCallback) {
+                const quotas = this.getSummary();
+                this.updateCallback(quotas);
+            }
+        }
+        catch (error) {
+            console.error('Error sending quota update:', error);
+        }
+    }
+}
+export const quotaTracker = new QuotaTracker();
+//# sourceMappingURL=quota_tracker.js.map

package/dist/utils/quota_tracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota_tracker.js","sourceRoot":"","sources":["../../utils/quota_tracker.ts"],"names":[],"mappings":"AA0CA,MAAM,OAAO,YAAY;IAEb,MAAM,GAAkC,EAAE,CAAC;IAG3C,cAAc,CAAuB;IAGrC,eAAe,GAAoB;QACvC,UAAU,EAAE;YACR,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,CAAC;SACV;QACD,aAAa,EAAE;YACX,KAAK,EAAE,QAAQ;YACf,IAAI,EAAE,CAAC;SACV;QAED,UAAU,EAAE,CAAC,iBAAiB,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC;QAEhE,aAAa,EAAE;YACX,cAAc;YACd,cAAc;YACd,aAAa;YACb,SAAS;YACT,SAAS;YACT,SAAS;SACZ;KACJ,CAAC;IAEF,YAAY,cAAoC;QAC5C,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,wBAAwB,EAAE,CAAC;IACpC,CAAC;IAKD,iBAAiB,CAAC,QAAyC;QACvD,IAAI,CAAC,cAAc,GAAG,QAAQ,CAAC;IACnC,CAAC;IAKO,wBAAwB;QAE5B,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG;YACpB,QAAQ,EAAE,QAAQ;YAClB,aAAa,EAAE,CAAC;YAChB,WAAW,EAAE,CAAC;YACd,aAAa,EAAE,IAAI,IAAI,EAAE;YACzB,MAAM,EAAE;gBAEJ,0BAA0B,EAAE;oBACxB,KAAK,EAAE,0BAA0B;oBACjC,eAAe,EAAE,OAAO;oBACxB,eAAe,EAAE,CAAC;oBAClB,iBAAiB,EAAE,EAAE;oBACrB,iBAAiB,EAAE,CAAC;oBACpB,SAAS,EAAE;wBACP,iBAAiB,EAAE,CAAC;wBACpB,eAAe,EAAE,MAAM;qBAC1B;oBACD,aAAa,EAAE,IAAI,IAAI,EAAE;iBAC5B;gBAED,gCAAgC,EAAE;oBAC9B,KAAK,EAAE,gCAAgC;oBACvC,eAAe,EAAE,CAAC;oBAClB,eAAe,EAAE,CAAC;oBAClB,iBAAiB,EAAE,GAAG;oBACtB,iBAAiB,EAAE,CAAC;oBACpB,SAAS,EAAE;wBACP,iBAAiB,EAAE,EAAE;wBACrB,eAAe,EAAE,MAAM;qBAC1B;oBACD,aAAa,EAAE,IAAI,IAAI,EAAE;iBAC5B;gBAED,kBAAkB,EAAE;oBAChB,KAAK,EAAE,kBAAkB;oBACzB,eAAe,EAAE,CAAC;oBAClB,eAAe,EAAE,CAAC;oBAClB,iBAAiB,EAAE,IAAI;oBACvB,iBAAiB,EAAE,CAAC;oBACpB,SAAS,EAAE;wBACP,iBAAiB,EAAE,EAAE;wBACrB,eAAe,EAAE,OAAO;qBAC3B;oBACD,aAAa,EAAE,IAAI,IAAI,EAAE;iBAC5B;gBAED,uBAAuB,EAAE;oBACrB,KAAK,EAAE,uBAAuB;oBAC9B,eAAe,EAAE,CAAC;oBAClB,eAAe,EAAE,CAAC;oBAClB,iBAAiB,EAAE,IAAI;oBACvB,iBAAiB,EAAE,CAAC;oBACpB,SAAS,EAAE;wBACP,iBAAiB,EAAE,EAAE;wBACrB,eAAe,EAAE,OAAO;qBAC3B;oBACD,aAAa,EAAE,IAAI,IAAI,EAAE;iBAC5B;aACJ;SACJ,CAAC;QAGF,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG;YACpB,QAAQ,EAAE,QAAQ;YAClB,aAAa,EAAE,CAAC;YAChB,WAAW,EAAE,CAAC;YACd,aAAa,EAAE,IAAI,IAAI,EAAE;YACzB,IAAI,EAAE;gBACF,SAAS,EAAE,IAAI,CAAC,eAAe;aAClC;YACD,MAAM,EAAE,EAEP;SACJ,CAAC;QAGF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG;YACjB,QAAQ,EAAE,KAAK;YACf,aAAa,EAAE,GAAG;YAClB,WAAW,EAAE,GAAG;YAChB,aAAa,EAAE,IAAI,IAAI,EAAE;YACzB,MAAM,EAAE,EAEP;SACJ,CAAC;IACN,CAAC;IAKD,gBAAgB,CAAC,QAAyB;QACtC,OAAO,CACH,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI;YACrB,QAAQ;YACR,MAAM,EAAE,EAAE;SACb,CACJ,CAAC;IACN,CAAC;IAMD,aAAa,CACT,QAAyB,EACzB,KAAa;QAEb,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAGtD,IAAI,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvC,CAAC;QAGD,OAAO,IAAI,CAAC;IAChB,CAAC;IAMD,UAAU,CACN,QAAyB,EACzB,KAAa,EACb,WAAmB,EACnB,YAAoB;QAGpB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC;QAChB,CAAC;QAGD,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,iBAAiB,GAAG,KAAK,CAAC;QAC9B,MAAM,WAAW,GAAG,WAAW,GAAG,YAAY,CAAC;QAG/C,MAAM,KAAK,GAAG,IAAI,IAAI,EAAE,CAAC;QACzB,IACI,aAAa,CAAC,aAAa;YAC3B,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,aAAa,CAAC,aAAa,CAAC,UAAU,EAAE;gBAC5D,KAAK,CAAC,WAAW,EAAE;oBACf,aAAa,CAAC,aAAa,CAAC,WAAW,EAAE;gBAC7C,KAAK,CAAC,cAAc,EAAE;oBAClB,aAAa,CAAC,aAAa,CAAC,cAAc,EAAE,CAAC,EACvD,CAAC;YAEC,KAAK,MAAM,QAAQ,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;gBAC1C,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBAClD,UAAU,CAAC,eAAe,GAAG,CAAC,CAAC;gBAC/B,UAAU,CAAC,iBAAiB,GAAG,CAAC,CAAC;gBACjC,UAAU,CAAC,aAAa,GAAG,KAAK,CAAC;YACrC,CAAC;YAGD,IAAI,QAAQ,KAAK,QAAQ,IAAI,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;gBACzD,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI;qBAC/B,SAA4B,CAAC;gBAClC,SAAS,CAAC,UAAU,CAAC,IAAI,GAAG,CAAC,CAAC;gBAC9B,SAAS,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC,CAAC;YACrC,CAAC;YAGD,aAAa,CAAC,aAAa,GAAG,KAAK,CAAC;YACpC,iBAAiB,GAAG,IAAI,CAAC;QAC7B,CAAC;QAGD,IAAI,QAAQ,KAAK,QAAQ,IAAI,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YACzD,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,CAAC,SAA4B,CAAC;YAGlE,IAAI,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBACvC,MAAM,QAAQ,GAAG,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC;gBAC3C,SAAS,CAAC,UAAU,CAAC,IAAI,IAAI,WAAW,CAAC;gBACzC,OAAO,CAAC,GAAG,CACP,6CAA6C,SAAS,CAAC,UAAU,CAAC,IAAI,IAAI,SAAS,CAAC,UAAU,CAAC,KAAK,EAAE,CACzG,CAAC;gBAGF,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAC1B,CAAC,QAAQ,GAAG,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE,CAC/C,CAAC;gBACF,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAC1B,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,GAAG,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC;oBACpD,EAAE,CACT,CAAC;gBACF,IAAI,WAAW,KAAK,WAAW,EAAE,CAAC;oBAC9B,iBAAiB,GAAG,IAAI,CAAC;gBAC7B,CAAC;gBAGD,IAAI,SAAS,CAAC,UAAU,CAAC,IAAI,IAAI,SAAS,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;oBAC1D,OAAO,CAAC,GAAG,CACP,2DAA2D,SAAS,CAAC,UAAU,CAAC,IAAI,MAAM,SAAS,CAAC,UAAU,CAAC,KAAK,EAAE,CACzH,CAAC;oBACF,iBAAiB,GAAG,IAAI,CAAC;oBACzB,IAAI,CAAC,eAAe,EAAE,CAAC;oBACvB,OAAO,KAAK,CAAC;gBACjB,CAAC;YACL,CAAC;YAGD,IAAI,SAAS,CAAC,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1C,MAAM,QAAQ,GAAG,SAAS,CAAC,aAAa,CAAC,IAAI,CAAC;gBAC9C,SAAS,CAAC,aAAa,CAAC,IAAI,IAAI,WAAW,CAAC;gBAC5C,OAAO,CAAC,GAAG,CACP,gDAAgD,SAAS,CAAC,aAAa,CAAC,IAAI,IAAI,SAAS,CAAC,aAAa,CAAC,KAAK,EAAE,CAClH,CAAC;gBAGF,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAC1B,CAAC,QAAQ,GAAG,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,GAAG,EAAE,CAClD,CAAC;gBACF,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAC1B,CAAC,SAAS,CAAC,aAAa,CAAC,IAAI;oBACzB,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC;oBAC9B,EAAE,CACT,CAAC;gBACF,IAAI,WAAW,KAAK,WAAW,EAAE,CAAC;oBAC9B,iBAAiB,GAAG,IAAI,CAAC;gBAC7B,CAAC;gBAGD,IACI,SAAS,CAAC,aAAa,CAAC,IAAI;oBAC5B,SAAS,CAAC,aAAa,CAAC,KAAK,EAC/B,CAAC;oBACC,OAAO,CAAC,GAAG,CACP,8DAA8D,SAAS,CAAC,aAAa,CAAC,IAAI,MAAM,SAAS,CAAC,aAAa,CAAC,KAAK,EAAE,CAClI,CAAC;oBACF,iBAAiB,GAAG,IAAI,CAAC;oBACzB,IAAI,CAAC,eAAe,EAAE,CAAC;oBACvB,OAAO,KAAK,CAAC;gBACjB,CAAC;YACL,CAAC;QACL,CAAC;QAGD,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAGvD,IAAI,CAAC,UAAU,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACvC,OAAO,IAAI,CAAC;QAChB,CAAC;QAGD,IAAI,UAAU,EAAE,CAAC;YAEb,MAAM,uBAAuB,GAAG,UAAU,CAAC,eAAe,CAAC;YAC3D,UAAU,CAAC,eAAe,IAAI,WAAW,CAAC;YAG1C,MAAM,yBAAyB,GAAG,UAAU,CAAC,iBAAiB,CAAC;YAC/D,UAAU,CAAC,iBAAiB,IAAI,CAAC,CAAC;YAGlC,IAAI,UAAU,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC;gBACjC,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAC9B,CAAC,uBAAuB,GAAG,UAAU,CAAC,eAAe,CAAC,GAAG,EAAE,CAC9D,CAAC;gBACF,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAC7B,CAAC,UAAU,CAAC,eAAe,GAAG,UAAU,CAAC,eAAe,CAAC;oBACrD,EAAE,CACT,CAAC;gBAEF,IAAI,eAAe,KAAK,cAAc,EAAE,CAAC;oBACrC,iBAAiB,GAAG,IAAI,CAAC;gBAC7B,CAAC;YACL,CAAC;YAGD,IAAI,UAAU,CAAC,iBAAiB,GAAG,CAAC,EAAE,CAAC;gBACnC,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAC9B,CAAC,yBAAyB,GAAG,UAAU,CAAC,iBAAiB,CAAC;oBACtD,EAAE,CACT,CAAC;gBACF,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAC7B,CAAC,UAAU,CAAC,iBAAiB;oBACzB,UAAU,CAAC,iBAAiB,CAAC;oBAC7B,EAAE,CACT,CAAC;gBAEF,IAAI,eAAe,KAAK,cAAc,EAAE,CAAC;oBACrC,iBAAiB,GAAG,IAAI,CAAC;gBAC7B,CAAC;YACL,CAAC;YAGD,IACI,UAAU,CAAC,eAAe,GAAG,CAAC;gBAC9B,UAAU,CAAC,eAAe,IAAI,UAAU,CAAC,eAAe,EAC1D,CAAC;gBACC,OAAO,CAAC,GAAG,CACP,kBAAkB,QAAQ,UAAU,KAAK,+BAA+B,UAAU,CAAC,eAAe,MAAM,UAAU,CAAC,eAAe,EAAE,CACvI,CAAC;gBACF,iBAAiB,GAAG,IAAI,CAAC;gBACzB,IAAI,CAAC,eAAe,EAAE,CAAC;gBACvB,OAAO,KAAK,CAAC;YACjB,CAAC;YAGD,IACI,UAAU,CAAC,iBAAiB,GAAG,CAAC;gBAChC,UAAU,CAAC,iBAAiB,IAAI,UAAU,CAAC,iBAAiB,EAC9D,CAAC;gBACC,OAAO,CAAC,GAAG,CACP,kBAAkB,QAAQ,UAAU,KAAK,iCAAiC,UAAU,CAAC,iBAAiB,MAAM,UAAU,CAAC,iBAAiB,EAAE,CAC7I,CAAC;gBACF,iBAAiB,GAAG,IAAI,CAAC;gBACzB,IAAI,CAAC,eAAe,EAAE,CAAC;gBACvB,OAAO,KAAK,CAAC;YACjB,CAAC;QACL,CAAC;QAGD,IAAI,iBAAiB,EAAE,CAAC;YACpB,IAAI,CAAC,eAAe,EAAE,CAAC;QAC3B,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAMD,QAAQ,CAAC,QAAyB,EAAE,KAAa;QAE7C,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC;QAGD,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACvD,IAAI,UAAU,EAAE,CAAC;YAEb,OAAO,CACH,CAAC,UAAU,CAAC,eAAe,KAAK,CAAC;gBAC7B,UAAU,CAAC,eAAe,GAAG,UAAU,CAAC,eAAe,CAAC;gBAC5D,CAAC,UAAU,CAAC,iBAAiB,KAAK,CAAC;oBAC/B,UAAU,CAAC,iBAAiB,GAAG,UAAU,CAAC,iBAAiB,CAAC,CACnE,CAAC;QACN,CAAC;QAGD,OAAO,IAAI,CAAC;IAChB,CAAC;IAKD,kBAAkB,CAAC,KAAa;QAC5B,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,CAAC,aAAa,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,SAAS;YAAE,OAAO,IAAI,CAAC;QAElE,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,CAAC,SAA4B,CAAC;QAGlE,IAAI,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACvC,OAAO,SAAS,CAAC,UAAU,CAAC,IAAI,GAAG,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC;QAClE,CAAC;QAGD,IAAI,SAAS,CAAC,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,SAAS,CAAC,aAAa,CAAC,IAAI,GAAG,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC;QACxE,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAKD,gBAAgB,CAAC,QAAyB,EAAE,YAAoB;QAC5D,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,CAAC,aAAa,IAAI,aAAa,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;YAC9D,OAAO;QACX,CAAC;QAED,IAAI,aAAa,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;YAC5C,aAAa,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,CAClC,CAAC,EACD,aAAa,CAAC,aAAa,GAAG,YAAY,CAC7C,CAAC;QACN,CAAC;IACL,CAAC;IAKD,gBAAgB,CAAC,QAAyB;QACtC,OAAO,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,aAAa,IAAI,CAAC,CAAC;IAC/D,CAAC;IAKD,UAAU;QACN,MAAM,OAAO,GAAwB,EAAE,CAAC;QAExC,KAAK,MAAM,CAAC,QAAQ,EAAE,aAAa,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAElE,OAAO,CAAC,QAAQ,CAAC,GAAG;gBAChB,aAAa,EAAE,aAAa,CAAC,aAAa;gBAC1C,WAAW,EAAE,aAAa,CAAC,WAAW;gBACtC,MAAM,EAAE,EAAE;aACb,CAAC;YAGF,KAAK,MAAM,CAAC,SAAS,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAChD,aAAa,CAAC,MAAM,CACvB,EAAE,CAAC;gBAEA,OAAO,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG;oBAClC,MAAM,EAAE;wBACJ,IAAI,EAAE,UAAU,CAAC,eAAe;wBAChC,KAAK,EAAE,UAAU,CAAC,eAAe;wBACjC,OAAO,EACH,UAAU,CAAC,eAAe,GAAG,CAAC;4BAC1B,CAAC,CAAC,CAAC,UAAU,CAAC,eAAe;gCACvB,UAAU,CAAC,eAAe,CAAC;gCAC/B,GAAG;4BACL,CAAC,CAAC,CAAC;qBACd;oBACD,QAAQ,EAAE;wBACN,IAAI,EAAE,UAAU,CAAC,iBAAiB;wBAClC,KAAK,EAAE,UAAU,CAAC,iBAAiB;wBACnC,OAAO,EACH,UAAU,CAAC,iBAAiB,GAAG,CAAC;4BAC5B,CAAC,CAAC,CAAC,UAAU,CAAC,iBAAiB;gCACzB,UAAU,CAAC,iBAAiB,CAAC;gCACjC,GAAG;4BACL,CAAC,CAAC,CAAC;qBACd;oBACD,SAAS,EAAE,UAAU,CAAC,aAAa;iBACtC,CAAC;YACN,CAAC;YAGD,IAAI,QAAQ,KAAK,QAAQ,IAAI,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;gBACzD,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI;qBAC/B,SAA4B,CAAC;gBAClC,OAAO,CAAC,QAAQ,CAAC,CAAC,QAAQ,GAAG;oBACzB,UAAU,EAAE;wBACR,IAAI,EAAE,SAAS,CAAC,UAAU,CAAC,IAAI;wBAC/B,KAAK,EAAE,SAAS,CAAC,UAAU,CAAC,KAAK;wBACjC,OAAO,EACH,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI;4BACtB,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC;4BAC/B,GAAG;qBACV;oBACD,aAAa,EAAE;wBACX,IAAI,EAAE,SAAS,CAAC,aAAa,CAAC,IAAI;wBAClC,KAAK,EAAE,SAAS,CAAC,aAAa,CAAC,KAAK;wBACpC,OAAO,EACH,CAAC,SAAS,CAAC,aAAa,CAAC,IAAI;4BACzB,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC;4BAClC,GAAG;qBACV;iBACJ,CAAC;YACN,CAAC;QACL,CAAC;QAED,OAAO,OAAO,CAAC;IACnB,CAAC;IAKD,eAAe;QACX,IAAI,CAAC;YACD,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtB,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;gBACjC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAChC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;QACxD,CAAC;IACL,CAAC;CACJ;AAGD,MAAM,CAAC,MAAM,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC"}

package/dist/utils/stream_converter.d.ts

@@ -0,0 +1,19 @@
+import type { EnsembleStreamEvent, ResponseInput, ResponseThinkingMessage, ResponseOutputMessage, ToolCall } from '../types.js';
+export interface ConversionOptions {
+    onThinking?: (message: ResponseThinkingMessage) => void | Promise<void>;
+    onResponse?: (message: ResponseOutputMessage) => void | Promise<void>;
+    processToolCall?: (toolCalls: ToolCall[]) => Promise<Record<string, any>>;
+    model?: string;
+}
+export interface ConversionResult {
+    messages: ResponseInput;
+    fullResponse: string;
+    toolCalls: ToolCall[];
+}
+export declare function convertStreamToMessages(stream: AsyncGenerator<EnsembleStreamEvent>, initialMessages?: ResponseInput, options?: ConversionOptions): Promise<ConversionResult>;
+export declare function chainRequests(requests: Array<{
+    model: string;
+    systemPrompt?: string;
+    options?: ConversionOptions;
+}>, initialMessages?: ResponseInput): Promise<ConversionResult>;
+//# sourceMappingURL=stream_converter.d.ts.map

package/dist/utils/stream_converter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream_converter.d.ts","sourceRoot":"","sources":["../../utils/stream_converter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACR,mBAAmB,EACnB,aAAa,EACb,uBAAuB,EACvB,qBAAqB,EAGrB,QAAQ,EACX,MAAM,aAAa,CAAC;AAErB,MAAM,WAAW,iBAAiB;IAE9B,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,uBAAuB,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAGxE,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,qBAAqB,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAGtE,eAAe,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,EAAE,KAAK,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,CAAC;IAG1E,KAAK,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,gBAAgB;IAC7B,QAAQ,EAAE,aAAa,CAAC;IACxB,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,QAAQ,EAAE,CAAC;CACzB;AAOD,wBAAsB,uBAAuB,CACzC,MAAM,EAAE,cAAc,CAAC,mBAAmB,CAAC,EAC3C,eAAe,GAAE,aAAkB,EACnC,OAAO,GAAE,iBAAsB,GAChC,OAAO,CAAC,gBAAgB,CAAC,CA0L3B;AAMD,wBAAsB,aAAa,CAC/B,QAAQ,EAAE,KAAK,CAAC;IACZ,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,iBAAiB,CAAC;CAC/B,CAAC,EACF,eAAe,GAAE,aAAkB,GACpC,OAAO,CAAC,gBAAgB,CAAC,CAqC3B"}