@blackbox_ai/blackbox-cli-core 0.0.9 → 0.8.1

package/dist/src/utils/git-worktree-utils.d.ts

@@ -0,0 +1,56 @@
+/**
+* @license
+* Copyright 2025 Blackbox
+* SPDX-License-Identifier: Apache-2.0
+*/
+export interface WorktreeInfo {
+    name: string;
+    path: string;
+    branch: string;
+}
+export declare class GitWorktreeManager {
+    private worktrees;
+    private baseDir;
+    private tempDir;
+    constructor(baseDir: string);
+    /**
+     * Check if we're in a git repository
+     */
+    isGitRepository(): Promise<boolean>;
+    /**
+     * Get the current branch name
+     */
+    getCurrentBranch(): string;
+    /**
+     * Check if there are uncommitted changes
+     */
+    hasUncommittedChanges(): boolean;
+    /**
+     * Create a worktree for an agent
+     */
+    createWorktree(agentName: string): Promise<WorktreeInfo>;
+    /**
+     * Remove a worktree
+     */
+    removeWorktree(agentName: string): Promise<void>;
+    /**
+     * Remove all worktrees
+     */
+    cleanup(): Promise<void>;
+    /**
+     * Get worktree info for an agent
+     */
+    getWorktree(agentName: string): WorktreeInfo | undefined;
+    /**
+     * Get all worktrees
+     */
+    getAllWorktrees(): WorktreeInfo[];
+    /**
+     * Check if gh CLI is available
+     */
+    isGitHubCliAvailable(): Promise<boolean>;
+    /**
+     * Create a pull request using gh CLI
+     */
+    createPullRequest(branchName: string, title: string, body: string): Promise<string>;
+}

package/dist/src/utils/git-worktree-utils.js

