@threaded/ai 1.0.13 → 1.0.15

package/dist/index.js

@@ -232,6 +232,7 @@ var generateOpenAICompatible = async (endpoint, modelName, prompt, apiKey, confi
     if (config?.outputFormat) body.output_format = config.outputFormat;
     if (config?.outputCompression != null) body.output_compression = config.outputCompression;
     if (config?.background) body.background = config.background;
+    if (config?.moderation) body.moderation = config.moderation;
   }
   const response = await fetch(endpoint, {
     method: "POST",
@@ -260,8 +261,11 @@ var generateGoogle = async (modelName, prompt, apiKey, config) => {
       responseModalities: ["TEXT", "IMAGE"]
     }
   };
-  if (config?.aspectRatio) {
-    body.generationConfig.aspectRatio = config.aspectRatio;
+  const imageConfig = {};
+  if (config?.aspectRatio) imageConfig.aspectRatio = config.aspectRatio;
+  if (config?.imageSize) imageConfig.imageSize = config.imageSize;
+  if (Object.keys(imageConfig).length > 0) {
+    body.generationConfig.imageGenerationConfig = imageConfig;
   }
   const response = await fetch(endpoint, {
     method: "POST",
@@ -315,6 +319,100 @@ var generateImage = async (model2, prompt, config) => {
   }
 };
 
+// src/image-model-schema.ts
+var IMAGE_MODEL_SCHEMA = {
+  openai: {
+    "dall-e-3": {
+      size: {
+        values: ["1024x1024", "1024x1792", "1792x1024"],
+        default: "1024x1024",
+        description: "Image dimensions"
+      },
+      quality: {
+        values: ["standard", "hd"],
+        default: "standard",
+        description: "Image quality level"
+      },
+      style: {
+        values: ["vivid", "natural"],
+        default: "vivid",
+        description: "Image style"
+      }
+    },
+    "gpt-image-1.5": {
+      size: {
+        values: ["1024x1024", "1536x1024", "1024x1536", "auto"],
+        default: "auto",
+        description: "Image dimensions"
+      },
+      quality: {
+        values: ["low", "medium", "high", "auto"],
+        default: "auto",
+        description: "Image quality level"
+      },
+      background: {
+        values: ["transparent", "opaque", "auto"],
+        default: "auto",
+        description: "Background type"
+      },
+      moderation: {
+        values: ["auto", "low"],
+        default: "auto",
+        description: "Content moderation level"
+      }
+    }
+  },
+  google: {
+    "gemini-2.5-flash-image": {
+      aspectRatio: {
+        values: ["1:1", "3:4", "4:3", "9:16", "16:9"],
+        default: "1:1",
+        description: "Image aspect ratio"
+      }
+    },
+    "gemini-3-pro-image-preview": {
+      aspectRatio: {
+        values: ["1:1", "3:4", "4:3", "9:16", "16:9"],
+        default: "1:1",
+        description: "Image aspect ratio"
+      },
+      imageSize: {
+        values: ["1K", "2K"],
+        default: "1K",
+        description: "Output image size"
+      }
+    },
+    "nano-banana-pro-preview": {
+      aspectRatio: {
+        values: ["1:1", "3:4", "4:3", "9:16", "16:9"],
+        default: "1:1",
+        description: "Image aspect ratio"
+      }
+    }
+  },
+  xai: {
+    "grok-2-image-1212": {
+      size: {
+        values: ["1024x1024"],
+        default: "1024x1024",
+        description: "Image dimensions"
+      }
+    }
+  }
+};
+function getModelConfig(provider, model2) {
+  return IMAGE_MODEL_SCHEMA[provider]?.[model2] || null;
+}
+function getDefaultConfig(provider, model2) {
+  const schema = getModelConfig(provider, model2);
+  if (!schema) return {};
+  const defaults = {};
+  for (const [key, option] of Object.entries(schema)) {
+    defaults[key] = option.default;
+  }
+  return defaults;
+}
+
 // src/providers/openai.ts
 var getApiKey2 = (configApiKey) => {
   if (configApiKey) return configApiKey;
@@ -1642,6 +1740,7 @@ var rateLimited = (config) => (fn) => {
   });
 };
 export {
+  IMAGE_MODEL_SCHEMA,
   Inherit,
   appendToLastRequest,
   compose,
@@ -1654,7 +1753,9 @@ export {
   everyNTokens,
   generateApprovalToken,
   generateImage,
+  getDefaultConfig,
   getKey,
+  getModelConfig,
   getOrCreateThread,
   isStandardSchema,
   maxCalls,