@ai-sdk/google 4.0.0-beta.47 → 4.0.0-beta.49

package/dist/internal/index.js

@@ -179,7 +179,8 @@ import {
 } from "@ai-sdk/provider";
 import {
   convertToBase64,
-  isProviderReference,
+  isFullMediaType,
+  resolveFullMediaType,
   resolveProviderReference
 } from "@ai-sdk/provider-utils";
 var dataUrlRegex = /^data:([^;,]+);base64,(.+)$/s;
@@ -289,13 +290,25 @@ function appendLegacyToolResultParts(parts, toolName, outputValue) {
   }
 }
 function convertToGoogleMessages(prompt, options) {
-  var _a, _b, _c, _d, _e, _f, _g, _h;
+  var _a, _b, _c, _d;
   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 providerOptionsNames = (_b = options == null ? void 0 : options.providerOptionsNames) != null ? _b : ["google"];
+  const isVertexLike = !providerOptionsNames.includes("google");
   const supportsFunctionResponseParts = (_c = options == null ? void 0 : options.supportsFunctionResponseParts) != null ? _c : true;
+  const readProviderOpts = (part) => {
+    var _a2, _b2, _c2, _d2, _e;
+    for (const name of providerOptionsNames) {
+      const v = (_a2 = part.providerOptions) == null ? void 0 : _a2[name];
+      if (v != null) return v;
+    }
+    if (isVertexLike) {
+      return (_b2 = part.providerOptions) == null ? void 0 : _b2.google;
+    }
+    return (_e = (_c2 = part.providerOptions) == null ? void 0 : _c2.googleVertex) != null ? _e : (_d2 = part.providerOptions) == null ? void 0 : _d2.vertex;
+  };
   for (const { role, content } of prompt) {
     switch (role) {
       case "system": {
@@ -317,36 +330,53 @@ function convertToGoogleMessages(prompt, options) {
               break;
             }
             case "file": {
-              const mediaType = part.mediaType === "image/*" ? "image/jpeg" : part.mediaType;
-              if (part.data instanceof URL) {
-                parts.push({
-                  fileData: {
-                    mimeType: mediaType,
-                    fileUri: part.data.toString()
-                  }
-                });
-              } else if (isProviderReference(part.data)) {
-                if (providerOptionsName === "vertex") {
-                  throw new UnsupportedFunctionalityError({
-                    functionality: "file parts with provider references"
+              switch (part.data.type) {
+                case "url": {
+                  parts.push({
+                    fileData: {
+                      mimeType: resolveFullMediaType({ part }),
+                      fileUri: part.data.url.toString()
+                    }
                   });
+                  break;
                 }
-                parts.push({
-                  fileData: {
-                    mimeType: mediaType,
-                    fileUri: resolveProviderReference({
-                      reference: part.data,
-                      provider: "google"
-                    })
-                  }
-                });
-              } else {
-                parts.push({
-                  inlineData: {
-                    mimeType: mediaType,
-                    data: convertToBase64(part.data)
+                case "reference": {
+                  if (isVertexLike) {
+                    throw new UnsupportedFunctionalityError({
+                      functionality: "file parts with provider references"
+                    });
                   }
-                });
+                  parts.push({
+                    fileData: {
+                      mimeType: resolveFullMediaType({ part }),
+                      fileUri: resolveProviderReference({
+                        reference: part.data.reference,
+                        provider: "google"
+                      })
+                    }
+                  });
+                  break;
+                }
+                case "text": {
+                  parts.push({
+                    inlineData: {
+                      mimeType: isFullMediaType(part.mediaType) ? part.mediaType : "text/plain",
+                      data: convertToBase64(
+                        new TextEncoder().encode(part.data.text)
+                      )
+                    }
+                  });
+                  break;
+                }
+                case "data": {
+                  parts.push({
+                    inlineData: {
+                      mimeType: resolveFullMediaType({ part }),
+                      data: convertToBase64(part.data.data)
+                    }
+                  });
+                  break;
+                }
               }
               break;
             }
@@ -360,8 +390,7 @@ function convertToGoogleMessages(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 = readProviderOpts(part);
             const thoughtSignature = (providerOpts == null ? void 0 : providerOpts.thoughtSignature) != null ? String(providerOpts.thoughtSignature) : void 0;
             switch (part.type) {
               case "text": {
@@ -378,52 +407,74 @@ function convertToGoogleMessages(prompt, options) {
                 };
               }
               case "reasoning-file": {
-                if (part.data instanceof URL) {
-                  throw new UnsupportedFunctionalityError({
-                    functionality: "File data URLs in assistant messages are not supported"
-                  });
+                switch (part.data.type) {
+                  case "url": {
+                    throw new UnsupportedFunctionalityError({
+                      functionality: "File data URLs in assistant messages are not supported"
+                    });
+                  }
+                  case "data": {
+                    return {
+                      inlineData: {
+                        mimeType: part.mediaType,
+                        data: convertToBase64(part.data.data)
+                      },
+                      thought: true,
+                      thoughtSignature
+                    };
+                  }
                 }
-                return {
-                  inlineData: {
-                    mimeType: part.mediaType,
-                    data: convertToBase64(part.data)
-                  },
-                  thought: true,
-                  thoughtSignature
-                };
+                break;
               }
               case "file": {
-                if (part.data instanceof URL) {
-                  throw new UnsupportedFunctionalityError({
-                    functionality: "File data URLs in assistant messages are not supported"
-                  });
-                }
-                if (isProviderReference(part.data)) {
-                  if (providerOptionsName === "vertex") {
+                switch (part.data.type) {
+                  case "url": {
                     throw new UnsupportedFunctionalityError({
-                      functionality: "file parts with provider references"
+                      functionality: "File data URLs in assistant messages are not supported"
                     });
                   }
-                  return {
-                    fileData: {
-                      mimeType: part.mediaType,
-                      fileUri: resolveProviderReference({
-                        reference: part.data,
-                        provider: "google"
-                      })
-                    },
-                    ...(providerOpts == null ? void 0 : providerOpts.thought) === true ? { thought: true } : {},
-                    thoughtSignature
-                  };
+                  case "reference": {
+                    if (isVertexLike) {
+                      throw new UnsupportedFunctionalityError({
+                        functionality: "file parts with provider references"
+                      });
+                    }
+                    return {
+                      fileData: {
+                        mimeType: part.mediaType,
+                        fileUri: resolveProviderReference({
+                          reference: part.data.reference,
+                          provider: "google"
+                        })
+                      },
+                      ...(providerOpts == null ? void 0 : providerOpts.thought) === true ? { thought: true } : {},
+                      thoughtSignature
+                    };
+                  }
+                  case "text": {
+                    return {
+                      inlineData: {
+                        mimeType: isFullMediaType(part.mediaType) ? part.mediaType : "text/plain",
+                        data: convertToBase64(
+                          new TextEncoder().encode(part.data.text)
+                        )
+                      },
+                      ...(providerOpts == null ? void 0 : providerOpts.thought) === true ? { thought: true } : {},
+                      thoughtSignature
+                    };
+                  }
+                  case "data": {
+                    return {
+                      inlineData: {
+                        mimeType: part.mediaType,
+                        data: convertToBase64(part.data.data)
+                      },
+                      ...(providerOpts == null ? void 0 : providerOpts.thought) === true ? { thought: true } : {},
+                      thoughtSignature
+                    };
+                  }
                 }
-                return {
-                  inlineData: {
-                    mimeType: part.mediaType,
-                    data: convertToBase64(part.data)
-                  },
-                  ...(providerOpts == null ? void 0 : providerOpts.thought) === true ? { thought: true } : {},
-                  thoughtSignature
-                };
+                break;
               }
               case "tool-call": {
                 const serverToolCallId = (providerOpts == null ? void 0 : providerOpts.serverToolCallId) != null ? String(providerOpts.serverToolCallId) : void 0;
@@ -473,7 +524,7 @@ function convertToGoogleMessages(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 = readProviderOpts(part);
           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) {
@@ -506,7 +557,7 @@ function convertToGoogleMessages(prompt, options) {
                 name: part.toolName,
                 response: {
                   name: part.toolName,
-                  content: output.type === "execution-denied" ? (_h = output.reason) != null ? _h : "Tool call execution denied." : output.value
+                  content: output.type === "execution-denied" ? (_d = output.reason) != null ? _d : "Tool call execution denied." : output.value
                 }
               }
             });
@@ -558,8 +609,11 @@ var googleFailedResponseHandler = createJsonErrorResponseHandler({
   errorToMessage: (data) => data.error.message
 });
 
-// src/google-options.ts
-import { lazySchema as lazySchema2, zodSchema as zodSchema2 } from "@ai-sdk/provider-utils";
+// src/google-language-model-options.ts
+import {
+  lazySchema as lazySchema2,
+  zodSchema as zodSchema2
+} from "@ai-sdk/provider-utils";
 import { z as z2 } from "zod/v4";
 var googleLanguageModelOptions = lazySchema2(
   () => zodSchema2(
@@ -1247,13 +1301,17 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
   }, { isStreaming = false } = {}) {
     var _a, _b;
     const warnings = [];
-    const providerOptionsName = this.config.provider.includes("vertex") ? "vertex" : "google";
-    let googleOptions = await parseProviderOptions({
-      provider: providerOptionsName,
-      providerOptions,
-      schema: googleLanguageModelOptions
-    });
-    if (googleOptions == null && providerOptionsName !== "google") {
+    const providerOptionsNames = this.config.provider.includes("vertex") ? ["googleVertex", "vertex"] : ["google"];
+    let googleOptions;
+    for (const name of providerOptionsNames) {
+      googleOptions = await parseProviderOptions({
+        provider: name,
+        providerOptions,
+        schema: googleLanguageModelOptions
+      });
+      if (googleOptions != null) break;
+    }
+    if (googleOptions == null && !providerOptionsNames.includes("google")) {
       googleOptions = await parseProviderOptions({
         provider: "google",
         providerOptions,
@@ -1283,7 +1341,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
     const supportsFunctionResponseParts = this.modelId.startsWith("gemini-3");
     const { contents, systemInstruction } = convertToGoogleMessages(prompt, {
       isGemmaModel,
-      providerOptionsName,
+      providerOptionsNames,
       supportsFunctionResponseParts
     });
     const {
@@ -1356,12 +1414,15 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
         serviceTier: sanitizedServiceTier
       },
       warnings: [...warnings, ...toolWarnings],
-      providerOptionsName
+      providerOptionsNames
     };
   }
   async doGenerate(options) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p;
-    const { args, warnings, providerOptionsName } = await this.getArgs(options);
+    const { args, warnings, providerOptionsNames } = await this.getArgs(options);
+    const wrapProviderMetadata = (payload) => Object.fromEntries(
+      providerOptionsNames.map((name) => [name, payload])
+    );
     const mergedHeaders = combineHeaders(
       this.config.headers ? await resolve(this.config.headers) : void 0,
       options.headers
@@ -1411,11 +1472,9 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
         });
         lastCodeExecutionToolCallId = void 0;
       } else if ("text" in part && part.text != null) {
-        const thoughtSignatureMetadata = part.thoughtSignature ? {
-          [providerOptionsName]: {
-            thoughtSignature: part.thoughtSignature
-          }
-        } : void 0;
+        const thoughtSignatureMetadata = part.thoughtSignature ? wrapProviderMetadata({
+          thoughtSignature: part.thoughtSignature
+        }) : void 0;
         if (part.text.length === 0) {
           if (thoughtSignatureMetadata != null && content.length > 0) {
             const lastContent = content[content.length - 1];
@@ -1434,24 +1493,20 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
           toolCallId: this.config.generateId(),
           toolName: part.functionCall.name,
           input: JSON.stringify(part.functionCall.args),
-          providerMetadata: part.thoughtSignature ? {
-            [providerOptionsName]: {
-              thoughtSignature: part.thoughtSignature
-            }
-          } : void 0
+          providerMetadata: part.thoughtSignature ? wrapProviderMetadata({
+            thoughtSignature: part.thoughtSignature
+          }) : void 0
         });
       } else if ("inlineData" in part) {
         const hasThought = part.thought === true;
         const hasThoughtSignature = !!part.thoughtSignature;
         content.push({
           type: hasThought ? "reasoning-file" : "file",
-          data: part.inlineData.data,
+          data: { type: "data", data: part.inlineData.data },
           mediaType: part.inlineData.mimeType,
-          providerMetadata: hasThoughtSignature ? {
-            [providerOptionsName]: {
-              thoughtSignature: part.thoughtSignature
-            }
-          } : void 0
+          providerMetadata: hasThoughtSignature ? wrapProviderMetadata({
+            thoughtSignature: part.thoughtSignature
+          }) : void 0
         });
       } else if ("toolCall" in part && part.toolCall) {
         const toolCallId = (_e = part.toolCall.id) != null ? _e : this.config.generateId();
@@ -1463,18 +1518,14 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
           input: JSON.stringify((_f = part.toolCall.args) != null ? _f : {}),
           providerExecuted: true,
           dynamic: true,
-          providerMetadata: part.thoughtSignature ? {
-            [providerOptionsName]: {
-              thoughtSignature: part.thoughtSignature,
-              serverToolCallId: toolCallId,
-              serverToolType: part.toolCall.toolType
-            }
-          } : {
-            [providerOptionsName]: {
-              serverToolCallId: toolCallId,
-              serverToolType: part.toolCall.toolType
-            }
-          }
+          providerMetadata: part.thoughtSignature ? wrapProviderMetadata({
+            thoughtSignature: part.thoughtSignature,
+            serverToolCallId: toolCallId,
+            serverToolType: part.toolCall.toolType
+          }) : wrapProviderMetadata({
+            serverToolCallId: toolCallId,
+            serverToolType: part.toolCall.toolType
+          })
         });
       } else if ("toolResponse" in part && part.toolResponse) {
         const responseToolCallId = (_g = lastServerToolCallId != null ? lastServerToolCallId : part.toolResponse.id) != null ? _g : this.config.generateId();
@@ -1483,18 +1534,14 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
           toolCallId: responseToolCallId,
           toolName: `server:${part.toolResponse.toolType}`,
           result: (_h = part.toolResponse.response) != null ? _h : {},
-          providerMetadata: part.thoughtSignature ? {
-            [providerOptionsName]: {
-              thoughtSignature: part.thoughtSignature,
-              serverToolCallId: responseToolCallId,
-              serverToolType: part.toolResponse.toolType
-            }
-          } : {
-            [providerOptionsName]: {
-              serverToolCallId: responseToolCallId,
-              serverToolType: part.toolResponse.toolType
-            }
-          }
+          providerMetadata: part.thoughtSignature ? wrapProviderMetadata({
+            thoughtSignature: part.thoughtSignature,
+            serverToolCallId: responseToolCallId,
+            serverToolType: part.toolResponse.toolType
+          }) : wrapProviderMetadata({
+            serverToolCallId: responseToolCallId,
+            serverToolType: part.toolResponse.toolType
+          })
         });
         lastServerToolCallId = void 0;
       }
@@ -1520,17 +1567,15 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
       },
       usage: convertGoogleUsage(usageMetadata),
       warnings,
-      providerMetadata: {
-        [providerOptionsName]: {
-          promptFeedback: (_k = response.promptFeedback) != null ? _k : null,
-          groundingMetadata: (_l = candidate.groundingMetadata) != null ? _l : null,
-          urlContextMetadata: (_m = candidate.urlContextMetadata) != null ? _m : null,
-          safetyRatings: (_n = candidate.safetyRatings) != null ? _n : null,
-          usageMetadata: usageMetadata != null ? usageMetadata : null,
-          finishMessage: (_o = candidate.finishMessage) != null ? _o : null,
-          serviceTier: (_p = response.serviceTier) != null ? _p : null
-        }
-      },
+      providerMetadata: wrapProviderMetadata({
+        promptFeedback: (_k = response.promptFeedback) != null ? _k : null,
+        groundingMetadata: (_l = candidate.groundingMetadata) != null ? _l : null,
+        urlContextMetadata: (_m = candidate.urlContextMetadata) != null ? _m : null,
+        safetyRatings: (_n = candidate.safetyRatings) != null ? _n : null,
+        usageMetadata: usageMetadata != null ? usageMetadata : null,
+        finishMessage: (_o = candidate.finishMessage) != null ? _o : null,
+        serviceTier: (_p = response.serviceTier) != null ? _p : null
+      }),
       request: { body: args },
       response: {
         // TODO timestamp, model id, id
@@ -1540,10 +1585,13 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
     };
   }
   async doStream(options) {
-    const { args, warnings, providerOptionsName } = await this.getArgs(
+    const { args, warnings, providerOptionsNames } = await this.getArgs(
       options,
       { isStreaming: true }
     );
+    const wrapProviderMetadata = (payload) => Object.fromEntries(
+      providerOptionsNames.map((name) => [name, payload])
+    );
     const headers = combineHeaders(
       this.config.headers ? await resolve(this.config.headers) : void 0,
       options.headers
@@ -1651,11 +1699,9 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
                     lastCodeExecutionToolCallId = void 0;
                   }
                 } else if ("text" in part && part.text != null) {
-                  const thoughtSignatureMetadata = part.thoughtSignature ? {
-                    [providerOptionsName]: {
-                      thoughtSignature: part.thoughtSignature
-                    }
-                  } : void 0;
+                  const thoughtSignatureMetadata = part.thoughtSignature ? wrapProviderMetadata({
+                    thoughtSignature: part.thoughtSignature
+                  }) : void 0;
                   if (part.text.length === 0) {
                     if (thoughtSignatureMetadata != null && currentTextBlockId !== null) {
                       controller.enqueue({
@@ -1727,27 +1773,23 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
                   }
                   const hasThought = part.thought === true;
                   const hasThoughtSignature = !!part.thoughtSignature;
-                  const fileMeta = hasThoughtSignature ? {
-                    [providerOptionsName]: {
-                      thoughtSignature: part.thoughtSignature
-                    }
-                  } : void 0;
+                  const fileMeta = hasThoughtSignature ? wrapProviderMetadata({
+                    thoughtSignature: part.thoughtSignature
+                  }) : void 0;
                   controller.enqueue({
                     type: hasThought ? "reasoning-file" : "file",
                     mediaType: part.inlineData.mimeType,
-                    data: part.inlineData.data,
+                    data: { type: "data", data: part.inlineData.data },
                     providerMetadata: fileMeta
                   });
                 } else if ("toolCall" in part && part.toolCall) {
                   const toolCallId = (_e = part.toolCall.id) != null ? _e : generateId2();
                   lastServerToolCallId = toolCallId;
-                  const serverMeta = {
-                    [providerOptionsName]: {
-                      ...part.thoughtSignature ? { thoughtSignature: part.thoughtSignature } : {},
-                      serverToolCallId: toolCallId,
-                      serverToolType: part.toolCall.toolType
-                    }
-                  };
+                  const serverMeta = wrapProviderMetadata({
+                    ...part.thoughtSignature ? { thoughtSignature: part.thoughtSignature } : {},
+                    serverToolCallId: toolCallId,
+                    serverToolType: part.toolCall.toolType
+                  });
                   controller.enqueue({
                     type: "tool-call",
                     toolCallId,
@@ -1759,13 +1801,11 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
                   });
                 } else if ("toolResponse" in part && part.toolResponse) {
                   const responseToolCallId = (_g = lastServerToolCallId != null ? lastServerToolCallId : part.toolResponse.id) != null ? _g : generateId2();
-                  const serverMeta = {
-                    [providerOptionsName]: {
-                      ...part.thoughtSignature ? { thoughtSignature: part.thoughtSignature } : {},
-                      serverToolCallId: responseToolCallId,
-                      serverToolType: part.toolResponse.toolType
-                    }
-                  };
+                  const serverMeta = wrapProviderMetadata({
+                    ...part.thoughtSignature ? { thoughtSignature: part.thoughtSignature } : {},
+                    serverToolCallId: responseToolCallId,
+                    serverToolType: part.toolResponse.toolType
+                  });
                   controller.enqueue({
                     type: "tool-result",
                     toolCallId: responseToolCallId,
@@ -1778,11 +1818,9 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
               }
               for (const part of parts) {
                 if (!("functionCall" in part)) continue;
-                const providerMeta = part.thoughtSignature ? {
-                  [providerOptionsName]: {
-                    thoughtSignature: part.thoughtSignature
-                  }
-                } : void 0;
+                const providerMeta = part.thoughtSignature ? wrapProviderMetadata({
+                  thoughtSignature: part.thoughtSignature
+                }) : void 0;
                 const isStreamingChunk = part.functionCall.partialArgs != null || part.functionCall.name != null && part.functionCall.willContinue === true;
                 const isTerminalChunk = part.functionCall.name == null && part.functionCall.args == null && part.functionCall.partialArgs == null && part.functionCall.willContinue == null;
                 const isCompleteCall = part.functionCall.name != null && part.functionCall.args != null && part.functionCall.partialArgs == null;
@@ -1893,17 +1931,15 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
                 }),
                 raw: candidate.finishReason
               };
-              providerMetadata = {
-                [providerOptionsName]: {
-                  promptFeedback: (_j = value.promptFeedback) != null ? _j : null,
-                  groundingMetadata: lastGroundingMetadata,
-                  urlContextMetadata: lastUrlContextMetadata,
-                  safetyRatings: (_k = candidate.safetyRatings) != null ? _k : null,
-                  usageMetadata: usageMetadata != null ? usageMetadata : null,
-                  finishMessage: (_l = candidate.finishMessage) != null ? _l : null,
-                  serviceTier
-                }
-              };
+              providerMetadata = wrapProviderMetadata({
+                promptFeedback: (_j = value.promptFeedback) != null ? _j : null,
+                groundingMetadata: lastGroundingMetadata,
+                urlContextMetadata: lastUrlContextMetadata,
+                safetyRatings: (_k = candidate.safetyRatings) != null ? _k : null,
+                usageMetadata: usageMetadata != null ? usageMetadata : null,
+                finishMessage: (_l = candidate.finishMessage) != null ? _l : null,
+                serviceTier
+              });
             }
           },
           flush(controller) {