@ai-sdk/google 2.0.33 → 2.0.34

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @ai-sdk/google
 
+## 2.0.34
+
+### Patch Changes
+
+- 9a66892: Add Google Vertex RAG Engine grounding provider tool
+
 ## 2.0.33
 
 ### Patch Changes

package/dist/index.d.mts

@@ -85,8 +85,6 @@ declare const responseSchema: _ai_sdk_provider_utils.LazySchema<{
                 retrievedContext?: {
                     uri: string;
                     title?: string | null | undefined;
-                } | {
-                    title?: string | null | undefined;
                     text?: string | null | undefined;
                 } | null | undefined;
             }[] | null | undefined;
@@ -214,6 +212,14 @@ declare const googleTools: {
         outcome: string;
         output: string;
     }, {}>;
+    /**
+     * 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: _ai_sdk_provider_utils.ProviderDefinedToolFactory<{}, {
+        ragCorpus: string;
+        topK?: number;
+    }>;
 };
 
 interface GoogleGenerativeAIProvider extends ProviderV2 {

package/dist/index.d.ts

@@ -85,8 +85,6 @@ declare const responseSchema: _ai_sdk_provider_utils.LazySchema<{
                 retrievedContext?: {
                     uri: string;
                     title?: string | null | undefined;
-                } | {
-                    title?: string | null | undefined;
                     text?: string | null | undefined;
                 } | null | undefined;
             }[] | null | undefined;
@@ -214,6 +212,14 @@ declare const googleTools: {
         outcome: string;
         output: string;
     }, {}>;
+    /**
+     * 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: _ai_sdk_provider_utils.ProviderDefinedToolFactory<{}, {
+        ragCorpus: string;
+        topK?: number;
+    }>;
 };
 
 interface GoogleGenerativeAIProvider extends ProviderV2 {

package/dist/index.js

@@ -27,10 +27,10 @@ __export(src_exports, {
 module.exports = __toCommonJS(src_exports);
 
 // src/google-provider.ts
-var import_provider_utils12 = require("@ai-sdk/provider-utils");
+var import_provider_utils13 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "2.0.33" : "0.0.0-test";
+var VERSION = true ? "2.0.34" : "0.0.0-test";
 
 // src/google-generative-ai-embedding-model.ts
 var import_provider = require("@ai-sdk/provider");
@@ -661,6 +661,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;
@@ -805,6 +825,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,
@@ -1248,16 +1276,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 = () => import_v45.z.object({
   webSearchQueries: import_v45.z.array(import_v45.z.string()).nullish(),
@@ -1266,13 +1342,11 @@ var getGroundingMetadataSchema = () => import_v45.z.object({
   groundingChunks: import_v45.z.array(
     import_v45.z.object({
       web: import_v45.z.object({ uri: import_v45.z.string(), title: import_v45.z.string().nullish() }).nullish(),
-      retrievedContext: import_v45.z.union([
-        import_v45.z.object({ uri: import_v45.z.string(), title: import_v45.z.string().nullish() }).nullish(),
-        import_v45.z.object({
-          title: import_v45.z.string().nullish(),
-          text: import_v45.z.string().nullish()
-        })
-      ])
+      retrievedContext: import_v45.z.object({
+        uri: import_v45.z.string(),
+        title: import_v45.z.string().nullish(),
+        text: import_v45.z.string().nullish()
+      }).nullish()
     })
   ).nullish(),
   groundingSupports: import_v45.z.array(
@@ -1464,6 +1538,18 @@ var urlContext = (0, import_provider_utils10.createProviderDefinedToolFactory)({
   inputSchema: (0, import_provider_utils10.lazySchema)(() => (0, import_provider_utils10.zodSchema)(import_v49.z.object({})))
 });
 
+// src/tool/vertex-rag-store.ts
+var import_provider_utils11 = require("@ai-sdk/provider-utils");
+var import_v410 = require("zod/v4");
+var vertexRagStore = (0, import_provider_utils11.createProviderDefinedToolFactory)({
+  id: "google.vertex_rag_store",
+  name: "vertex_rag_store",
+  inputSchema: import_v410.z.object({
+    ragCorpus: import_v410.z.string(),
+    topK: import_v410.z.number().optional()
+  })
+});
+
 // src/google-tools.ts
 var googleTools = {
   /**
@@ -1497,12 +1583,17 @@ 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
-var import_provider_utils11 = require("@ai-sdk/provider-utils");
-var import_v410 = require("zod/v4");
+var import_provider_utils12 = require("@ai-sdk/provider-utils");
+var import_v411 = require("zod/v4");
 var GoogleGenerativeAIImageModel = class {
   constructor(modelId, settings, config) {
     this.modelId = modelId;
@@ -1544,7 +1635,7 @@ var GoogleGenerativeAIImageModel = class {
         details: "This model does not support the `seed` option through this provider."
       });
     }
-    const googleOptions = await (0, import_provider_utils11.parseProviderOptions)({
+    const googleOptions = await (0, import_provider_utils12.parseProviderOptions)({
       provider: "google",
       providerOptions,
       schema: googleImageProviderOptionsSchema
@@ -1563,12 +1654,12 @@ var GoogleGenerativeAIImageModel = class {
       instances: [{ prompt }],
       parameters
     };
-    const { responseHeaders, value: response } = await (0, import_provider_utils11.postJsonToApi)({
+    const { responseHeaders, value: response } = await (0, import_provider_utils12.postJsonToApi)({
       url: `${this.config.baseURL}/models/${this.modelId}:predict`,
-      headers: (0, import_provider_utils11.combineHeaders)(await (0, import_provider_utils11.resolve)(this.config.headers), headers),
+      headers: (0, import_provider_utils12.combineHeaders)(await (0, import_provider_utils12.resolve)(this.config.headers), headers),
       body,
       failedResponseHandler: googleFailedResponseHandler,
-      successfulResponseHandler: (0, import_provider_utils11.createJsonResponseHandler)(
+      successfulResponseHandler: (0, import_provider_utils12.createJsonResponseHandler)(
         googleImageResponseSchema
       ),
       abortSignal,
@@ -1594,18 +1685,18 @@ var GoogleGenerativeAIImageModel = class {
     };
   }
 };
-var googleImageResponseSchema = (0, import_provider_utils11.lazySchema)(
-  () => (0, import_provider_utils11.zodSchema)(
-    import_v410.z.object({
-      predictions: import_v410.z.array(import_v410.z.object({ bytesBase64Encoded: import_v410.z.string() })).default([])
+var googleImageResponseSchema = (0, import_provider_utils12.lazySchema)(
+  () => (0, import_provider_utils12.zodSchema)(
+    import_v411.z.object({
+      predictions: import_v411.z.array(import_v411.z.object({ bytesBase64Encoded: import_v411.z.string() })).default([])
     })
   )
 );
-var googleImageProviderOptionsSchema = (0, import_provider_utils11.lazySchema)(
-  () => (0, import_provider_utils11.zodSchema)(
-    import_v410.z.object({
-      personGeneration: import_v410.z.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
-      aspectRatio: import_v410.z.enum(["1:1", "3:4", "4:3", "9:16", "16:9"]).nullish()
+var googleImageProviderOptionsSchema = (0, import_provider_utils12.lazySchema)(
+  () => (0, import_provider_utils12.zodSchema)(
+    import_v411.z.object({
+      personGeneration: import_v411.z.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
+      aspectRatio: import_v411.z.enum(["1:1", "3:4", "4:3", "9:16", "16:9"]).nullish()
     })
   )
 );
@@ -1613,11 +1704,11 @@ var googleImageProviderOptionsSchema = (0, import_provider_utils11.lazySchema)(
 // src/google-provider.ts
 function createGoogleGenerativeAI(options = {}) {
   var _a, _b;
-  const baseURL = (_a = (0, import_provider_utils12.withoutTrailingSlash)(options.baseURL)) != null ? _a : "https://generativelanguage.googleapis.com/v1beta";
+  const baseURL = (_a = (0, import_provider_utils13.withoutTrailingSlash)(options.baseURL)) != null ? _a : "https://generativelanguage.googleapis.com/v1beta";
   const providerName = (_b = options.name) != null ? _b : "google.generative-ai";
-  const getHeaders = () => (0, import_provider_utils12.withUserAgentSuffix)(
+  const getHeaders = () => (0, import_provider_utils13.withUserAgentSuffix)(
     {
-      "x-goog-api-key": (0, import_provider_utils12.loadApiKey)({
+      "x-goog-api-key": (0, import_provider_utils13.loadApiKey)({
         apiKey: options.apiKey,
         environmentVariableName: "GOOGLE_GENERATIVE_AI_API_KEY",
         description: "Google Generative AI"
@@ -1632,7 +1723,7 @@ function createGoogleGenerativeAI(options = {}) {
       provider: providerName,
       baseURL,
       headers: getHeaders,
-      generateId: (_a2 = options.generateId) != null ? _a2 : import_provider_utils12.generateId,
+      generateId: (_a2 = options.generateId) != null ? _a2 : import_provider_utils13.generateId,
       supportedUrls: () => ({
         "*": [
           // Google Generative Language "files" endpoint