@ai-sdk/google 2.0.28 → 2.0.29

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @ai-sdk/google
 
+## 2.0.29
+
+### Patch Changes
+
+- a929a32: Add Google File search tool
+
 ## 2.0.28
 
 ### Patch Changes

package/dist/index.d.mts

@@ -85,6 +85,9 @@ 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;
             groundingSupports?: {
@@ -177,6 +180,22 @@ declare const googleTools: {
      * Must have name "url_context".
      */
     urlContext: _ai_sdk_provider_utils.ProviderDefinedToolFactory<{}, {}>;
+    /**
+     * Enables Retrieval Augmented Generation (RAG) via the Gemini File Search tool.
+     * Must have name "file_search".
+     *
+     * @param fileSearchStoreNames - Fully-qualified File Search store resource names.
+     * @param metadataFilter - Optional filter expression to restrict the files that can be retrieved.
+     * @param topK - Optional result limit for the number of chunks returned from File Search.
+     *
+     * @see https://ai.google.dev/gemini-api/docs/file-search
+     */
+    fileSearch: _ai_sdk_provider_utils.ProviderDefinedToolFactory<{}, {
+        [x: string]: unknown;
+        fileSearchStoreNames: string[];
+        topK?: number | undefined;
+        metadataFilter?: string | undefined;
+    }>;
     /**
      * A tool that enables the model to generate and run Python code.
      * Must have name "code_execution".

package/dist/index.d.ts

@@ -85,6 +85,9 @@ 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;
             groundingSupports?: {
@@ -177,6 +180,22 @@ declare const googleTools: {
      * Must have name "url_context".
      */
     urlContext: _ai_sdk_provider_utils.ProviderDefinedToolFactory<{}, {}>;
+    /**
+     * Enables Retrieval Augmented Generation (RAG) via the Gemini File Search tool.
+     * Must have name "file_search".
+     *
+     * @param fileSearchStoreNames - Fully-qualified File Search store resource names.
+     * @param metadataFilter - Optional filter expression to restrict the files that can be retrieved.
+     * @param topK - Optional result limit for the number of chunks returned from File Search.
+     *
+     * @see https://ai.google.dev/gemini-api/docs/file-search
+     */
+    fileSearch: _ai_sdk_provider_utils.ProviderDefinedToolFactory<{}, {
+        [x: string]: unknown;
+        fileSearchStoreNames: string[];
+        topK?: number | undefined;
+        metadataFilter?: string | undefined;
+    }>;
     /**
      * A tool that enables the model to generate and run Python code.
      * Must have name "code_execution".

package/dist/index.js

@@ -27,10 +27,10 @@ __export(src_exports, {
 module.exports = __toCommonJS(src_exports);
 
 // src/google-provider.ts
-var import_provider_utils11 = require("@ai-sdk/provider-utils");
+var import_provider_utils12 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "2.0.28" : "0.0.0-test";
+var VERSION = true ? "2.0.29" : "0.0.0-test";
 
 // src/google-generative-ai-embedding-model.ts
 var import_provider = require("@ai-sdk/provider");
@@ -589,6 +589,7 @@ function prepareTools({
   const toolWarnings = [];
   const isGemini2 = modelId.includes("gemini-2");
   const supportsDynamicRetrieval = modelId.includes("gemini-1.5-flash") && !modelId.includes("-8b");
+  const supportsFileSearch = modelId.includes("gemini-2.5");
   if (tools == null) {
     return { tools: void 0, toolConfig: void 0, toolWarnings };
   }
@@ -648,6 +649,17 @@ function prepareTools({
             });
           }
           break;
+        case "google.file_search":
+          if (supportsFileSearch) {
+            googleTools2.push({ fileSearch: { ...tool.args } });
+          } else {
+            toolWarnings.push({
+              type: "unsupported-tool",
+              tool,
+              details: "The file search tool is only supported with Gemini 2.5 models."
+            });
+          }
+          break;
         default:
           toolWarnings.push({ type: "unsupported-tool", tool });
           break;
@@ -1253,7 +1265,13 @@ 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.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()
+        })
+      ])
     })
   ).nullish(),
   groundingSupports: import_v45.z.array(
@@ -1390,29 +1408,57 @@ var codeExecution = (0, import_provider_utils7.createProviderDefinedToolFactoryW
   })
 });
 
-// src/tool/google-search.ts
+// src/tool/file-search.ts
 var import_provider_utils8 = require("@ai-sdk/provider-utils");
 var import_v47 = require("zod/v4");
-var googleSearch = (0, import_provider_utils8.createProviderDefinedToolFactory)({
+var fileSearchArgsBaseSchema = import_v47.z.object({
+  /** The names of the file_search_stores to retrieve from.
+   *  Example: `fileSearchStores/my-file-search-store-123`
+   */
+  fileSearchStoreNames: import_v47.z.array(import_v47.z.string()).describe(
+    "The names of the file_search_stores to retrieve from. Example: `fileSearchStores/my-file-search-store-123`"
+  ),
+  /** The number of file search retrieval chunks to retrieve. */
+  topK: import_v47.z.number().int().positive().describe("The number of file search retrieval chunks to retrieve.").optional(),
+  /** Metadata filter to apply to the file search retrieval documents.
+   *  See https://google.aip.dev/160 for the syntax of the filter expression.
+   */
+  metadataFilter: import_v47.z.string().describe(
+    "Metadata filter to apply to the file search retrieval documents. See https://google.aip.dev/160 for the syntax of the filter expression."
+  ).optional()
+}).passthrough();
+var fileSearchArgsSchema = (0, import_provider_utils8.lazySchema)(
+  () => (0, import_provider_utils8.zodSchema)(fileSearchArgsBaseSchema)
+);
+var fileSearch = (0, import_provider_utils8.createProviderDefinedToolFactory)({
+  id: "google.file_search",
+  name: "file_search",
+  inputSchema: fileSearchArgsSchema
+});
+
+// src/tool/google-search.ts
+var import_provider_utils9 = require("@ai-sdk/provider-utils");
+var import_v48 = require("zod/v4");
+var googleSearch = (0, import_provider_utils9.createProviderDefinedToolFactory)({
   id: "google.google_search",
   name: "google_search",
-  inputSchema: (0, import_provider_utils8.lazySchema)(
-    () => (0, import_provider_utils8.zodSchema)(
-      import_v47.z.object({
-        mode: import_v47.z.enum(["MODE_DYNAMIC", "MODE_UNSPECIFIED"]).default("MODE_UNSPECIFIED"),
-        dynamicThreshold: import_v47.z.number().default(1)
+  inputSchema: (0, import_provider_utils9.lazySchema)(
+    () => (0, import_provider_utils9.zodSchema)(
+      import_v48.z.object({
+        mode: import_v48.z.enum(["MODE_DYNAMIC", "MODE_UNSPECIFIED"]).default("MODE_UNSPECIFIED"),
+        dynamicThreshold: import_v48.z.number().default(1)
       })
     )
   )
 });
 
 // src/tool/url-context.ts
-var import_provider_utils9 = require("@ai-sdk/provider-utils");
-var import_v48 = require("zod/v4");
-var urlContext = (0, import_provider_utils9.createProviderDefinedToolFactory)({
+var import_provider_utils10 = require("@ai-sdk/provider-utils");
+var import_v49 = require("zod/v4");
+var urlContext = (0, import_provider_utils10.createProviderDefinedToolFactory)({
   id: "google.url_context",
   name: "url_context",
-  inputSchema: (0, import_provider_utils9.lazySchema)(() => (0, import_provider_utils9.zodSchema)(import_v48.z.object({})))
+  inputSchema: (0, import_provider_utils10.lazySchema)(() => (0, import_provider_utils10.zodSchema)(import_v49.z.object({})))
 });
 
 // src/google-tools.ts
@@ -1427,6 +1473,17 @@ var googleTools = {
    * Must have name "url_context".
    */
   urlContext,
+  /**
+   * Enables Retrieval Augmented Generation (RAG) via the Gemini File Search tool.
+   * Must have name "file_search".
+   *
+   * @param fileSearchStoreNames - Fully-qualified File Search store resource names.
+   * @param metadataFilter - Optional filter expression to restrict the files that can be retrieved.
+   * @param topK - Optional result limit for the number of chunks returned from File Search.
+   *
+   * @see https://ai.google.dev/gemini-api/docs/file-search
+   */
+  fileSearch,
   /**
    * A tool that enables the model to generate and run Python code.
    * Must have name "code_execution".
@@ -1441,8 +1498,8 @@ var googleTools = {
 };
 
 // src/google-generative-ai-image-model.ts
-var import_provider_utils10 = require("@ai-sdk/provider-utils");
-var import_v49 = require("zod/v4");
+var import_provider_utils11 = require("@ai-sdk/provider-utils");
+var import_v410 = require("zod/v4");
 var GoogleGenerativeAIImageModel = class {
   constructor(modelId, settings, config) {
     this.modelId = modelId;
@@ -1484,7 +1541,7 @@ var GoogleGenerativeAIImageModel = class {
         details: "This model does not support the `seed` option through this provider."
       });
     }
-    const googleOptions = await (0, import_provider_utils10.parseProviderOptions)({
+    const googleOptions = await (0, import_provider_utils11.parseProviderOptions)({
       provider: "google",
       providerOptions,
       schema: googleImageProviderOptionsSchema
@@ -1503,12 +1560,12 @@ var GoogleGenerativeAIImageModel = class {
       instances: [{ prompt }],
       parameters
     };
-    const { responseHeaders, value: response } = await (0, import_provider_utils10.postJsonToApi)({
+    const { responseHeaders, value: response } = await (0, import_provider_utils11.postJsonToApi)({
       url: `${this.config.baseURL}/models/${this.modelId}:predict`,
-      headers: (0, import_provider_utils10.combineHeaders)(await (0, import_provider_utils10.resolve)(this.config.headers), headers),
+      headers: (0, import_provider_utils11.combineHeaders)(await (0, import_provider_utils11.resolve)(this.config.headers), headers),
       body,
       failedResponseHandler: googleFailedResponseHandler,
-      successfulResponseHandler: (0, import_provider_utils10.createJsonResponseHandler)(
+      successfulResponseHandler: (0, import_provider_utils11.createJsonResponseHandler)(
         googleImageResponseSchema
       ),
       abortSignal,
@@ -1534,18 +1591,18 @@ var GoogleGenerativeAIImageModel = class {
     };
   }
 };
-var googleImageResponseSchema = (0, import_provider_utils10.lazySchema)(
-  () => (0, import_provider_utils10.zodSchema)(
-    import_v49.z.object({
-      predictions: import_v49.z.array(import_v49.z.object({ bytesBase64Encoded: import_v49.z.string() })).default([])
+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 googleImageProviderOptionsSchema = (0, import_provider_utils10.lazySchema)(
-  () => (0, import_provider_utils10.zodSchema)(
-    import_v49.z.object({
-      personGeneration: import_v49.z.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
-      aspectRatio: import_v49.z.enum(["1:1", "3:4", "4:3", "9:16", "16:9"]).nullish()
+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()
     })
   )
 );
@@ -1553,10 +1610,10 @@ var googleImageProviderOptionsSchema = (0, import_provider_utils10.lazySchema)(
 // src/google-provider.ts
 function createGoogleGenerativeAI(options = {}) {
   var _a;
-  const baseURL = (_a = (0, import_provider_utils11.withoutTrailingSlash)(options.baseURL)) != null ? _a : "https://generativelanguage.googleapis.com/v1beta";
-  const getHeaders = () => (0, import_provider_utils11.withUserAgentSuffix)(
+  const baseURL = (_a = (0, import_provider_utils12.withoutTrailingSlash)(options.baseURL)) != null ? _a : "https://generativelanguage.googleapis.com/v1beta";
+  const getHeaders = () => (0, import_provider_utils12.withUserAgentSuffix)(
     {
-      "x-goog-api-key": (0, import_provider_utils11.loadApiKey)({
+      "x-goog-api-key": (0, import_provider_utils12.loadApiKey)({
         apiKey: options.apiKey,
         environmentVariableName: "GOOGLE_GENERATIVE_AI_API_KEY",
         description: "Google Generative AI"
@@ -1571,7 +1628,7 @@ function createGoogleGenerativeAI(options = {}) {
       provider: "google.generative-ai",
       baseURL,
       headers: getHeaders,
-      generateId: (_a2 = options.generateId) != null ? _a2 : import_provider_utils11.generateId,
+      generateId: (_a2 = options.generateId) != null ? _a2 : import_provider_utils12.generateId,
       supportedUrls: () => ({
         "*": [
           // Google Generative Language "files" endpoint