@threaded/ai 1.0.12 → 1.0.14

package/dist/index.cjs

@@ -30,6 +30,7 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  IMAGE_MODEL_SCHEMA: () => IMAGE_MODEL_SCHEMA,
   Inherit: () => Inherit,
   appendToLastRequest: () => appendToLastRequest,
   compose: () => compose,
@@ -42,7 +43,9 @@ __export(index_exports, {
   everyNTokens: () => everyNTokens,
   generateApprovalToken: () => generateApprovalToken,
   generateImage: () => generateImage,
+  getDefaultConfig: () => getDefaultConfig,
   getKey: () => getKey,
+  getModelConfig: () => getModelConfig,
   getOrCreateThread: () => getOrCreateThread,
   isStandardSchema: () => isStandardSchema,
   maxCalls: () => maxCalls,
@@ -301,6 +304,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",
@@ -332,6 +336,9 @@ var generateGoogle = async (modelName, prompt, apiKey, config) => {
   if (config?.aspectRatio) {
     body.generationConfig.aspectRatio = config.aspectRatio;
   }
+  if (config?.imageSize) {
+    body.generationConfig.imageSize = config.imageSize;
+  }
   const response = await fetch(endpoint, {
     method: "POST",
     headers: {
@@ -384,6 +391,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;
@@ -760,23 +861,27 @@ var callGoogle = async (config, ctx) => {
   for (let i = 0; i < ctx.history.length; i++) {
     const msg2 = ctx.history[i];
     if (msg2.role === "assistant") {
-      const parts = [];
+      const parts2 = [];
       if (msg2.content) {
-        parts.push({ text: msg2.content });
+        parts2.push({ text: msg2.content });
       }
       if (msg2.tool_calls?.length) {
         for (const tc of msg2.tool_calls) {
           toolCallMap.set(tc.id, tc.function.name);
-          parts.push({
+          const part = {
             functionCall: {
               name: tc.function.name,
               args: JSON.parse(tc.function.arguments)
             }
-          });
+          };
+          if (tc.thoughtSignature) {
+            part.thoughtSignature = tc.thoughtSignature;
+          }
+          parts2.push(part);
         }
       }
-      if (parts.length > 0) {
-        contents.push({ role: "model", parts });
+      if (parts2.length > 0) {
+        contents.push({ role: "model", parts: parts2 });
       }
     } else if (msg2.role === "tool") {
       const responseParts = [];
@@ -845,22 +950,33 @@ var callGoogle = async (config, ctx) => {
   }
   const data = await response.json();
   const candidate = data.candidates[0];
-  const part = candidate.content.parts[0];
+  const parts = candidate.content.parts || [];
   const msg = {
     role: "assistant",
-    content: part.text || ""
+    content: ""
   };
-  if (part.functionCall) {
-    msg.tool_calls = [
-      {
+  const toolCalls = [];
+  for (const part of parts) {
+    if (part.text) {
+      msg.content += part.text;
+    }
+    if (part.functionCall) {
+      const tc = {
         id: Math.random().toString(36).substring(2, 9),
         type: "function",
         function: {
           name: part.functionCall.name,
           arguments: JSON.stringify(part.functionCall.args)
         }
+      };
+      if (part.thoughtSignature) {
+        tc.thoughtSignature = part.thoughtSignature;
       }
-    ];
+      toolCalls.push(tc);
+    }
+  }
+  if (toolCalls.length > 0) {
+    msg.tool_calls = toolCalls;
   }
   return {
     ...ctx,
@@ -900,14 +1016,18 @@ var handleGoogleStream = async (response, ctx) => {
                 }
               }
               if (part?.functionCall) {
-                toolCalls.push({
+                const tc = {
                   id: Math.random().toString(36).substring(2, 9),
                   type: "function",
                   function: {
                     name: part.functionCall.name,
                     arguments: JSON.stringify(part.functionCall.args)
                   }
-                });
+                };
+                if (part.thoughtSignature) {
+                  tc.thoughtSignature = part.thoughtSignature;
+                }
+                toolCalls.push(tc);
               }
             }
           } catch (e) {
@@ -1693,6 +1813,7 @@ var rateLimited = (config) => (fn) => {
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  IMAGE_MODEL_SCHEMA,
   Inherit,
   appendToLastRequest,
   compose,
@@ -1705,7 +1826,9 @@ var rateLimited = (config) => (fn) => {
   everyNTokens,
   generateApprovalToken,
   generateImage,
+  getDefaultConfig,
   getKey,
+  getModelConfig,
   getOrCreateThread,
   isStandardSchema,
   maxCalls,