@ai-sdk/google 2.0.32 → 2.0.34

package/dist/internal/index.d.ts

@@ -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

@@ -443,10 +443,11 @@ function prepareTools({
     (tool) => tool.type === "provider-defined"
   );
   if (hasFunctionTools && hasProviderDefinedTools) {
+    const functionTools = tools.filter((tool) => tool.type === "function");
     toolWarnings.push({
       type: "unsupported-tool",
       tool: tools.find((tool) => tool.type === "function"),
-      details: "Cannot mix function tools with provider-defined tools in the same request. Please use either function tools or provider-defined tools, but not both."
+      details: `Cannot mix function tools with provider-defined tools in the same request. Falling back to provider-defined tools only. The following function tools will be ignored: ${functionTools.map((t) => t.name).join(", ")}. Please use either function tools or provider-defined tools, but not both.`
     });
   }
   if (hasProviderDefinedTools) {
@@ -505,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;
@@ -649,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,
@@ -1092,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(),
@@ -1110,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(
@@ -1308,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 = {
   /**
@@ -1341,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 = {