@ai-sdk/google 4.0.63 → 4.0.65

package/dist/internal/index.d.ts

@@ -1,5 +1,5 @@
 import * as _ai_sdk_provider_utils from '@ai-sdk/provider-utils';
-import { WORKFLOW_SERIALIZE, WORKFLOW_DESERIALIZE, Resolvable, FetchFunction, InferSchema } from '@ai-sdk/provider-utils';
+import { WORKFLOW_SERIALIZE, WORKFLOW_DESERIALIZE, Resolvable, FetchFunction, InferSchema, createToolNameMapping } from '@ai-sdk/provider-utils';
 import * as _ai_sdk_provider from '@ai-sdk/provider';
 import { LanguageModelV4, JSONObject, LanguageModelV4CallOptions, SharedV4Warning, LanguageModelV4GenerateResult, LanguageModelV4StreamResult, SpeechModelV4 } from '@ai-sdk/provider';
 import { z } from 'zod/v4';
@@ -96,7 +96,10 @@ declare class GoogleLanguageModel implements LanguageModelV4 {
     constructor(modelId: GoogleModelId, config: GoogleLanguageModelConfig);
     get provider(): string;
     get supportedUrls(): Record<string, RegExp[]> | PromiseLike<Record<string, RegExp[]>>;
-    protected getArgs({ prompt, maxOutputTokens, temperature, topP, topK, frequencyPenalty, presencePenalty, stopSequences, responseFormat, seed, tools, toolChoice, reasoning, providerOptions, }: LanguageModelV4CallOptions, { isStreaming }?: {
+    static prepareRequest({ modelId, config, options: { prompt, maxOutputTokens, temperature, topP, topK, frequencyPenalty, presencePenalty, stopSequences, responseFormat, seed, tools, toolChoice, reasoning, providerOptions, }, isStreaming, }: {
+        modelId: GoogleModelId;
+        config: GoogleLanguageModelConfig;
+        options: LanguageModelV4CallOptions;
         isStreaming?: boolean;
     }): Promise<{
         args: {
@@ -169,11 +172,15 @@ declare class GoogleLanguageModel implements LanguageModelV4 {
         warnings: SharedV4Warning[];
         providerOptionsNames: readonly string[];
         extraHeaders: Record<string, string> | undefined;
+        toolNameMapping: _ai_sdk_provider_utils.ToolNameMapping;
     }>;
-    protected convertGenerateContentResponse({ response, warnings, providerOptionsNames, }: {
+    private getArgs;
+    static convertGenerateContentResponse({ config, response, warnings, providerOptionsNames, toolNameMapping, }: {
+        config: GoogleLanguageModelConfig;
         response: InferSchema<typeof responseSchema>;
         warnings: SharedV4Warning[];
         providerOptionsNames: readonly string[];
+        toolNameMapping?: ReturnType<typeof createToolNameMapping>;
     }): LanguageModelV4GenerateResult;
     doGenerate(options: LanguageModelV4CallOptions): Promise<LanguageModelV4GenerateResult>;
     doStream(options: LanguageModelV4CallOptions): Promise<LanguageModelV4StreamResult>;

package/dist/internal/index.js

@@ -1,6 +1,7 @@
 // src/google-language-model.ts
 import {
   combineHeaders,
+  createToolNameMapping,
   createEventSourceResponseHandler,
   createJsonResponseHandler,
   generateId,
@@ -97,6 +98,8 @@ function convertJSONSchemaDefinition(jsonSchema, isRoot, referenceContext) {
     format,
     const: constValue,
     minLength,
+    minItems,
+    maxItems,
     enum: enumValues
   } = jsonSchema;
   const result = {};
@@ -179,6 +182,12 @@ function convertJSONSchemaDefinition(jsonSchema, isRoot, referenceContext) {
   if (minLength !== void 0) {
     result.minLength = minLength;
   }
+  if (minItems !== void 0) {
+    result.minItems = minItems;
+  }
+  if (maxItems !== void 0) {
+    result.maxItems = maxItems;
+  }
   return result;
 }
 function convertJSONSchemaReference({
@@ -1578,25 +1587,32 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
     var _a, _b, _c;
     return (_c = (_b = (_a = this.config).supportedUrls) == null ? void 0 : _b.call(_a)) != null ? _c : {};
   }
-  async getArgs({
-    prompt,
-    maxOutputTokens,
-    temperature,
-    topP,
-    topK,
-    frequencyPenalty,
-    presencePenalty,
-    stopSequences,
-    responseFormat,
-    seed,
-    tools,
-    toolChoice,
-    reasoning,
-    providerOptions
-  }, { isStreaming = false } = {}) {
+  static async prepareRequest({
+    modelId,
+    config,
+    options: {
+      prompt,
+      maxOutputTokens,
+      temperature,
+      topP,
+      topK,
+      frequencyPenalty,
+      presencePenalty,
+      stopSequences,
+      responseFormat,
+      seed,
+      tools,
+      toolChoice,
+      reasoning,
+      providerOptions
+    },
+    isStreaming = false
+  }) {
     var _a, _b, _c;
     const warnings = [];
-    const providerOptionsNames = this.config.provider.includes("vertex") ? ["googleVertex", "vertex"] : ["google"];
+    const providerOptionsNames = config.provider.includes(
+      "vertex"
+    ) ? ["googleVertex", "vertex"] : ["google"];
     let googleOptions;
     for (const name of providerOptionsNames) {
       googleOptions = await parseProviderOptions({
@@ -1613,19 +1629,19 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
         schema: googleLanguageModelOptions
       });
     }
-    const isVertexProvider = this.config.provider.startsWith("google.vertex.");
+    const isVertexProvider = config.provider.startsWith("google.vertex.");
     if ((tools == null ? void 0 : tools.some(
       (tool) => tool.type === "provider" && tool.id === "google.vertex_rag_store"
     )) && !isVertexProvider) {
       warnings.push({
         type: "other",
-        message: `The 'vertex_rag_store' tool is only supported with the Google Vertex provider and might not be supported or could behave unexpectedly with the current Google provider (${this.config.provider}).`
+        message: `The 'vertex_rag_store' tool is only supported with the Google Vertex provider and might not be supported or could behave unexpectedly with the current Google provider (${config.provider}).`
       });
     }
     if ((googleOptions == null ? void 0 : googleOptions.streamFunctionCallArguments) && !isVertexProvider) {
       warnings.push({
         type: "other",
-        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`
+        message: `'streamFunctionCallArguments' is only supported on the Vertex AI API and will be ignored with the current Google provider (${config.provider}). See https://docs.cloud.google.com/vertex-ai/generative-ai/docs/multimodal/function-calling#streaming-fc`
       });
     }
     if ((googleOptions == null ? void 0 : googleOptions.serviceTier) && isVertexProvider) {
@@ -1637,7 +1653,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
     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}).`
+        message: `'sharedRequestType' and 'requestType' are Vertex AI options and are ignored with the current Google provider (${config.provider}).`
       });
     }
     const vertexPaygoHeaders = isVertexProvider && ((googleOptions == null ? void 0 : googleOptions.sharedRequestType) || (googleOptions == null ? void 0 : googleOptions.requestType)) ? {
@@ -1665,13 +1681,13 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
       if (droppedImageConfigFields.length > 0) {
         warnings.push({
           type: "other",
-          message: `${droppedImageConfigFields.join(", ")} ${droppedImageConfigFields.length === 1 ? "is a Vertex AI option and is" : "are Vertex AI options and are"} ignored with the current Google provider (${this.config.provider}).`
+          message: `${droppedImageConfigFields.join(", ")} ${droppedImageConfigFields.length === 1 ? "is a Vertex AI option and is" : "are Vertex AI options and are"} ignored with the current Google provider (${config.provider}).`
         });
         imageConfig = geminiApiImageConfig;
       }
     }
-    const isGemmaModel = this.modelId.toLowerCase().startsWith("gemma-");
-    const isGemini25DeveloperApiModel = !isVertexProvider && gemini25ModelPattern2.test(this.modelId);
+    const isGemmaModel = modelId.toLowerCase().startsWith("gemma-");
+    const isGemini25DeveloperApiModel = !isVertexProvider && gemini25ModelPattern2.test(modelId);
     if (isGemini25DeveloperApiModel && frequencyPenalty != null) {
       warnings.push({
         type: "unsupported",
@@ -1684,7 +1700,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
         feature: "presencePenalty"
       });
     }
-    const { usesGemini3Features } = getGoogleModelCapabilities(this.modelId);
+    const { usesGemini3Features } = getGoogleModelCapabilities(modelId);
     const { contents, systemInstruction } = convertToGoogleMessages(prompt, {
       isGemmaModel,
       isGemini3Model: usesGemini3Features,
@@ -1700,12 +1716,18 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
     } = prepareTools({
       tools,
       toolChoice,
-      modelId: this.modelId,
+      modelId,
       isVertexProvider
     });
+    const toolNameMapping = createToolNameMapping({
+      tools,
+      providerToolNames: {
+        "google.code_execution": "code_execution"
+      }
+    });
     const resolvedThinking = resolveThinkingConfig({
       reasoning,
-      modelId: this.modelId,
+      modelId,
       warnings
     });
     const thinkingConfig = (googleOptions == null ? void 0 : googleOptions.thinkingConfig) || resolvedThinking ? { ...resolvedThinking, ...googleOptions == null ? void 0 : googleOptions.thinkingConfig } : void 0;
@@ -1767,15 +1789,26 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
       },
       warnings: [...warnings, ...toolWarnings],
       providerOptionsNames,
-      extraHeaders: vertexPaygoHeaders
+      extraHeaders: vertexPaygoHeaders,
+      toolNameMapping
     };
   }
-  convertGenerateContentResponse({
+  getArgs(options, { isStreaming = false } = {}) {
+    return _GoogleLanguageModel.prepareRequest({
+      modelId: this.modelId,
+      config: this.config,
+      options,
+      isStreaming
+    });
+  }
+  static convertGenerateContentResponse({
+    config,
     response,
     warnings,
-    providerOptionsNames
+    providerOptionsNames,
+    toolNameMapping
   }) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q, _r, _s, _t, _u;
     const wrapProviderMetadata = (payload) => Object.fromEntries(
       providerOptionsNames.map((name) => [name, payload])
     );
@@ -1790,12 +1823,12 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
     let lastServerToolCallId;
     for (const part of parts) {
       if ("executableCode" in part && ((_g = part.executableCode) == null ? void 0 : _g.code)) {
-        const toolCallId = this.config.generateId();
+        const toolCallId = config.generateId();
         lastCodeExecutionToolCallId = toolCallId;
         content.push({
           type: "tool-call",
           toolCallId,
-          toolName: "code_execution",
+          toolName: (_h = toolNameMapping == null ? void 0 : toolNameMapping.toCustomToolName("code_execution")) != null ? _h : "code_execution",
           input: JSON.stringify(part.executableCode),
           providerExecuted: true
         });
@@ -1804,10 +1837,10 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
           type: "tool-result",
           // Results correspond to the most recent executable code part.
           toolCallId: lastCodeExecutionToolCallId,
-          toolName: "code_execution",
+          toolName: (_i = toolNameMapping == null ? void 0 : toolNameMapping.toCustomToolName("code_execution")) != null ? _i : "code_execution",
           result: {
             outcome: part.codeExecutionResult.outcome,
-            output: (_h = part.codeExecutionResult.output) != null ? _h : ""
+            output: (_j = part.codeExecutionResult.output) != null ? _j : ""
           }
         });
       } else if ("text" in part && part.text != null) {
@@ -1829,9 +1862,9 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
       } else if ("functionCall" in part && part.functionCall.name != null) {
         content.push({
           type: "tool-call",
-          toolCallId: part.functionCall.id || this.config.generateId(),
+          toolCallId: part.functionCall.id || config.generateId(),
           toolName: part.functionCall.name,
-          input: JSON.stringify((_i = part.functionCall.args) != null ? _i : {}),
+          input: JSON.stringify((_k = part.functionCall.args) != null ? _k : {}),
           providerMetadata: part.thoughtSignature ? wrapProviderMetadata({
             thoughtSignature: part.thoughtSignature
           }) : void 0
@@ -1848,13 +1881,13 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
           }) : void 0
         });
       } else if ("toolCall" in part && part.toolCall) {
-        const toolCallId = part.toolCall.id || this.config.generateId();
+        const toolCallId = part.toolCall.id || config.generateId();
         lastServerToolCallId = toolCallId;
         content.push({
           type: "tool-call",
           toolCallId,
           toolName: `server:${part.toolCall.toolType}`,
-          input: JSON.stringify((_j = part.toolCall.args) != null ? _j : {}),
+          input: JSON.stringify((_l = part.toolCall.args) != null ? _l : {}),
           providerExecuted: true,
           dynamic: true,
           providerMetadata: part.thoughtSignature ? wrapProviderMetadata({
@@ -1867,12 +1900,12 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
           })
         });
       } else if ("toolResponse" in part && part.toolResponse) {
-        const responseToolCallId = lastServerToolCallId || part.toolResponse.id || this.config.generateId();
+        const responseToolCallId = lastServerToolCallId || part.toolResponse.id || config.generateId();
         content.push({
           type: "tool-result",
           toolCallId: responseToolCallId,
           toolName: `server:${part.toolResponse.toolType}`,
-          result: (_k = part.toolResponse.response) != null ? _k : {},
+          result: (_m = part.toolResponse.response) != null ? _m : {},
           providerMetadata: part.thoughtSignature ? wrapProviderMetadata({
             thoughtSignature: part.thoughtSignature,
             serverToolCallId: responseToolCallId,
@@ -1885,10 +1918,10 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
         lastServerToolCallId = void 0;
       }
     }
-    const sources = (_l = extractSources({
+    const sources = (_n = extractSources({
       groundingMetadata: candidate == null ? void 0 : candidate.groundingMetadata,
-      generateId: this.config.generateId
-    })) != null ? _l : [];
+      generateId: config.generateId
+    })) != null ? _n : [];
     for (const source of sources) {
       content.push(source);
     }
@@ -1907,22 +1940,28 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
       usage: convertGoogleUsage(usageMetadata),
       warnings,
       providerMetadata: wrapProviderMetadata({
-        promptFeedback: (_m = response.promptFeedback) != null ? _m : null,
-        groundingMetadata: (_n = candidate == null ? void 0 : candidate.groundingMetadata) != null ? _n : null,
-        urlContextMetadata: (_o = candidate == null ? void 0 : candidate.urlContextMetadata) != null ? _o : null,
-        safetyRatings: (_p = candidate == null ? void 0 : candidate.safetyRatings) != null ? _p : null,
+        promptFeedback: (_o = response.promptFeedback) != null ? _o : null,
+        groundingMetadata: (_p = candidate == null ? void 0 : candidate.groundingMetadata) != null ? _p : null,
+        urlContextMetadata: (_q = candidate == null ? void 0 : candidate.urlContextMetadata) != null ? _q : null,
+        safetyRatings: (_r = candidate == null ? void 0 : candidate.safetyRatings) != null ? _r : null,
         usageMetadata: usageMetadata != null ? usageMetadata : null,
-        finishMessage: (_q = candidate == null ? void 0 : candidate.finishMessage) != null ? _q : null,
-        serviceTier: (_r = usageMetadata == null ? void 0 : usageMetadata.serviceTier) != null ? _r : null
+        finishMessage: (_s = candidate == null ? void 0 : candidate.finishMessage) != null ? _s : null,
+        serviceTier: (_t = usageMetadata == null ? void 0 : usageMetadata.serviceTier) != null ? _t : null
       }),
       response: {
         // TODO timestamp, model id
-        id: (_s = response.responseId) != null ? _s : void 0
+        id: (_u = response.responseId) != null ? _u : void 0
       }
     };
   }
   async doGenerate(options) {
-    const { args, warnings, providerOptionsNames, extraHeaders } = await this.getArgs(options);
+    const {
+      args,
+      warnings,
+      providerOptionsNames,
+      extraHeaders,
+      toolNameMapping
+    } = await this.getArgs(options);
     const mergedHeaders = combineHeaders(
       this.config.headers ? await resolve(this.config.headers) : void 0,
       options.headers,
@@ -1943,10 +1982,12 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
-    const result = this.convertGenerateContentResponse({
+    const result = _GoogleLanguageModel.convertGenerateContentResponse({
+      config: this.config,
       response,
       warnings,
-      providerOptionsNames
+      providerOptionsNames,
+      toolNameMapping
     });
     return {
       ...result,
@@ -1959,7 +2000,13 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
     };
   }
   async doStream(options) {
-    const { args, warnings, providerOptionsNames, extraHeaders } = await this.getArgs(options, { isStreaming: true });
+    const {
+      args,
+      warnings,
+      providerOptionsNames,
+      extraHeaders,
+      toolNameMapping
+    } = await this.getArgs(options, { isStreaming: true });
     const wrapProviderMetadata = (payload) => Object.fromEntries(
       providerOptionsNames.map((name) => [name, payload])
     );
@@ -2100,7 +2147,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
                   controller.enqueue({
                     type: "tool-call",
                     toolCallId,
-                    toolName: "code_execution",
+                    toolName: toolNameMapping.toCustomToolName("code_execution"),
                     input: JSON.stringify(part.executableCode),
                     providerExecuted: true
                   });
@@ -2110,7 +2157,7 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
                     controller.enqueue({
                       type: "tool-result",
                       toolCallId,
-                      toolName: "code_execution",
+                      toolName: toolNameMapping.toCustomToolName("code_execution"),
                       result: {
                         outcome: part.codeExecutionResult.outcome,
                         output: (_g = part.codeExecutionResult.output) != null ? _g : ""