@ai-sdk/google 4.0.0-canary.68 → 4.0.0-canary.70

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # @ai-sdk/google
 
+## 4.0.0-canary.70
+
+### Patch Changes
+
+- aeea161: feat(google): read `serviceTier` from `x-gemini-service-tier` response header in Gemini API and use PayGo for Vertex
+
+## 4.0.0-canary.69
+
+### Patch Changes
+
+- 7f04802: feat(provider/google): add support for managed agents in the Interactions API
+
 ## 4.0.0-canary.68
 
 ### Patch Changes

package/dist/index.d.ts

@@ -41,6 +41,8 @@ declare const googleLanguageModelOptions: _ai_sdk_provider_utils.LazySchema<{
     } | undefined;
     streamFunctionCallArguments?: boolean | undefined;
     serviceTier?: "standard" | "flex" | "priority" | undefined;
+    sharedRequestType?: "standard" | "flex" | "priority" | undefined;
+    requestType?: "shared" | undefined;
 }>;
 type GoogleLanguageModelOptions = InferSchema<typeof googleLanguageModelOptions>;
 
@@ -189,7 +191,6 @@ declare const responseSchema: _ai_sdk_provider_utils.LazySchema<{
             blocked?: boolean | null | undefined;
         }[] | null | undefined;
     } | null | undefined;
-    serviceTier?: string | null | undefined;
 }>;
 type GroundingMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['groundingMetadata']>;
 type UrlContextMetadataSchema = NonNullable<InferSchema<typeof responseSchema>['candidates'][number]['urlContextMetadata']>;
@@ -318,6 +319,29 @@ declare const googleInteractionsLanguageModelOptions: _ai_sdk_provider_utils.Laz
     signature?: string | null | undefined;
     interactionId?: string | null | undefined;
     pollingTimeoutMs?: number | null | undefined;
+    background?: boolean | null | undefined;
+    environment?: string | {
+        type: "remote";
+        sources?: ({
+            type: "gcs";
+            source: string;
+            target?: string | null | undefined;
+        } | {
+            type: "repository";
+            source: string;
+            target?: string | null | undefined;
+        } | {
+            type: "inline";
+            content: string;
+            target: string;
+        })[] | null | undefined;
+        network?: "disabled" | {
+            allowlist: {
+                domain: string;
+                transform?: Record<string, string>[] | null | undefined;
+            }[];
+        } | null | undefined;
+    } | null | undefined;
 }>;
 type GoogleLanguageModelInteractionsOptions = InferSchema<typeof googleInteractionsLanguageModelOptions>;
 
