@ai-sdk/google 2.0.0-beta.17 → 2.0.0-beta.18

package/dist/index.mjs

@@ -522,6 +522,17 @@ function prepareTools({
             });
           }
           break;
+        case "google.code_execution":
+          if (isGemini2) {
+            googleTools2.codeExecution = {};
+          } else {
+            toolWarnings.push({
+              type: "unsupported-tool",
+              tool,
+              details: "The code execution tools is not supported with other Gemini models than Gemini 2."
+            });
+          }
+          break;
         default:
           toolWarnings.push({ type: "unsupported-tool", tool });
           break;
@@ -774,7 +785,7 @@ var GoogleGenerativeAILanguageModel = class {
     };
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
     const { args, warnings } = await this.getArgs(options);
     const body = JSON.stringify(args);
     const mergedHeaders = combineHeaders2(
@@ -798,10 +809,34 @@ var GoogleGenerativeAILanguageModel = class {
     });
     const candidate = response.candidates[0];
     const content = [];
-    const parts = candidate.content == null || typeof candidate.content !== "object" || !("parts" in candidate.content) ? [] : (_a = candidate.content.parts) != null ? _a : [];
+    const parts = (_b = (_a = candidate.content) == null ? void 0 : _a.parts) != null ? _b : [];
     const usageMetadata = response.usageMetadata;
+    let lastCodeExecutionToolCallId;
     for (const part of parts) {
-      if ("text" in part && part.text != null && part.text.length > 0) {
+      if ("executableCode" in part && ((_c = part.executableCode) == null ? void 0 : _c.code)) {
+        const toolCallId = this.config.generateId();
+        lastCodeExecutionToolCallId = toolCallId;
+        content.push({
+          type: "tool-call",
+          toolCallId,
+          toolName: "code_execution",
+          input: JSON.stringify(part.executableCode),
+          providerExecuted: true
+        });
+      } else if ("codeExecutionResult" in part && part.codeExecutionResult) {
+        content.push({
+          type: "tool-result",
+          // Assumes a result directly follows its corresponding call part.
+          toolCallId: lastCodeExecutionToolCallId,
+          toolName: "code_execution",
+          result: {
+            outcome: part.codeExecutionResult.outcome,
+            output: part.codeExecutionResult.output
+          },
+          providerExecuted: true
+        });
+        lastCodeExecutionToolCallId = void 0;
+      } else if ("text" in part && part.text != null && part.text.length > 0) {
         if (part.thought === true) {
           content.push({ type: "reasoning", text: part.text });
         } else {
@@ -822,10 +857,10 @@ var GoogleGenerativeAILanguageModel = class {
         });
       }
     }
-    const sources = (_b = extractSources({
+    const sources = (_d = extractSources({
       groundingMetadata: candidate.groundingMetadata,
       generateId: this.config.generateId
-    })) != null ? _b : [];
+    })) != null ? _d : [];
     for (const source of sources) {
       content.push(source);
     }
@@ -836,18 +871,18 @@ var GoogleGenerativeAILanguageModel = class {
         hasToolCalls: content.some((part) => part.type === "tool-call")
       }),
       usage: {
-        inputTokens: (_c = usageMetadata == null ? void 0 : usageMetadata.promptTokenCount) != null ? _c : void 0,
-        outputTokens: (_d = usageMetadata == null ? void 0 : usageMetadata.candidatesTokenCount) != null ? _d : void 0,
-        totalTokens: (_e = usageMetadata == null ? void 0 : usageMetadata.totalTokenCount) != null ? _e : void 0,
-        reasoningTokens: (_f = usageMetadata == null ? void 0 : usageMetadata.thoughtsTokenCount) != null ? _f : void 0,
-        cachedInputTokens: (_g = usageMetadata == null ? void 0 : usageMetadata.cachedContentTokenCount) != null ? _g : void 0
+        inputTokens: (_e = usageMetadata == null ? void 0 : usageMetadata.promptTokenCount) != null ? _e : void 0,
+        outputTokens: (_f = usageMetadata == null ? void 0 : usageMetadata.candidatesTokenCount) != null ? _f : void 0,
+        totalTokens: (_g = usageMetadata == null ? void 0 : usageMetadata.totalTokenCount) != null ? _g : void 0,
+        reasoningTokens: (_h = usageMetadata == null ? void 0 : usageMetadata.thoughtsTokenCount) != null ? _h : void 0,
+        cachedInputTokens: (_i = usageMetadata == null ? void 0 : usageMetadata.cachedContentTokenCount) != null ? _i : void 0
       },
       warnings,
       providerMetadata: {
         google: {
-          groundingMetadata: (_h = candidate.groundingMetadata) != null ? _h : null,
-          urlContextMetadata: (_i = candidate.urlContextMetadata) != null ? _i : null,
-          safetyRatings: (_j = candidate.safetyRatings) != null ? _j : null,
+          groundingMetadata: (_j = candidate.groundingMetadata) != null ? _j : null,
+          urlContextMetadata: (_k = candidate.urlContextMetadata) != null ? _k : null,
+          safetyRatings: (_l = candidate.safetyRatings) != null ? _l : null,
           usageMetadata: usageMetadata != null ? usageMetadata : null
         }
       },
