@ai-sdk/google 2.0.33 → 2.0.35

package/dist/internal/index.d.mts

@@ -3,7 +3,7 @@ import { Resolvable, FetchFunction, InferValidator } from '@ai-sdk/provider-util
 import { LanguageModelV2 } from '@ai-sdk/provider';
 import { z } from 'zod/v4';
 
-type GoogleGenerativeAIModelId = 'gemini-1.5-flash' | 'gemini-1.5-flash-latest' | 'gemini-1.5-flash-001' | 'gemini-1.5-flash-002' | 'gemini-1.5-flash-8b' | 'gemini-1.5-flash-8b-latest' | 'gemini-1.5-flash-8b-001' | 'gemini-1.5-pro' | 'gemini-1.5-pro-latest' | 'gemini-1.5-pro-001' | 'gemini-1.5-pro-002' | 'gemini-2.0-flash' | 'gemini-2.0-flash-001' | 'gemini-2.0-flash-live-001' | 'gemini-2.0-flash-lite' | 'gemini-2.0-pro-exp-02-05' | 'gemini-2.0-flash-thinking-exp-01-21' | 'gemini-2.0-flash-exp' | 'gemini-2.5-pro' | 'gemini-2.5-flash' | 'gemini-2.5-flash-image-preview' | 'gemini-2.5-flash-lite' | 'gemini-2.5-flash-lite-preview-09-2025' | 'gemini-2.5-flash-preview-04-17' | 'gemini-2.5-flash-preview-09-2025' | 'gemini-2.5-pro-exp-03-25' | 'gemini-exp-1206' | 'gemma-3-12b-it' | 'gemma-3-27b-it' | (string & {});
+type GoogleGenerativeAIModelId = 'gemini-1.5-flash' | 'gemini-1.5-flash-latest' | 'gemini-1.5-flash-001' | 'gemini-1.5-flash-002' | 'gemini-1.5-flash-8b' | 'gemini-1.5-flash-8b-latest' | 'gemini-1.5-flash-8b-001' | 'gemini-1.5-pro' | 'gemini-1.5-pro-latest' | 'gemini-1.5-pro-001' | 'gemini-1.5-pro-002' | 'gemini-2.0-flash' | 'gemini-2.0-flash-001' | 'gemini-2.0-flash-live-001' | 'gemini-2.0-flash-lite' | 'gemini-2.0-pro-exp-02-05' | 'gemini-2.0-flash-thinking-exp-01-21' | 'gemini-2.0-flash-exp' | 'gemini-2.5-pro' | 'gemini-2.5-flash' | 'gemini-2.5-flash-image-preview' | 'gemini-2.5-flash-lite' | 'gemini-2.5-flash-lite-preview-09-2025' | 'gemini-2.5-flash-preview-04-17' | 'gemini-2.5-flash-preview-09-2025' | 'gemini-3-pro-preview' | 'gemini-pro-latest' | 'gemini-flash-latest' | 'gemini-flash-lite-latest' | 'gemini-2.5-pro-exp-03-25' | 'gemini-exp-1206' | 'gemma-3-12b-it' | 'gemma-3-27b-it' | (string & {});
 
 type GoogleGenerativeAIConfig = {
     provider: string;
@@ -39,13 +39,11 @@ declare const getGroundingMetadataSchema: () => z.ZodObject<{
             uri: z.ZodString;
             title: z.ZodOptional<z.ZodNullable<z.ZodString>>;
         }, z.core.$strip>>>;
-        retrievedContext: z.ZodUnion<readonly [z.ZodOptional<z.ZodNullable<z.ZodObject<{
+        retrievedContext: z.ZodOptional<z.ZodNullable<z.ZodObject<{
             uri: z.ZodString;
             title: z.ZodOptional<z.ZodNullable<z.ZodString>>;
-        }, z.core.$strip>>>, z.ZodObject<{
-            title: z.ZodOptional<z.ZodNullable<z.ZodString>>;
             text: z.ZodOptional<z.ZodNullable<z.ZodString>>;
-        }, z.core.$strip>]>;
+        }, z.core.$strip>>>;
     }, z.core.$strip>>>>;
     groundingSupports: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodObject<{
         segment: z.ZodObject<{
@@ -120,8 +118,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;
@@ -219,6 +215,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;
+    }>;
 };
 
 export { GoogleGenerativeAILanguageModel, type GoogleGenerativeAIModelId, type GroundingMetadataSchema, type SafetyRatingSchema, type UrlContextMetadataSchema, getGroundingMetadataSchema, getUrlContextMetadataSchema, googleTools };

