@just-every/ensemble 0.2.257 → 0.2.259

package/dist/model_providers/assemblyai.js

@@ -0,0 +1,323 @@
+import { BaseModelProvider } from './base_provider.js';
+const ASSEMBLYAI_STREAMING_URL = 'wss://streaming.assemblyai.com/v3/ws';
+const ASSEMBLYAI_STREAMING_MODELS = new Set(['u3-rt-pro']);
+const DEFAULT_SAMPLE_RATE = 16000;
+const DEFAULT_CHANNELS = 1;
+const BYTES_PER_SAMPLE = 2;
+const TARGET_CHUNK_MS = 50;
+const WEBSOCKET_OPEN_STATE = 1;
+const DONE = Symbol('done');
+function normalizeAudioSource(source) {
+    if (source instanceof ReadableStream) {
+        return source;
+    }
+    if (typeof source === 'object' && source !== null && Symbol.asyncIterator in source) {
+        return new ReadableStream({
+            async start(controller) {
+                try {
+                    for await (const chunk of source) {
+                        controller.enqueue(chunk);
+                    }
+                    controller.close();
+                }
+                catch (error) {
+                    controller.error(error);
+                }
+            },
+        });
+    }
+    if (typeof source === 'function') {
+        return normalizeAudioSource(source());
+    }
+    if (source instanceof ArrayBuffer || source instanceof Uint8Array) {
+        const data = source instanceof ArrayBuffer ? new Uint8Array(source) : source;
+        return new ReadableStream({
+            start(controller) {
+                controller.enqueue(data);
+                controller.close();
+            },
+        });
+    }
+    throw new Error(`Unsupported audio source type: ${typeof source}`);
+}
+function appendBuffer(left, right) {
+    if (left.length === 0)
+        return right;
+    const merged = new Uint8Array(left.length + right.length);
+    merged.set(left);
+    merged.set(right, left.length);
+    return merged;
+}
+function copyChunk(chunk) {
+    const copy = new Uint8Array(chunk.length);
+    copy.set(chunk);
+    return copy;
+}
+function wait(ms) {
+    return new Promise(resolve => setTimeout(resolve, ms));
+}
+function getTranscriptDelta(previous, next) {
+    if (!previous)
+        return next;
+    if (next.startsWith(previous))
+        return next.slice(previous.length);
+    let commonPrefixLength = 0;
+    const limit = Math.min(previous.length, next.length);
+    while (commonPrefixLength < limit && previous[commonPrefixLength] === next[commonPrefixLength]) {
+        commonPrefixLength++;
+    }
+    return next.slice(commonPrefixLength);
+}
+function setParam(params, name, value) {
+    if (value === undefined || value === null || value === '')
+        return;
+    if (Array.isArray(value)) {
+        if (value.length === 0)
+            return;
+        params.set(name, JSON.stringify(value));
+        return;
+    }
+    params.set(name, String(value));
+}
+function buildStreamingUrl(model, opts) {
+    const params = new URLSearchParams();
+    params.set('sample_rate', String(opts.audioFormat?.sampleRate || DEFAULT_SAMPLE_RATE));
+    params.set('speech_model', model);
+    setParam(params, 'mode', opts.mode);
+    setParam(params, 'prompt', opts.prompt);
+    setParam(params, 'language_code', opts.language_code || opts.languageCode || opts.language);
+    setParam(params, 'keyterms_prompt', opts.keyterms_prompt || opts.keytermsPrompt);
+    setParam(params, 'min_turn_silence', opts.min_turn_silence || opts.minTurnSilence);
+    setParam(params, 'max_turn_silence', opts.max_turn_silence || opts.maxTurnSilence);
+    setParam(params, 'vad_threshold', opts.vad_threshold || opts.vadThreshold);
+    setParam(params, 'agent_context', opts.agent_context || opts.agentContext);
+    return `${ASSEMBLYAI_STREAMING_URL}?${params.toString()}`;
+}
+function isSocketOpen(ws) {
+    return ws.readyState === WEBSOCKET_OPEN_STATE;
+}
+export class AssemblyAIProvider extends BaseModelProvider {
+    apiKey;
+    constructor(apiKey) {
+        super('assemblyai');
+        this.apiKey = apiKey;
+    }
+    async *createResponseStream() {
+        yield* [];
+        throw new Error('AssemblyAI provider only supports streaming transcription');
+    }
+    async *createTranscription(audio, _agent, model, opts = {}) {
+        if (!ASSEMBLYAI_STREAMING_MODELS.has(model)) {
+            throw new Error(`Model ${model} does not support AssemblyAI streaming transcription. Supported models: ${Array.from(ASSEMBLYAI_STREAMING_MODELS).join(', ')}`);
+        }
+        const apiKey = this.apiKey || process.env.ASSEMBLYAI_API_KEY;
+        if (!apiKey) {
+            throw new Error('Failed to initialize AssemblyAI transcription. Make sure ASSEMBLYAI_API_KEY is set.');
+        }
+        const { WebSocket } = await import('ws');
+        const ws = new WebSocket(buildStreamingUrl(model, opts), {
+            headers: { Authorization: apiKey },
+        });
+        const queue = [];
+        let wake;
+        let finished = false;
+        let socketError;
+        const previousTurnTranscripts = new Map();
+        const push = (item) => {
+            queue.push(item);
+            wake?.();
+            wake = undefined;
+        };
+        const markDone = () => {
+            if (finished)
+                return;
+            finished = true;
+            push(DONE);
+        };
+        const connected = new Promise((resolve, reject) => {
+            const timeout = setTimeout(() => {
+                reject(new Error('AssemblyAI streaming connection timeout'));
+            }, 10000);
+            ws.once('open', () => {
+                clearTimeout(timeout);
+                resolve();
+            });
+            ws.once('error', error => {
+                clearTimeout(timeout);
+                reject(error);
+            });
+        });
+        ws.on('message', data => {
+            let message;
+            try {
+                message = JSON.parse(data.toString());
+            }
+            catch (error) {
+                socketError = error instanceof Error ? error : new Error(String(error));
+                push({
+                    type: 'error',
+                    timestamp: new Date().toISOString(),
+                    error: `Failed to parse AssemblyAI message: ${socketError.message}`,
+                });
+                return;
+            }
+            const timestamp = new Date().toISOString();
+            switch (message.type) {
+                case 'Begin':
+                    break;
+                case 'SpeechStarted':
+                    push({
+                        type: 'transcription_turn_start',
+                        timestamp,
+                    });
+                    break;
+                case 'Turn': {
+                    const transcript = typeof message.transcript === 'string' ? message.transcript : '';
+                    const turnOrder = typeof message.turn_order === 'number' ? message.turn_order : 0;
+                    const previousTranscript = previousTurnTranscripts.get(turnOrder) || '';
+                    const delta = getTranscriptDelta(previousTranscript, transcript);
+                    previousTurnTranscripts.set(turnOrder, transcript);
+                    if (delta) {
+                        push({
+                            type: 'transcription_turn_delta',
+                            timestamp,
+                            delta,
+                            partial: !message.end_of_turn,
+                        });
+                    }
+                    if (message.end_of_turn) {
+                        previousTurnTranscripts.delete(turnOrder);
+                        push({
+                            type: 'transcription_turn_complete',
+                            timestamp,
+                            text: transcript,
+                        });
+                    }
+                    break;
+                }
+                case 'Termination':
+                    markDone();
+                    break;
+                case 'Error':
+                    socketError = new Error(message.error || 'AssemblyAI streaming error');
+                    push({
+                        type: 'error',
+                        timestamp,
+                        error: socketError.message,
+                    });
+                    markDone();
+                    break;
+                default:
+                    break;
+            }
+        });
+        ws.on('error', error => {
+            socketError = error instanceof Error ? error : new Error(String(error));
+            push({
+                type: 'error',
+                timestamp: new Date().toISOString(),
+                error: socketError.message,
+            });
+            markDone();
+        });
+        ws.on('close', (code, reason) => {
+            if (!finished && code !== 1000) {
+                const detail = reason.toString() || `AssemblyAI streaming socket closed with code ${code}`;
+                push({
+                    type: 'error',
+                    timestamp: new Date().toISOString(),
+                    error: detail,
+                });
+            }
+            markDone();
+        });
+        let audioPump;
+        try {
+            await connected;
+            audioPump = this.streamAudio(ws, audio, opts)
+                .then(() => {
+                if (isSocketOpen(ws)) {
+                    ws.send(JSON.stringify({ type: 'Terminate' }));
+                }
+            })
+                .catch(error => {
+                socketError = error instanceof Error ? error : new Error(String(error));
+                push({
+                    type: 'error',
+                    timestamp: new Date().toISOString(),
+                    error: socketError.message,
+                });
+                if (isSocketOpen(ws)) {
+                    ws.close();
+                }
+                markDone();
+            });
+            while (true) {
+                if (queue.length === 0) {
+                    await new Promise(resolve => {
+                        wake = resolve;
+                    });
+                }
+                const item = queue.shift();
+                if (!item)
+                    continue;
+                if (item === DONE)
+                    break;
+                yield item;
+            }
+            if (socketError) {
+                return;
+            }
+        }
+        finally {
+            await audioPump?.catch(() => undefined);
+            if (isSocketOpen(ws)) {
+                ws.close();
+            }
+        }
+    }
+    async streamAudio(ws, audio, opts) {
+        const stream = normalizeAudioSource(audio);
+        const reader = stream.getReader();
+        const sampleRate = opts.audioFormat?.sampleRate || DEFAULT_SAMPLE_RATE;
+        const channels = opts.audioFormat?.channels || DEFAULT_CHANNELS;
+        const throttleAudio = opts.throttleAudio ??
+            (!opts.disableRealtimeThrottle && (audio instanceof ArrayBuffer || audio instanceof Uint8Array));
+        const targetBytes = Math.max(1, Math.floor((sampleRate * channels * BYTES_PER_SAMPLE * TARGET_CHUNK_MS) / 1000));
+        let buffer = new Uint8Array(0);
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            if (value?.length) {
+                buffer = appendBuffer(buffer, copyChunk(value));
+            }
+            while (buffer.length >= targetBytes) {
+                const chunk = buffer.slice(0, targetBytes);
+                buffer = buffer.slice(targetBytes);
+                await this.sendAudioChunk(ws, chunk, throttleAudio);
+            }
+        }
+        if (buffer.length > 0) {
+            await this.sendAudioChunk(ws, buffer, throttleAudio);
+        }
+    }
+    async sendAudioChunk(ws, chunk, throttleAudio) {
+        if (!isSocketOpen(ws)) {
+            throw new Error('AssemblyAI streaming socket closed before audio could be sent');
+        }
+        await new Promise((resolve, reject) => {
+            ws.send(chunk, error => {
+                if (error)
+                    reject(error);
+                else
+                    resolve();
+            });
+        });
+        if (throttleAudio) {
+            await wait(TARGET_CHUNK_MS);
+        }
+    }
+}
+export const assemblyAIProvider = new AssemblyAIProvider();
+//# sourceMappingURL=assemblyai.js.map

