@ai-sdk/google 4.0.0-beta.16 → 4.0.0-beta.18

package/dist/index.mjs

@@ -7,7 +7,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "4.0.0-beta.16" : "0.0.0-test";
+var VERSION = true ? "4.0.0-beta.18" : "0.0.0-test";
 
 // src/google-generative-ai-embedding-model.ts
 import {
@@ -795,7 +795,15 @@ var googleLanguageModelOptions = lazySchema4(
           latitude: z4.number(),
           longitude: z4.number()
         }).optional()
-      }).optional()
+      }).optional(),
+      /**
+       * Optional. The service tier to use for the request.
+       */
+      serviceTier: z4.enum([
+        "SERVICE_TIER_STANDARD",
+        "SERVICE_TIER_FLEX",
+        "SERVICE_TIER_PRIORITY"
+      ]).optional()
     })
   )
 );
@@ -1158,14 +1166,15 @@ var GoogleGenerativeAILanguageModel = class {
           retrievalConfig: googleOptions.retrievalConfig
         } : googleToolConfig,
         cachedContent: googleOptions == null ? void 0 : googleOptions.cachedContent,
-        labels: googleOptions == null ? void 0 : googleOptions.labels
+        labels: googleOptions == null ? void 0 : googleOptions.labels,
+        serviceTier: googleOptions == null ? void 0 : googleOptions.serviceTier
       },
       warnings: [...warnings, ...toolWarnings],
       providerOptionsName
     };
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
     const { args, warnings, providerOptionsName } = await this.getArgs(options);
     const mergedHeaders = combineHeaders2(
       await resolve2(this.config.headers),
@@ -1287,7 +1296,8 @@ var GoogleGenerativeAILanguageModel = class {
           urlContextMetadata: (_i = candidate.urlContextMetadata) != null ? _i : null,
           safetyRatings: (_j = candidate.safetyRatings) != null ? _j : null,
           usageMetadata: usageMetadata != null ? usageMetadata : null,
-          finishMessage: (_k = candidate.finishMessage) != null ? _k : null
+          finishMessage: (_k = candidate.finishMessage) != null ? _k : null,
+          serviceTier: (_l = response.serviceTier) != null ? _l : null
         }
       },
       request: { body: args },
@@ -1323,6 +1333,7 @@ var GoogleGenerativeAILanguageModel = class {
     let providerMetadata = void 0;
     let lastGroundingMetadata = null;
     let lastUrlContextMetadata = null;
+    let serviceTier = null;
     const generateId3 = this.config.generateId;
     let hasToolCalls = false;
     let currentTextBlockId = null;
@@ -1350,6 +1361,9 @@ var GoogleGenerativeAILanguageModel = class {
             if (usageMetadata != null) {
               usage = usageMetadata;
             }
+            if (value.serviceTier != null) {
+              serviceTier = value.serviceTier;
+            }
             const candidate = (_a = value.candidates) == null ? void 0 : _a[0];
             if (candidate == null) {
               return;
@@ -1540,7 +1554,8 @@ var GoogleGenerativeAILanguageModel = class {
                   urlContextMetadata: lastUrlContextMetadata,
                   safetyRatings: (_f = candidate.safetyRatings) != null ? _f : null,
                   usageMetadata: usageMetadata != null ? usageMetadata : null,
-                  finishMessage: (_g = candidate.finishMessage) != null ? _g : null
+                  finishMessage: (_g = candidate.finishMessage) != null ? _g : null,
+                  serviceTier
                 }
               };
             }
@@ -1881,7 +1896,8 @@ var responseSchema = lazySchema5(
       promptFeedback: z5.object({
         blockReason: z5.string().nullish(),
         safetyRatings: z5.array(getSafetyRatingSchema()).nullish()
-      }).nullish()
+      }).nullish(),
+      serviceTier: z5.string().nullish()
     })
   )
 );
@@ -1902,7 +1918,8 @@ var chunkSchema = lazySchema5(
       promptFeedback: z5.object({
         blockReason: z5.string().nullish(),
         safetyRatings: z5.array(getSafetyRatingSchema()).nullish()
-      }).nullish()
+      }).nullish(),
+      serviceTier: z5.string().nullish()
     })
   )
 );