@ai-sdk/google 3.0.53 → 3.0.55

package/dist/internal/index.mjs

@@ -277,7 +277,7 @@ function appendLegacyToolResultParts(parts, toolName, outputValue) {
   }
 }
 function convertToGoogleGenerativeAIMessages(prompt, options) {
-  var _a, _b, _c, _d;
+  var _a, _b, _c, _d, _e, _f, _g, _h;
   const systemInstructionParts = [];
   const contents = [];
   let systemMessagesAllowed = true;
@@ -364,6 +364,18 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
                 };
               }
               case "tool-call": {
+                const serverToolCallId = (providerOpts == null ? void 0 : providerOpts.serverToolCallId) != null ? String(providerOpts.serverToolCallId) : void 0;
+                const serverToolType = (providerOpts == null ? void 0 : providerOpts.serverToolType) != null ? String(providerOpts.serverToolType) : void 0;
+                if (serverToolCallId && serverToolType) {
+                  return {
+                    toolCall: {
+                      toolType: serverToolType,
+                      args: typeof part.input === "string" ? JSON.parse(part.input) : part.input,
+                      id: serverToolCallId
+                    },
+                    thoughtSignature
+                  };
+                }
                 return {
                   functionCall: {
                     name: part.toolName,
@@ -372,6 +384,21 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
                   thoughtSignature
                 };
               }
+              case "tool-result": {
+                const serverToolCallId = (providerOpts == null ? void 0 : providerOpts.serverToolCallId) != null ? String(providerOpts.serverToolCallId) : void 0;
+                const serverToolType = (providerOpts == null ? void 0 : providerOpts.serverToolType) != null ? String(providerOpts.serverToolType) : void 0;
+                if (serverToolCallId && serverToolType) {
+                  return {
+                    toolResponse: {
+                      toolType: serverToolType,
+                      response: part.output.type === "json" ? part.output.value : {},
+                      id: serverToolCallId
+                    },
+                    thoughtSignature
+                  };
+                }
+                return void 0;
+              }
             }
           }).filter((part) => part !== void 0)
         });
@@ -384,6 +411,26 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
           if (part.type === "tool-approval-response") {
             continue;
           }
+          const partProviderOpts = (_g = (_d = part.providerOptions) == null ? void 0 : _d[providerOptionsName]) != null ? _g : providerOptionsName !== "google" ? (_e = part.providerOptions) == null ? void 0 : _e.google : (_f = part.providerOptions) == null ? void 0 : _f.vertex;
+          const serverToolCallId = (partProviderOpts == null ? void 0 : partProviderOpts.serverToolCallId) != null ? String(partProviderOpts.serverToolCallId) : void 0;
+          const serverToolType = (partProviderOpts == null ? void 0 : partProviderOpts.serverToolType) != null ? String(partProviderOpts.serverToolType) : void 0;
+          if (serverToolCallId && serverToolType) {
+            const serverThoughtSignature = (partProviderOpts == null ? void 0 : partProviderOpts.thoughtSignature) != null ? String(partProviderOpts.thoughtSignature) : void 0;
+            if (contents.length > 0) {
+              const lastContent = contents[contents.length - 1];
+              if (lastContent.role === "model") {
+                lastContent.parts.push({
+                  toolResponse: {
+                    toolType: serverToolType,
+                    response: part.output.type === "json" ? part.output.value : {},
+                    id: serverToolCallId
+                  },
+                  thoughtSignature: serverThoughtSignature
+                });
+                continue;
+              }
+            }
+          }
           const output = part.output;
           if (output.type === "content") {
             if (supportsFunctionResponseParts) {
@@ -397,7 +444,7 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
                 name: part.toolName,
                 response: {
                   name: part.toolName,
-                  content: output.type === "execution-denied" ? (_d = output.reason) != null ? _d : "Tool execution denied." : output.value
+                  content: output.type === "execution-denied" ? (_h = output.reason) != null ? _h : "Tool execution denied." : output.value
                 }
               }
             });
@@ -566,7 +613,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()
     })
   )
 );
