@ai-sdk/google 2.0.41 → 2.0.43

package/dist/index.mjs

@@ -7,7 +7,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "2.0.41" : "0.0.0-test";
+var VERSION = true ? "2.0.43" : "0.0.0-test";
 
 // src/google-generative-ai-embedding-model.ts
 import {
@@ -240,14 +240,16 @@ function convertJSONSchemaToOpenAPISchema(jsonSchema) {
   }
   if (type) {
     if (Array.isArray(type)) {
-      if (type.includes("null")) {
-        result.type = type.filter((t) => t !== "null")[0];
-        result.nullable = true;
+      const hasNull = type.includes("null");
+      const nonNullTypes = type.filter((t) => t !== "null");
+      if (nonNullTypes.length === 0) {
+        result.type = "null";
       } else {
-        result.type = type;
+        result.anyOf = nonNullTypes.map((t) => ({ type: t }));
+        if (hasNull) {
+          result.nullable = true;
+        }
       }
-    } else if (type === "null") {
-      result.type = "null";
     } else {
       result.type = type;
     }
@@ -729,7 +731,7 @@ function prepareTools({
   }
   if (toolChoice == null) {
     return {
-      tools: { functionDeclarations },
+      tools: [{ functionDeclarations }],
       toolConfig: void 0,
       toolWarnings
     };
@@ -738,25 +740,25 @@ function prepareTools({
   switch (type) {
     case "auto":
       return {
-        tools: { functionDeclarations },
+        tools: [{ functionDeclarations }],
         toolConfig: { functionCallingConfig: { mode: "AUTO" } },
         toolWarnings
       };
     case "none":
       return {
-        tools: { functionDeclarations },
+        tools: [{ functionDeclarations }],
         toolConfig: { functionCallingConfig: { mode: "NONE" } },
         toolWarnings
       };
     case "required":
       return {
-        tools: { functionDeclarations },
+        tools: [{ functionDeclarations }],
         toolConfig: { functionCallingConfig: { mode: "ANY" } },
         toolWarnings
       };
     case "tool":
       return {
-        tools: { functionDeclarations },
+        tools: [{ functionDeclarations }],
         toolConfig: {
           functionCallingConfig: {
             mode: "ANY",
@@ -1175,11 +1177,7 @@ var GoogleGenerativeAILanguageModel = class {
                       } : void 0
                     });
                   }
-                }
-              }
-              const inlineDataParts = getInlineDataParts(content.parts);
-              if (inlineDataParts != null) {
-                for (const part of inlineDataParts) {
+                } else if ("inlineData" in part) {
                   controller.enqueue({
                     type: "file",
                     mediaType: part.inlineData.mimeType,
@@ -1281,16 +1279,11 @@ function getToolCallsFromParts({
     providerMetadata: part.thoughtSignature ? { google: { thoughtSignature: part.thoughtSignature } } : void 0
   }));
 }
-function getInlineDataParts(parts) {
-  return parts == null ? void 0 : parts.filter(
-    (part) => "inlineData" in part
-  );
-}
 function extractSources({
   groundingMetadata,
   generateId: generateId3
 }) {
-  var _a, _b, _c;
+  var _a, _b, _c, _d;
   if (!(groundingMetadata == null ? void 0 : groundingMetadata.groundingChunks)) {
     return void 0;
   }
@@ -1306,7 +1299,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",
@@ -1314,7 +1308,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;
@@ -1344,6 +1338,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()
+        });
       }
     }
   }
@@ -1357,9 +1361,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(),