@@ -347,14 +371,12 @@ type GoogleInteractionsProviderMetadata = {
  * Type-only module: declares the union of supported Gemini Interactions agent
  * names. Used by the `google.interactions({ agent })` factory branch.
  *
- * Sourced from `googleapis/js-genai` `src/interactions/resources/interactions.ts`
- * (`Interaction.agent` enum). Subject to expansion as Google adds new agents.
- *
- * This is a strict string-literal union (no `string` escape hatch) so that
- * passing an unknown agent name is a compile-time error. Add new agents here
- * as Google publishes them.
+ * Strict string-literal union: unknown agent names are a compile-time error.
+ * User-defined agents (created via the `/agents` endpoint) are addressed by
+ * a separate `{ managedAgent: string }` factory shape — see
+ * `GoogleInteractionsModelInput`.
  */
-type GoogleInteractionsAgentName = 'deep-research-pro-preview-12-2025' | 'deep-research-preview-04-2026' | 'deep-research-max-preview-04-2026';
+type GoogleInteractionsAgentName = 'deep-research-pro-preview-12-2025' | 'deep-research-preview-04-2026' | 'deep-research-max-preview-04-2026' | 'antigravity-preview-05-2026';
 
 declare const googleTools: {
     /**
@@ -486,11 +508,16 @@ interface GoogleProvider extends ProviderV4 {
     files(): FilesV4;
     /**
      * Creates a language model targeting the Gemini Interactions API
-     * (`POST /v1beta/interactions`). Pass either a model ID (string) or
-     * `{ agent: <name> }` to use a Gemini agent preset.
+     * (`POST /v1beta/interactions`). Pass:
+     *   - a model ID (string),
+     *   - `{ agent: <name> }` to use a known Gemini agent preset, or
+     *   - `{ managedAgent: <name> }` to use a user-defined agent created via
+     *     the `/v1beta/agents` endpoint.
      */
     interactions(modelIdOrAgent: GoogleInteractionsModelId | {
         agent: GoogleInteractionsAgentName;
+    } | {
+        managedAgent: string;
     }): LanguageModelV4;
     tools: typeof googleTools;
 }

package/dist/index.js

@@ -7,7 +7,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "4.0.0-canary.68" : "0.0.0-test";
+var VERSION = true ? "4.0.0-canary.70" : "0.0.0-test";
 
 // src/google-embedding-model.ts
 import {
@@ -989,17 +989,32 @@ var googleLanguageModelOptions = lazySchema4(
        */
       streamFunctionCallArguments: z4.boolean().optional(),
       /**
-       * Optional. The service tier to use for the request.
+       * Optional. The service tier to use for the request. Sent as the
+       * `serviceTier` body field. Gemini API only.
        */
-      serviceTier: z4.enum(["standard", "flex", "priority"]).optional()
+      serviceTier: z4.enum(["standard", "flex", "priority"]).optional(),
+      /**
+       * Optional. Vertex AI only. Sent as the
+       * `X-Vertex-AI-LLM-Shared-Request-Type` request header to select a
+       * shared (PayGo) tier. With Provisioned Throughput allocated and
+       * `requestType` unset, the request falls back to this tier only if
+       * PT capacity is exhausted.
+       *
+       * https://docs.cloud.google.com/vertex-ai/generative-ai/docs/priority-paygo
+       * https://docs.cloud.google.com/vertex-ai/generative-ai/docs/flex-paygo
+       */
+      sharedRequestType: z4.enum(["priority", "flex", "standard"]).optional(),
+      /**
+       * Optional. Vertex AI only. Sent as the `X-Vertex-AI-LLM-Request-Type`
+       * request header. Set to `'shared'` together with `sharedRequestType`
+       * to bypass Provisioned Throughput entirely.
+       *
+       * https://docs.cloud.google.com/vertex-ai/generative-ai/docs/priority-paygo
+       */
+      requestType: z4.enum(["shared"]).optional()
     })
   )
 );
-var VertexServiceTierMap = {
-  standard: "SERVICE_TIER_STANDARD",
-  flex: "SERVICE_TIER_FLEX",
-  priority: "SERVICE_TIER_PRIORITY"
-};
 
 // src/google-prepare-tools.ts
 import {
@@ -1580,10 +1595,27 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
         message: `'streamFunctionCallArguments' is only supported on the Vertex AI API and will be ignored with the current Google provider (${this.config.provider}). See https://docs.cloud.google.com/vertex-ai/generative-ai/docs/multimodal/function-calling#streaming-fc`
       });
     }
-    let sanitizedServiceTier = googleOptions == null ? void 0 : googleOptions.serviceTier;
     if ((googleOptions == null ? void 0 : googleOptions.serviceTier) && isVertexProvider) {
-      sanitizedServiceTier = VertexServiceTierMap[googleOptions.serviceTier];
+      warnings.push({
+        type: "other",
+        message: "'serviceTier' is a Gemini API option and is not supported on Vertex AI. Use 'sharedRequestType' (and optionally 'requestType') instead. See https://docs.cloud.google.com/vertex-ai/generative-ai/docs/priority-paygo"
+      });
+    }
+    if (((googleOptions == null ? void 0 : googleOptions.sharedRequestType) || (googleOptions == null ? void 0 : googleOptions.requestType)) && !isVertexProvider) {
+      warnings.push({
+        type: "other",
+        message: `'sharedRequestType' and 'requestType' are Vertex AI options and are ignored with the current Google provider (${this.config.provider}).`
+      });
     }
