@loonylabs/tti-middleware 1.6.0 → 1.8.0

package/dist/middleware/services/tti/providers/base-tti-provider.js

@@ -450,7 +450,8 @@ class BaseTTIProvider {
             message.includes('epipe') ||
             message.includes('ehostunreach') ||
             message.includes('enetunreach') ||
-            message.includes('socket hang up')) {
+            message.includes('socket hang up') ||
+            message.includes('fetch failed')) {
             return true;
         }
         return false;
@@ -482,7 +483,8 @@ class BaseTTIProvider {
         }
         if (errorMessage.includes('timeout') ||
             errorMessage.includes('econnrefused') ||
-            errorMessage.includes('enotfound')) {
+            errorMessage.includes('enotfound') ||
+            errorMessage.includes('fetch failed')) {
             return new NetworkError(this.providerName, `Network error${context ? `: ${context}` : ''}`, error);
         }
         return new GenerationFailedError(this.providerName, `Generation failed${context ? `: ${context}` : ''}: ${error.message}`, error);

package/dist/middleware/services/tti/providers/google-cloud-provider.d.ts

@@ -8,6 +8,7 @@
  * - Imagen 4 Ultra (imagen-4.0-ultra-generate-001) - Highest quality variant
  * - Gemini 2.5 Flash Image - Text-to-image with character consistency
  * - Gemini 3 Pro Image (gemini-3-pro-image-preview) - 4K, text rendering
+ * - Gemini 3.1 Flash Image (gemini-3.1-flash-image-preview) - 4K, improved text rendering (global endpoint)
  *
  * All requests go through Google Cloud (Vertex AI) with proper DPA.
  * EU-compliant when using EU regions.

package/dist/middleware/services/tti/providers/google-cloud-provider.js

@@ -9,6 +9,7 @@
  * - Imagen 4 Ultra (imagen-4.0-ultra-generate-001) - Highest quality variant
  * - Gemini 2.5 Flash Image - Text-to-image with character consistency
  * - Gemini 3 Pro Image (gemini-3-pro-image-preview) - 4K, text rendering
+ * - Gemini 3.1 Flash Image (gemini-3.1-flash-image-preview) - 4K, improved text rendering (global endpoint)
  *
  * All requests go through Google Cloud (Vertex AI) with proper DPA.
  * EU-compliant when using EU regions.
@@ -163,6 +164,20 @@ const GOOGLE_CLOUD_MODELS = [
         availableRegions: ['global'],
         pricingUrl: 'https://cloud.google.com/vertex-ai/generative-ai/pricing',
     },
+    {
+        id: 'gemini-flash-image-2',
+        displayName: 'Gemini 3.1 Flash Image',
+        capabilities: {
+            textToImage: true,
+            characterConsistency: true, // Up to 5 characters + 14 objects
+            imageEditing: false,
+            maxImagesPerRequest: 1,
+        },
+        // Preview model — requires global endpoint (same as gemini-pro-image).
+        // Will likely get regional endpoints once GA.
+        availableRegions: ['global'],
+        pricingUrl: 'https://cloud.google.com/vertex-ai/generative-ai/pricing',
+    },
 ];
 // Internal model IDs used in Vertex AI API calls
 const MODEL_ID_MAP = {
@@ -172,9 +187,10 @@ const MODEL_ID_MAP = {
     'imagen-4-ultra': 'imagen-4.0-ultra-generate-001',
     'gemini-flash-image': 'gemini-2.5-flash-image',
     'gemini-pro-image': 'gemini-3-pro-image-preview',
+    'gemini-flash-image-2': 'gemini-3.1-flash-image-preview',
 };
 // Models that use the Gemini generateContent API (vs Imagen predict API)
-const GEMINI_API_MODELS = new Set(['gemini-flash-image', 'gemini-pro-image']);
+const GEMINI_API_MODELS = new Set(['gemini-flash-image', 'gemini-pro-image', 'gemini-flash-image-2']);
 // ============================================================
 // PROVIDER IMPLEMENTATION
 // ============================================================
@@ -553,11 +569,17 @@ class GoogleCloudTTIProvider extends base_tti_provider_1.BaseTTIProvider {
             const config = {
                 responseModalities: ['TEXT', 'IMAGE'],
             };
-            // Add imageConfig with aspectRatio if provided
-            if (request.aspectRatio) {
-                config.imageConfig = {
-                    aspectRatio: request.aspectRatio,
-                };
+            // Add imageConfig with aspectRatio and/or imageSize if provided
+            if (request.aspectRatio || request.providerOptions?.imageSize) {
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                const imageConfig = {};
+                if (request.aspectRatio) {
+                    imageConfig.aspectRatio = request.aspectRatio;
+                }
+                if (request.providerOptions?.imageSize) {
+                    imageConfig.imageSize = request.providerOptions.imageSize;
+                }
+                config.imageConfig = imageConfig;
             }
             // Add temperature if provided
             if (request.providerOptions?.temperature !== undefined) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@loonylabs/tti-middleware",
-  "version": "1.6.0",
+  "version": "1.8.0",
   "description": "Provider-agnostic Text-to-Image middleware with GDPR compliance. Supports Google Cloud (Imagen, Gemini), Eden AI, and IONOS.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",