@@ -580,7 +635,7 @@ function prepareTools({
   toolChoice,
   modelId
 }) {
-  var _a;
+  var _a, _b;
   tools = (tools == null ? void 0 : tools.length) ? tools : void 0;
   const toolWarnings = [];
   const isLatest = [
@@ -589,13 +644,14 @@ function prepareTools({
     "gemini-pro-latest"
   ].some((id) => id === modelId);
   const isGemini2orNewer = modelId.includes("gemini-2") || modelId.includes("gemini-3") || modelId.includes("nano-banana") || isLatest;
+  const isGemini3orNewer = modelId.includes("gemini-3");
   const supportsFileSearch = modelId.includes("gemini-2.5") || modelId.includes("gemini-3");
   if (tools == null) {
     return { tools: void 0, toolConfig: void 0, toolWarnings };
   }
   const hasFunctionTools = tools.some((tool) => tool.type === "function");
   const hasProviderTools = tools.some((tool) => tool.type === "provider");
-  if (hasFunctionTools && hasProviderTools) {
+  if (hasFunctionTools && hasProviderTools && !isGemini3orNewer) {
     toolWarnings.push({
       type: "unsupported",
       feature: `combination of function and provider-defined tools`
@@ -700,6 +756,45 @@ function prepareTools({
           break;
       }
     });
+    if (hasFunctionTools && isGemini3orNewer && googleTools2.length > 0) {
+      const functionDeclarations2 = [];
+      for (const tool of tools) {
+        if (tool.type === "function") {
+          functionDeclarations2.push({
+            name: tool.name,
+            description: (_a = tool.description) != null ? _a : "",
+            parameters: convertJSONSchemaToOpenAPISchema(tool.inputSchema)
+          });
+        }
+      }
+      const combinedToolConfig = {
+        functionCallingConfig: { mode: "VALIDATED" },
+        includeServerSideToolInvocations: true
+      };
+      if (toolChoice != null) {
+        switch (toolChoice.type) {
+          case "auto":
+            break;
+          case "none":
+            combinedToolConfig.functionCallingConfig = { mode: "NONE" };
+            break;
+          case "required":
+            combinedToolConfig.functionCallingConfig = { mode: "ANY" };
+            break;
+          case "tool":
+            combinedToolConfig.functionCallingConfig = {
+              mode: "ANY",
+              allowedFunctionNames: [toolChoice.toolName]
+            };
+            break;
+        }
+      }
+      return {
+        tools: [...googleTools2, { functionDeclarations: functionDeclarations2 }],
+        toolConfig: combinedToolConfig,
+        toolWarnings
+      };
+    }
     return {
       tools: googleTools2.length > 0 ? googleTools2 : void 0,
       toolConfig: void 0,
@@ -713,7 +808,7 @@ function prepareTools({
       case "function":
         functionDeclarations.push({
           name: tool.name,
-          description: (_a = tool.description) != null ? _a : "",
+          description: (_b = tool.description) != null ? _b : "",
           parameters: convertJSONSchemaToOpenAPISchema(tool.inputSchema)
         });
         if (tool.strict === true) {
@@ -922,14 +1017,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, _m, _n, _o, _p;
     const { args, warnings, providerOptionsName } = await this.getArgs(options);
     const mergedHeaders = combineHeaders(
       await resolve(this.config.headers),
@@ -955,6 +1051,7 @@ var GoogleGenerativeAILanguageModel = class {
     const parts = (_b = (_a = candidate.content) == null ? void 0 : _a.parts) != null ? _b : [];
     const usageMetadata = response.usageMetadata;
     let lastCodeExecutionToolCallId;
+    let lastServerToolCallId;
     for (const part of parts) {
       if ("executableCode" in part && ((_c = part.executableCode) == null ? void 0 : _c.code)) {
         const toolCallId = this.config.generateId();
@@ -1022,12 +1119,56 @@ var GoogleGenerativeAILanguageModel = class {
             }
           } : void 0
         });
+      } else if ("toolCall" in part && part.toolCall) {
+        const toolCallId = (_e = part.toolCall.id) != null ? _e : this.config.generateId();
+        lastServerToolCallId = toolCallId;
+        content.push({
+          type: "tool-call",
+          toolCallId,
+          toolName: `server:${part.toolCall.toolType}`,
+          input: JSON.stringify((_f = part.toolCall.args) != null ? _f : {}),
+          providerExecuted: true,
+          dynamic: true,
+          providerMetadata: part.thoughtSignature ? {
+            [providerOptionsName]: {
+              thoughtSignature: part.thoughtSignature,
+              serverToolCallId: toolCallId,
+              serverToolType: part.toolCall.toolType
+            }
+          } : {
+            [providerOptionsName]: {
+              serverToolCallId: toolCallId,
+              serverToolType: part.toolCall.toolType
+            }
+          }
+        });
+      } else if ("toolResponse" in part && part.toolResponse) {
+        const responseToolCallId = (_g = lastServerToolCallId != null ? lastServerToolCallId : part.toolResponse.id) != null ? _g : this.config.generateId();
+        content.push({
+          type: "tool-result",
+          toolCallId: responseToolCallId,
+          toolName: `server:${part.toolResponse.toolType}`,
+          result: (_h = part.toolResponse.response) != null ? _h : {},
+          providerMetadata: part.thoughtSignature ? {
+            [providerOptionsName]: {
+              thoughtSignature: part.thoughtSignature,
+              serverToolCallId: responseToolCallId,
+              serverToolType: part.toolResponse.toolType
+            }
+          } : {
+            [providerOptionsName]: {
+              serverToolCallId: responseToolCallId,
+              serverToolType: part.toolResponse.toolType
+            }
+          }
+        });
+        lastServerToolCallId = void 0;
       }
     }
-    const sources = (_e = extractSources({
+    const sources = (_i = extractSources({
       groundingMetadata: candidate.groundingMetadata,
       generateId: this.config.generateId
-    })) != null ? _e : [];
+    })) != null ? _i : [];
     for (const source of sources) {
       content.push(source);
     }
@@ -1041,18 +1182,19 @@ var GoogleGenerativeAILanguageModel = class {
             (part) => part.type === "tool-call" && !part.providerExecuted
           )
         }),
-        raw: (_f = candidate.finishReason) != null ? _f : void 0
+        raw: (_j = candidate.finishReason) != null ? _j : void 0
       },
       usage: convertGoogleGenerativeAIUsage(usageMetadata),
       warnings,
       providerMetadata: {
         [providerOptionsName]: {
-          promptFeedback: (_g = response.promptFeedback) != null ? _g : null,
-          groundingMetadata: (_h = candidate.groundingMetadata) != null ? _h : null,
-          urlContextMetadata: (_i = candidate.urlContextMetadata) != null ? _i : null,
-          safetyRatings: (_j = candidate.safetyRatings) != null ? _j : null,
+          promptFeedback: (_k = response.promptFeedback) != null ? _k : null,
+          groundingMetadata: (_l = candidate.groundingMetadata) != null ? _l : null,
+          urlContextMetadata: (_m = candidate.urlContextMetadata) != null ? _m : null,
+          safetyRatings: (_n = candidate.safetyRatings) != null ? _n : null,
           usageMetadata: usageMetadata != null ? usageMetadata : null,
-          finishMessage: (_k = candidate.finishMessage) != null ? _k : null
+          finishMessage: (_o = candidate.finishMessage) != null ? _o : null,
+          serviceTier: (_p = response.serviceTier) != null ? _p : null
         }
       },
       request: { body: args },
@@ -1088,6 +1230,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;
@@ -1095,6 +1238,7 @@ var GoogleGenerativeAILanguageModel = class {
     let blockCounter = 0;
     const emittedSourceUrls = /* @__PURE__ */ new Set();
     let lastCodeExecutionToolCallId;
+    let lastServerToolCallId;
     return {
       stream: response.pipeThrough(
         new TransformStream({
@@ -1102,7 +1246,7 @@ var GoogleGenerativeAILanguageModel = class {
             controller.enqueue({ type: "stream-start", warnings });
           },
           transform(chunk, controller) {
-            var _a, _b, _c, _d, _e, _f, _g;
+            var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
             if (options.includeRawChunks) {
               controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
             }
@@ -1115,6 +1259,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;
@@ -1254,6 +1401,42 @@ var GoogleGenerativeAILanguageModel = class {
                     data: part.inlineData.data,
                     providerMetadata: fileMeta
                   });
+                } else if ("toolCall" in part && part.toolCall) {
+                  const toolCallId = (_e = part.toolCall.id) != null ? _e : generateId2();
+                  lastServerToolCallId = toolCallId;
+                  const serverMeta = {
+                    [providerOptionsName]: {
+                      ...part.thoughtSignature ? { thoughtSignature: part.thoughtSignature } : {},
+                      serverToolCallId: toolCallId,
+                      serverToolType: part.toolCall.toolType
+                    }
+                  };
+                  controller.enqueue({
+                    type: "tool-call",
+                    toolCallId,
+                    toolName: `server:${part.toolCall.toolType}`,
+                    input: JSON.stringify((_f = part.toolCall.args) != null ? _f : {}),
+                    providerExecuted: true,
+                    dynamic: true,
+                    providerMetadata: serverMeta
+                  });
+                } else if ("toolResponse" in part && part.toolResponse) {
+                  const responseToolCallId = (_g = lastServerToolCallId != null ? lastServerToolCallId : part.toolResponse.id) != null ? _g : generateId2();
+                  const serverMeta = {
+                    [providerOptionsName]: {
+                      ...part.thoughtSignature ? { thoughtSignature: part.thoughtSignature } : {},
+                      serverToolCallId: responseToolCallId,
+                      serverToolType: part.toolResponse.toolType
+                    }
+                  };
+                  controller.enqueue({
+                    type: "tool-result",
+                    toolCallId: responseToolCallId,
+                    toolName: `server:${part.toolResponse.toolType}`,
+                    result: (_h = part.toolResponse.response) != null ? _h : {},
+                    providerMetadata: serverMeta
+                  });
+                  lastServerToolCallId = void 0;
                 }
               }
               const toolCallDeltas = getToolCallsFromParts({
@@ -1301,12 +1484,13 @@ var GoogleGenerativeAILanguageModel = class {
               };
               providerMetadata = {
                 [providerOptionsName]: {
-                  promptFeedback: (_e = value.promptFeedback) != null ? _e : null,
+                  promptFeedback: (_i = value.promptFeedback) != null ? _i : null,
                   groundingMetadata: lastGroundingMetadata,
                   urlContextMetadata: lastUrlContextMetadata,
-                  safetyRatings: (_f = candidate.safetyRatings) != null ? _f : null,
+                  safetyRatings: (_j = candidate.safetyRatings) != null ? _j : null,
                   usageMetadata: usageMetadata != null ? usageMetadata : null,
-                  finishMessage: (_g = candidate.finishMessage) != null ? _g : null
+                  finishMessage: (_k = candidate.finishMessage) != null ? _k : null,
+                  serviceTier
                 }
               };
             }
@@ -1520,6 +1704,22 @@ var getContentSchema = () => z3.object({
         thought: z3.boolean().nullish(),
         thoughtSignature: z3.string().nullish()
       }),
+      z3.object({
+        toolCall: z3.object({
+          toolType: z3.string(),
+          args: z3.unknown().nullish(),
+          id: z3.string()
+        }),
+        thoughtSignature: z3.string().nullish()
+      }),
+      z3.object({
+        toolResponse: z3.object({
+          toolType: z3.string(),
+          response: z3.unknown().nullish(),
+          id: z3.string()
+        }),
+        thoughtSignature: z3.string().nullish()
+      }),
       z3.object({
         executableCode: z3.object({
           language: z3.string(),
@@ -1578,7 +1778,8 @@ var responseSchema = lazySchema3(
       promptFeedback: z3.object({
         blockReason: z3.string().nullish(),
         safetyRatings: z3.array(getSafetyRatingSchema()).nullish()
-      }).nullish()
+      }).nullish(),
+      serviceTier: z3.string().nullish()
     })
   )
 );
@@ -1599,7 +1800,8 @@ var chunkSchema = lazySchema3(
       promptFeedback: z3.object({
         blockReason: z3.string().nullish(),
         safetyRatings: z3.array(getSafetyRatingSchema()).nullish()
-      }).nullish()
+      }).nullish(),
+      serviceTier: z3.string().nullish()
     })
   )
 );