package/dist/model_providers/assemblyai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"assemblyai.js","sourceRoot":"","sources":["../../model_providers/assemblyai.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAUvD,MAAM,wBAAwB,GAAG,sCAAsC,CAAC;AACxE,MAAM,2BAA2B,GAAG,IAAI,GAAG,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;AAC3D,MAAM,mBAAmB,GAAG,KAAK,CAAC;AAClC,MAAM,gBAAgB,GAAG,CAAC,CAAC;AAC3B,MAAM,gBAAgB,GAAG,CAAC,CAAC;AAC3B,MAAM,eAAe,GAAG,EAAE,CAAC;AAC3B,MAAM,oBAAoB,GAAG,CAAC,CAAC;AAsB/B,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;AAE5B,SAAS,oBAAoB,CAAC,MAAgC;IAC1D,IAAI,MAAM,YAAY,cAAc,EAAE,CAAC;QACnC,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,IAAI,IAAI,MAAM,CAAC,aAAa,IAAI,MAAM,EAAE,CAAC;QAClF,OAAO,IAAI,cAAc,CAAC;YACtB,KAAK,CAAC,KAAK,CAAC,UAAU;gBAClB,IAAI,CAAC;oBACD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAmC,EAAE,CAAC;wBAC5D,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;oBAC9B,CAAC;oBACD,UAAU,CAAC,KAAK,EAAE,CAAC;gBACvB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACb,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBAC5B,CAAC;YACL,CAAC;SACJ,CAAC,CAAC;IACP,CAAC;IAED,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;QAC/B,OAAO,oBAAoB,CAAC,MAAM,EAA8B,CAAC,CAAC;IACtE,CAAC;IAED,IAAI,MAAM,YAAY,WAAW,IAAI,MAAM,YAAY,UAAU,EAAE,CAAC;QAChE,MAAM,IAAI,GAAG,MAAM,YAAY,WAAW,CAAC,CAAC,CAAC,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;QAC7E,OAAO,IAAI,cAAc,CAAC;YACtB,KAAK,CAAC,UAAU;gBACZ,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;gBACzB,UAAU,CAAC,KAAK,EAAE,CAAC;YACvB,CAAC;SACJ,CAAC,CAAC;IACP,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,kCAAkC,OAAO,MAAM,EAAE,CAAC,CAAC;AACvE,CAAC;AAED,SAAS,YAAY,CACjB,IAAiC,EACjC,KAAkC;IAElC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IACpC,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;IAC1D,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACjB,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IAC/B,OAAO,MAAM,CAAC;AAClB,CAAC;AAED,SAAS,SAAS,CAAC,KAAkC;IACjD,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IAC1C,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAChB,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,IAAI,CAAC,EAAU;IACpB,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC;AAED,SAAS,kBAAkB,CAAC,QAAgB,EAAE,IAAY;IACtD,IAAI,CAAC,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC3B,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;QAAE,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAElE,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACrD,OAAO,kBAAkB,GAAG,KAAK,IAAI,QAAQ,CAAC,kBAAkB,CAAC,KAAK,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC;QAC7F,kBAAkB,EAAE,CAAC;IACzB,CAAC;IAED,OAAO,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;AAC1C,CAAC;AAED,SAAS,QAAQ,CAAC,MAAuB,EAAE,IAAY,EAAE,KAAc;IACnE,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,EAAE;QAAE,OAAO;IAClE,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACvB,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAC/B,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;QACxC,OAAO;IACX,CAAC;IACD,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;AACpC,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAa,EAAE,IAAgC;IACtE,MAAM,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;IACrC,MAAM,CAAC,GAAG,CAAC,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,UAAU,IAAI,mBAAmB,CAAC,CAAC,CAAC;IACvF,MAAM,CAAC,GAAG,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;IAClC,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;IACpC,QAAQ,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACxC,QAAQ,CAAC,MAAM,EAAE,eAAe,EAAE,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC5F,QAAQ,CAAC,MAAM,EAAE,iBAAiB,EAAE,IAAI,CAAC,eAAe,IAAI,IAAI,CAAC,cAAc,CAAC,CAAC;IACjF,QAAQ,CAAC,MAAM,EAAE,kBAAkB,EAAE,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC,cAAc,CAAC,CAAC;IACnF,QAAQ,CAAC,MAAM,EAAE,kBAAkB,EAAE,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC,cAAc,CAAC,CAAC;IACnF,QAAQ,CAAC,MAAM,EAAE,eAAe,EAAE,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,YAAY,CAAC,CAAC;IAC3E,QAAQ,CAAC,MAAM,EAAE,eAAe,EAAE,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,YAAY,CAAC,CAAC;IAC3E,OAAO,GAAG,wBAAwB,IAAI,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC;AAC9D,CAAC;AAED,SAAS,YAAY,CAAC,EAAa;IAC/B,OAAO,EAAE,CAAC,UAAU,KAAK,oBAAoB,CAAC;AAClD,CAAC;AAED,MAAM,OAAO,kBAAmB,SAAQ,iBAAiB;IACjC;IAApB,YAAoB,MAAe;QAC/B,KAAK,CAAC,YAAY,CAAC,CAAC;QADJ,WAAM,GAAN,MAAM,CAAS;IAEnC,CAAC;IAED,KAAK,CAAC,CAAC,oBAAoB;QACvB,KAAK,CAAC,CAAC,EAA2B,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;IACjF,CAAC;IAED,KAAK,CAAC,CAAC,mBAAmB,CACtB,KAA+B,EAC/B,MAAuB,EACvB,KAAa,EACb,OAAmC,EAAE;QAErC,IAAI,CAAC,2BAA2B,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1C,MAAM,IAAI,KAAK,CACX,SAAS,KAAK,2EAA2E,KAAK,CAAC,IAAI,CAC/F,2BAA2B,CAC9B,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACjB,CAAC;QACN,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC;QAC7D,IAAI,CAAC,MAAM,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,qFAAqF,CAAC,CAAC;QAC3G,CAAC;QAED,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,EAAE,GAAG,IAAI,SAAS,CAAC,iBAAiB,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE;YACrD,OAAO,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE;SACrC,CAAC,CAAC;QAEH,MAAM,KAAK,GAAgB,EAAE,CAAC;QAC9B,IAAI,IAA8B,CAAC;QACnC,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,WAA8B,CAAC;QACnC,MAAM,uBAAuB,GAAG,IAAI,GAAG,EAAkB,CAAC;QAE1D,MAAM,IAAI,GAAG,CAAC,IAAe,EAAE,EAAE;YAC7B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjB,IAAI,EAAE,EAAE,CAAC;YACT,IAAI,GAAG,SAAS,CAAC;QACrB,CAAC,CAAC;QAEF,MAAM,QAAQ,GAAG,GAAG,EAAE;YAClB,IAAI,QAAQ;gBAAE,OAAO;YACrB,QAAQ,GAAG,IAAI,CAAC;YAChB,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,CAAC,CAAC;QAEF,MAAM,SAAS,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACpD,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,MAAM,CAAC,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC,CAAC;YACjE,CAAC,EAAE,KAAK,CAAC,CAAC;YAEV,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE;gBACjB,YAAY,CAAC,OAAO,CAAC,CAAC;gBACtB,OAAO,EAAE,CAAC;YACd,CAAC,CAAC,CAAC;YACH,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE;gBACrB,YAAY,CAAC,OAAO,CAAC,CAAC;gBACtB,MAAM,CAAC,KAAK,CAAC,CAAC;YAClB,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,IAAI,CAAC,EAAE;YACpB,IAAI,OAAY,CAAC;YACjB,IAAI,CAAC;gBACD,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;YAC1C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,WAAW,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;gBACxE,IAAI,CAAC;oBACD,IAAI,EAAE,OAAO;oBACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACnC,KAAK,EAAE,uCAAuC,WAAW,CAAC,OAAO,EAAE;iBACtE,CAAC,CAAC;gBACH,OAAO;YACX,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YAC3C,QAAQ,OAAO,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,OAAO;oBACR,MAAM;gBACV,KAAK,eAAe;oBAChB,IAAI,CAAC;wBACD,IAAI,EAAE,0BAA0B;wBAChC,SAAS;qBACZ,CAAC,CAAC;oBACH,MAAM;gBACV,KAAK,MAAM,CAAC,CAAC,CAAC;oBACV,MAAM,UAAU,GAAG,OAAO,OAAO,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;oBACpF,MAAM,SAAS,GAAG,OAAO,OAAO,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;oBAClF,MAAM,kBAAkB,GAAG,uBAAuB,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;oBACxE,MAAM,KAAK,GAAG,kBAAkB,CAAC,kBAAkB,EAAE,UAAU,CAAC,CAAC;oBACjE,uBAAuB,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;oBAEnD,IAAI,KAAK,EAAE,CAAC;wBACR,IAAI,CAAC;4BACD,IAAI,EAAE,0BAA0B;4BAChC,SAAS;4BACT,KAAK;4BACL,OAAO,EAAE,CAAC,OAAO,CAAC,WAAW;yBACV,CAAC,CAAC;oBAC7B,CAAC;oBAED,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC;wBACtB,uBAAuB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;wBAC1C,IAAI,CAAC;4BACD,IAAI,EAAE,6BAA6B;4BACnC,SAAS;4BACT,IAAI,EAAE,UAAU;yBACnB,CAAC,CAAC;oBACP,CAAC;oBACD,MAAM;gBACV,CAAC;gBACD,KAAK,aAAa;oBACd,QAAQ,EAAE,CAAC;oBACX,MAAM;gBACV,KAAK,OAAO;oBACR,WAAW,GAAG,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,IAAI,4BAA4B,CAAC,CAAC;oBACvE,IAAI,CAAC;wBACD,IAAI,EAAE,OAAO;wBACb,SAAS;wBACT,KAAK,EAAE,WAAW,CAAC,OAAO;qBAC7B,CAAC,CAAC;oBACH,QAAQ,EAAE,CAAC;oBACX,MAAM;gBACV;oBACI,MAAM;YACd,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE;YACnB,WAAW,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;YACxE,IAAI,CAAC;gBACD,IAAI,EAAE,OAAO;gBACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,KAAK,EAAE,WAAW,CAAC,OAAO;aAC7B,CAAC,CAAC;YACH,QAAQ,EAAE,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE;YAC5B,IAAI,CAAC,QAAQ,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;gBAC7B,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,EAAE,IAAI,gDAAgD,IAAI,EAAE,CAAC;gBAC3F,IAAI,CAAC;oBACD,IAAI,EAAE,OAAO;oBACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACnC,KAAK,EAAE,MAAM;iBAChB,CAAC,CAAC;YACP,CAAC;YACD,QAAQ,EAAE,CAAC;QACf,CAAC,CAAC,CAAC;QAEH,IAAI,SAAoC,CAAC;QAEzC,IAAI,CAAC;YACD,MAAM,SAAS,CAAC;YAChB,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC;iBACxC,IAAI,CAAC,GAAG,EAAE;gBACP,IAAI,YAAY,CAAC,EAAE,CAAC,EAAE,CAAC;oBACnB,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;gBACnD,CAAC;YACL,CAAC,CAAC;iBACD,KAAK,CAAC,KAAK,CAAC,EAAE;gBACX,WAAW,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;gBACxE,IAAI,CAAC;oBACD,IAAI,EAAE,OAAO;oBACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACnC,KAAK,EAAE,WAAW,CAAC,OAAO;iBAC7B,CAAC,CAAC;gBACH,IAAI,YAAY,CAAC,EAAE,CAAC,EAAE,CAAC;oBACnB,EAAE,CAAC,KAAK,EAAE,CAAC;gBACf,CAAC;gBACD,QAAQ,EAAE,CAAC;YACf,CAAC,CAAC,CAAC;YAEP,OAAO,IAAI,EAAE,CAAC;gBACV,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBACrB,MAAM,IAAI,OAAO,CAAO,OAAO,CAAC,EAAE;wBAC9B,IAAI,GAAG,OAAO,CAAC;oBACnB,CAAC,CAAC,CAAC;gBACP,CAAC;gBAED,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;gBAC3B,IAAI,CAAC,IAAI;oBAAE,SAAS;gBACpB,IAAI,IAAI,KAAK,IAAI;oBAAE,MAAM;gBACzB,MAAM,IAAI,CAAC;YACf,CAAC;YAED,IAAI,WAAW,EAAE,CAAC;gBACd,OAAO;YACX,CAAC;QACL,CAAC;gBAAS,CAAC;YACP,MAAM,SAAS,EAAE,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YACxC,IAAI,YAAY,CAAC,EAAE,CAAC,EAAE,CAAC;gBACnB,EAAE,CAAC,KAAK,EAAE,CAAC;YACf,CAAC;QACL,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,WAAW,CACrB,EAAa,EACb,KAA+B,EAC/B,IAAgC;QAEhC,MAAM,MAAM,GAAG,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;QAClC,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,EAAE,UAAU,IAAI,mBAAmB,CAAC;QACvE,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,EAAE,QAAQ,IAAI,gBAAgB,CAAC;QAChE,MAAM,aAAa,GACf,IAAI,CAAC,aAAa;YAClB,CAAC,CAAC,IAAI,CAAC,uBAAuB,IAAI,CAAC,KAAK,YAAY,WAAW,IAAI,KAAK,YAAY,UAAU,CAAC,CAAC,CAAC;QACrG,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CACxB,CAAC,EACD,IAAI,CAAC,KAAK,CAAC,CAAC,UAAU,GAAG,QAAQ,GAAG,gBAAgB,GAAG,eAAe,CAAC,GAAG,IAAI,CAAC,CAClF,CAAC;QACF,IAAI,MAAM,GAAgC,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;QAE5D,OAAO,IAAI,EAAE,CAAC;YACV,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAChB,IAAI,KAAK,EAAE,MAAM,EAAE,CAAC;gBAChB,MAAM,GAAG,YAAY,CAAC,MAAM,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;YACpD,CAAC;YAED,OAAO,MAAM,CAAC,MAAM,IAAI,WAAW,EAAE,CAAC;gBAClC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC;gBAC3C,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;gBACnC,MAAM,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,KAAK,EAAE,aAAa,CAAC,CAAC;YACxD,CAAC;QACL,CAAC;QAED,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACpB,MAAM,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;QACzD,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,cAAc,CACxB,EAAa,EACb,KAAkC,EAClC,aAAsB;QAEtB,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;QACrF,CAAC;QAED,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxC,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE;gBACnB,IAAI,KAAK;oBAAE,MAAM,CAAC,KAAK,CAAC,CAAC;;oBACpB,OAAO,EAAE,CAAC;YACnB,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QAEH,IAAI,aAAa,EAAE,CAAC;YAChB,MAAM,IAAI,CAAC,eAAe,CAAC,CAAC;QAChC,CAAC;IACL,CAAC;CACJ;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,IAAI,kBAAkB,EAAE,CAAC"}

