@ai-sdk/google 3.0.0-beta.54 → 3.0.0-beta.56

package/dist/index.mjs

@@ -7,7 +7,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "3.0.0-beta.54" : "0.0.0-test";
+var VERSION = true ? "3.0.0-beta.56" : "0.0.0-test";
 
 // src/google-generative-ai-embedding-model.ts
 import {
@@ -380,11 +380,6 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
                 };
               }
               case "file": {
-                if (part.mediaType !== "image/png") {
-                  throw new UnsupportedFunctionalityError({
-                    functionality: "Only PNG images are supported in assistant messages"
-                  });
-                }
                 if (part.data instanceof URL) {
                   throw new UnsupportedFunctionalityError({
                     functionality: "File data URLs in assistant messages are not supported"
@@ -394,7 +389,8 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
                   inlineData: {
                     mimeType: part.mediaType,
                     data: convertToBase64(part.data)
-                  }
+                  },
+                  thoughtSignature
                 };
               }
               case "tool-call": {
@@ -971,7 +967,8 @@ var GoogleGenerativeAILanguageModel = class {
         content.push({
           type: "file",
           data: part.inlineData.data,
-          mediaType: part.inlineData.mimeType
+          mediaType: part.inlineData.mimeType,
+          providerMetadata: part.thoughtSignature ? { google: { thoughtSignature: part.thoughtSignature } } : void 0
         });
       }
     }
@@ -1278,7 +1275,7 @@ function extractSources({
   groundingMetadata,
   generateId: generateId3
 }) {
-  var _a, _b, _c;
+  var _a, _b, _c, _d;
   if (!(groundingMetadata == null ? void 0 : groundingMetadata.groundingChunks)) {
     return void 0;
   }
@@ -1294,7 +1291,8 @@ function extractSources({
       });
     } else if (chunk.retrievedContext != null) {
       const uri = chunk.retrievedContext.uri;
-      if (uri.startsWith("http://") || uri.startsWith("https://")) {
+      const fileSearchStore = chunk.retrievedContext.fileSearchStore;
+      if (uri && (uri.startsWith("http://") || uri.startsWith("https://"))) {
         sources.push({
           type: "source",
           sourceType: "url",
@@ -1302,7 +1300,7 @@ function extractSources({
           url: uri,
           title: (_b = chunk.retrievedContext.title) != null ? _b : void 0
         });
-      } else {
+      } else if (uri) {
         const title = (_c = chunk.retrievedContext.title) != null ? _c : "Unknown Document";
         let mediaType = "application/octet-stream";
         let filename = void 0;
@@ -1332,6 +1330,16 @@ function extractSources({
           title,
           filename
         });
+      } else if (fileSearchStore) {
+        const title = (_d = chunk.retrievedContext.title) != null ? _d : "Unknown Document";
+        sources.push({
+          type: "source",
+          sourceType: "document",
+          id: generateId3(),
+          mediaType: "application/octet-stream",
+          title,
+          filename: fileSearchStore.split("/").pop()
+        });
       }
     }
   }
@@ -1345,9 +1353,10 @@ var getGroundingMetadataSchema = () => z5.object({
     z5.object({
       web: z5.object({ uri: z5.string(), title: z5.string().nullish() }).nullish(),
       retrievedContext: z5.object({
-        uri: z5.string(),
+        uri: z5.string().nullish(),
         title: z5.string().nullish(),
-        text: z5.string().nullish()
+        text: z5.string().nullish(),
+        fileSearchStore: z5.string().nullish()
       }).nullish()
     })
   ).nullish(),
@@ -1387,7 +1396,8 @@ var getContentSchema = () => z5.object({
         inlineData: z5.object({
           mimeType: z5.string(),
           data: z5.string()
-        })
+        }),
+        thoughtSignature: z5.string().nullish()
       }),
       z5.object({
         executableCode: z5.object({