@just-every/ensemble 0.2.257 → 0.2.258

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;