@ai-sdk/google 3.0.78 → 3.0.80

package/dist/index.mjs

@@ -7,7 +7,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "3.0.78" : "0.0.0-test";
+var VERSION = true ? "3.0.80" : "0.0.0-test";
 
 // src/google-generative-ai-embedding-model.ts
 import {
@@ -412,6 +412,23 @@ import {
   UnsupportedFunctionalityError
 } from "@ai-sdk/provider";
 import { convertToBase64 } from "@ai-sdk/provider-utils";
+var SKIP_THOUGHT_SIGNATURE_VALIDATOR = "skip_thought_signature_validator";
+function getGoogleProviderOptions(providerOptions, providerOptionsName) {
+  const namespaces = [
+    providerOptionsName,
+    "google",
+    "googleVertex",
+    "vertex"
+  ].filter((namespace, index, allNamespaces) => {
+    return allNamespaces.indexOf(namespace) === index;
+  });
+  for (const namespace of namespaces) {
+    const options = providerOptions == null ? void 0 : providerOptions[namespace];
+    if (options != null) {
+      return options;
+    }
+  }
+}
 var dataUrlRegex = /^data:([^;,]+);base64,(.+)$/s;
 function parseBase64DataUrl(value) {
   const match = dataUrlRegex.exec(value);
@@ -519,13 +536,22 @@ function appendLegacyToolResultParts(parts, toolName, outputValue, toolCallId) {
   }
 }
 function convertToGoogleGenerativeAIMessages(prompt, options) {
-  var _a, _b, _c, _d, _e, _f, _g, _h;
+  var _a, _b, _c, _d, _e;
   const systemInstructionParts = [];
   const contents = [];
   let systemMessagesAllowed = true;
   const isGemmaModel = (_a = options == null ? void 0 : options.isGemmaModel) != null ? _a : false;
-  const providerOptionsName = (_b = options == null ? void 0 : options.providerOptionsName) != null ? _b : "google";
-  const supportsFunctionResponseParts = (_c = options == null ? void 0 : options.supportsFunctionResponseParts) != null ? _c : true;
+  const isGemini3Model = (_b = options == null ? void 0 : options.isGemini3Model) != null ? _b : false;
+  const providerOptionsName = (_c = options == null ? void 0 : options.providerOptionsName) != null ? _c : "google";
+  const supportsFunctionResponseParts = (_d = options == null ? void 0 : options.supportsFunctionResponseParts) != null ? _d : true;
+  const onWarning = options == null ? void 0 : options.onWarning;
+  let sentinelInjected = false;
+  const missingSignatureToolNames = [];
+  const injectSkipSignature = (toolName) => {
+    missingSignatureToolNames.push(toolName);
+    sentinelInjected = true;
+    return SKIP_THOUGHT_SIGNATURE_VALIDATOR;
+  };
   for (const { role, content } of prompt) {
     switch (role) {
       case "system": {
@@ -573,8 +599,10 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
         contents.push({
           role: "model",
           parts: content.map((part) => {
-            var _a2, _b2, _c2, _d2;
-            const providerOpts = (_d2 = (_a2 = part.providerOptions) == null ? void 0 : _a2[providerOptionsName]) != null ? _d2 : providerOptionsName !== "google" ? (_b2 = part.providerOptions) == null ? void 0 : _b2.google : (_c2 = part.providerOptions) == null ? void 0 : _c2.vertex;
+            const providerOpts = getGoogleProviderOptions(
+              part.providerOptions,
+              providerOptionsName
+            );
             const thoughtSignature = (providerOpts == null ? void 0 : providerOpts.thoughtSignature) != null ? String(providerOpts.thoughtSignature) : void 0;
             switch (part.type) {
               case "text": {
@@ -608,6 +636,7 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
               case "tool-call": {
                 const serverToolCallId = (providerOpts == null ? void 0 : providerOpts.serverToolCallId) != null ? String(providerOpts.serverToolCallId) : void 0;
                 const serverToolType = (providerOpts == null ? void 0 : providerOpts.serverToolType) != null ? String(providerOpts.serverToolType) : void 0;
+                const effectiveThoughtSignature = thoughtSignature != null ? thoughtSignature : isGemini3Model ? injectSkipSignature(part.toolName) : void 0;
                 if (serverToolCallId && serverToolType) {
                   return {
                     toolCall: {
@@ -615,7 +644,7 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
                       args: typeof part.input === "string" ? JSON.parse(part.input) : part.input,
                       id: serverToolCallId
                     },
-                    thoughtSignature
+                    thoughtSignature: effectiveThoughtSignature
                   };
                 }
                 return {
@@ -624,7 +653,7 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
                     name: part.toolName,
                     args: part.input
                   },
-                  thoughtSignature
+                  thoughtSignature: effectiveThoughtSignature
                 };
               }
               case "tool-result": {
@@ -654,7 +683,10 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
           if (part.type === "tool-approval-response") {
             continue;
           }
-          const partProviderOpts = (_g = (_d = part.providerOptions) == null ? void 0 : _d[providerOptionsName]) != null ? _g : providerOptionsName !== "google" ? (_e = part.providerOptions) == null ? void 0 : _e.google : (_f = part.providerOptions) == null ? void 0 : _f.vertex;
+          const partProviderOpts = getGoogleProviderOptions(
+            part.providerOptions,
+            providerOptionsName
+          );
           const serverToolCallId = (partProviderOpts == null ? void 0 : partProviderOpts.serverToolCallId) != null ? String(partProviderOpts.serverToolCallId) : void 0;
           const serverToolType = (partProviderOpts == null ? void 0 : partProviderOpts.serverToolType) != null ? String(partProviderOpts.serverToolType) : void 0;
           if (serverToolCallId && serverToolType) {
@@ -698,7 +730,7 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
                 name: part.toolName,
                 response: {
                   name: part.toolName,
-                  content: output.type === "execution-denied" ? (_h = output.reason) != null ? _h : "Tool execution denied." : output.value
+                  content: output.type === "execution-denied" ? (_e = output.reason) != null ? _e : "Tool execution denied." : output.value
                 }
               }
             });
@@ -716,6 +748,13 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
     const systemText = systemInstructionParts.map((part) => part.text).join("\n\n");
     contents[0].parts.unshift({ text: systemText + "\n\n" });
   }
+  if (sentinelInjected && onWarning != null) {
+    const uniqueToolNames = Array.from(new Set(missingSignatureToolNames));
+    onWarning({
+      type: "other",
+      message: `Replayed ${missingSignatureToolNames.length} \`functionCall\` part(s) for a Gemini 3 model without a \`thoughtSignature\` (tools: ${uniqueToolNames.map((name) => `\`${name}\``).join(", ")}). Injected the documented \`skip_thought_signature_validator\` sentinel to keep the request from failing with HTTP 400. The likely cause is application code that drops \`providerOptions.google.thoughtSignature\` when persisting or serializing assistant tool-call messages. See https://ai.google.dev/gemini-api/docs/thought-signatures.`
+    });
+  }
   return {
     systemInstruction: systemInstructionParts.length > 0 && !isGemmaModel ? { parts: systemInstructionParts } : void 0,
     contents
@@ -1474,13 +1513,16 @@ var GoogleGenerativeAILanguageModel = class {
     } : void 0;
     const bodyServiceTier = isVertexProvider ? void 0 : googleOptions == null ? void 0 : googleOptions.serviceTier;
     const isGemmaModel = this.modelId.toLowerCase().startsWith("gemma-");
-    const supportsFunctionResponseParts = this.modelId.startsWith("gemini-3");
+    const isGemini3Model = /^gemini-3[.-]/.test(this.modelId);
+    const supportsFunctionResponseParts = isGemini3Model;
     const { contents, systemInstruction } = convertToGoogleGenerativeAIMessages(
       prompt,
       {
         isGemmaModel,
+        isGemini3Model,
         providerOptionsName,
-        supportsFunctionResponseParts
+        supportsFunctionResponseParts,
+        onWarning: (warning) => warnings.push(warning)
       }
     );
     const {
@@ -2708,7 +2750,15 @@ var GoogleGenerativeAIImageModel = class {
       parameters.aspectRatio = aspectRatio;
     }
     if (googleOptions) {
-      Object.assign(parameters, googleOptions);
+      const { googleSearch: imagenGoogleSearch, ...imagenOptions } = googleOptions;
+      if (imagenGoogleSearch != null) {
+        warnings.push({
+          type: "unsupported",
+          feature: "googleSearch",
+          details: "Google Search grounding is only supported on Gemini image models."
+        });
+      }
+      Object.assign(parameters, imagenOptions);
     }
     const body = {
       instances: [{ prompt }],
@@ -2745,7 +2795,7 @@ var GoogleGenerativeAIImageModel = class {
     };
   }
   async doGenerateGemini(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
     const {
       prompt,
       n,
@@ -2800,12 +2850,18 @@ var GoogleGenerativeAIImageModel = class {
     const languageModelPrompt = [
       { role: "user", content: userContent }
     ];
+    const googleImageOptions = await parseProviderOptions3({
+      provider: "google",
+      providerOptions,
+      schema: googleImageModelOptionsSchema
+    });
+    const { googleSearch: _strippedGoogleSearch, ...passthroughGoogleOptions } = (_a = providerOptions == null ? void 0 : providerOptions.google) != null ? _a : {};
     const languageModel = new GoogleGenerativeAILanguageModel(this.modelId, {
       provider: this.config.provider,
       baseURL: this.config.baseURL,
-      headers: (_a = this.config.headers) != null ? _a : {},
+      headers: (_b = this.config.headers) != null ? _b : {},
       fetch: this.config.fetch,
-      generateId: (_b = this.config.generateId) != null ? _b : defaultGenerateId
+      generateId: (_c = this.config.generateId) != null ? _c : defaultGenerateId
     });
     const result = await languageModel.doGenerate({
       prompt: languageModelPrompt,
@@ -2816,9 +2872,17 @@ var GoogleGenerativeAIImageModel = class {
           imageConfig: aspectRatio ? {
             aspectRatio
           } : void 0,
-          ...(_c = providerOptions == null ? void 0 : providerOptions.google) != null ? _c : {}
+          ...passthroughGoogleOptions
         }
       },
+      tools: (googleImageOptions == null ? void 0 : googleImageOptions.googleSearch) != null ? [
+        {
+          type: "provider",
+          id: "google.google_search",
+          name: "google_search",
+          args: googleImageOptions.googleSearch
+        }
+      ] : void 0,
       headers,
       abortSignal
     });
@@ -2829,23 +2893,25 @@ var GoogleGenerativeAIImageModel = class {
         images.push(convertToBase642(part.data));
       }
     }
+    const languageModelGoogleMetadata = (_h = (_g = result.providerMetadata) == null ? void 0 : _g.google) != null ? _h : {};
     return {
       images,
       warnings,
       providerMetadata: {
         google: {
+          ...languageModelGoogleMetadata,
           images: images.map(() => ({}))
         }
       },
       response: {
         timestamp: currentDate,
         modelId: this.modelId,
-        headers: (_g = result.response) == null ? void 0 : _g.headers
+        headers: (_i = result.response) == null ? void 0 : _i.headers
       },
       usage: result.usage ? {
         inputTokens: result.usage.inputTokens.total,
         outputTokens: result.usage.outputTokens.total,
-        totalTokens: ((_h = result.usage.inputTokens.total) != null ? _h : 0) + ((_i = result.usage.outputTokens.total) != null ? _i : 0)
+        totalTokens: ((_j = result.usage.inputTokens.total) != null ? _j : 0) + ((_k = result.usage.outputTokens.total) != null ? _k : 0)
       } : void 0
     };
   }
@@ -2864,7 +2930,17 @@ var googleImageModelOptionsSchema = lazySchema11(
   () => zodSchema11(
     z13.object({
       personGeneration: z13.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
-      aspectRatio: z13.enum(["1:1", "3:4", "4:3", "9:16", "16:9"]).nullish()
+      aspectRatio: z13.enum(["1:1", "3:4", "4:3", "9:16", "16:9"]).nullish(),
+      /**
+       * Enable Google Search grounding for Gemini image models. The value is
+       * forwarded as the args of the `google.tools.googleSearch` provider
+       * tool on the underlying language-model call. Pass `{}` for defaults.
+       *
+       * `generateImage` does not accept a `tools` parameter, so this is the
+       * dedicated escape hatch for grounding image generation the same way
+       * `generateText` does.
+       */
+      googleSearch: googleSearchToolArgsBaseSchema.optional()
     })
   )
 );