package/dist/internal/index.d.ts

@@ -3,7 +3,7 @@ import { Resolvable, FetchFunction, InferValidator } from '@ai-sdk/provider-util
 import { LanguageModelV2 } from '@ai-sdk/provider';
 import { z } from 'zod/v4';
 
-type GoogleGenerativeAIModelId = 'gemini-1.5-flash' | 'gemini-1.5-flash-latest' | 'gemini-1.5-flash-001' | 'gemini-1.5-flash-002' | 'gemini-1.5-flash-8b' | 'gemini-1.5-flash-8b-latest' | 'gemini-1.5-flash-8b-001' | 'gemini-1.5-pro' | 'gemini-1.5-pro-latest' | 'gemini-1.5-pro-001' | 'gemini-1.5-pro-002' | 'gemini-2.0-flash' | 'gemini-2.0-flash-001' | 'gemini-2.0-flash-live-001' | 'gemini-2.0-flash-lite' | 'gemini-2.0-pro-exp-02-05' | 'gemini-2.0-flash-thinking-exp-01-21' | 'gemini-2.0-flash-exp' | 'gemini-2.5-pro' | 'gemini-2.5-flash' | 'gemini-2.5-flash-image-preview' | 'gemini-2.5-flash-lite' | 'gemini-2.5-flash-lite-preview-09-2025' | 'gemini-2.5-flash-preview-04-17' | 'gemini-2.5-flash-preview-09-2025' | 'gemini-2.5-pro-exp-03-25' | 'gemini-exp-1206' | 'gemma-3-12b-it' | 'gemma-3-27b-it' | (string & {});
+type GoogleGenerativeAIModelId = 'gemini-1.5-flash' | 'gemini-1.5-flash-latest' | 'gemini-1.5-flash-001' | 'gemini-1.5-flash-002' | 'gemini-1.5-flash-8b' | 'gemini-1.5-flash-8b-latest' | 'gemini-1.5-flash-8b-001' | 'gemini-1.5-pro' | 'gemini-1.5-pro-latest' | 'gemini-1.5-pro-001' | 'gemini-1.5-pro-002' | 'gemini-2.0-flash' | 'gemini-2.0-flash-001' | 'gemini-2.0-flash-live-001' | 'gemini-2.0-flash-lite' | 'gemini-2.0-pro-exp-02-05' | 'gemini-2.0-flash-thinking-exp-01-21' | 'gemini-2.0-flash-exp' | 'gemini-2.5-pro' | 'gemini-2.5-flash' | 'gemini-2.5-flash-image-preview' | 'gemini-2.5-flash-lite' | 'gemini-2.5-flash-lite-preview-09-2025' | 'gemini-2.5-flash-preview-04-17' | 'gemini-2.5-flash-preview-09-2025' | 'gemini-3-pro-preview' | 'gemini-pro-latest' | 'gemini-flash-latest' | 'gemini-flash-lite-latest' | 'gemini-2.5-pro-exp-03-25' | 'gemini-exp-1206' | 'gemma-3-12b-it' | 'gemma-3-27b-it' | (string & {});
 
 type GoogleGenerativeAIConfig = {
     provider: string;
@@ -39,13 +39,11 @@ declare const getGroundingMetadataSchema: () => z.ZodObject<{
             uri: z.ZodString;
             title: z.ZodOptional<z.ZodNullable<z.ZodString>>;
         }, z.core.$strip>>>;
-        retrievedContext: z.ZodUnion<readonly [z.ZodOptional<z.ZodNullable<z.ZodObject<{
+        retrievedContext: z.ZodOptional<z.ZodNullable<z.ZodObject<{
             uri: z.ZodString;
             title: z.ZodOptional<z.ZodNullable<z.ZodString>>;
-        }, z.core.$strip>>>, z.ZodObject<{
-            title: z.ZodOptional<z.ZodNullable<z.ZodString>>;
             text: z.ZodOptional<z.ZodNullable<z.ZodString>>;
-        }, z.core.$strip>]>;
+        }, z.core.$strip>>>;
     }, z.core.$strip>>>>;
     groundingSupports: z.ZodOptional<z.ZodNullable<z.ZodArray<z.ZodObject<{
         segment: z.ZodObject<{
@@ -120,8 +118,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;
@@ -219,6 +215,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;
+    }>;
 };
 
 export { GoogleGenerativeAILanguageModel, type GoogleGenerativeAIModelId, type GroundingMetadataSchema, type SafetyRatingSchema, type UrlContextMetadataSchema, getGroundingMetadataSchema, getUrlContextMetadataSchema, googleTools };

