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

package/dist/internal/index.mjs

@@ -584,7 +584,15 @@ var googleLanguageModelOptions = lazySchema2(
           latitude: z2.number(),
           longitude: z2.number()
         }).optional()
-      }).optional()
+      }).optional(),
+      /**
+       * Optional. The service tier to use for the request.
+       */
+      serviceTier: z2.enum([
+        "SERVICE_TIER_STANDARD",
+        "SERVICE_TIER_FLEX",
+        "SERVICE_TIER_PRIORITY"
+      ]).optional()
     })
   )
 );
@@ -947,14 +955,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 = combineHeaders(
       await resolve(this.config.headers),
@@ -1076,7 +1085,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 },
@@ -1112,6 +1122,7 @@ var GoogleGenerativeAILanguageModel = class {
     let providerMetadata = void 0;
     let lastGroundingMetadata = null;
     let lastUrlContextMetadata = null;
+    let serviceTier = null;
     const generateId2 = this.config.generateId;
     let hasToolCalls = false;
     let currentTextBlockId = null;
@@ -1139,6 +1150,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;
@@ -1329,7 +1343,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
                 }
               };
             }
@@ -1670,7 +1685,8 @@ var responseSchema = lazySchema3(
       promptFeedback: z3.object({
         blockReason: z3.string().nullish(),
         safetyRatings: z3.array(getSafetyRatingSchema()).nullish()
-      }).nullish()
+      }).nullish(),
+      serviceTier: z3.string().nullish()
     })
   )
 );
@@ -1691,7 +1707,8 @@ var chunkSchema = lazySchema3(
       promptFeedback: z3.object({
         blockReason: z3.string().nullish(),
         safetyRatings: z3.array(getSafetyRatingSchema()).nullish()
-      }).nullish()
+      }).nullish(),
+      serviceTier: z3.string().nullish()
     })
   )
 );