@ai-sdk/google 2.0.103 → 2.0.105

package/dist/index.mjs

@@ -7,7 +7,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "2.0.103" : "0.0.0-test";
+var VERSION = true ? "2.0.105" : "0.0.0-test";
 
 // src/google-generative-ai-embedding-model.ts
 import {
@@ -54,7 +54,7 @@ import {
   zodSchema as zodSchema2
 } from "@ai-sdk/provider-utils";
 import { z as z2 } from "zod/v4";
-var googleEmbeddingContentPartSchema = z2.union([
+var googleEmbeddingContentPartSchema = () => z2.union([
   z2.object({ text: z2.string() }),
   z2.object({
     inlineData: z2.object({
@@ -102,7 +102,7 @@ var googleGenerativeAIEmbeddingProviderOptions = lazySchema2(
        * The array length must match the number of values being embedded. In
        * the case of a single embedding, the array length must be 1.
        */
-      content: z2.array(z2.array(googleEmbeddingContentPartSchema).min(1).nullable()).optional()
+      content: z2.array(z2.array(googleEmbeddingContentPartSchema()).min(1).nullable()).optional()
     })
   )
 );
@@ -1497,7 +1497,7 @@ var GoogleGenerativeAILanguageModel = class {
     };
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u, _v, _w;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u, _v, _w, _x;
     const { args, warnings } = await this.getArgs(options);
     const body = JSON.stringify(args);
     const mergedHeaders = combineHeaders2(
@@ -1521,18 +1521,20 @@ var GoogleGenerativeAILanguageModel = class {
       rawValue: rawResponse
     } = await generateContent(args);
     const continuedParts = [];
+    const groundingMetadataPerRequest = [];
     let continuedUsageMetadata;
     const firstPromptTokenCount = (_a = response.usageMetadata) == null ? void 0 : _a.promptTokenCount;
     while (true) {
       const candidate2 = (_b = response.candidates) == null ? void 0 : _b[0];
+      groundingMetadataPerRequest.push((_c = candidate2 == null ? void 0 : candidate2.groundingMetadata) != null ? _c : null);
       const continuationToken = (candidate2 == null ? void 0 : candidate2.finishReason) === "CONTINUATION" ? candidate2.continuationToken : void 0;
       if (continuationToken == null) {
         break;
       }
-      continuedParts.push(...(_d = (_c = candidate2 == null ? void 0 : candidate2.content) == null ? void 0 : _c.parts) != null ? _d : []);
+      continuedParts.push(...(_e = (_d = candidate2 == null ? void 0 : candidate2.content) == null ? void 0 : _d.parts) != null ? _e : []);
       continuedUsageMetadata = addGoogleGenerativeAIUsageMetadata(
         continuedUsageMetadata,
-        (_e = response.usageMetadata) != null ? _e : void 0
+        (_f = response.usageMetadata) != null ? _f : void 0
       );
       ({
         responseHeaders,
@@ -1540,18 +1542,18 @@ var GoogleGenerativeAILanguageModel = class {
         rawValue: rawResponse
       } = await generateContent({ ...args, continuationToken }));
     }
-    const candidate = (_f = response.candidates) == null ? void 0 : _f[0];
-    const promptBlockReason = (_g = response.promptFeedback) == null ? void 0 : _g.blockReason;
+    const candidate = (_g = response.candidates) == null ? void 0 : _g[0];
+    const promptBlockReason = (_h = response.promptFeedback) == null ? void 0 : _h.blockReason;
     const isPromptBlocked = (candidate == null ? void 0 : candidate.finishReason) == null && promptBlockReason != null;
     const content = [];
-    const parts = [...continuedParts, ...(_i = (_h = candidate == null ? void 0 : candidate.content) == null ? void 0 : _h.parts) != null ? _i : []];
+    const parts = [...continuedParts, ...(_j = (_i = candidate == null ? void 0 : candidate.content) == null ? void 0 : _i.parts) != null ? _j : []];
     const usageMetadata = addGoogleGenerativeAIUsageMetadata(
       continuedUsageMetadata,
-      (_j = response.usageMetadata) != null ? _j : void 0
+      (_k = response.usageMetadata) != null ? _k : void 0
     );
     let lastCodeExecutionToolCallId;
     for (const part of parts) {
-      if ("executableCode" in part && ((_k = part.executableCode) == null ? void 0 : _k.code)) {
+      if ("executableCode" in part && ((_l = part.executableCode) == null ? void 0 : _l.code)) {
         const toolCallId = this.config.generateId();
         lastCodeExecutionToolCallId = toolCallId;
         content.push({
@@ -1592,7 +1594,7 @@ var GoogleGenerativeAILanguageModel = class {
       } else if ("functionCall" in part) {
         content.push({
           type: "tool-call",
-          toolCallId: (_l = part.functionCall.id) != null ? _l : this.config.generateId(),
+          toolCallId: (_m = part.functionCall.id) != null ? _m : this.config.generateId(),
           toolName: part.functionCall.name,
           input: JSON.stringify(part.functionCall.args),
           providerMetadata: part.thoughtSignature ? { google: { thoughtSignature: part.thoughtSignature } } : void 0
@@ -1605,10 +1607,10 @@ var GoogleGenerativeAILanguageModel = class {
         });
       }
     }
-    const sources = (_m = extractSources({
+    const sources = (_n = extractSources({
       groundingMetadata: candidate == null ? void 0 : candidate.groundingMetadata,
       generateId: this.config.generateId
-    })) != null ? _m : [];
+    })) != null ? _n : [];
     for (const source of sources) {
       content.push(source);
     }
@@ -1619,21 +1621,22 @@ var GoogleGenerativeAILanguageModel = class {
         hasToolCalls: content.some((part) => part.type === "tool-call")
       }),
       usage: {
-        inputTokens: (_n = usageMetadata == null ? void 0 : usageMetadata.promptTokenCount) != null ? _n : void 0,
-        outputTokens: (_o = usageMetadata == null ? void 0 : usageMetadata.candidatesTokenCount) != null ? _o : void 0,
-        totalTokens: (_p = usageMetadata == null ? void 0 : usageMetadata.totalTokenCount) != null ? _p : void 0,
-        reasoningTokens: (_q = usageMetadata == null ? void 0 : usageMetadata.thoughtsTokenCount) != null ? _q : void 0,
-        cachedInputTokens: (_r = usageMetadata == null ? void 0 : usageMetadata.cachedContentTokenCount) != null ? _r : void 0
+        inputTokens: (_o = usageMetadata == null ? void 0 : usageMetadata.promptTokenCount) != null ? _o : void 0,
+        outputTokens: (_p = usageMetadata == null ? void 0 : usageMetadata.candidatesTokenCount) != null ? _p : void 0,
+        totalTokens: (_q = usageMetadata == null ? void 0 : usageMetadata.totalTokenCount) != null ? _q : void 0,
+        reasoningTokens: (_r = usageMetadata == null ? void 0 : usageMetadata.thoughtsTokenCount) != null ? _r : void 0,
+        cachedInputTokens: (_s = usageMetadata == null ? void 0 : usageMetadata.cachedContentTokenCount) != null ? _s : void 0
       },
       warnings,
       providerMetadata: {
         google: {
-          promptFeedback: (_s = response.promptFeedback) != null ? _s : null,
-          groundingMetadata: (_t = candidate == null ? void 0 : candidate.groundingMetadata) != null ? _t : null,
-          urlContextMetadata: (_u = candidate == null ? void 0 : candidate.urlContextMetadata) != null ? _u : null,
-          safetyRatings: (_v = candidate == null ? void 0 : candidate.safetyRatings) != null ? _v : null,
-          serviceTier: (_w = response.serviceTier) != null ? _w : null,
-          ...(candidate == null ? void 0 : candidate.continuationToken) != null ? { continuationToken: candidate.continuationToken } : {},
+          promptFeedback: (_t = response.promptFeedback) != null ? _t : null,
+          groundingMetadata: (_u = candidate == null ? void 0 : candidate.groundingMetadata) != null ? _u : null,
+          ...groundingMetadataPerRequest.length > 1 ? { groundingMetadataPerRequest } : {},
+          urlContextMetadata: (_v = candidate == null ? void 0 : candidate.urlContextMetadata) != null ? _v : null,
+          safetyRatings: (_w = candidate == null ? void 0 : candidate.safetyRatings) != null ? _w : null,
+          serviceTier: (_x = response.serviceTier) != null ? _x : null,
+          ...(candidate == null ? void 0 : candidate.continuationToken) != null && candidate.finishReason === "MAX_TOKENS" ? { continuationToken: candidate.continuationToken } : {},
           ...continuedUsageMetadata != null && firstPromptTokenCount != null ? { initialPromptTokenCount: firstPromptTokenCount } : {},
           usageMetadata: usageMetadata != null ? usageMetadata : null
         }
@@ -1665,14 +1668,27 @@ var GoogleGenerativeAILanguageModel = class {
       fetch: this.config.fetch
     });
     const { responseHeaders, value: firstResponse } = await streamGenerateContent(args);
+    const groundingMetadataPerRequest = [];
+    let requestGroundingMetadata = null;
     const response = continueGoogleStream({
       stream: firstResponse,
       getContinuationToken: (chunk) => {
         var _a, _b;
         const candidate = (_a = chunk.candidates) == null ? void 0 : _a[0];
+        if ((candidate == null ? void 0 : candidate.groundingMetadata) != null) {
+          requestGroundingMetadata = candidate.groundingMetadata;
+        }
         return (candidate == null ? void 0 : candidate.finishReason) === "CONTINUATION" ? (_b = candidate.continuationToken) != null ? _b : void 0 : void 0;
       },
-      continueStream: async (continuationToken) => (await streamGenerateContent({ ...args, continuationToken })).value
+      continueStream: async (continuationToken) => {
+        const { value } = await streamGenerateContent({
+          ...args,
+          continuationToken
+        });
+        groundingMetadataPerRequest.push(requestGroundingMetadata);
+        requestGroundingMetadata = null;
+        return value;
+      }
     });
     let finishReason = "unknown";
     const usage = {
@@ -1950,6 +1966,18 @@ var GoogleGenerativeAILanguageModel = class {
                 id: currentReasoningBlockId
               });
             }
+            if (groundingMetadataPerRequest.length > 0) {
+              providerMetadata = {
+                ...providerMetadata,
+                google: {
+                  ...providerMetadata == null ? void 0 : providerMetadata.google,
+                  groundingMetadataPerRequest: [
+                    ...groundingMetadataPerRequest,
+                    requestGroundingMetadata
+                  ]
+                }
+              };
+            }
             controller.enqueue({
               type: "finish",
               finishReason,
@@ -2167,13 +2195,13 @@ var getSafetyRatingSchema = () => z5.object({
   severityScore: z5.number().nullish(),
   blocked: z5.boolean().nullish()
 });
-var tokenDetailsSchema = z5.array(
+var tokenDetailsSchema = () => z5.array(
   z5.object({
     modality: z5.string(),
     tokenCount: z5.number()
   })
 ).nullish();
-var usageSchema = z5.object({
+var usageSchema = () => z5.object({
   cachedContentTokenCount: z5.number().nullish(),
   thoughtsTokenCount: z5.number().nullish(),
   promptTokenCount: z5.number().nullish(),
@@ -2182,8 +2210,8 @@ var usageSchema = z5.object({
   // https://cloud.google.com/vertex-ai/generative-ai/docs/reference/rest/v1/GenerateContentResponse#TrafficType
   trafficType: z5.string().nullish(),
   // https://ai.google.dev/api/generate-content#Modality
-  promptTokensDetails: tokenDetailsSchema,
-  candidatesTokensDetails: tokenDetailsSchema
+  promptTokensDetails: tokenDetailsSchema(),
+  candidatesTokensDetails: tokenDetailsSchema()
 });
 var getUrlContextMetadataSchema = () => z5.object({
   urlMetadata: z5.array(
@@ -2206,7 +2234,7 @@ var responseSchema = lazySchema5(
           urlContextMetadata: getUrlContextMetadataSchema().nullish()
         })
       ).nullish(),
-      usageMetadata: usageSchema.nullish(),
+      usageMetadata: usageSchema().nullish(),
       promptFeedback: z5.object({
         blockReason: z5.string().nullish(),
         safetyRatings: z5.array(getSafetyRatingSchema()).nullish()
@@ -2228,7 +2256,7 @@ var chunkSchema = lazySchema5(
           urlContextMetadata: getUrlContextMetadataSchema().nullish()
         })
       ).nullish(),
-      usageMetadata: usageSchema.nullish(),
+      usageMetadata: usageSchema().nullish(),
       promptFeedback: z5.object({
         blockReason: z5.string().nullish(),
         safetyRatings: z5.array(getSafetyRatingSchema()).nullish()
@@ -2274,7 +2302,7 @@ import {
   zodSchema as zodSchema7
 } from "@ai-sdk/provider-utils";
 import { z as z8 } from "zod/v4";
-var fileSearchArgsBaseSchema = z8.object({
+var fileSearchArgsBaseSchema = () => z8.object({
   /** The names of the file_search_stores to retrieve from.
    *  Example: `fileSearchStores/my-file-search-store-123`
    */
@@ -2291,7 +2319,7 @@ var fileSearchArgsBaseSchema = z8.object({
   ).optional()
 }).passthrough();
 var fileSearchArgsSchema = lazySchema7(
-  () => zodSchema7(fileSearchArgsBaseSchema)
+  () => zodSchema7(fileSearchArgsBaseSchema())
 );
 var fileSearch = createProviderDefinedToolFactory2({
   id: "google.file_search",
@@ -2319,7 +2347,7 @@ import {
   zodSchema as zodSchema9
 } from "@ai-sdk/provider-utils";
 import { z as z10 } from "zod/v4";
-var googleSearchToolArgsBaseSchema = z10.object({
+var googleSearchToolArgsBaseSchema = () => z10.object({
   searchTypes: z10.object({
     webSearch: z10.object({}).optional(),
     imageSearch: z10.object({}).optional()
@@ -2330,7 +2358,7 @@ var googleSearchToolArgsBaseSchema = z10.object({
   }).optional()
 }).passthrough();
 var googleSearchToolArgsSchema = lazySchema9(
-  () => zodSchema9(googleSearchToolArgsBaseSchema)
+  () => zodSchema9(googleSearchToolArgsBaseSchema())
 );
 var googleSearch = createProviderDefinedToolFactory4({
   id: "google.google_search",