package/dist/internal/index.js

@@ -506,6 +506,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;
@@ -650,6 +670,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,
@@ -1093,16 +1121,64 @@ function extractSources({
   groundingMetadata,
   generateId: generateId2
 }) {
-  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: generateId2(),
-    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: generateId2(),
+        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: generateId2(),
+          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: generateId2(),
+          mediaType,
+          title,
+          filename
+        });
+      }
+    }
+  }
+  return sources.length > 0 ? sources : void 0;
 }
 var getGroundingMetadataSchema = () => import_v43.z.object({
   webSearchQueries: import_v43.z.array(import_v43.z.string()).nullish(),
@@ -1111,13 +1187,11 @@ var getGroundingMetadataSchema = () => import_v43.z.object({
   groundingChunks: import_v43.z.array(
     import_v43.z.object({
       web: import_v43.z.object({ uri: import_v43.z.string(), title: import_v43.z.string().nullish() }).nullish(),
-      retrievedContext: import_v43.z.union([
-        import_v43.z.object({ uri: import_v43.z.string(), title: import_v43.z.string().nullish() }).nullish(),
-        import_v43.z.object({
-          title: import_v43.z.string().nullish(),
-          text: import_v43.z.string().nullish()
-        })
-      ])
+      retrievedContext: import_v43.z.object({
+        uri: import_v43.z.string(),
+        title: import_v43.z.string().nullish(),
+        text: import_v43.z.string().nullish()
+      }).nullish()
     })
   ).nullish(),
   groundingSupports: import_v43.z.array(
@@ -1309,6 +1383,18 @@ var urlContext = (0, import_provider_utils8.createProviderDefinedToolFactory)({
   inputSchema: (0, import_provider_utils8.lazySchema)(() => (0, import_provider_utils8.zodSchema)(import_v47.z.object({})))
 });
 
+// src/tool/vertex-rag-store.ts
+var import_provider_utils9 = require("@ai-sdk/provider-utils");
+var import_v48 = require("zod/v4");
+var vertexRagStore = (0, import_provider_utils9.createProviderDefinedToolFactory)({
+  id: "google.vertex_rag_store",
+  name: "vertex_rag_store",
+  inputSchema: import_v48.z.object({
+    ragCorpus: import_v48.z.string(),
+    topK: import_v48.z.number().optional()
+  })
+});
+
 // src/google-tools.ts
 var googleTools = {
   /**
@@ -1342,7 +1428,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
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {