@ai-sdk/google 3.0.50 → 3.0.52

package/dist/internal/index.mjs

@@ -172,13 +172,118 @@ import {
   UnsupportedFunctionalityError
 } from "@ai-sdk/provider";
 import { convertToBase64 } from "@ai-sdk/provider-utils";
+var dataUrlRegex = /^data:([^;,]+);base64,(.+)$/s;
+function parseBase64DataUrl(value) {
+  const match = dataUrlRegex.exec(value);
+  if (match == null) {
+    return void 0;
+  }
+  return {
+    mediaType: match[1],
+    data: match[2]
+  };
+}
+function convertUrlToolResultPart(url) {
+  const parsedDataUrl = parseBase64DataUrl(url);
+  if (parsedDataUrl == null) {
+    return void 0;
+  }
+  return {
+    inlineData: {
+      mimeType: parsedDataUrl.mediaType,
+      data: parsedDataUrl.data
+    }
+  };
+}
+function appendToolResultParts(parts, toolName, outputValue) {
+  const functionResponseParts = [];
+  const responseTextParts = [];
+  for (const contentPart of outputValue) {
+    switch (contentPart.type) {
+      case "text": {
+        responseTextParts.push(contentPart.text);
+        break;
+      }
+      case "image-data":
+      case "file-data": {
+        functionResponseParts.push({
+          inlineData: {
+            mimeType: contentPart.mediaType,
+            data: contentPart.data
+          }
+        });
+        break;
+      }
+      case "image-url":
+      case "file-url": {
+        const functionResponsePart = convertUrlToolResultPart(
+          contentPart.url
+        );
+        if (functionResponsePart != null) {
+          functionResponseParts.push(functionResponsePart);
+        } else {
+          responseTextParts.push(JSON.stringify(contentPart));
+        }
+        break;
+      }
+      default: {
+        responseTextParts.push(JSON.stringify(contentPart));
+        break;
+      }
+    }
+  }
+  parts.push({
+    functionResponse: {
+      name: toolName,
+      response: {
+        name: toolName,
+        content: responseTextParts.length > 0 ? responseTextParts.join("\n") : "Tool executed successfully."
+      },
+      ...functionResponseParts.length > 0 ? { parts: functionResponseParts } : {}
+    }
+  });
+}
+function appendLegacyToolResultParts(parts, toolName, outputValue) {
+  for (const contentPart of outputValue) {
+    switch (contentPart.type) {
+      case "text":
+        parts.push({
+          functionResponse: {
+            name: toolName,
+            response: {
+              name: toolName,
+              content: contentPart.text
+            }
+          }
+        });
+        break;
+      case "image-data":
+        parts.push(
+          {
+            inlineData: {
+              mimeType: String(contentPart.mediaType),
+              data: String(contentPart.data)
+            }
+          },
+          {
+            text: "Tool executed successfully and returned this image as a response"
+          }
+        );
+        break;
+      default:
+        parts.push({ text: JSON.stringify(contentPart) });
+        break;
+    }
+  }
+}
 function convertToGoogleGenerativeAIMessages(prompt, options) {
-  var _a, _b, _c;
+  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 supportsFunctionResponseParts = (_c = options == null ? void 0 : options.supportsFunctionResponseParts) != null ? _c : true;
   for (const { role, content } of prompt) {
     switch (role) {
       case "system": {
@@ -226,8 +331,8 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
         contents.push({
           role: "model",
           parts: content.map((part) => {
-            var _a2, _b2, _c2, _d;
-            const providerOpts = (_d = (_a2 = part.providerOptions) == null ? void 0 : _a2[providerOptionsName]) != null ? _d : providerOptionsName !== "google" ? (_b2 = part.providerOptions) == null ? void 0 : _b2.google : (_c2 = part.providerOptions) == null ? void 0 : _c2.vertex;
+            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 thoughtSignature = (providerOpts == null ? void 0 : providerOpts.thoughtSignature) != null ? String(providerOpts.thoughtSignature) : void 0;
             switch (part.type) {
               case "text": {
@@ -281,36 +386,10 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
           }
           const output = part.output;
           if (output.type === "content") {
-            for (const contentPart of output.value) {
-              switch (contentPart.type) {
-                case "text":
-                  parts.push({
-                    functionResponse: {
-                      name: part.toolName,
-                      response: {
-                        name: part.toolName,
-                        content: contentPart.text
-                      }
-                    }
-                  });
-                  break;
-                case "image-data":
-                  parts.push(
-                    {
-                      inlineData: {
-                        mimeType: contentPart.mediaType,
-                        data: contentPart.data
-                      }
-                    },
-                    {
-                      text: "Tool executed successfully and returned this image as a response"
-                    }
-                  );
-                  break;
-                default:
-                  parts.push({ text: JSON.stringify(contentPart) });
-                  break;
-              }
+            if (supportsFunctionResponseParts) {
+              appendToolResultParts(parts, part.toolName, output.value);
+            } else {
+              appendLegacyToolResultParts(parts, part.toolName, output.value);
             }
           } else {
             parts.push({
@@ -318,7 +397,7 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
                 name: part.toolName,
                 response: {
                   name: part.toolName,
-                  content: output.type === "execution-denied" ? (_c = output.reason) != null ? _c : "Tool execution denied." : output.value
+                  content: output.type === "execution-denied" ? (_d = output.reason) != null ? _d : "Tool execution denied." : output.value
                 }
               }
             });
@@ -785,9 +864,14 @@ var GoogleGenerativeAILanguageModel = class {
       });
     }
     const isGemmaModel = this.modelId.toLowerCase().startsWith("gemma-");
+    const supportsFunctionResponseParts = this.modelId.startsWith("gemini-3");
     const { contents, systemInstruction } = convertToGoogleGenerativeAIMessages(
       prompt,
-      { isGemmaModel, providerOptionsName }
+      {
+        isGemmaModel,
+        providerOptionsName,
+        supportsFunctionResponseParts
+      }
     );
     const {
       tools: googleTools2,
@@ -845,7 +929,7 @@ var GoogleGenerativeAILanguageModel = class {
     };
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
     const { args, warnings, providerOptionsName } = await this.getArgs(options);
     const mergedHeaders = combineHeaders(
       await resolve(this.config.headers),
@@ -967,7 +1051,8 @@ var GoogleGenerativeAILanguageModel = class {
           groundingMetadata: (_h = candidate.groundingMetadata) != null ? _h : null,
           urlContextMetadata: (_i = candidate.urlContextMetadata) != null ? _i : null,
           safetyRatings: (_j = candidate.safetyRatings) != null ? _j : null,
-          usageMetadata: usageMetadata != null ? usageMetadata : null
+          usageMetadata: usageMetadata != null ? usageMetadata : null,
+          finishMessage: (_k = candidate.finishMessage) != null ? _k : null
         }
       },
       request: { body: args },
@@ -1017,7 +1102,7 @@ var GoogleGenerativeAILanguageModel = class {
             controller.enqueue({ type: "stream-start", warnings });
           },
           transform(chunk, controller) {
-            var _a, _b, _c, _d, _e, _f;
+            var _a, _b, _c, _d, _e, _f, _g;
             if (options.includeRawChunks) {
               controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
             }
@@ -1219,12 +1304,11 @@ var GoogleGenerativeAILanguageModel = class {
                   promptFeedback: (_e = value.promptFeedback) != null ? _e : null,
                   groundingMetadata: lastGroundingMetadata,
                   urlContextMetadata: lastUrlContextMetadata,
-                  safetyRatings: (_f = candidate.safetyRatings) != null ? _f : null
+                  safetyRatings: (_f = candidate.safetyRatings) != null ? _f : null,
+                  usageMetadata: usageMetadata != null ? usageMetadata : null,
+                  finishMessage: (_g = candidate.finishMessage) != null ? _g : null
                 }
               };
-              if (usageMetadata != null) {
-                providerMetadata[providerOptionsName].usageMetadata = usageMetadata;
-              }
             }
           },
           flush(controller) {
@@ -1484,6 +1568,7 @@ var responseSchema = lazySchema3(
         z3.object({
           content: getContentSchema().nullish().or(z3.object({}).strict()),
           finishReason: z3.string().nullish(),
+          finishMessage: z3.string().nullish(),
           safetyRatings: z3.array(getSafetyRatingSchema()).nullish(),
           groundingMetadata: getGroundingMetadataSchema().nullish(),
           urlContextMetadata: getUrlContextMetadataSchema().nullish()
@@ -1504,6 +1589,7 @@ var chunkSchema = lazySchema3(
         z3.object({
           content: getContentSchema().nullish(),
           finishReason: z3.string().nullish(),
+          finishMessage: z3.string().nullish(),
           safetyRatings: z3.array(getSafetyRatingSchema()).nullish(),
           groundingMetadata: getGroundingMetadataSchema().nullish(),
           urlContextMetadata: getUrlContextMetadataSchema().nullish()