@@ -890,6 +925,7 @@ var GoogleGenerativeAILanguageModel = class {
     let currentReasoningBlockId = null;
     let blockCounter = 0;
     const emittedSourceUrls = /* @__PURE__ */ new Set();
+    let lastCodeExecutionToolCallId;
     return {
       stream: response.pipeThrough(
         new TransformStream({
@@ -897,7 +933,7 @@ var GoogleGenerativeAILanguageModel = class {
             controller.enqueue({ type: "stream-start", warnings });
           },
           transform(chunk, controller) {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+            var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
             if (options.includeRawChunks) {
               controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
             }
@@ -934,7 +970,33 @@ var GoogleGenerativeAILanguageModel = class {
             if (content != null) {
               const parts = (_g = content.parts) != null ? _g : [];
               for (const part of parts) {
-                if ("text" in part && part.text != null && part.text.length > 0) {
+                if ("executableCode" in part && ((_h = part.executableCode) == null ? void 0 : _h.code)) {
+                  const toolCallId = generateId3();
+                  lastCodeExecutionToolCallId = toolCallId;
+                  controller.enqueue({
+                    type: "tool-call",
+                    toolCallId,
+                    toolName: "code_execution",
+                    input: JSON.stringify(part.executableCode),
+                    providerExecuted: true
+                  });
+                  hasToolCalls = true;
+                } else if ("codeExecutionResult" in part && part.codeExecutionResult) {
+                  const toolCallId = lastCodeExecutionToolCallId;
+                  if (toolCallId) {
+                    controller.enqueue({
+                      type: "tool-result",
+                      toolCallId,
+                      toolName: "code_execution",
+                      result: {
+                        outcome: part.codeExecutionResult.outcome,
+                        output: part.codeExecutionResult.output
+                      },
+                      providerExecuted: true
+                    });
+                    lastCodeExecutionToolCallId = void 0;
+                  }
+                } else if ("text" in part && part.text != null && part.text.length > 0) {
                   if (part.thought === true) {
                     if (currentTextBlockId !== null) {
                       controller.enqueue({
@@ -1025,9 +1087,9 @@ var GoogleGenerativeAILanguageModel = class {
               });
               providerMetadata = {
                 google: {
-                  groundingMetadata: (_h = candidate.groundingMetadata) != null ? _h : null,
-                  urlContextMetadata: (_i = candidate.urlContextMetadata) != null ? _i : null,
-                  safetyRatings: (_j = candidate.safetyRatings) != null ? _j : null
+                  groundingMetadata: (_i = candidate.groundingMetadata) != null ? _i : null,
+                  urlContextMetadata: (_j = candidate.urlContextMetadata) != null ? _j : null,
+                  safetyRatings: (_k = candidate.safetyRatings) != null ? _k : null
                 }
               };
               if (usageMetadata != null) {
@@ -1113,6 +1175,14 @@ var contentSchema = z7.object({
         })
       }),
       z7.object({
+        executableCode: z7.object({
+          language: z7.string(),
+          code: z7.string()
+        }).nullish(),
+        codeExecutionResult: z7.object({
+          outcome: z7.string(),
+          output: z7.string()
+        }).nullish(),
         text: z7.string().nullish(),
         thought: z7.boolean().nullish()
       })
@@ -1159,6 +1229,22 @@ var chunkSchema = z7.object({
   usageMetadata: usageSchema.nullish()
 });
 
+// src/tool/code-execution.ts
+import { createProviderDefinedToolFactoryWithOutputSchema } from "@ai-sdk/provider-utils";
+import { z as z8 } from "zod/v4";
+var codeExecution = createProviderDefinedToolFactoryWithOutputSchema({
+  id: "google.code_execution",
+  name: "code_execution",
+  inputSchema: z8.object({
+    language: z8.string().describe("The programming language of the code."),
+    code: z8.string().describe("The code to be executed.")
+  }),
+  outputSchema: z8.object({
+    outcome: z8.string().describe('The outcome of the execution (e.g., "OUTCOME_OK").'),
+    output: z8.string().describe("The output from the code execution.")
+  })
+});
+
 // src/google-tools.ts
 var googleTools = {
   /**
@@ -1170,7 +1256,18 @@ var googleTools = {
    * Creates a URL context tool that gives Google direct access to real-time web content.
    * Must have name "url_context".
    */
-  urlContext
+  urlContext,
+  /**
+   * A tool that enables the model to generate and run Python code.
+   * Must have name "code_execution".
+   *
+   * @note Ensure the selected model supports Code Execution.
+   * Multi-tool usage with the code execution tool is typically compatible with Gemini >=2 models.
+   *
+   * @see https://ai.google.dev/gemini-api/docs/code-execution (Google AI)
+   * @see https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/code-execution-api (Vertex AI)
+   */
+  codeExecution
 };
 
 // src/google-generative-ai-image-model.ts
@@ -1181,7 +1278,7 @@ import {
   postJsonToApi as postJsonToApi3,
   resolve as resolve3
 } from "@ai-sdk/provider-utils";
-import { z as z8 } from "zod/v4";
+import { z as z9 } from "zod/v4";
 var GoogleGenerativeAIImageModel = class {
   constructor(modelId, settings, config) {
     this.modelId = modelId;
@@ -1273,12 +1370,12 @@ var GoogleGenerativeAIImageModel = class {
     };
   }
 };
-var googleImageResponseSchema = z8.object({
-  predictions: z8.array(z8.object({ bytesBase64Encoded: z8.string() })).default([])
+var googleImageResponseSchema = z9.object({
+  predictions: z9.array(z9.object({ bytesBase64Encoded: z9.string() })).default([])
 });
-var googleImageProviderOptionsSchema = z8.object({
-  personGeneration: z8.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
-  aspectRatio: z8.enum(["1:1", "3:4", "4:3", "9:16", "16:9"]).nullish()
+var googleImageProviderOptionsSchema = z9.object({
+  personGeneration: z9.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
+  aspectRatio: z9.enum(["1:1", "3:4", "4:3", "9:16", "16:9"]).nullish()
 });
 
 // src/google-provider.ts