@ai-sdk/google 4.0.0-beta.6 → 4.0.0-beta.8

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # @ai-sdk/google
 
+## 4.0.0-beta.8
+
+### Patch Changes
+
+- 4dac142: feat(google): add new finishMessage field in providerMetadata
+
+## 4.0.0-beta.7
+
+### Patch Changes
+
+- 82288b0: feat(provider/google): add `gemini-embedding-2-preview` and fix multimodal embedding support with `embedMany`
+
 ## 4.0.0-beta.6
 
 ### Patch Changes

package/dist/index.d.mts

@@ -73,6 +73,7 @@ declare const responseSchema: _ai_sdk_provider_utils.LazySchema<{
             })[] | null | undefined;
         } | null | undefined;
         finishReason?: string | null | undefined;
+        finishMessage?: string | null | undefined;
         safetyRatings?: {
             category?: string | null | undefined;
             probability?: string | null | undefined;
@@ -158,14 +159,21 @@ declare const responseSchema: _ai_sdk_provider_utils.LazySchema<{
 type GroundingMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['groundingMetadata']>;
 type UrlContextMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['urlContextMetadata']>;
 type SafetyRatingSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['safetyRatings']>[number];
+type PromptFeedbackSchema = NonNullable<InferSchema<typeof responseSchema>['promptFeedback']>;
+type UsageMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['usageMetadata']>;
 
 type GoogleGenerativeAIGroundingMetadata = GroundingMetadataSchema;
 type GoogleGenerativeAIUrlContextMetadata = UrlContextMetadataSchema;
 type GoogleGenerativeAISafetyRating = SafetyRatingSchema;
+type GoogleGenerativeAIPromptFeedback = PromptFeedbackSchema;
+type GoogleGenerativeAIUsageMetadata = UsageMetadataSchema;
 interface GoogleGenerativeAIProviderMetadata {
+    promptFeedback: GoogleGenerativeAIPromptFeedback | null;
     groundingMetadata: GoogleGenerativeAIGroundingMetadata | null;
     urlContextMetadata: GoogleGenerativeAIUrlContextMetadata | null;
     safetyRatings: GoogleGenerativeAISafetyRating[] | null;
+    usageMetadata: GoogleGenerativeAIUsageMetadata | null;
+    finishMessage: string | null;
 }
 
 type GoogleGenerativeAIImageModelId = 'imagen-4.0-generate-001' | 'imagen-4.0-ultra-generate-001' | 'imagen-4.0-fast-generate-001' | 'gemini-2.5-flash-image' | 'gemini-3-pro-image-preview' | 'gemini-3.1-flash-image-preview' | (string & {});
@@ -182,18 +190,18 @@ declare const googleImageModelOptionsSchema: _ai_sdk_provider_utils.LazySchema<{
 }>;
 type GoogleImageModelOptions = InferSchema<typeof googleImageModelOptionsSchema>;
 
-type GoogleGenerativeAIEmbeddingModelId = 'gemini-embedding-001' | (string & {});
+type GoogleGenerativeAIEmbeddingModelId = 'gemini-embedding-001' | 'gemini-embedding-2-preview' | (string & {});
 declare const googleEmbeddingModelOptions: _ai_sdk_provider_utils.LazySchema<{
     outputDimensionality?: number | undefined;
     taskType?: "SEMANTIC_SIMILARITY" | "CLASSIFICATION" | "CLUSTERING" | "RETRIEVAL_DOCUMENT" | "RETRIEVAL_QUERY" | "QUESTION_ANSWERING" | "FACT_VERIFICATION" | "CODE_RETRIEVAL_QUERY" | undefined;
-    content?: ({
+    content?: (({
         text: string;
     } | {
         inlineData: {
             mimeType: string;
             data: string;
         };
-    })[] | undefined;
+    })[] | null)[] | undefined;
 }>;
 type GoogleEmbeddingModelOptions = InferSchema<typeof googleEmbeddingModelOptions>;
 

package/dist/index.d.ts

@@ -73,6 +73,7 @@ declare const responseSchema: _ai_sdk_provider_utils.LazySchema<{
             })[] | null | undefined;
         } | null | undefined;
         finishReason?: string | null | undefined;