@@ -0,0 +1,176 @@
+/**
+* @license
+* Copyright 2025 Blackbox
+* SPDX-License-Identifier: Apache-2.0
+*/
+import { execSync } from 'node:child_process';
+import path from 'node:path';
+import os from 'node:os';
+import fs from 'node:fs';
+export class GitWorktreeManager {
+    worktrees = new Map();
+    baseDir;
+    tempDir;
+    constructor(baseDir) {
+        this.baseDir = baseDir;
+        this.tempDir = path.join(os.tmpdir(), 'blackbox-multi-agent');
+    }
+    /**
+     * Check if we're in a git repository
+     */
+    async isGitRepository() {
+        try {
+            execSync('git rev-parse --git-dir', {
+                cwd: this.baseDir,
+                stdio: 'pipe',
+            });
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+    /**
+     * Get the current branch name
+     */
+    getCurrentBranch() {
+        try {
+            return execSync('git rev-parse --abbrev-ref HEAD', {
+                cwd: this.baseDir,
+                encoding: 'utf-8',
+                stdio: 'pipe',
+            }).trim();
+        }
+        catch (error) {
+            throw new Error(`Failed to get current branch: ${error}`);
+        }
+    }
+    /**
+     * Check if there are uncommitted changes
+     */
+    hasUncommittedChanges() {
+        try {
+            const status = execSync('git status --porcelain', {
+                cwd: this.baseDir,
+                encoding: 'utf-8',
+                stdio: 'pipe',
+            });
+            return status.trim().length > 0;
+        }
+        catch {
+            return false;
+        }
+    }
+    /**
+     * Create a worktree for an agent
+     */
+    async createWorktree(agentName) {
+        const timestamp = Date.now();
+        const branchName = `blackboxai/multi-agent-${agentName}-${timestamp}`;
+        const worktreePath = path.join(this.tempDir, `${agentName}-${timestamp}`);
+        try {
+            // Ensure temp directory exists
+            if (!fs.existsSync(this.tempDir)) {
+                fs.mkdirSync(this.tempDir, { recursive: true });
+            }
+            // Create new branch from current HEAD
+            execSync(`git branch ${branchName}`, {
+                cwd: this.baseDir,
+                stdio: 'pipe',
+            });
+            // Create worktree
+            execSync(`git worktree add "${worktreePath}" ${branchName}`, {
+                cwd: this.baseDir,
+                stdio: 'pipe',
+            });
+            const worktreeInfo = {
+                name: agentName,
+                path: worktreePath,
+                branch: branchName,
+            };
+            this.worktrees.set(agentName, worktreeInfo);
+            return worktreeInfo;
+        }
+        catch (error) {
+            throw new Error(`Failed to create worktree for ${agentName}: ${error}`);
+        }
+    }
+    /**
+     * Remove a worktree
+     */
+    async removeWorktree(agentName) {
+        const worktree = this.worktrees.get(agentName);
+        if (!worktree) {
+            return;
+        }
+        try {
+            execSync(`git worktree remove "${worktree.path}" --force`, {
+                cwd: this.baseDir,
+                stdio: 'pipe',
+            });
+            this.worktrees.delete(agentName);
+        }
+        catch (error) {
+            console.error(`Failed to remove worktree for ${agentName}:`, error);
+        }
+    }
+    /**
+     * Remove all worktrees
+     */
+    async cleanup() {
+        const agentNames = Array.from(this.worktrees.keys());
+        for (const agentName of agentNames) {
+            await this.removeWorktree(agentName);
+        }
+        // Clean up temp directory
+        try {
+            if (fs.existsSync(this.tempDir)) {
+                fs.rmSync(this.tempDir, { recursive: true, force: true });
+            }
+        }
+        catch (error) {
+            console.error('Failed to clean up temp directory:', error);
+        }
+    }
+    /**
+     * Get worktree info for an agent
+     */
+    getWorktree(agentName) {
+        return this.worktrees.get(agentName);
+    }
+    /**
+     * Get all worktrees
+     */
+    getAllWorktrees() {
+        return Array.from(this.worktrees.values());
+    }
+    /**
+     * Check if gh CLI is available
+     */
+    async isGitHubCliAvailable() {
+        try {
+            execSync('gh --version', { stdio: 'pipe' });
+            return true;
+        }
+        catch {
+            return false;
+        }
+    }
+    /**
+     * Create a pull request using gh CLI
+     */
+    async createPullRequest(branchName, title, body) {
+        try {
+            const result = execSync(`gh pr create --head ${branchName} --title "${title}" --body "${body}"`, {
+                cwd: this.baseDir,
+                encoding: 'utf-8',
+                stdio: 'pipe',
+            });
+            return result.trim();
+        }
+        catch (error) {
+            throw new Error(`Failed to create PR: ${error}`);
+        }
+    }
+}
+//# sourceMappingURL=git-worktree-utils.js.map

package/dist/src/utils/git-worktree-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"git-worktree-utils.js","sourceRoot":"","sources":["../../../src/utils/git-worktree-utils.ts"],"names":[],"mappings":"AAAC;;;;EAIE;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC9C,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,MAAM,SAAS,CAAC;AAQzB,MAAM,OAAO,kBAAkB;IACrB,SAAS,GAA8B,IAAI,GAAG,EAAE,CAAC;IACjD,OAAO,CAAS;IAChB,OAAO,CAAS;IAExB,YAAY,OAAe;QACzB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,sBAAsB,CAAC,CAAC;IAChE,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe;QACnB,IAAI,CAAC;YACH,QAAQ,CAAC,yBAAyB,EAAE;gBAClC,GAAG,EAAE,IAAI,CAAC,OAAO;gBACjB,KAAK,EAAE,MAAM;aACd,CAAC,CAAC;YACH,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,IAAI,CAAC;YACH,OAAO,QAAQ,CAAC,iCAAiC,EAAE;gBACjD,GAAG,EAAE,IAAI,CAAC,OAAO;gBACjB,QAAQ,EAAE,OAAO;gBACjB,KAAK,EAAE,MAAM;aACd,CAAC,CAAC,IAAI,EAAE,CAAC;QACZ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,iCAAiC,KAAK,EAAE,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAED;;OAEG;IACH,qBAAqB;QACnB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,QAAQ,CAAC,wBAAwB,EAAE;gBAChD,GAAG,EAAE,IAAI,CAAC,OAAO;gBACjB,QAAQ,EAAE,OAAO;gBACjB,KAAK,EAAE,MAAM;aACd,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;QAClC,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,SAAiB;QACpC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,MAAM,UAAU,GAAG,0BAA0B,SAAS,IAAI,SAAS,EAAE,CAAC;QACtE,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,SAAS,IAAI,SAAS,EAAE,CAAC,CAAC;QAE1E,IAAI,CAAC;YACH,+BAA+B;YAC/B,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;gBACjC,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAClD,CAAC;YAED,sCAAsC;YACtC,QAAQ,CAAC,cAAc,UAAU,EAAE,EAAE;gBACnC,GAAG,EAAE,IAAI,CAAC,OAAO;gBACjB,KAAK,EAAE,MAAM;aACd,CAAC,CAAC;YAEH,kBAAkB;YAClB,QAAQ,CAAC,qBAAqB,YAAY,KAAK,UAAU,EAAE,EAAE;gBAC3D,GAAG,EAAE,IAAI,CAAC,OAAO;gBACjB,KAAK,EAAE,MAAM;aACd,CAAC,CAAC;YAEH,MAAM,YAAY,GAAiB;gBACjC,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,UAAU;aACnB,CAAC;YAEF,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;YAC5C,OAAO,YAAY,CAAC;QACtB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CACb,iCAAiC,SAAS,KAAK,KAAK,EAAE,CACvD,CAAC;QACJ,CAAC;IACH,CAAC;IAGD;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,SAAiB;QACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC/C,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,QAAQ,CAAC,wBAAwB,QAAQ,CAAC,IAAI,WAAW,EAAE;gBACzD,GAAG,EAAE,IAAI,CAAC,OAAO;gBACjB,KAAK,EAAE,MAAM;aACd,CAAC,CAAC;YACH,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,iCAAiC,SAAS,GAAG,EAAE,KAAK,CAAC,CAAC;QACtE,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO;QACX,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QACrD,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,MAAM,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,CAAC;QACvC,CAAC;QAED,0BAA0B;QAC1B,IAAI,CAAC;YACH,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;gBAChC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5D,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;QAC7D,CAAC;IACH,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,SAAiB;QAC3B,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,eAAe;QACb,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,oBAAoB;QACxB,IAAI,CAAC;YACH,QAAQ,CAAC,cAAc,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5C,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,iBAAiB,CACrB,UAAkB,EAClB,KAAa,EACb,IAAY;QAEZ,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,QAAQ,CACrB,uBAAuB,UAAU,aAAa,KAAK,aAAa,IAAI,GAAG,EACvE;gBACE,GAAG,EAAE,IAAI,CAAC,OAAO;gBACjB,QAAQ,EAAE,OAAO;gBACjB,KAAK,EAAE,MAAM;aACd,CACF,CAAC;YACF,OAAO,MAAM,CAAC,IAAI,EAAE,CAAC;QACvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,wBAAwB,KAAK,EAAE,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;CACF"}

package/dist/src/utils/imageCompression.d.ts

@@ -0,0 +1,34 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * Result of image compression operation
+ */
+export interface CompressionResult {
+    buffer: Buffer;
+    mimeType: string;
+    originalSize: number;
+    compressedSize: number;
+    compressionRatio: number;
+    wasResized: boolean;
+    finalQuality?: number;
+}
+/**
+ * Compresses an image buffer to ensure it stays under the size limit.
+ * Converts to JPEG format and applies quality reduction and/or resizing as needed.
+ *
+ * @param inputBuffer The original image buffer
+ * @param originalMimeType The original MIME type of the image
+ * @returns Promise resolving to CompressionResult with compressed image data
+ */
+export declare function compressImage(inputBuffer: Buffer, originalMimeType: string): Promise<CompressionResult>;
+/**
+ * Checks if an image should be compressed based on its size and type.
+ *
+ * @param buffer The image buffer
+ * @param mimeType The MIME type of the image
+ * @returns true if the image should be compressed
+ */
+export declare function shouldCompressImage(buffer: Buffer, mimeType: string): boolean;

package/dist/src/utils/imageCompression.js

@@ -0,0 +1,170 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import sharp from 'sharp';
+/**
+ * Maximum size for compressed images in bytes (4.5MB to stay safely under 5MB limit)
+ * This accounts for base64 encoding overhead (~33% increase)
+ */
+const MAX_COMPRESSED_SIZE_BYTES = 4.5 * 1024 * 1024;
+/**
+ * Maximum dimension (width or height) for images
+ */
+const MAX_IMAGE_DIMENSION = 2048;
+/**
+ * Initial JPEG quality for compression
+ */
+const INITIAL_JPEG_QUALITY = 85;
+/**
+ * Minimum JPEG quality to try before giving up
+ */
+const MIN_JPEG_QUALITY = 60;
+/**
+ * Quality reduction step when image is still too large
+ */
+const QUALITY_STEP = 5;
+/**
+ * Compresses an image buffer to ensure it stays under the size limit.
+ * Converts to JPEG format and applies quality reduction and/or resizing as needed.
+ *
+ * @param inputBuffer The original image buffer
+ * @param originalMimeType The original MIME type of the image
+ * @returns Promise resolving to CompressionResult with compressed image data
+ */
+export async function compressImage(inputBuffer, originalMimeType) {
+    const originalSize = inputBuffer.length;
+    try {
+        // Initialize sharp instance
+        let image = sharp(inputBuffer);
+        const metadata = await image.metadata();
+        // Check if resizing is needed based on dimensions
+        let wasResized = false;
+        if (metadata.width && metadata.height) {
+            const maxDimension = Math.max(metadata.width, metadata.height);
+            if (maxDimension > MAX_IMAGE_DIMENSION) {
+                wasResized = true;
+                // Resize maintaining aspect ratio
+                image = image.resize(MAX_IMAGE_DIMENSION, MAX_IMAGE_DIMENSION, {
+                    fit: 'inside',
+                    withoutEnlargement: true,
+                });
+            }
+        }
+        // Try different quality levels until we get under the size limit
+        let quality = INITIAL_JPEG_QUALITY;
+        let compressedBuffer;
+        let finalQuality = quality;
+        while (quality >= MIN_JPEG_QUALITY) {
+            // Convert to JPEG with current quality
+            compressedBuffer = await image
+                .jpeg({ quality, mozjpeg: true }) // mozjpeg provides better compression
+                .toBuffer();
+            finalQuality = quality;
+            // Check if size is acceptable
+            if (compressedBuffer.length <= MAX_COMPRESSED_SIZE_BYTES) {
+                const compressionRatio = (1 - compressedBuffer.length / originalSize) * 100;
+                console.log(`[Image Compression] Original: ${(originalSize / 1024 / 1024).toFixed(2)}MB, ` +
+                    `Compressed: ${(compressedBuffer.length / 1024 / 1024).toFixed(2)}MB, ` +
+                    `Ratio: ${compressionRatio.toFixed(1)}%, Quality: ${quality}%, ` +
+                    `Resized: ${wasResized}`);
+                return {
+                    buffer: compressedBuffer,
+                    mimeType: 'image/jpeg',
+                    originalSize,
+                    compressedSize: compressedBuffer.length,
+                    compressionRatio,
+                    wasResized,
+                    finalQuality,
+                };
+            }
+            // Reduce quality and try again
+            quality -= QUALITY_STEP;
+        }
+        // If we still can't get under the limit, try more aggressive resizing
+        if (metadata.width && metadata.height) {
+            console.warn(`[Image Compression] Could not compress to under ${(MAX_COMPRESSED_SIZE_BYTES / 1024 / 1024).toFixed(1)}MB with quality reduction. Trying aggressive resize...`);
+            // Try progressively smaller sizes
+            const sizes = [1600, 1280, 1024, 800];
+            for (const size of sizes) {
+                image = sharp(inputBuffer).resize(size, size, {
+                    fit: 'inside',
+                    withoutEnlargement: true,
+                });
+                compressedBuffer = await image
+                    .jpeg({ quality: MIN_JPEG_QUALITY, mozjpeg: true })
+                    .toBuffer();
+                if (compressedBuffer.length <= MAX_COMPRESSED_SIZE_BYTES) {
+                    const compressionRatio = (1 - compressedBuffer.length / originalSize) * 100;
+                    console.log(`[Image Compression] Aggressive resize successful. ` +
+                        `Original: ${(originalSize / 1024 / 1024).toFixed(2)}MB, ` +
+                        `Compressed: ${(compressedBuffer.length / 1024 / 1024).toFixed(2)}MB, ` +
+                        `Ratio: ${compressionRatio.toFixed(1)}%, Max dimension: ${size}px`);
+                    return {
+                        buffer: compressedBuffer,
+                        mimeType: 'image/jpeg',
+                        originalSize,
+                        compressedSize: compressedBuffer.length,
+                        compressionRatio,
+                        wasResized: true,
+                        finalQuality: MIN_JPEG_QUALITY,
+                    };
+                }
+            }
+        }
+        // Last resort: return the best we could do and warn
+        compressedBuffer = await sharp(inputBuffer)
+            .resize(800, 800, { fit: 'inside', withoutEnlargement: true })
+            .jpeg({ quality: MIN_JPEG_QUALITY, mozjpeg: true })
+            .toBuffer();
+        console.warn(`[Image Compression] Warning: Could not compress image to under ${(MAX_COMPRESSED_SIZE_BYTES / 1024 / 1024).toFixed(1)}MB. ` +
+            `Final size: ${(compressedBuffer.length / 1024 / 1024).toFixed(2)}MB. ` +
+            `This may exceed API limits.`);
+        return {
+            buffer: compressedBuffer,
+            mimeType: 'image/jpeg',
+            originalSize,
+            compressedSize: compressedBuffer.length,
+            compressionRatio: (1 - compressedBuffer.length / originalSize) * 100,
+            wasResized: true,
+            finalQuality: MIN_JPEG_QUALITY,
+        };
+    }
+    catch (error) {
+        // If compression fails, log error and return original
+        console.error(`[Image Compression] Failed to compress image: ${error instanceof Error ? error.message : String(error)}`);
+        // Return original buffer as fallback
+        return {
+            buffer: inputBuffer,
+            mimeType: originalMimeType,
+            originalSize,
+            compressedSize: originalSize,
+            compressionRatio: 0,
+            wasResized: false,
+        };
+    }
+}
+/**
+ * Checks if an image should be compressed based on its size and type.
+ *
+ * @param buffer The image buffer
+ * @param mimeType The MIME type of the image
+ * @returns true if the image should be compressed
+ */
+export function shouldCompressImage(buffer, mimeType) {
+    // Always compress if over 3MB to be safe
+    if (buffer.length > 3 * 1024 * 1024) {
+        return true;
+    }
+    // Compress PNG files over 1MB (they often compress well to JPEG)
+    if (mimeType === 'image/png' && buffer.length > 1 * 1024 * 1024) {
+        return true;
+    }
+    // Don't compress small images or GIFs (to preserve animation)
+    if (mimeType === 'image/gif') {
+        return false;
+    }
+    return false;
+}
+//# sourceMappingURL=imageCompression.js.map

package/dist/src/utils/imageCompression.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"imageCompression.js","sourceRoot":"","sources":["../../../src/utils/imageCompression.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B;;;GAGG;AACH,MAAM,yBAAyB,GAAG,GAAG,GAAG,IAAI,GAAG,IAAI,CAAC;AAEpD;;GAEG;AACH,MAAM,mBAAmB,GAAG,IAAI,CAAC;AAEjC;;GAEG;AACH,MAAM,oBAAoB,GAAG,EAAE,CAAC;AAEhC;;GAEG;AACH,MAAM,gBAAgB,GAAG,EAAE,CAAC;AAE5B;;GAEG;AACH,MAAM,YAAY,GAAG,CAAC,CAAC;AAevB;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,WAAmB,EACnB,gBAAwB;IAExB,MAAM,YAAY,GAAG,WAAW,CAAC,MAAM,CAAC;IAExC,IAAI,CAAC;QACH,4BAA4B;QAC5B,IAAI,KAAK,GAAG,KAAK,CAAC,WAAW,CAAC,CAAC;QAC/B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,EAAE,CAAC;QAExC,kDAAkD;QAClD,IAAI,UAAU,GAAG,KAAK,CAAC;QACvB,IAAI,QAAQ,CAAC,KAAK,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACtC,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC/D,IAAI,YAAY,GAAG,mBAAmB,EAAE,CAAC;gBACvC,UAAU,GAAG,IAAI,CAAC;gBAClB,kCAAkC;gBAClC,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,mBAAmB,EAAE,mBAAmB,EAAE;oBAC7D,GAAG,EAAE,QAAQ;oBACb,kBAAkB,EAAE,IAAI;iBACzB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,iEAAiE;QACjE,IAAI,OAAO,GAAG,oBAAoB,CAAC;QACnC,IAAI,gBAAwB,CAAC;QAC7B,IAAI,YAAY,GAAG,OAAO,CAAC;QAE3B,OAAO,OAAO,IAAI,gBAAgB,EAAE,CAAC;YACnC,uCAAuC;YACvC,gBAAgB,GAAG,MAAM,KAAK;iBAC3B,IAAI,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,sCAAsC;iBACvE,QAAQ,EAAE,CAAC;YAEd,YAAY,GAAG,OAAO,CAAC;YAEvB,8BAA8B;YAC9B,IAAI,gBAAgB,CAAC,MAAM,IAAI,yBAAyB,EAAE,CAAC;gBACzD,MAAM,gBAAgB,GAAG,CAAC,CAAC,GAAG,gBAAgB,CAAC,MAAM,GAAG,YAAY,CAAC,GAAG,GAAG,CAAC;gBAE5E,OAAO,CAAC,GAAG,CACT,iCAAiC,CAAC,YAAY,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;oBAC9E,eAAe,CAAC,gBAAgB,CAAC,MAAM,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;oBACvE,UAAU,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,OAAO,KAAK;oBAChE,YAAY,UAAU,EAAE,CACzB,CAAC;gBAEF,OAAO;oBACL,MAAM,EAAE,gBAAgB;oBACxB,QAAQ,EAAE,YAAY;oBACtB,YAAY;oBACZ,cAAc,EAAE,gBAAgB,CAAC,MAAM;oBACvC,gBAAgB;oBAChB,UAAU;oBACV,YAAY;iBACb,CAAC;YACJ,CAAC;YAED,+BAA+B;YAC/B,OAAO,IAAI,YAAY,CAAC;QAC1B,CAAC;QAED,sEAAsE;QACtE,IAAI,QAAQ,CAAC,KAAK,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;YACtC,OAAO,CAAC,IAAI,CACV,mDAAmD,CAAC,yBAAyB,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,wDAAwD,CAChK,CAAC;YAEF,kCAAkC;YAClC,MAAM,KAAK,GAAG,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;YACtC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,KAAK,GAAG,KAAK,CAAC,WAAW,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE;oBAC5C,GAAG,EAAE,QAAQ;oBACb,kBAAkB,EAAE,IAAI;iBACzB,CAAC,CAAC;gBAEH,gBAAgB,GAAG,MAAM,KAAK;qBAC3B,IAAI,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;qBAClD,QAAQ,EAAE,CAAC;gBAEd,IAAI,gBAAgB,CAAC,MAAM,IAAI,yBAAyB,EAAE,CAAC;oBACzD,MAAM,gBAAgB,GAAG,CAAC,CAAC,GAAG,gBAAgB,CAAC,MAAM,GAAG,YAAY,CAAC,GAAG,GAAG,CAAC;oBAE5E,OAAO,CAAC,GAAG,CACT,oDAAoD;wBACpD,aAAa,CAAC,YAAY,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;wBAC1D,eAAe,CAAC,gBAAgB,CAAC,MAAM,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;wBACvE,UAAU,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC,qBAAqB,IAAI,IAAI,CACnE,CAAC;oBAEF,OAAO;wBACL,MAAM,EAAE,gBAAgB;wBACxB,QAAQ,EAAE,YAAY;wBACtB,YAAY;wBACZ,cAAc,EAAE,gBAAgB,CAAC,MAAM;wBACvC,gBAAgB;wBAChB,UAAU,EAAE,IAAI;wBAChB,YAAY,EAAE,gBAAgB;qBAC/B,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAED,oDAAoD;QACpD,gBAAgB,GAAG,MAAM,KAAK,CAAC,WAAW,CAAC;aACxC,MAAM,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,GAAG,EAAE,QAAQ,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC;aAC7D,IAAI,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;aAClD,QAAQ,EAAE,CAAC;QAEd,OAAO,CAAC,IAAI,CACV,kEAAkE,CAAC,yBAAyB,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;YAC5H,eAAe,CAAC,gBAAgB,CAAC,MAAM,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;YACvE,6BAA6B,CAC9B,CAAC;QAEF,OAAO;YACL,MAAM,EAAE,gBAAgB;YACxB,QAAQ,EAAE,YAAY;YACtB,YAAY;YACZ,cAAc,EAAE,gBAAgB,CAAC,MAAM;YACvC,gBAAgB,EAAE,CAAC,CAAC,GAAG,gBAAgB,CAAC,MAAM,GAAG,YAAY,CAAC,GAAG,GAAG;YACpE,UAAU,EAAE,IAAI;YAChB,YAAY,EAAE,gBAAgB;SAC/B,CAAC;IAEJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,sDAAsD;QACtD,OAAO,CAAC,KAAK,CACX,iDAAiD,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAC1G,CAAC;QAEF,qCAAqC;QACrC,OAAO;YACL,MAAM,EAAE,WAAW;YACnB,QAAQ,EAAE,gBAAgB;YAC1B,YAAY;YACZ,cAAc,EAAE,YAAY;YAC5B,gBAAgB,EAAE,CAAC;YACnB,UAAU,EAAE,KAAK;SAClB,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,mBAAmB,CAAC,MAAc,EAAE,QAAgB;IAClE,yCAAyC;IACzC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,EAAE,CAAC;QACpC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,iEAAiE;IACjE,IAAI,QAAQ,KAAK,WAAW,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,EAAE,CAAC;QAChE,OAAO,IAAI,CAAC;IACd,CAAC;IAED,8DAA8D;IAC9D,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;QAC7B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/src/utils/messageTruncator.d.ts

@@ -0,0 +1,51 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+interface Message {
+    role: string;
+    content: string | ContentBlock[];
+    tool_calls?: ToolCall[];
+    [key: string]: unknown;
+}
+interface ContentBlock {
+    type: string;
+    text?: string;
+    [key: string]: unknown;
+}
+interface ToolCall {
+    function: {
+        arguments: string;
+        name?: string;
+    };
+    [key: string]: unknown;
+}
+export declare class MessageTruncator {
+    private debugMode;
+    constructor(debugMode?: boolean);
+    /**
+     * Helper to get the tokenizer for a model.
+     * Handles generic fallback if the model string isn't an exact match.
+     */
+    private getTokenizer;
+    /**
+     * Helper to count tokens for text or a list of messages.
+     * This mimics the self.count_tokens behavior.
+     */
+    private countTokens;
+    /**
+     * Returns max token limits for models that may not be available in LiteLLM's get_model_info().
+     * Based on the comprehensive model mapping from the Python implementation.
+     */
+    private getModelMaxTokens;
+    /**
+     * Smartly truncates message content by targeting longer texts first.
+     */
+    private truncateMessageContent;
+    /**
+     * Main truncation method
+     */
+    truncateLongMessages(model: string, messages: Message[], maxOutputTokens?: number, systemPrompt?: string | null): Message[];
+}
+export {};