@ai-sdk/google 2.0.33 → 2.0.35

package/dist/index.mjs

@@ -7,7 +7,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "2.0.33" : "0.0.0-test";
+var VERSION = true ? "2.0.35" : "0.0.0-test";
 
 // src/google-generative-ai-embedding-model.ts
 import {
@@ -672,6 +672,26 @@ function prepareTools({
             });
           }
           break;
+        case "google.vertex_rag_store":
+          if (isGemini2) {
+            googleTools2.push({
+              retrieval: {
+                vertex_rag_store: {
+                  rag_resources: {
+                    rag_corpus: tool.args.ragCorpus
+                  },
+                  similarity_top_k: tool.args.topK
+                }
+              }
+            });
+          } else {
+            toolWarnings.push({
+              type: "unsupported-tool",
+              tool,
+              details: "The RAG store tool is not supported with other Gemini models than Gemini 2."
+            });
+          }
+          break;
         default:
           toolWarnings.push({ type: "unsupported-tool", tool });
           break;
@@ -816,6 +836,14 @@ var GoogleGenerativeAILanguageModel = class {
         message: `The 'includeThoughts' option is only supported with the Google Vertex provider and might not be supported or could behave unexpectedly with the current Google provider (${this.config.provider}).`
       });
     }
+    if ((tools == null ? void 0 : tools.some(
+      (tool) => tool.type === "provider-defined" && tool.id === "google.vertex_rag_store"
+    )) && !this.config.provider.startsWith("google.vertex.")) {
+      warnings.push({
+        type: "other",
+        message: `The 'vertex_rag_store' tool is only supported with the Google Vertex provider and might not be supported or could behave unexpectedly with the current Google provider (${this.config.provider}).`
+      });
+    }
     const isGemmaModel = this.modelId.toLowerCase().startsWith("gemma-");
     const { contents, systemInstruction } = convertToGoogleGenerativeAIMessages(
       prompt,
@@ -1259,16 +1287,64 @@ function extractSources({
   groundingMetadata,
   generateId: generateId3
 }) {
-  var _a;
-  return (_a = groundingMetadata == null ? void 0 : groundingMetadata.groundingChunks) == null ? void 0 : _a.filter(
-    (chunk) => chunk.web != null
-  ).map((chunk) => ({
-    type: "source",
-    sourceType: "url",
-    id: generateId3(),
-    url: chunk.web.uri,
-    title: chunk.web.title
-  }));
+  var _a, _b, _c;
+  if (!(groundingMetadata == null ? void 0 : groundingMetadata.groundingChunks)) {
+    return void 0;
+  }
+  const sources = [];
+  for (const chunk of groundingMetadata.groundingChunks) {
+    if (chunk.web != null) {
+      sources.push({
+        type: "source",
+        sourceType: "url",
+        id: generateId3(),
+        url: chunk.web.uri,
+        title: (_a = chunk.web.title) != null ? _a : void 0
+      });
+    } else if (chunk.retrievedContext != null) {
+      const uri = chunk.retrievedContext.uri;
+      if (uri.startsWith("http://") || uri.startsWith("https://")) {
+        sources.push({
+          type: "source",
+          sourceType: "url",
+          id: generateId3(),
+          url: uri,
+          title: (_b = chunk.retrievedContext.title) != null ? _b : void 0
+        });
+      } else {
+        const title = (_c = chunk.retrievedContext.title) != null ? _c : "Unknown Document";
+        let mediaType = "application/octet-stream";
+        let filename = void 0;
+        if (uri.endsWith(".pdf")) {
+          mediaType = "application/pdf";
+          filename = uri.split("/").pop();
+        } else if (uri.endsWith(".txt")) {
+          mediaType = "text/plain";
+          filename = uri.split("/").pop();
+        } else if (uri.endsWith(".docx")) {
+          mediaType = "application/vnd.openxmlformats-officedocument.wordprocessingml.document";
+          filename = uri.split("/").pop();
+        } else if (uri.endsWith(".doc")) {
+          mediaType = "application/msword";
+          filename = uri.split("/").pop();
+        } else if (uri.match(/\.(md|markdown)$/)) {
+          mediaType = "text/markdown";
+          filename = uri.split("/").pop();
+        } else {
+          filename = uri.split("/").pop();
+        }
+        sources.push({
+          type: "source",
+          sourceType: "document",
+          id: generateId3(),
+          mediaType,
+          title,
+          filename
+        });
+      }
+    }
+  }
+  return sources.length > 0 ? sources : void 0;
 }
 var getGroundingMetadataSchema = () => z5.object({
   webSearchQueries: z5.array(z5.string()).nullish(),
@@ -1277,13 +1353,11 @@ var getGroundingMetadataSchema = () => z5.object({
   groundingChunks: z5.array(
     z5.object({
       web: z5.object({ uri: z5.string(), title: z5.string().nullish() }).nullish(),
-      retrievedContext: z5.union([
-        z5.object({ uri: z5.string(), title: z5.string().nullish() }).nullish(),
-        z5.object({
-          title: z5.string().nullish(),
-          text: z5.string().nullish()
-        })
-      ])
+      retrievedContext: z5.object({
+        uri: z5.string(),
+        title: z5.string().nullish(),
+        text: z5.string().nullish()
+      }).nullish()
     })
   ).nullish(),
   groundingSupports: z5.array(
@@ -1487,6 +1561,18 @@ var urlContext = createProviderDefinedToolFactory3({
   inputSchema: lazySchema8(() => zodSchema8(z9.object({})))
 });
 
+// src/tool/vertex-rag-store.ts
+import { createProviderDefinedToolFactory as createProviderDefinedToolFactory4 } from "@ai-sdk/provider-utils";
+import { z as z10 } from "zod/v4";
+var vertexRagStore = createProviderDefinedToolFactory4({
+  id: "google.vertex_rag_store",
+  name: "vertex_rag_store",
+  inputSchema: z10.object({
+    ragCorpus: z10.string(),
+    topK: z10.number().optional()
+  })
+});
+
 // src/google-tools.ts
 var googleTools = {
   /**
@@ -1520,7 +1606,12 @@ var googleTools = {
    * @see https://ai.google.dev/gemini-api/docs/code-execution (Google AI)
    * @see https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/code-execution-api (Vertex AI)
    */
-  codeExecution
+  codeExecution,
+  /**
+   * Creates a Vertex RAG Store tool that enables the model to perform RAG searches against a Vertex RAG Store.
+   * Must have name "vertex_rag_store".
+   */
+  vertexRagStore
 };
 
 // src/google-generative-ai-image-model.ts
@@ -1533,7 +1624,7 @@ import {
   resolve as resolve3,
   zodSchema as zodSchema9
 } from "@ai-sdk/provider-utils";
-import { z as z10 } from "zod/v4";
+import { z as z11 } from "zod/v4";
 var GoogleGenerativeAIImageModel = class {
   constructor(modelId, settings, config) {
     this.modelId = modelId;
@@ -1627,16 +1718,16 @@ var GoogleGenerativeAIImageModel = class {
 };
 var googleImageResponseSchema = lazySchema9(
   () => zodSchema9(
-    z10.object({
-      predictions: z10.array(z10.object({ bytesBase64Encoded: z10.string() })).default([])
+    z11.object({
+      predictions: z11.array(z11.object({ bytesBase64Encoded: z11.string() })).default([])
     })
   )
 );
 var googleImageProviderOptionsSchema = lazySchema9(
   () => zodSchema9(
-    z10.object({
-      personGeneration: z10.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
-      aspectRatio: z10.enum(["1:1", "3:4", "4:3", "9:16", "16:9"]).nullish()
+    z11.object({
+      personGeneration: z11.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
+      aspectRatio: z11.enum(["1:1", "3:4", "4:3", "9:16", "16:9"]).nullish()
     })
   )
 );