+        finishMessage?: string | null | undefined;
         safetyRatings?: {
             category?: string | null | undefined;
             probability?: string | null | undefined;
@@ -158,14 +159,21 @@ declare const responseSchema: _ai_sdk_provider_utils.LazySchema<{
 type GroundingMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['groundingMetadata']>;
 type UrlContextMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['urlContextMetadata']>;
 type SafetyRatingSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['safetyRatings']>[number];
+type PromptFeedbackSchema = NonNullable<InferSchema<typeof responseSchema>['promptFeedback']>;
+type UsageMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['usageMetadata']>;
 
 type GoogleGenerativeAIGroundingMetadata = GroundingMetadataSchema;
 type GoogleGenerativeAIUrlContextMetadata = UrlContextMetadataSchema;
 type GoogleGenerativeAISafetyRating = SafetyRatingSchema;
+type GoogleGenerativeAIPromptFeedback = PromptFeedbackSchema;
+type GoogleGenerativeAIUsageMetadata = UsageMetadataSchema;
 interface GoogleGenerativeAIProviderMetadata {
+    promptFeedback: GoogleGenerativeAIPromptFeedback | null;
     groundingMetadata: GoogleGenerativeAIGroundingMetadata | null;
     urlContextMetadata: GoogleGenerativeAIUrlContextMetadata | null;
     safetyRatings: GoogleGenerativeAISafetyRating[] | null;
+    usageMetadata: GoogleGenerativeAIUsageMetadata | null;
+    finishMessage: string | null;
 }
 
 type GoogleGenerativeAIImageModelId = 'imagen-4.0-generate-001' | 'imagen-4.0-ultra-generate-001' | 'imagen-4.0-fast-generate-001' | 'gemini-2.5-flash-image' | 'gemini-3-pro-image-preview' | 'gemini-3.1-flash-image-preview' | (string & {});
@@ -182,18 +190,18 @@ declare const googleImageModelOptionsSchema: _ai_sdk_provider_utils.LazySchema<{
 }>;
 type GoogleImageModelOptions = InferSchema<typeof googleImageModelOptionsSchema>;
 
-type GoogleGenerativeAIEmbeddingModelId = 'gemini-embedding-001' | (string & {});
+type GoogleGenerativeAIEmbeddingModelId = 'gemini-embedding-001' | 'gemini-embedding-2-preview' | (string & {});
 declare const googleEmbeddingModelOptions: _ai_sdk_provider_utils.LazySchema<{
     outputDimensionality?: number | undefined;
     taskType?: "SEMANTIC_SIMILARITY" | "CLASSIFICATION" | "CLUSTERING" | "RETRIEVAL_DOCUMENT" | "RETRIEVAL_QUERY" | "QUESTION_ANSWERING" | "FACT_VERIFICATION" | "CODE_RETRIEVAL_QUERY" | undefined;
-    content?: ({
+    content?: (({
         text: string;
     } | {
         inlineData: {
             mimeType: string;
             data: string;
         };
-    })[] | undefined;
+    })[] | null)[] | undefined;
 }>;
 type GoogleEmbeddingModelOptions = InferSchema<typeof googleEmbeddingModelOptions>;
 

package/dist/index.js

@@ -30,7 +30,7 @@ module.exports = __toCommonJS(src_exports);
 var import_provider_utils16 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "4.0.0-beta.6" : "0.0.0-test";
+var VERSION = true ? "4.0.0-beta.8" : "0.0.0-test";
 
 // src/google-generative-ai-embedding-model.ts
 var import_provider = require("@ai-sdk/provider");
@@ -99,11 +99,15 @@ var googleEmbeddingModelOptions = (0, import_provider_utils2.lazySchema)(
         "CODE_RETRIEVAL_QUERY"
       ]).optional(),
       /**
-       * Optional. Multimodal content parts for embedding non-text content
-       * (images, video, PDF, audio). When provided, these parts are merged
-       * with the text values in the embedding request.
+       * Optional. Per-value multimodal content parts for embedding non-text
+       * content (images, video, PDF, audio). Each entry corresponds to the
+       * embedding value at the same index and its parts are merged with the
+       * text value in the request. Use `null` for entries that are text-only.
+       *
+       * 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: import_v42.z.array(googleEmbeddingContentPartSchema).min(1).optional()
+      content: import_v42.z.array(import_v42.z.array(googleEmbeddingContentPartSchema).min(1).nullable()).optional()
     })
   )
 );
@@ -126,7 +130,6 @@ var GoogleGenerativeAIEmbeddingModel = class {
     abortSignal,
     providerOptions
   }) {
-    var _a;
     const googleOptions = await (0, import_provider_utils3.parseProviderOptions)({
       provider: "google",
       providerOptions,
@@ -144,10 +147,16 @@ var GoogleGenerativeAIEmbeddingModel = class {
       await (0, import_provider_utils3.resolve)(this.config.headers),
       headers
     );
-    const multimodalContent = (_a = googleOptions == null ? void 0 : googleOptions.content) != null ? _a : [];
+    const multimodalContent = googleOptions == null ? void 0 : googleOptions.content;
+    if (multimodalContent != null && multimodalContent.length !== values.length) {
+      throw new Error(
+        `The number of multimodal content entries (${multimodalContent.length}) must match the number of values (${values.length}).`
+      );
+    }
     if (values.length === 1) {
+      const valueParts = multimodalContent == null ? void 0 : multimodalContent[0];
       const textPart = values[0] ? [{ text: values[0] }] : [];
-      const parts = multimodalContent.length > 0 ? [...textPart, ...multimodalContent] : [{ text: values[0] }];
+      const parts = valueParts != null ? [...textPart, ...valueParts] : [{ text: values[0] }];
       const {
         responseHeaders: responseHeaders2,
         value: response2,
@@ -185,13 +194,14 @@ var GoogleGenerativeAIEmbeddingModel = class {
       url: `${this.config.baseURL}/models/${this.modelId}:batchEmbedContents`,
       headers: mergedHeaders,
       body: {
-        requests: values.map((value) => {
+        requests: values.map((value, index) => {
+          const valueParts = multimodalContent == null ? void 0 : multimodalContent[index];
           const textPart = value ? [{ text: value }] : [];
           return {
             model: `models/${this.modelId}`,
             content: {
               role: "user",
-              parts: multimodalContent.length > 0 ? [...textPart, ...multimodalContent] : [{ text: value }]
+              parts: valueParts != null ? [...textPart, ...valueParts] : [{ text: value }]
             },
             outputDimensionality: googleOptions == null ? void 0 : googleOptions.outputDimensionality,
             taskType: googleOptions == null ? void 0 : googleOptions.taskType
@@ -1038,7 +1048,7 @@ var GoogleGenerativeAILanguageModel = class {
     };
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
     const { args, warnings, providerOptionsName } = await this.getArgs(options);
     const mergedHeaders = (0, import_provider_utils6.combineHeaders)(
       await (0, import_provider_utils6.resolve)(this.config.headers),
@@ -1160,7 +1170,8 @@ var GoogleGenerativeAILanguageModel = class {
           groundingMetadata: (_h = candidate.groundingMetadata) != null ? _h : null,
           urlContextMetadata: (_i = candidate.urlContextMetadata) != null ? _i : null,
           safetyRatings: (_j = candidate.safetyRatings) != null ? _j : null,
-          usageMetadata: usageMetadata != null ? usageMetadata : null
+          usageMetadata: usageMetadata != null ? usageMetadata : null,
+          finishMessage: (_k = candidate.finishMessage) != null ? _k : null
         }
       },
       request: { body: args },
@@ -1210,7 +1221,7 @@ var GoogleGenerativeAILanguageModel = class {
             controller.enqueue({ type: "stream-start", warnings });
           },
           transform(chunk, controller) {
-            var _a, _b, _c, _d, _e, _f;
+            var _a, _b, _c, _d, _e, _f, _g;
             if (options.includeRawChunks) {
               controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
             }
@@ -1412,12 +1423,11 @@ var GoogleGenerativeAILanguageModel = class {
                   promptFeedback: (_e = value.promptFeedback) != null ? _e : null,
                   groundingMetadata: lastGroundingMetadata,
                   urlContextMetadata: lastUrlContextMetadata,
-                  safetyRatings: (_f = candidate.safetyRatings) != null ? _f : null
+                  safetyRatings: (_f = candidate.safetyRatings) != null ? _f : null,
+                  usageMetadata: usageMetadata != null ? usageMetadata : null,
+                  finishMessage: (_g = candidate.finishMessage) != null ? _g : null
                 }
               };
-              if (usageMetadata != null) {
-                providerMetadata[providerOptionsName].usageMetadata = usageMetadata;
-              }
             }
           },
           flush(controller) {
@@ -1677,6 +1687,7 @@ var responseSchema = (0, import_provider_utils6.lazySchema)(
         import_v45.z.object({
           content: getContentSchema().nullish().or(import_v45.z.object({}).strict()),
           finishReason: import_v45.z.string().nullish(),
+          finishMessage: import_v45.z.string().nullish(),
           safetyRatings: import_v45.z.array(getSafetyRatingSchema()).nullish(),
           groundingMetadata: getGroundingMetadataSchema().nullish(),
           urlContextMetadata: getUrlContextMetadataSchema().nullish()
@@ -1697,6 +1708,7 @@ var chunkSchema = (0, import_provider_utils6.lazySchema)(
         import_v45.z.object({
           content: getContentSchema().nullish(),
           finishReason: import_v45.z.string().nullish(),
+          finishMessage: import_v45.z.string().nullish(),
           safetyRatings: import_v45.z.array(getSafetyRatingSchema()).nullish(),
           groundingMetadata: getGroundingMetadataSchema().nullish(),
           urlContextMetadata: getUrlContextMetadataSchema().nullish()