+    const vertexPaygoHeaders = isVertexProvider && ((googleOptions == null ? void 0 : googleOptions.sharedRequestType) || (googleOptions == null ? void 0 : googleOptions.requestType)) ? {
+      ...googleOptions.sharedRequestType && {
+        "X-Vertex-AI-LLM-Shared-Request-Type": googleOptions.sharedRequestType
+      },
+      ...googleOptions.requestType && {
+        "X-Vertex-AI-LLM-Request-Type": googleOptions.requestType
+      }
+    } : void 0;
+    const bodyServiceTier = isVertexProvider ? void 0 : googleOptions == null ? void 0 : googleOptions.serviceTier;
     const isGemmaModel = this.modelId.toLowerCase().startsWith("gemma-");
     const supportsFunctionResponseParts = this.modelId.startsWith("gemini-3");
     const { contents, systemInstruction } = convertToGoogleMessages(prompt, {
@@ -1658,21 +1690,23 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
         toolConfig,
         cachedContent: googleOptions == null ? void 0 : googleOptions.cachedContent,
         labels: googleOptions == null ? void 0 : googleOptions.labels,
-        serviceTier: sanitizedServiceTier
+        serviceTier: bodyServiceTier
       },
       warnings: [...warnings, ...toolWarnings],
-      providerOptionsNames
+      providerOptionsNames,
+      extraHeaders: vertexPaygoHeaders
     };
   }
   async doGenerate(options) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r;