package/dist/model_providers/gemini.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../model_providers/gemini.ts"],"names":[],"mappings":"AA2BA,OAAO,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAEhD,OAAO,EAGH,mBAAmB,EACnB,aAAa,EACb,eAAe,EACf,mBAAmB,EAInB,mBAAmB,EAEnB,iBAAiB,EACjB,wBAAwB,EACxB,kBAAkB,EAClB,UAAU,EACV,WAAW,EACX,WAAW,EAOd,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AA6NvD,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAO1D;AAmCD,wBAAgB,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAEzD;AAqgBD,qBAAa,cAAe,SAAQ,iBAAiB;IACjD,OAAO,CAAC,OAAO,CAAC,CAAc;IAC9B,OAAO,CAAC,MAAM,CAAC,CAAS;gBAEZ,MAAM,CAAC,EAAE,MAAM;IAS3B,OAAO,KAAK,MAAM,GAejB;IASK,eAAe,CACjB,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,EACxB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,eAAe,EACtB,IAAI,CAAC,EAAE,SAAS,GACjB,OAAO,CAAC,MAAM,EAAE,GAAG,MAAM,EAAE,EAAE,CAAC;YAmKlB,2BAA2B;IA+BnC,oBAAoB,CACvB,QAAQ,EAAE,aAAa,EACvB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,eAAe,EACtB,SAAS,CAAC,EAAE,MAAM,GACnB,cAAc,CAAC,mBAAmB,CAAC;IAukBhC,WAAW,CACb,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,eAAe,EACtB,IAAI,CAAC,EAAE,mBAAmB,GAC3B,OAAO,CAAC,MAAM,EAAE,CAAC;IA4hBpB,OAAO,CAAC,YAAY;IAmCd,WAAW,CACb,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,eAAe,EACtB,IAAI,CAAC,EAAE,mBAAmB,GAC3B,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,GAAG,WAAW,CAAC;IAgMpD,OAAO,CAAC,gBAAgB;IA6DjB,mBAAmB,CACtB,KAAK,EAAE,wBAAwB,EAC/B,KAAK,EAAE,eAAe,EACtB,KAAK,EAAE,MAAM,EACb,IAAI,CAAC,EAAE,iBAAiB,GACzB,cAAc,CAAC,kBAAkB,CAAC;IA6S/B,iBAAiB,CACnB,MAAM,EAAE,UAAU,EAClB,KAAK,EAAE,eAAe,EACtB,KAAK,EAAE,MAAM,EACb,IAAI,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC,WAAW,CAAC;CAyB1B;AAyfD,eAAO,MAAM,cAAc,gBAAuB,CAAC"}
+{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../model_providers/gemini.ts"],"names":[],"mappings":"AA2BA,OAAO,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAEhD,OAAO,EAGH,mBAAmB,EACnB,aAAa,EACb,eAAe,EACf,mBAAmB,EAInB,mBAAmB,EAEnB,iBAAiB,EACjB,wBAAwB,EACxB,kBAAkB,EAClB,UAAU,EACV,WAAW,EACX,WAAW,EAOd,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AA6NvD,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAO1D;AAmCD,wBAAgB,eAAe,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAEzD;AA8hBD,qBAAa,cAAe,SAAQ,iBAAiB;IACjD,OAAO,CAAC,OAAO,CAAC,CAAc;IAC9B,OAAO,CAAC,MAAM,CAAC,CAAS;gBAEZ,MAAM,CAAC,EAAE,MAAM;IAS3B,OAAO,KAAK,MAAM,GAejB;IASK,eAAe,CACjB,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,EACxB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,eAAe,EACtB,IAAI,CAAC,EAAE,SAAS,GACjB,OAAO,CAAC,MAAM,EAAE,GAAG,MAAM,EAAE,EAAE,CAAC;YAmKlB,2BAA2B;IA+BnC,oBAAoB,CACvB,QAAQ,EAAE,aAAa,EACvB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,eAAe,EACtB,SAAS,CAAC,EAAE,MAAM,GACnB,cAAc,CAAC,mBAAmB,CAAC;IAukBhC,WAAW,CACb,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,eAAe,EACtB,IAAI,CAAC,EAAE,mBAAmB,GAC3B,OAAO,CAAC,MAAM,EAAE,CAAC;IAwkBpB,OAAO,CAAC,YAAY;IAsCd,WAAW,CACb,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,eAAe,EACtB,IAAI,CAAC,EAAE,mBAAmB,GAC3B,OAAO,CAAC,cAAc,CAAC,UAAU,CAAC,GAAG,WAAW,CAAC;IAgMpD,OAAO,CAAC,gBAAgB;IA6DjB,mBAAmB,CACtB,KAAK,EAAE,wBAAwB,EAC/B,KAAK,EAAE,eAAe,EACtB,KAAK,EAAE,MAAM,EACb,IAAI,CAAC,EAAE,iBAAiB,GACzB,cAAc,CAAC,kBAAkB,CAAC;IA6S/B,iBAAiB,CACnB,MAAM,EAAE,UAAU,EAClB,KAAK,EAAE,eAAe,EACtB,KAAK,EAAE,MAAM,EACb,IAAI,CAAC,EAAE,WAAW,GACnB,OAAO,CAAC,WAAW,CAAC;CAyB1B;AAyfD,eAAO,MAAM,cAAc,gBAAuB,CAAC"}

package/dist/model_providers/gemini.js

@@ -582,6 +582,24 @@ function isGemmaModel(model) {
     const modelId = model.startsWith('models/') ? model.slice('models/'.length) : model;
     return modelId.startsWith('gemma-4-');
 }
+function isGemini25FlashImageModel(model) {
+    return model.includes('gemini-2.5-flash-image');
+}
+function isGemini31FlashImageModel(model) {
+    return model.includes('gemini-3.1-flash-image');
+}
+function isGemini31FlashLiteImageModel(model) {
+    return model.includes('gemini-3.1-flash-lite-image');
+}
+function isGemini3ProImageModel(model) {
+    return model.includes('gemini-3-pro-image');
+}
+function isGeminiStreamingImageModel(model) {
+    return (isGemini25FlashImageModel(model) ||
+        isGemini31FlashImageModel(model) ||
+        isGemini31FlashLiteImageModel(model) ||
+        isGemini3ProImageModel(model));
+}
 function getSupportedThinkingLevels(model) {
     if (isGemmaModel(model)) {
         return new Set(['HIGH']);
@@ -589,17 +607,17 @@ function getSupportedThinkingLevels(model) {
     if (model.includes('gemini-3.1-pro-preview') || model.includes('gemini-3-pro-preview')) {
         return new Set(['LOW', 'MEDIUM', 'HIGH']);
     }
+    if (isGemini31FlashImageModel(model) || isGemini31FlashLiteImageModel(model)) {
+        return new Set(['MINIMAL', 'HIGH']);
+    }
+    if (isGemini3ProImageModel(model)) {
+        return new Set(['HIGH']);
+    }
     if (model.includes('gemini-3.5-flash') ||
         model.includes('gemini-3.1-flash-lite') ||
         model.includes('gemini-3-flash-preview')) {
         return new Set(['MINIMAL', 'LOW', 'MEDIUM', 'HIGH']);
     }
-    if (model.includes('gemini-3.1-flash-image-preview')) {
-        return new Set(['MINIMAL', 'HIGH']);
-    }
-    if (model.includes('gemini-3-pro-image-preview')) {
-        return new Set(['HIGH']);
-    }
     return null;
 }
 const GEMINI_3_PRO_IMAGE_DIMENSION_PRESETS = {
@@ -1230,7 +1248,7 @@ export class GeminiProvider extends BaseModelProvider {
         const requestId = `req_${Date.now()}_${Math.random().toString(36).slice(2, 9)}`;
         let finalRequestId = requestId;
         try {
-            model = model || 'gemini-2.5-flash-image-preview';
+            model = model || 'gemini-3.1-flash-image';
             const numberOfImages = opts?.n ?? 1;
             if (!Number.isInteger(numberOfImages) || numberOfImages < 1) {
                 throw new Error('[Gemini] ImageGenerationOpts.n must be a positive integer.');
@@ -1244,11 +1262,17 @@ export class GeminiProvider extends BaseModelProvider {
             }
             const explicitWebGrounding = opts?.grounding?.web_search;
             const explicitImageGrounding = opts?.grounding?.image_search;
-            const enableWebGrounding = explicitWebGrounding ?? hasGoogleWebSearch ?? false;
-            const isGemini31FlashImageModel = model.includes('gemini-3.1-flash-image-preview');
-            const enableImageGrounding = explicitImageGrounding === true && isGemini31FlashImageModel;
-            if (explicitImageGrounding && !isGemini31FlashImageModel) {
-                console.warn('[Gemini] Image Search grounding is only available for gemini-3.1-flash-image-preview. Ignoring image_search=true.');
+            const isGemini31FlashImage = isGemini31FlashImageModel(model);
+            const isGemini31FlashLiteImage = isGemini31FlashLiteImageModel(model);
+            const isGemini3ProImage = isGemini3ProImageModel(model);
+            const supportsWebGrounding = isGemini31FlashImage || isGemini3ProImage;
+            const enableWebGrounding = supportsWebGrounding && (explicitWebGrounding ?? hasGoogleWebSearch ?? false);
+            const enableImageGrounding = explicitImageGrounding === true && isGemini31FlashImage;
+            if ((explicitWebGrounding || hasGoogleWebSearch) && !supportsWebGrounding) {
+                console.warn('[Gemini] Google Search grounding is only available for gemini-3.1-flash-image and gemini-3-pro-image. Ignoring web_search=true.');
+            }
+            if (explicitImageGrounding && !isGemini31FlashImage) {
+                console.warn('[Gemini] Image Search grounding is only available for gemini-3.1-flash-image. Ignoring image_search=true.');
             }
             const thinkingOptions = opts?.thinking;
             const hasThinkingOptionsObject = thinkingOptions !== null && typeof thinkingOptions === 'object' && !Array.isArray(thinkingOptions);
@@ -1258,11 +1282,12 @@ export class GeminiProvider extends BaseModelProvider {
                 ? thinkingOptions.level
                 : undefined;
             const thinkingLevel = requestedThinkingLevel === 'high' ? 'High' : requestedThinkingLevel ? 'Minimal' : undefined;
-            if (requestedThinkingLevel && !isGemini31FlashImageModel) {
-                console.warn('[Gemini] thinking.level is currently supported for gemini-3.1-flash-image-preview only. Ignoring thinking level.');
+            const supportsImageThinkingControls = isGemini31FlashImage || isGemini31FlashLiteImage;
+            if (requestedThinkingLevel && !supportsImageThinkingControls) {
+                console.warn('[Gemini] thinking.level is currently supported for gemini-3.1-flash-image and gemini-3.1-flash-lite-image only. Ignoring thinking level.');
             }
-            if (hasThinkingOptionsObject && 'include_thoughts' in thinkingOptions && !isGemini31FlashImageModel) {
-                console.warn('[Gemini] thinking.include_thoughts is currently supported for gemini-3.1-flash-image-preview only. Ignoring include_thoughts.');
+            if (hasThinkingOptionsObject && 'include_thoughts' in thinkingOptions && !supportsImageThinkingControls) {
+                console.warn('[Gemini] thinking.include_thoughts is currently supported for gemini-3.1-flash-image and gemini-3.1-flash-lite-image only. Ignoring include_thoughts.');
             }
             let aspectRatio = '1:1';
             if (opts?.size === 'landscape')
@@ -1270,9 +1295,7 @@ export class GeminiProvider extends BaseModelProvider {
             else if (opts?.size === 'portrait')
                 aspectRatio = '2:3';
             console.log(`[Gemini] Generating ${numberOfImages} image(s) with model ${model}, prompt: "${prompt.substring(0, 100)}${prompt.length > 100 ? '...' : ''}"`);
-            if (model.includes('gemini-2.5-flash-image-preview') ||
-                model.includes('gemini-3.1-flash-image-preview') ||
-                model.includes('gemini-3-pro-image-preview')) {
+            if (isGeminiStreamingImageModel(model)) {
                 let aggregateMetadata = { model };
                 const sizeMap = {
                     '1:1': { ar: '1:1' },
@@ -1303,7 +1326,7 @@ export class GeminiProvider extends BaseModelProvider {
                     '1024x1792': { ar: '9:16' },
                 };
                 const sm = opts?.size ? sizeMap[String(opts.size)] : undefined;
-                const gemini3ProDimensionPreset = model.includes('gemini-3-pro-image-preview')
+                const gemini3ProDimensionPreset = isGemini3ProImage
                     ? GEMINI_3_PRO_IMAGE_DIMENSION_PRESETS[String(opts?.size)]
                     : undefined;
                 const imageConfig = {};
@@ -1312,7 +1335,24 @@ export class GeminiProvider extends BaseModelProvider {
                 if (gemini3ProDimensionPreset?.ar)
                     imageConfig.aspectRatio = gemini3ProDimensionPreset.ar;
                 const qualityKey = typeof opts?.quality === 'string' ? opts.quality.toLowerCase() : '';
-                const imageSizeMap = isGemini31FlashImageModel
+                if (isGemini31FlashLiteImage && imageConfig.aspectRatio) {
+                    const liteAspectRatios = new Set([
+                        '1:1',
+                        '2:3',
+                        '3:2',
+                        '3:4',
+                        '4:3',
+                        '4:5',
+                        '5:4',
+                        '9:16',
+                        '16:9',
+                        '21:9',
+                    ]);
+                    if (!liteAspectRatios.has(imageConfig.aspectRatio)) {
+                        throw new Error(`[Gemini] gemini-3.1-flash-lite-image does not support aspect ratio ${imageConfig.aspectRatio}.`);
+                    }
+                }
+                const imageSizeMap = isGemini31FlashImage
                     ? {
                         low: '0.5K',
                         standard: '1K',
@@ -1320,28 +1360,44 @@ export class GeminiProvider extends BaseModelProvider {
                         hd: '4K',
                         high: '4K',
                     }
-                    : {
-                        low: '1K',
-                        standard: '2K',
-                        medium: '2K',
-                        hd: '4K',
-                        high: '4K',
-                    };
+                    : isGemini31FlashLiteImage
+                        ? {
+                            low: '1K',
+                            standard: '1K',
+                            medium: '1K',
+                            hd: '1K',
+                            high: '1K',
+                        }
+                        : {
+                            low: '1K',
+                            standard: '2K',
+                            medium: '2K',
+                            hd: '4K',
+                            high: '4K',
+                        };
                 let imageSize = imageSizeMap[qualityKey];
                 if (gemini3ProDimensionPreset?.imageSize) {
                     imageSize = gemini3ProDimensionPreset.imageSize;
                 }
-                if (isGemini31FlashImageModel && opts?.size === '512x512') {
+                if (isGemini31FlashImage && opts?.size === '512x512') {
                     imageSize = '0.5K';
                 }
+                if (isGemini31FlashLiteImage) {
+                    imageSize = '1K';
+                    if (opts?.size === '512x512') {
+                        throw new Error('[Gemini] gemini-3.1-flash-lite-image only supports 1K image output.');
+                    }
+                }
                 const requestImageSize = imageSize === '0.5K' ? '512' : imageSize;
                 if (requestImageSize)
                     imageConfig.imageSize = requestImageSize;
                 const thinkingConfig = {};
-                if (hasThinkingOptionsObject && 'include_thoughts' in thinkingOptions && isGemini31FlashImageModel) {
+                if (hasThinkingOptionsObject &&
+                    'include_thoughts' in thinkingOptions &&
+                    supportsImageThinkingControls) {
                     thinkingConfig.includeThoughts = includeThoughts;
                 }
-                if (thinkingLevel && isGemini31FlashImageModel) {
+                if (thinkingLevel && supportsImageThinkingControls) {
                     thinkingConfig.thinkingLevel = thinkingLevel;
                 }
                 const searchTypes = {};
@@ -1657,7 +1713,7 @@ export class GeminiProvider extends BaseModelProvider {
         }
     }
     getImageCost(model, imageSize) {
-        if (model.includes('gemini-3.1-flash-image-preview')) {
+        if (isGemini31FlashImageModel(model)) {
             if (imageSize === '4K')
                 return 0.151;
             if (imageSize === '2K')
@@ -1666,10 +1722,13 @@ export class GeminiProvider extends BaseModelProvider {
                 return 0.045;
             return 0.067;
         }
-        else if (model.includes('gemini-2.5-flash-image-preview')) {
+        else if (isGemini31FlashLiteImageModel(model)) {
+            return 0.0336;
+        }
+        else if (isGemini25FlashImageModel(model)) {
             return 0.039;
         }
-        else if (model.includes('gemini-3-pro-image-preview')) {
+        else if (isGemini3ProImageModel(model)) {
             if (imageSize === '4K')
                 return 0.24;
             return 0.134;