@blackbox_ai/blackbox-cli-core 0.0.4

package/dist/src/core/tokenLimits.js

@@ -0,0 +1,175 @@
+export const DEFAULT_TOKEN_LIMIT = 131_072; // 128K (power-of-two)
+export const DEFAULT_OUTPUT_TOKEN_LIMIT = 4_096; // 4K tokens
+/**
+ * Accurate numeric limits:
+ * - power-of-two approximations (128K -> 131072, 256K -> 262144, etc.)
+ * - vendor-declared exact values (e.g., 200k -> 200000) are used as stated in docs.
+ */
+const LIMITS = {
+    '32k': 32_768,
+    '64k': 65_536,
+    '128k': 131_072,
+    '200k': 200_000, // vendor-declared decimal (OpenAI / Anthropic use 200k)
+    '256k': 262_144,
+    '512k': 524_288,
+    '1m': 1_048_576,
+    '2m': 2_097_152,
+    '10m': 10_485_760, // 10 million tokens
+    // Output token limits (typically much smaller than input limits)
+    '4k': 4_096,
+    '8k': 8_192,
+    '16k': 16_384,
+};
+/** Robust normalizer: strips provider prefixes, pipes/colons, date/version suffixes, etc. */
+export function normalize(model) {
+    let s = (model ?? '').toLowerCase().trim();
+    // keep final path segment (strip provider prefixes), handle pipe/colon
+    s = s.replace(/^.*\//, '');
+    s = s.split('|').pop() ?? s;
+    s = s.split(':').pop() ?? s;
+    // collapse whitespace to single hyphen
+    s = s.replace(/\s+/g, '-');
+    // remove trailing build / date / revision suffixes:
+    // - dates (e.g., -20250219), -v1, version numbers, 'latest', 'preview' etc.
+    s = s.replace(/-preview/g, '');
+    // Special handling for Blackbox model names that include "-latest" as part of the model name
+    if (!s.match(/^blackbox-(?:plus|flash|vl-max)-latest$/)) {
+        // \d{6,} - Match 6 or more digits (dates) like -20250219 (6+ digit dates)
+        // \d+x\d+b - Match patterns like 4x8b, -7b, -70b
+        // v\d+(?:\.\d+)* - Match version patterns starting with 'v' like -v1, -v1.2, -v2.1.3
+        // -\d+(?:\.\d+)+ - Match version numbers with dots (that are preceded by a dash),
+        //   like -1.1, -2.0.1 but only when they're suffixes, Example: model-test-1.1 → model-test;
+        //   Note: this does NOT match 4.1 in gpt-4.1 because there's no dash before 4.1 in that context.
+        // latest - Match the literal string "latest"
+        s = s.replace(/-(?:\d{6,}|\d+x\d+b|v\d+(?:\.\d+)*|-\d+(?:\.\d+)+|latest)$/g, '');
+    }
+    // remove quantization / numeric / precision suffixes common in local/community models
+    s = s.replace(/-(?:\d?bit|int[48]|bf16|fp16|q[45]|quantized)$/g, '');
+    return s;
+}
+/** Ordered regex patterns: most specific -> most general (first match wins). */
+const PATTERNS = [
+    // -------------------
+    // Google Gemini
+    // -------------------
+    [/^gemini-1\.5-pro$/, LIMITS['2m']],
+    [/^gemini-1\.5-flash$/, LIMITS['1m']],
+    [/^gemini-2\.5-pro.*$/, LIMITS['1m']],
+    [/^gemini-2\.5-flash.*$/, LIMITS['1m']],
+    [/^gemini-2\.0-flash-image-generation$/, LIMITS['32k']],
+    [/^gemini-2\.0-flash.*$/, LIMITS['1m']],
+    // -------------------
+    // OpenAI (o3 / o4-mini / gpt-4.1 / gpt-4o family)
+    // o3 and o4-mini document a 200,000-token context window (decimal).
+    // Note: GPT-4.1 models typically report 1_048_576 (1M) context in OpenAI announcements.
+    [/^o3(?:-mini|$).*$/, LIMITS['200k']],
+    [/^o3.*$/, LIMITS['200k']],
+    [/^o4-mini.*$/, LIMITS['200k']],
+    [/^gpt-4\.1-mini.*$/, LIMITS['1m']],
+    [/^gpt-4\.1.*$/, LIMITS['1m']],
+    [/^gpt-4o-mini.*$/, LIMITS['128k']],
+    [/^gpt-4o.*$/, LIMITS['128k']],
+    [/^gpt-4.*$/, LIMITS['128k']],
+    // -------------------
+    // Anthropic Claude
+    // - Claude Sonnet / Sonnet 3.5 and related Sonnet variants: 200,000 tokens documented.
+    // - Some Sonnet/Opus models offer 1M in beta/enterprise tiers (handled separately if needed).
+    [/^claude-3\.5-sonnet.*$/, LIMITS['200k']],
+    [/^claude-3\.7-sonnet.*$/, LIMITS['1m']], // some Sonnet 3.7/Opus variants advertise 1M beta in docs
+    [/^claude-sonnet-4.*$/, LIMITS['1m']],
+    [/^claude-opus-4.*$/, LIMITS['1m']],
+    // -------------------
+    // Alibaba / Blackbox
+    // -------------------
+    // Commercial Blackbox3-Coder-Plus: 1M token context
+    [/^blackbox3-coder-plus(-.*)?$/, LIMITS['1m']], // catches "blackbox3-coder-plus" and date variants
+    // Commercial Blackbox3-Coder-Flash: 1M token context
+    [/^blackbox3-coder-flash(-.*)?$/, LIMITS['1m']], // catches "blackbox3-coder-flash" and date variants
+    // Generic coder-model: same as blackbox3-coder-plus (1M token context)
+    [/^coder-model$/, LIMITS['1m']],
+    // Commercial Blackbox3-Max-Preview: 256K token context
+    [/^blackbox3-max-preview(-.*)?$/, LIMITS['256k']], // catches "blackbox3-max-preview" and date variants
+    // Open-source Blackbox3-Coder variants: 256K native
+    [/^blackbox3-coder-.*$/, LIMITS['256k']],
+    // Open-source Blackbox3 2507 variants: 256K native
+    [/^blackbox3-.*-2507-.*$/, LIMITS['256k']],
+    // Open-source long-context Blackbox2.5-1M
+    [/^blackbox2\.5-1m.*$/, LIMITS['1m']],
+    // Standard Blackbox2.5: 128K
+    [/^blackbox2\.5.*$/, LIMITS['128k']],
+    // Studio commercial Blackbox-Plus / Blackbox-Flash / Blackbox-Turbo
+    [/^blackbox-plus-latest$/, LIMITS['1m']], // Commercial latest: 1M
+    [/^blackbox-plus.*$/, LIMITS['128k']], // Standard: 128K
+    [/^blackbox-flash-latest$/, LIMITS['1m']],
+    [/^blackbox-turbo.*$/, LIMITS['128k']],
+    // Blackbox Vision Models
+    [/^blackbox3-vl-plus$/, LIMITS['256k']], // Blackbox3-VL-Plus: 256K input
+    [/^blackbox-vl-max.*$/, LIMITS['128k']],
+    // Generic vision-model: same as blackbox-vl-max (128K token context)
+    [/^vision-model$/, LIMITS['128k']],
+    // -------------------
+    // ByteDance Seed-OSS (512K)
+    // -------------------
+    [/^seed-oss.*$/, LIMITS['512k']],
+    // -------------------
+    // Zhipu GLM
+    // -------------------
+    [/^glm-4\.5v.*$/, LIMITS['64k']],
+    [/^glm-4\.5-air.*$/, LIMITS['128k']],
+    [/^glm-4\.5.*$/, LIMITS['128k']],
+    // -------------------
+    // DeepSeek / GPT-OSS / Kimi / Llama & Mistral examples
+    // -------------------
+    [/^deepseek-r1.*$/, LIMITS['128k']],
+    [/^deepseek-v3(?:\.1)?.*$/, LIMITS['128k']],
+    [/^kimi-k2-instruct.*$/, LIMITS['128k']],
+    [/^gpt-oss.*$/, LIMITS['128k']],
+    [/^llama-4-scout.*$/, LIMITS['10m']], // ultra-long variants - handle carefully
+    [/^mistral-large-2.*$/, LIMITS['128k']],
+];
+/**
+ * Output token limit patterns for specific model families.
+ * These patterns define the maximum number of tokens that can be generated
+ * in a single response for specific models.
+ */
+const OUTPUT_PATTERNS = [
+    // -------------------
+    // Alibaba / Blackbox - DashScope Models
+    // -------------------
+    // Blackbox3-Coder-Plus: 65,536 max output tokens
+    [/^blackbox3-coder-plus(-.*)?$/, LIMITS['64k']],
+    // Generic coder-model: same as blackbox3-coder-plus (64K max output tokens)
+    [/^coder-model$/, LIMITS['64k']],
+    // Blackbox3-Max-Preview: 65,536 max output tokens
+    [/^blackbox3-max-preview(-.*)?$/, LIMITS['64k']],
+    // Blackbox-VL-Max-Latest: 8,192 max output tokens
+    [/^blackbox-vl-max-latest$/, LIMITS['8k']],
+    // Generic vision-model: same as blackbox-vl-max-latest (8K max output tokens)
+    [/^vision-model$/, LIMITS['8k']],
+    // Blackbox3-VL-Plus: 32K max output tokens
+    [/^blackbox3-vl-plus$/, LIMITS['32k']],
+];
+/**
+ * Return the token limit for a model string based on the specified type.
+ *
+ * This function determines the maximum number of tokens for either input context
+ * or output generation based on the model and token type. It uses the same
+ * normalization logic for consistency across both input and output limits.
+ *
+ * @param model - The model name to get the token limit for
+ * @param type - The type of token limit ('input' for context window, 'output' for generation)
+ * @returns The maximum number of tokens allowed for this model and type
+ */
+export function tokenLimit(model, type = 'input') {
+    const norm = normalize(model);
+    // Choose the appropriate patterns based on token type
+    const patterns = type === 'output' ? OUTPUT_PATTERNS : PATTERNS;
+    for (const [regex, limit] of patterns) {
+        if (regex.test(norm)) {
+            return limit;
+        }
+    }
+    // Return appropriate default based on token type
+    return type === 'output' ? DEFAULT_OUTPUT_TOKEN_LIMIT : DEFAULT_TOKEN_LIMIT;
+}
+//# sourceMappingURL=tokenLimits.js.map

package/dist/src/core/tokenLimits.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokenLimits.js","sourceRoot":"","sources":["../../../src/core/tokenLimits.ts"],"names":[],"mappings":"AAUA,MAAM,CAAC,MAAM,mBAAmB,GAAe,OAAO,CAAC,CAAC,sBAAsB;AAC9E,MAAM,CAAC,MAAM,0BAA0B,GAAe,KAAK,CAAC,CAAC,YAAY;AAEzE;;;;GAIG;AACH,MAAM,MAAM,GAAG;IACb,KAAK,EAAE,MAAM;IACb,KAAK,EAAE,MAAM;IACb,MAAM,EAAE,OAAO;IACf,MAAM,EAAE,OAAO,EAAE,wDAAwD;IACzE,MAAM,EAAE,OAAO;IACf,MAAM,EAAE,OAAO;IACf,IAAI,EAAE,SAAS;IACf,IAAI,EAAE,SAAS;IACf,KAAK,EAAE,UAAU,EAAE,oBAAoB;IACvC,iEAAiE;IACjE,IAAI,EAAE,KAAK;IACX,IAAI,EAAE,KAAK;IACX,KAAK,EAAE,MAAM;CACL,CAAC;AAEX,6FAA6F;AAC7F,MAAM,UAAU,SAAS,CAAC,KAAa;IACrC,IAAI,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;IAE3C,uEAAuE;IACvE,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC3B,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IAC5B,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IAE5B,uCAAuC;IACvC,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAE3B,oDAAoD;IACpD,4EAA4E;IAC5E,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;IAC/B,6FAA6F;IAC7F,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,yCAAyC,CAAC,EAAE,CAAC;QACxD,0EAA0E;QAC1E,iDAAiD;QACjD,qFAAqF;QACrF,kFAAkF;QAClF,4FAA4F;QAC5F,iGAAiG;QACjG,6CAA6C;QAC7C,CAAC,GAAG,CAAC,CAAC,OAAO,CACX,6DAA6D,EAC7D,EAAE,CACH,CAAC;IACJ,CAAC;IAED,sFAAsF;IACtF,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,iDAAiD,EAAE,EAAE,CAAC,CAAC;IAErE,OAAO,CAAC,CAAC;AACX,CAAC;AAED,gFAAgF;AAChF,MAAM,QAAQ,GAAgC;IAC5C,sBAAsB;IACtB,gBAAgB;IAChB,sBAAsB;IACtB,CAAC,mBAAmB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC,qBAAqB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC,qBAAqB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC,uBAAuB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC,sCAAsC,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;IACvD,CAAC,uBAAuB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IAEvC,sBAAsB;IACtB,kDAAkD;IAClD,oEAAoE;IACpE,wFAAwF;IACxF,CAAC,mBAAmB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC1B,CAAC,aAAa,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC/B,CAAC,mBAAmB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC,iBAAiB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC,YAAY,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC9B,CAAC,WAAW,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAE7B,sBAAsB;IACtB,mBAAmB;IACnB,uFAAuF;IACvF,8FAA8F;IAC9F,CAAC,wBAAwB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC1C,CAAC,wBAAwB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,EAAE,0DAA0D;IACpG,CAAC,qBAAqB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC,mBAAmB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IAEnC,sBAAsB;IACtB,qBAAqB;IACrB,sBAAsB;IACtB,oDAAoD;IACpD,CAAC,8BAA8B,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,EAAE,mDAAmD;IAEnG,qDAAqD;IACrD,CAAC,+BAA+B,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,EAAE,oDAAoD;IAErG,uEAAuE;IACvE,CAAC,eAAe,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IAE/B,uDAAuD;IACvD,CAAC,+BAA+B,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,oDAAoD;IAEvG,oDAAoD;IACpD,CAAC,sBAAsB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IACxC,mDAAmD;IACnD,CAAC,wBAAwB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAE1C,0CAA0C;IAC1C,CAAC,qBAAqB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IAErC,6BAA6B;IAC7B,CAAC,kBAAkB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAEpC,oEAAoE;IACpE,CAAC,wBAAwB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,EAAE,wBAAwB;IAClE,CAAC,mBAAmB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,iBAAiB;IACxD,CAAC,yBAAyB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IACzC,CAAC,oBAAoB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAEtC,yBAAyB;IACzB,CAAC,qBAAqB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,gCAAgC;IACzE,CAAC,qBAAqB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAEvC,qEAAqE;IACrE,CAAC,gBAAgB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAElC,sBAAsB;IACtB,4BAA4B;IAC5B,sBAAsB;IACtB,CAAC,cAAc,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAEhC,sBAAsB;IACtB,YAAY;IACZ,sBAAsB;IACtB,CAAC,eAAe,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC,kBAAkB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IACpC,CAAC,cAAc,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAEhC,sBAAsB;IACtB,uDAAuD;IACvD,sBAAsB;IACtB,CAAC,iBAAiB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC,yBAAyB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC3C,CAAC,sBAAsB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IACxC,CAAC,aAAa,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC/B,CAAC,mBAAmB,EAAE,MAAM,CAAC,KAAK,CAA0B,CAAC,EAAE,yCAAyC;IACxG,CAAC,qBAAqB,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;CACxC,CAAC;AAEF;;;;GAIG;AACH,MAAM,eAAe,GAAgC;IACnD,sBAAsB;IACtB,wCAAwC;IACxC,sBAAsB;IACtB,iDAAiD;IACjD,CAAC,8BAA8B,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;IAE/C,4EAA4E;IAC5E,CAAC,eAAe,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;IAEhC,kDAAkD;IAClD,CAAC,+BAA+B,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;IAEhD,kDAAkD;IAClD,CAAC,0BAA0B,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IAE1C,8EAA8E;IAC9E,CAAC,gBAAgB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IAEhC,2CAA2C;IAC3C,CAAC,qBAAqB,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;CACvC,CAAC;AAEF;;;;;;;;;;GAUG;AACH,MAAM,UAAU,UAAU,CACxB,KAAY,EACZ,OAAuB,OAAO;IAE9B,MAAM,IAAI,GAAG,SAAS,CAAC,KAAK,CAAC,CAAC;IAE9B,sDAAsD;IACtD,MAAM,QAAQ,GAAG,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,QAAQ,CAAC;IAEhE,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,QAAQ,EAAE,CAAC;QACtC,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACrB,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED,iDAAiD;IACjD,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,0BAA0B,CAAC,CAAC,CAAC,mBAAmB,CAAC;AAC9E,CAAC"}

package/dist/src/core/tokenLimits.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/src/core/tokenLimits.test.js

@@ -0,0 +1,261 @@
+import { describe, it, expect } from 'vitest';
+import { normalize, tokenLimit, DEFAULT_TOKEN_LIMIT, DEFAULT_OUTPUT_TOKEN_LIMIT, } from './tokenLimits.js';
+describe('normalize', () => {
+    it('should lowercase and trim the model string', () => {
+        expect(normalize('  GEMINI-1.5-PRO  ')).toBe('gemini-1.5-pro');
+    });
+    it('should strip provider prefixes', () => {
+        expect(normalize('google/gemini-1.5-pro')).toBe('gemini-1.5-pro');
+        expect(normalize('anthropic/claude-3.5-sonnet')).toBe('claude-3.5-sonnet');
+    });
+    it('should handle pipe and colon separators', () => {
+        expect(normalize('blackbox|blackbox2.5:blackbox2.5-1m')).toBe('blackbox2.5-1m');
+    });
+    it('should collapse whitespace to a single hyphen', () => {
+        expect(normalize('claude 3.5 sonnet')).toBe('claude-3.5-sonnet');
+    });
+    it('should remove date and version suffixes', () => {
+        expect(normalize('gemini-1.5-pro-20250219')).toBe('gemini-1.5-pro');
+        expect(normalize('gpt-4o-mini-v1')).toBe('gpt-4o-mini');
+        expect(normalize('claude-3.7-sonnet-20240715')).toBe('claude-3.7-sonnet');
+        expect(normalize('gpt-4.1-latest')).toBe('gpt-4.1');
+        expect(normalize('gemini-2.0-flash-preview-20250520')).toBe('gemini-2.0-flash');
+    });
+    it('should remove quantization and numeric suffixes', () => {
+        expect(normalize('blackbox3-coder-7b-4bit')).toBe('blackbox3-coder-7b');
+        expect(normalize('llama-4-scout-int8')).toBe('llama-4-scout');
+        expect(normalize('mistral-large-2-bf16')).toBe('mistral-large-2');
+        expect(normalize('deepseek-v3.1-q4')).toBe('deepseek-v3.1');
+        expect(normalize('blackbox2.5-quantized')).toBe('blackbox2.5');
+    });
+    it('should handle a combination of normalization rules', () => {
+        expect(normalize('  Google/GEMINI-2.5-PRO:gemini-2.5-pro-20250605  ')).toBe('gemini-2.5-pro');
+    });
+    it('should handle empty or null input', () => {
+        expect(normalize('')).toBe('');
+        expect(normalize(undefined)).toBe('');
+        expect(normalize(null)).toBe('');
+    });
+    it('should remove preview suffixes', () => {
+        expect(normalize('gemini-2.0-flash-preview')).toBe('gemini-2.0-flash');
+    });
+    it('should remove version numbers with dots when they are at the end', () => {
+        expect(normalize('gpt-4.1.1-latest')).toBe('gpt-4.1.1');
+        expect(normalize('gpt-4.1-latest')).toBe('gpt-4.1');
+    });
+});
+describe('tokenLimit', () => {
+    // Test cases for each model family
+    describe('Google Gemini', () => {
+        it('should return the correct limit for Gemini 1.5 Pro', () => {
+            expect(tokenLimit('gemini-1.5-pro')).toBe(2097152);
+        });
+        it('should return the correct limit for Gemini 1.5 Flash', () => {
+            expect(tokenLimit('gemini-1.5-flash')).toBe(1048576);
+        });
+        it('should return the correct limit for Gemini 2.5 Pro', () => {
+            expect(tokenLimit('gemini-2.5-pro')).toBe(1048576);
+        });
+        it('should return the correct limit for Gemini 2.5 Flash', () => {
+            expect(tokenLimit('gemini-2.5-flash')).toBe(1048576);
+        });
+        it('should return the correct limit for Gemini 2.0 Flash with image generation', () => {
+            expect(tokenLimit('gemini-2.0-flash-image-generation')).toBe(32768);
+        });
+        it('should return the correct limit for Gemini 2.0 Flash', () => {
+            expect(tokenLimit('gemini-2.0-flash')).toBe(1048576);
+        });
+    });
+    describe('OpenAI', () => {
+        it('should return the correct limit for o3-mini', () => {
+            expect(tokenLimit('o3-mini')).toBe(200000);
+        });
+        it('should return the correct limit for o3 models', () => {
+            expect(tokenLimit('o3')).toBe(200000);
+        });
+        it('should return the correct limit for o4-mini', () => {
+            expect(tokenLimit('o4-mini')).toBe(200000);
+        });
+        it('should return the correct limit for gpt-4o-mini', () => {
+            expect(tokenLimit('gpt-4o-mini')).toBe(131072);
+        });
+        it('should return the correct limit for gpt-4o', () => {
+            expect(tokenLimit('gpt-4o')).toBe(131072);
+        });
+        it('should return the correct limit for gpt-4.1-mini', () => {
+            expect(tokenLimit('gpt-4.1-mini')).toBe(1048576);
+        });
+        it('should return the correct limit for gpt-4.1 models', () => {
+            expect(tokenLimit('gpt-4.1')).toBe(1048576);
+        });
+        it('should return the correct limit for gpt-4', () => {
+            expect(tokenLimit('gpt-4')).toBe(131072);
+        });
+    });
+    describe('Anthropic Claude', () => {
+        it('should return the correct limit for Claude 3.5 Sonnet', () => {
+            expect(tokenLimit('claude-3.5-sonnet')).toBe(200000);
+        });
+        it('should return the correct limit for Claude 3.7 Sonnet', () => {
+            expect(tokenLimit('claude-3.7-sonnet')).toBe(1048576);
+        });
+        it('should return the correct limit for Claude Sonnet 4', () => {
+            expect(tokenLimit('claude-sonnet-4')).toBe(1048576);
+        });
+        it('should return the correct limit for Claude Opus 4', () => {
+            expect(tokenLimit('claude-opus-4')).toBe(1048576);
+        });
+    });
+    describe('Alibaba Blackbox', () => {
+        it('should return the correct limit for blackbox3-coder commercial models', () => {
+            expect(tokenLimit('blackbox3-coder-plus')).toBe(1048576);
+            expect(tokenLimit('blackbox3-coder-plus-20250601')).toBe(1048576);
+            expect(tokenLimit('blackbox3-coder-flash')).toBe(1048576);
+            expect(tokenLimit('blackbox3-coder-flash-20250601')).toBe(1048576);
+        });
+        it('should return the correct limit for blackbox3-coder open source models', () => {
+            expect(tokenLimit('blackbox3-coder-7b')).toBe(262144);
+            expect(tokenLimit('blackbox3-coder-480b-a35b-instruct')).toBe(262144);
+            expect(tokenLimit('blackbox3-coder-30b-a3b-instruct')).toBe(262144);
+        });
+        it('should return the correct limit for blackbox3 2507 variants', () => {
+            expect(tokenLimit('blackbox3-some-model-2507-instruct')).toBe(262144);
+        });
+        it('should return the correct limit for blackbox2.5-1m', () => {
+            expect(tokenLimit('blackbox2.5-1m')).toBe(1048576);
+            expect(tokenLimit('blackbox2.5-1m-instruct')).toBe(1048576);
+        });
+        it('should return the correct limit for blackbox2.5', () => {
+            expect(tokenLimit('blackbox2.5')).toBe(131072);
+            expect(tokenLimit('blackbox2.5-instruct')).toBe(131072);
+        });
+        it('should return the correct limit for blackbox-plus', () => {
+            expect(tokenLimit('blackbox-plus-latest')).toBe(1048576);
+            expect(tokenLimit('blackbox-plus')).toBe(131072);
+        });
+        it('should return the correct limit for blackbox-flash', () => {
+            expect(tokenLimit('blackbox-flash-latest')).toBe(1048576);
+        });
+        it('should return the correct limit for blackbox-turbo', () => {
+            expect(tokenLimit('blackbox-turbo')).toBe(131072);
+            expect(tokenLimit('blackbox-turbo-latest')).toBe(131072);
+        });
+    });
+    describe('ByteDance Seed-OSS', () => {
+        it('should return the correct limit for seed-oss', () => {
+            expect(tokenLimit('seed-oss')).toBe(524288);
+        });
+    });
+    describe('Zhipu GLM', () => {
+        it('should return the correct limit for glm-4.5v', () => {
+            expect(tokenLimit('glm-4.5v')).toBe(65536);
+        });
+        it('should return the correct limit for glm-4.5-air', () => {
+            expect(tokenLimit('glm-4.5-air')).toBe(131072);
+        });
+        it('should return the correct limit for glm-4.5', () => {
+            expect(tokenLimit('glm-4.5')).toBe(131072);
+        });
+    });
+    describe('Other models', () => {
+        it('should return the correct limit for deepseek-r1', () => {
+            expect(tokenLimit('deepseek-r1')).toBe(131072);
+        });
+        it('should return the correct limit for deepseek-v3', () => {
+            expect(tokenLimit('deepseek-v3')).toBe(131072);
+        });
+        it('should return the correct limit for deepseek-v3.1', () => {
+            expect(tokenLimit('deepseek-v3.1')).toBe(131072);
+        });
+        it('should return the correct limit for kimi-k2-instruct', () => {
+            expect(tokenLimit('kimi-k2-instruct')).toBe(131072);
+        });
+        it('should return the correct limit for gpt-oss', () => {
+            expect(tokenLimit('gpt-oss')).toBe(131072);
+        });
+        it('should return the correct limit for llama-4-scout', () => {
+            expect(tokenLimit('llama-4-scout')).toBe(10485760);
+        });
+        it('should return the correct limit for mistral-large-2', () => {
+            expect(tokenLimit('mistral-large-2')).toBe(131072);
+        });
+    });
+    // Test for default limit
+    it('should return the default token limit for an unknown model', () => {
+        expect(tokenLimit('unknown-model-v1.0')).toBe(DEFAULT_TOKEN_LIMIT);
+    });
+    // Test with complex model string
+    it('should return the correct limit for a complex model string', () => {
+        expect(tokenLimit('  a/b/c|GPT-4o:gpt-4o-2024-05-13-q4  ')).toBe(131072);
+    });
+    // Test case-insensitive matching
+    it('should handle case-insensitive model names', () => {
+        expect(tokenLimit('GPT-4O')).toBe(131072);
+        expect(tokenLimit('CLAUDE-3.5-SONNET')).toBe(200000);
+    });
+});
+describe('tokenLimit with output type', () => {
+    describe('Blackbox models with output limits', () => {
+        it('should return the correct output limit for blackbox3-coder-plus', () => {
+            expect(tokenLimit('blackbox3-coder-plus', 'output')).toBe(65536);
+            expect(tokenLimit('blackbox3-coder-plus-20250601', 'output')).toBe(65536);
+        });
+        it('should return the correct output limit for blackbox-vl-max-latest', () => {
+            expect(tokenLimit('blackbox-vl-max-latest', 'output')).toBe(8192);
+        });
+    });
+    describe('Default output limits', () => {
+        it('should return the default output limit for unknown models', () => {
+            expect(tokenLimit('unknown-model', 'output')).toBe(DEFAULT_OUTPUT_TOKEN_LIMIT);
+            expect(tokenLimit('gpt-4', 'output')).toBe(DEFAULT_OUTPUT_TOKEN_LIMIT);
+            expect(tokenLimit('claude-3.5-sonnet', 'output')).toBe(DEFAULT_OUTPUT_TOKEN_LIMIT);
+        });
+        it('should return the default output limit for models without specific output patterns', () => {
+            expect(tokenLimit('blackbox3-coder-7b', 'output')).toBe(DEFAULT_OUTPUT_TOKEN_LIMIT);
+            expect(tokenLimit('blackbox-plus', 'output')).toBe(DEFAULT_OUTPUT_TOKEN_LIMIT);
+            expect(tokenLimit('blackbox-vl-max', 'output')).toBe(DEFAULT_OUTPUT_TOKEN_LIMIT);
+        });
+    });
+    describe('Input vs Output limits comparison', () => {
+        it('should return different limits for input vs output for blackbox3-coder-plus', () => {
+            expect(tokenLimit('blackbox3-coder-plus', 'input')).toBe(1048576); // 1M input
+            expect(tokenLimit('blackbox3-coder-plus', 'output')).toBe(65536); // 64K output
+        });
+        it('should return different limits for input vs output for blackbox-vl-max-latest', () => {
+            expect(tokenLimit('blackbox-vl-max-latest', 'input')).toBe(131072); // 128K input
+            expect(tokenLimit('blackbox-vl-max-latest', 'output')).toBe(8192); // 8K output
+        });
+        it('should return different limits for input vs output for blackbox3-vl-plus', () => {
+            expect(tokenLimit('blackbox3-vl-plus', 'input')).toBe(262144); // 256K input
+            expect(tokenLimit('blackbox3-vl-plus', 'output')).toBe(32768); // 32K output
+        });
+        it('should return same default limits for unknown models', () => {
+            expect(tokenLimit('unknown-model', 'input')).toBe(DEFAULT_TOKEN_LIMIT); // 128K input
+            expect(tokenLimit('unknown-model', 'output')).toBe(DEFAULT_OUTPUT_TOKEN_LIMIT); // 4K output
+        });
+    });
+    describe('Backward compatibility', () => {
+        it('should default to input type when no type is specified', () => {
+            expect(tokenLimit('blackbox3-coder-plus')).toBe(1048576); // Should be input limit
+            expect(tokenLimit('blackbox-vl-max-latest')).toBe(131072); // Should be input limit
+            expect(tokenLimit('unknown-model')).toBe(DEFAULT_TOKEN_LIMIT); // Should be input default
+        });
+        it('should work with explicit input type', () => {
+            expect(tokenLimit('blackbox3-coder-plus', 'input')).toBe(1048576);
+            expect(tokenLimit('blackbox-vl-max-latest', 'input')).toBe(131072);
+            expect(tokenLimit('unknown-model', 'input')).toBe(DEFAULT_TOKEN_LIMIT);
+        });
+    });
+    describe('Model normalization with output limits', () => {
+        it('should handle normalized model names for output limits', () => {
+            expect(tokenLimit('BLACKBOX3-CODER-PLUS', 'output')).toBe(65536);
+            expect(tokenLimit('blackbox3-coder-plus-20250601', 'output')).toBe(65536);
+            expect(tokenLimit('BLACKBOX-VL-MAX-LATEST', 'output')).toBe(8192);
+        });
+        it('should handle complex model strings for output limits', () => {
+            expect(tokenLimit('  a/b/c|BLACKBOX3-CODER-PLUS:blackbox3-coder-plus-2024-05-13  ', 'output')).toBe(65536);
+            expect(tokenLimit('provider/blackbox-vl-max-latest:blackbox-vl-max-latest-v1', 'output')).toBe(8192);
+        });
+    });
+});
+//# sourceMappingURL=tokenLimits.test.js.map

package/dist/src/core/tokenLimits.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tokenLimits.test.js","sourceRoot":"","sources":["../../../src/core/tokenLimits.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,SAAS,EACT,UAAU,EACV,mBAAmB,EACnB,0BAA0B,GAC3B,MAAM,kBAAkB,CAAC;AAE1B,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IACzB,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAClE,MAAM,CAAC,SAAS,CAAC,6BAA6B,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,CAAC,SAAS,CAAC,qCAAqC,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAClF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACpE,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACxD,MAAM,CAAC,SAAS,CAAC,4BAA4B,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC1E,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACpD,MAAM,CAAC,SAAS,CAAC,mCAAmC,CAAC,CAAC,CAAC,IAAI,CACzD,kBAAkB,CACnB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,CAAC,SAAS,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACxE,MAAM,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC9D,MAAM,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAClE,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC5D,MAAM,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,CAAC,SAAS,CAAC,mDAAmD,CAAC,CAAC,CAAC,IAAI,CACzE,gBAAgB,CACjB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/B,MAAM,CAAC,SAAS,CAAC,SAA8B,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC3D,MAAM,CAAC,SAAS,CAAC,IAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACxD,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,mCAAmC;IACnC,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,MAAM,CAAC,UAAU,CAAC,mCAAmC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;YAC/E,MAAM,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACzD,MAAM,CAAC,UAAU,CAAC,+BAA+B,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAClE,MAAM,CAAC,UAAU,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC1D,MAAM,CAAC,UAAU,CAAC,gCAAgC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wEAAwE,EAAE,GAAG,EAAE;YAChF,MAAM,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtD,MAAM,CAAC,UAAU,CAAC,oCAAoC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtE,MAAM,CAAC,UAAU,CAAC,kCAAkC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,MAAM,CAAC,UAAU,CAAC,oCAAoC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACnD,MAAM,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACzD,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,CAAC,UAAU,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAClD,MAAM,CAAC,UAAU,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;YACzD,MAAM,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,CAAC,UAAU,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,MAAM,CAAC,UAAU,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,yBAAyB;IACzB,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,iCAAiC;IACjC,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,CAAC,UAAU,CAAC,uCAAuC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,iCAAiC;IACjC,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,MAAM,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAClD,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,MAAM,CAAC,UAAU,CAAC,sBAAsB,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACjE,MAAM,CAAC,UAAU,CAAC,+BAA+B,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;YAC3E,MAAM,CAAC,UAAU,CAAC,wBAAwB,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,CAAC,UAAU,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAChD,0BAA0B,CAC3B,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YACvE,MAAM,CAAC,UAAU,CAAC,mBAAmB,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CACpD,0BAA0B,CAC3B,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oFAAoF,EAAE,GAAG,EAAE;YAC5F,MAAM,CAAC,UAAU,CAAC,oBAAoB,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CACrD,0BAA0B,CAC3B,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAChD,0BAA0B,CAC3B,CAAC;YACF,MAAM,CAAC,UAAU,CAAC,iBAAiB,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAClD,0BAA0B,CAC3B,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;QACjD,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;YACrF,MAAM,CAAC,UAAU,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW;YAC9E,MAAM,CAAC,UAAU,CAAC,sBAAsB,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,aAAa;QACjF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+EAA+E,EAAE,GAAG,EAAE;YACvF,MAAM,CAAC,UAAU,CAAC,wBAAwB,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa;YACjF,MAAM,CAAC,UAAU,CAAC,wBAAwB,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,YAAY;QACjF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YAClF,MAAM,CAAC,UAAU,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa;YAC5E,MAAM,CAAC,UAAU,CAAC,mBAAmB,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,aAAa;QAC9E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,CAAC,UAAU,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC,aAAa;YACrF,MAAM,CAAC,UAAU,CAAC,eAAe,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAChD,0BAA0B,CAC3B,CAAC,CAAC,YAAY;QACjB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,CAAC,UAAU,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB;YAClF,MAAM,CAAC,UAAU,CAAC,wBAAwB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,wBAAwB;YACnF,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC,CAAC,0BAA0B;QAC3F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,CAAC,UAAU,CAAC,sBAAsB,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAClE,MAAM,CAAC,UAAU,CAAC,wBAAwB,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACnE,MAAM,CAAC,UAAU,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wCAAwC,EAAE,GAAG,EAAE;QACtD,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,CAAC,UAAU,CAAC,sBAAsB,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACjE,MAAM,CAAC,UAAU,CAAC,+BAA+B,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1E,MAAM,CAAC,UAAU,CAAC,wBAAwB,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;YAC/D,MAAM,CACJ,UAAU,CACR,gEAAgE,EAChE,QAAQ,CACT,CACF,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACd,MAAM,CACJ,UAAU,CACR,2DAA2D,EAC3D,QAAQ,CACT,CACF,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/core/turn.d.ts

@@ -0,0 +1,140 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { Part, PartListUnion, GenerateContentResponse, FunctionDeclaration, FinishReason } from '@google/genai';
+import type { ToolCallConfirmationDetails, ToolResult, ToolResultDisplay } from '../tools/tools.js';
+import type { ToolErrorType } from '../tools/tool-error.js';
+import type { GeminiChat } from './geminiChat.js';
+export interface ServerTool {
+    name: string;
+    schema: FunctionDeclaration;
+    execute(params: Record<string, unknown>, signal?: AbortSignal): Promise<ToolResult>;
+    shouldConfirmExecute(params: Record<string, unknown>, abortSignal: AbortSignal): Promise<ToolCallConfirmationDetails | false>;
+}
+export declare enum GeminiEventType {
+    Content = "content",
+    ToolCallRequest = "tool_call_request",
+    ToolCallResponse = "tool_call_response",
+    ToolCallConfirmation = "tool_call_confirmation",
+    UserCancelled = "user_cancelled",
+    Error = "error",
+    ChatCompressed = "chat_compressed",
+    Thought = "thought",
+    MaxSessionTurns = "max_session_turns",
+    SessionTokenLimitExceeded = "session_token_limit_exceeded",
+    Finished = "finished",
+    LoopDetected = "loop_detected",
+    Citation = "citation",
+    Retry = "retry"
+}
+export type ServerGeminiRetryEvent = {
+    type: GeminiEventType.Retry;
+};
+export interface StructuredError {
+    message: string;
+    status?: number;
+}
+export interface GeminiErrorEventValue {
+    error: StructuredError;
+}
+export interface SessionTokenLimitExceededValue {
+    currentTokens: number;
+    limit: number;
+    message: string;
+}
+export interface ToolCallRequestInfo {
+    callId: string;
+    name: string;
+    args: Record<string, unknown>;
+    isClientInitiated: boolean;
+    prompt_id: string;
+}
+export interface ToolCallResponseInfo {
+    callId: string;
+    responseParts: Part[];
+    resultDisplay: ToolResultDisplay | undefined;
+    error: Error | undefined;
+    errorType: ToolErrorType | undefined;
+}
+export interface ServerToolCallConfirmationDetails {
+    request: ToolCallRequestInfo;
+    details: ToolCallConfirmationDetails;
+}
+export type ThoughtSummary = {
+    subject: string;
+    description: string;
+};
+export type ServerGeminiContentEvent = {
+    type: GeminiEventType.Content;
+    value: string;
+};
+export type ServerGeminiThoughtEvent = {
+    type: GeminiEventType.Thought;
+    value: ThoughtSummary;
+};
+export type ServerGeminiToolCallRequestEvent = {
+    type: GeminiEventType.ToolCallRequest;
+    value: ToolCallRequestInfo;
+};
+export type ServerGeminiToolCallResponseEvent = {
+    type: GeminiEventType.ToolCallResponse;
+    value: ToolCallResponseInfo;
+};
+export type ServerGeminiToolCallConfirmationEvent = {
+    type: GeminiEventType.ToolCallConfirmation;
+    value: ServerToolCallConfirmationDetails;
+};
+export type ServerGeminiUserCancelledEvent = {
+    type: GeminiEventType.UserCancelled;
+};
+export type ServerGeminiErrorEvent = {
+    type: GeminiEventType.Error;
+    value: GeminiErrorEventValue;
+};
+export declare enum CompressionStatus {
+    /** The compression was successful */
+    COMPRESSED = 1,
+    /** The compression failed due to the compression inflating the token count */
+    COMPRESSION_FAILED_INFLATED_TOKEN_COUNT = 2,
+    /** The compression failed due to an error counting tokens */
+    COMPRESSION_FAILED_TOKEN_COUNT_ERROR = 3,
+    /** The compression was not necessary and no action was taken */
+    NOOP = 4
+}
+export interface ChatCompressionInfo {
+    originalTokenCount: number;
+    newTokenCount: number;
+    compressionStatus: CompressionStatus;
+}
+export type ServerGeminiChatCompressedEvent = {
+    type: GeminiEventType.ChatCompressed;
+    value: ChatCompressionInfo | null;
+};
+export type ServerGeminiMaxSessionTurnsEvent = {
+    type: GeminiEventType.MaxSessionTurns;
+};
+export type ServerGeminiSessionTokenLimitExceededEvent = {
+    type: GeminiEventType.SessionTokenLimitExceeded;
+    value: SessionTokenLimitExceededValue;
+};
+export type ServerGeminiFinishedEvent = {
+    type: GeminiEventType.Finished;
+    value: FinishReason;
+};
+export type ServerGeminiLoopDetectedEvent = {
+    type: GeminiEventType.LoopDetected;
+};
+export type ServerGeminiStreamEvent = ServerGeminiContentEvent | ServerGeminiToolCallRequestEvent | ServerGeminiToolCallResponseEvent | ServerGeminiToolCallConfirmationEvent | ServerGeminiUserCancelledEvent | ServerGeminiErrorEvent | ServerGeminiChatCompressedEvent | ServerGeminiThoughtEvent | ServerGeminiMaxSessionTurnsEvent | ServerGeminiSessionTokenLimitExceededEvent | ServerGeminiFinishedEvent | ServerGeminiLoopDetectedEvent | ServerGeminiRetryEvent;
+export declare class Turn {
+    private readonly chat;
+    private readonly prompt_id;
+    readonly pendingToolCalls: ToolCallRequestInfo[];
+    private debugResponses;
+    finishReason: FinishReason | undefined;
+    constructor(chat: GeminiChat, prompt_id: string);
+    run(req: PartListUnion, signal: AbortSignal): AsyncGenerator<ServerGeminiStreamEvent>;
+    private handlePendingFunctionCall;
+    getDebugResponses(): GenerateContentResponse[];
+}