-    const { args, warnings, providerOptionsNames } = await this.getArgs(options);
+    const { args, warnings, providerOptionsNames, extraHeaders } = await this.getArgs(options);
     const wrapProviderMetadata = (payload) => Object.fromEntries(
       providerOptionsNames.map((name) => [name, payload])
     );
     const mergedHeaders = combineHeaders2(
       this.config.headers ? await resolve2(this.config.headers) : void 0,
-      options.headers
+      options.headers,
+      extraHeaders
     );
     const {
       responseHeaders,
@@ -1821,7 +1855,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
         safetyRatings: (_p = candidate.safetyRatings) != null ? _p : null,
         usageMetadata: usageMetadata != null ? usageMetadata : null,
         finishMessage: (_q = candidate.finishMessage) != null ? _q : null,
-        serviceTier: (_r = response.serviceTier) != null ? _r : null
+        serviceTier: (_r = responseHeaders == null ? void 0 : responseHeaders["x-gemini-service-tier"]) != null ? _r : null
       }),
       request: { body: args },
       response: {
@@ -1832,16 +1866,15 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
     };
   }
   async doStream(options) {
-    const { args, warnings, providerOptionsNames } = await this.getArgs(
-      options,
-      { isStreaming: true }
-    );
+    var _a;
+    const { args, warnings, providerOptionsNames, extraHeaders } = await this.getArgs(options, { isStreaming: true });
     const wrapProviderMetadata = (payload) => Object.fromEntries(
       providerOptionsNames.map((name) => [name, payload])
     );
     const headers = combineHeaders2(
       this.config.headers ? await resolve2(this.config.headers) : void 0,
-      options.headers
+      options.headers,
+      extraHeaders
     );
     const { responseHeaders, value: response } = await postJsonToApi2({
       url: `${this.config.baseURL}/${getModelPath(
@@ -1862,7 +1895,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
     let providerMetadata = void 0;
     let lastGroundingMetadata = null;
     let lastUrlContextMetadata = null;
-    let serviceTier = null;
+    const serviceTier = (_a = responseHeaders == null ? void 0 : responseHeaders["x-gemini-service-tier"]) != null ? _a : null;
     const generateId3 = this.config.generateId;
     let hasToolCalls = false;
     let currentTextBlockId = null;
@@ -1907,7 +1940,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
             controller.enqueue({ type: "stream-start", warnings });
           },
           transform(chunk, controller) {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o;
+            var _a2, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o;
             if (options.includeRawChunks) {
               controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
             }
@@ -1920,10 +1953,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
             if (usageMetadata != null) {
               usage = usageMetadata;
             }
-            if (value.serviceTier != null) {
-              serviceTier = value.serviceTier;
-            }
-            const candidate = (_a = value.candidates) == null ? void 0 : _a[0];
+            const candidate = (_a2 = value.candidates) == null ? void 0 : _a2[0];
             if (candidate == null) {
               return;
             }
@@ -2574,8 +2604,7 @@ var responseSchema = lazySchema5(
       promptFeedback: z5.object({
         blockReason: z5.string().nullish(),
         safetyRatings: z5.array(getSafetyRatingSchema()).nullish()
-      }).nullish(),
-      serviceTier: z5.string().nullish()
+      }).nullish()
     })
   )
 );
@@ -2596,8 +2625,7 @@ var chunkSchema = lazySchema5(
       promptFeedback: z5.object({
         blockReason: z5.string().nullish(),
         safetyRatings: z5.array(getSafetyRatingSchema()).nullish()
-      }).nullish(),
-      serviceTier: z5.string().nullish()
+      }).nullish()
     })
   )
 );
@@ -5049,7 +5077,61 @@ var googleInteractionsLanguageModelOptions = lazySchema17(
        * call) before giving up. Defaults to 30 minutes. Long-running agents
        * such as deep research can take tens of minutes — increase if needed.
        */
-      pollingTimeoutMs: z19.number().int().positive().nullish()
+      pollingTimeoutMs: z19.number().int().positive().nullish(),
+      /**
+       * Run the interaction in the background. Required for agents whose
+       * server-side workflow cannot complete within a single request/response.
+       * When `true`, the POST returns with a non-terminal status and the SDK
+       * polls `GET /interactions/{id}` until the work completes. Some agents
+       * reject `true`; see the agent's documentation for which mode it
+       * requires.
+       */
+      background: z19.boolean().nullish(),
+      /**
+       * Environment configuration for the agent sandbox. Only applies to agent
+       * calls (`google.interactions({ agent })`); ignored on model-id calls.
+       *
+       *   - `"remote"`: provision a fresh sandbox for this call.
+       *   - any other string: an existing `environment_id` to reuse.
+       *   - object: provision a fresh sandbox and optionally preload `sources`
+       *     and/or constrain outbound traffic via `network`.
+       */
+      environment: z19.union([
+        z19.string(),
+        z19.object({
+          type: z19.literal("remote"),
+          sources: z19.array(
+            z19.union([
+              z19.object({
+                type: z19.literal("gcs"),
+                source: z19.string(),
+                target: z19.string().nullish()
+              }),
+              z19.object({
+                type: z19.literal("repository"),
+                source: z19.string(),
+                target: z19.string().nullish()
+              }),
+              z19.object({
+                type: z19.literal("inline"),
+                content: z19.string(),
+                target: z19.string()
+              })
+            ])
+          ).nullish(),
+          network: z19.union([
+            z19.literal("disabled"),
+            z19.object({
+              allowlist: z19.array(
+                z19.object({
+                  domain: z19.string(),
+                  transform: z19.array(z19.record(z19.string(), z19.string())).nullish()
+                })
+              )
+            })
+          ]).nullish()
+        })
+      ]).nullish()
     })
   )
 );
@@ -5786,6 +5868,9 @@ var GoogleInteractionsLanguageModel = class _GoogleInteractionsLanguageModel {
     if (typeof modelOrAgent === "string") {
       this.modelId = modelOrAgent;
       this.agent = void 0;
+    } else if ("managedAgent" in modelOrAgent) {
+      this.modelId = modelOrAgent.managedAgent;
+      this.agent = modelOrAgent.managedAgent;
     } else {
       this.modelId = modelOrAgent.agent;
       this.agent = modelOrAgent.agent;
@@ -5826,7 +5911,7 @@ var GoogleInteractionsLanguageModel = class _GoogleInteractionsLanguageModel {
     };
   }
   async getArgs(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, _x;
+    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, _y, _z;
     const warnings = [];
     const opts = await parseProviderOptions6({
       provider: "google",
@@ -5983,6 +6068,55 @@ var GoogleInteractionsLanguageModel = class _GoogleInteractionsLanguageModel {
         agentConfig = { type: "dynamic" };
       }
     }
+    let environment;
+    if ((opts == null ? void 0 : opts.environment) != null) {
+      if (!isAgent) {
+        warnings.push({
+          type: "other",
+          message: "google.interactions: environment is only supported when an agent is set; environment will be omitted from the request body."
+        });
+      } else if (typeof opts.environment === "string") {
+        environment = opts.environment;
+      } else {
+        const env = opts.environment;
+        const sources = (_u = env.sources) == null ? void 0 : _u.map((s) => {
+          var _a2;
+          if (s.type === "inline") {
+            return {
+              type: "inline",
+              content: s.content,
+              target: s.target
+            };
+          }
+          return pruneUndefined({
+            type: s.type,
+            source: s.source,
+            target: (_a2 = s.target) != null ? _a2 : void 0
+          });
+        });
+        let network;
+        if (env.network === "disabled") {
+          network = "disabled";
+        } else if (env.network != null) {
+          network = {
+            allowlist: env.network.allowlist.map(
+              (entry) => {
+                var _a2;
+                return pruneUndefined({
+                  domain: entry.domain,
+                  transform: (_a2 = entry.transform) != null ? _a2 : void 0
+                });
+              }
+            )
+          };
+        }
+        environment = pruneUndefined({
+          type: "remote",
+          sources: sources != null && sources.length > 0 ? sources : void 0,
+          network
+        });
+      }
+    }
     const args = pruneUndefined({
       ...isAgent ? { agent: this.agent } : { model: this.modelId },
       input,
@@ -5990,18 +6124,20 @@ var GoogleInteractionsLanguageModel = class _GoogleInteractionsLanguageModel {
       tools: toolsForBody,
       response_format: responseFormatEntries.length > 0 ? responseFormatEntries : void 0,
       response_modalities: (opts == null ? void 0 : opts.responseModalities) != null ? opts.responseModalities : void 0,
-      previous_interaction_id: (_u = opts == null ? void 0 : opts.previousInteractionId) != null ? _u : void 0,
-      service_tier: (_v = opts == null ? void 0 : opts.serviceTier) != null ? _v : void 0,
-      store: (_w = opts == null ? void 0 : opts.store) != null ? _w : void 0,
+      previous_interaction_id: (_v = opts == null ? void 0 : opts.previousInteractionId) != null ? _v : void 0,
+      service_tier: (_w = opts == null ? void 0 : opts.serviceTier) != null ? _w : void 0,
+      store: (_x = opts == null ? void 0 : opts.store) != null ? _x : void 0,
       generation_config: generationConfig != null && Object.keys(generationConfig).length > 0 ? generationConfig : void 0,
       agent_config: agentConfig,
-      ...isAgent ? { background: true } : {}
+      environment,
+      background: (_y = opts == null ? void 0 : opts.background) != null ? _y : void 0
     });
     return {
       args,
       warnings,
       isAgent,
-      pollingTimeoutMs: (_x = opts == null ? void 0 : opts.pollingTimeoutMs) != null ? _x : void 0
+      isBackground: (opts == null ? void 0 : opts.background) === true,
+      pollingTimeoutMs: (_z = opts == null ? void 0 : opts.pollingTimeoutMs) != null ? _z : void 0
     };
   }
   async doGenerate(options) {
@@ -6087,14 +6223,14 @@ var GoogleInteractionsLanguageModel = class _GoogleInteractionsLanguageModel {
   }
   async doStream(options) {
     var _a;
-    const { args, warnings, isAgent, pollingTimeoutMs } = await this.getArgs(options);
+    const { args, warnings, isBackground, pollingTimeoutMs } = await this.getArgs(options);
     const url = `${this.config.baseURL}/interactions`;
     const mergedHeaders = combineHeaders7(
       INTERACTIONS_API_REVISION_HEADER,
       this.config.headers ? await resolve5(this.config.headers) : void 0,
       options.headers
     );
-    if (isAgent) {
+    if (isBackground) {
       return this.doStreamBackground({
         args,
         warnings,