@ai-sdk/google 2.0.107 → 2.0.109

package/dist/internal/index.mjs

@@ -1,4 +1,7 @@
 // src/google-generative-ai-language-model.ts
+import {
+  APICallError
+} from "@ai-sdk/provider";
 import {
   combineHeaders,
   createEventSourceResponseHandler,
@@ -387,7 +390,11 @@ function isEmptyObjectSchema(jsonSchema) {
 import {
   UnsupportedFunctionalityError as UnsupportedFunctionalityError2
 } from "@ai-sdk/provider";
-import { convertToBase64 } from "@ai-sdk/provider-utils";
+import {
+  stringifyWithBigIntAndCycles,
+  secureJsonParse,
+  convertToBase64
+} from "@ai-sdk/provider-utils";
 var SKIP_THOUGHT_SIGNATURE_VALIDATOR = "skip_thought_signature_validator";
 function convertToGoogleGenerativeAIMessages(prompt, options) {
   var _a, _b, _c;
@@ -488,7 +495,7 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
                   functionCall: {
                     ...part.toolCallId != null ? { id: part.toolCallId } : {},
                     name: part.toolName,
-                    args: part.input
+                    args: parseToolCallInput(part.input)
                   },
                   thoughtSignature: effectiveThoughtSignature
                 };
@@ -516,7 +523,7 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
                 name: part.toolName,
                 response: {
                   name: part.toolName,
-                  content: output.value
+                  content: serializeFunctionResponseContent(output.value)
                 }
               }
             });
@@ -607,11 +614,46 @@ function appendLegacyToolResultParts({
         );
         break;
       default:
-        parts.push({ text: JSON.stringify(contentPart) });
+        parts.push({ text: stringifyWithBigIntAndCycles(contentPart) });
         break;
     }
   }
 }
+function serializeFunctionResponseContent(value) {
+  return containsNonJsonValue(value) ? secureJsonParse(stringifyWithBigIntAndCycles(value)) : value;
+}
+function containsNonJsonValue(value, ancestors = /* @__PURE__ */ new Set()) {
+  if (typeof value === "bigint") {
+    return true;
+  }
+  if (typeof value !== "object" || value === null) {
+    return false;
+  }
+  if (typeof value.toJSON === "function") {
+    return true;
+  }
+  if (ancestors.has(value)) {
+    return true;
+  }
+  ancestors.add(value);
+  try {
+    return Object.values(value).some(
+      (item) => containsNonJsonValue(item, ancestors)
+    );
+  } finally {
+    ancestors.delete(value);
+  }
+}
+function parseToolCallInput(input) {
+  if (typeof input !== "string") {
+    return input;
+  }
+  try {
+    return secureJsonParse(input);
+  } catch (e) {
+    return input;
+  }
+}
 
 // src/get-model-path.ts
 function getModelPath(modelId) {
@@ -625,15 +667,16 @@ import {
   zodSchema
 } from "@ai-sdk/provider-utils";
 import { z } from "zod/v4";
+var getGoogleErrorSchema = () => z.object({
+  code: z.number().nullable(),
+  message: z.string(),
+  status: z.string(),
+  details: z.array(z.unknown()).nullish()
+});
 var googleErrorDataSchema = lazySchema(
   () => zodSchema(
     z.object({
-      error: z.object({
-        code: z.number().nullable(),
-        message: z.string(),
-        status: z.string(),
-        details: z.array(z.unknown()).nullish()
-      })
+      error: getGoogleErrorSchema()
     })
   )
 );
@@ -641,6 +684,39 @@ var googleFailedResponseHandler = createJsonErrorResponseHandler({
   errorSchema: googleErrorDataSchema,
   errorToMessage: (data) => data.error.message
 });
+function mapGoogleErrorStatusToHttpStatusCode(status) {
+  switch (status) {
+    case "CANCELLED":
+      return 499;
+    case "INVALID_ARGUMENT":
+    case "FAILED_PRECONDITION":
+    case "OUT_OF_RANGE":
+      return 400;
+    case "DEADLINE_EXCEEDED":
+      return 504;
+    case "NOT_FOUND":
+      return 404;
+    case "ALREADY_EXISTS":
+    case "ABORTED":
+      return 409;
+    case "PERMISSION_DENIED":
+      return 403;
+    case "UNAUTHENTICATED":
+      return 401;
+    case "RESOURCE_EXHAUSTED":
+      return 429;
+    case "UNIMPLEMENTED":
+      return 501;
+    case "UNKNOWN":
+    case "INTERNAL":
+    case "DATA_LOSS":
+      return 500;
+    case "UNAVAILABLE":
+      return 503;
+    default:
+      return void 0;
+  }
+}
 
 // src/google-generative-ai-options.ts
 import {
@@ -848,6 +924,9 @@ function prepareTools({
   modelId,
   isVertexProvider = false
 }) {
+  if ((toolChoice == null ? void 0 : toolChoice.type) === "none") {
+    tools = tools == null ? void 0 : tools.filter((tool) => tool.type === "function");
+  }
   tools = (tools == null ? void 0 : tools.length) ? tools : void 0;
   const toolWarnings = [];
   const { supportsGemini2Tools, supportsFileSearch, usesGemini3Features } = getGoogleModelCapabilities(modelId);
@@ -863,7 +942,9 @@ function prepareTools({
     toolWarnings.push({
       type: "unsupported-tool",
       tool: tools.find((tool) => tool.type === "function"),
-      details: `Cannot mix function tools with provider-defined tools in the same request. Falling back to provider-defined tools only. The following function tools will be ignored: ${functionTools.map((t) => t.name).join(", ")}. Please use either function tools or provider-defined tools, but not both.`
+      details: `Cannot mix function tools with provider-defined tools in the same request. Falling back to provider-defined tools only. The following function tools will be ignored: ${functionTools.map((t) => t.name).join(
+        ", "
+      )}. Please use either function tools or provider-defined tools, but not both.`
     });
   }
   if (hasProviderDefinedTools) {
@@ -1452,10 +1533,11 @@ var GoogleGenerativeAILanguageModel = class {
       await resolve(this.config.headers),
       options.headers
     );
+    const url = `${this.config.baseURL}/${getModelPath(
+      this.modelId
+    )}:streamGenerateContent?alt=sse`;
     const streamGenerateContent = (body2) => postJsonToApi({
-      url: `${this.config.baseURL}/${getModelPath(
-        this.modelId
-      )}:streamGenerateContent?alt=sse`,
+      url,
       headers,
       body: body2,
       failedResponseHandler: googleFailedResponseHandler,
@@ -1514,7 +1596,7 @@ var GoogleGenerativeAILanguageModel = class {
             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 _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p;
             if (options.includeRawChunks) {
               controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
             }
@@ -1523,6 +1605,23 @@ var GoogleGenerativeAILanguageModel = class {
               return;
             }
             const value = chunk.value;
+            if (value.error != null) {
+              finishReason = "error";
+              const error = value.error;
+              controller.enqueue({
+                type: "error",
+                error: new APICallError({
+                  message: error.message,
+                  url,
+                  requestBodyValues: args,
+                  statusCode: (_a = error.code) != null ? _a : mapGoogleErrorStatusToHttpStatusCode(error.status),
+                  responseHeaders,
+                  responseBody: JSON.stringify({ error }),
+                  data: error
+                })
+              });
+              return;
+            }
             if (value.usageMetadata != null) {
               requestUsageMetadata = value.usageMetadata;
             }
@@ -1531,16 +1630,16 @@ var GoogleGenerativeAILanguageModel = class {
               requestUsageMetadata
             );
             if (usageMetadata != null) {
-              usage.inputTokens = (_a = usageMetadata.promptTokenCount) != null ? _a : void 0;
-              usage.outputTokens = (_b = usageMetadata.candidatesTokenCount) != null ? _b : void 0;
-              usage.totalTokens = (_c = usageMetadata.totalTokenCount) != null ? _c : void 0;
-              usage.reasoningTokens = (_d = usageMetadata.thoughtsTokenCount) != null ? _d : void 0;
-              usage.cachedInputTokens = (_e = usageMetadata.cachedContentTokenCount) != null ? _e : void 0;
+              usage.inputTokens = (_b = usageMetadata.promptTokenCount) != null ? _b : void 0;
+              usage.outputTokens = (_c = usageMetadata.candidatesTokenCount) != null ? _c : void 0;
+              usage.totalTokens = (_d = usageMetadata.totalTokenCount) != null ? _d : void 0;
+              usage.reasoningTokens = (_e = usageMetadata.thoughtsTokenCount) != null ? _e : void 0;
+              usage.cachedInputTokens = (_f = usageMetadata.cachedContentTokenCount) != null ? _f : void 0;
             }
             if (value.serviceTier != null) {
               serviceTier = value.serviceTier;
             }
-            const candidate = (_f = value.candidates) == null ? void 0 : _f[0];
+            const candidate = (_g = value.candidates) == null ? void 0 : _g[0];
             if ((candidate == null ? void 0 : candidate.continuationToken) != null) {
               lastContinuationToken = candidate.continuationToken;
             }
@@ -1555,12 +1654,12 @@ var GoogleGenerativeAILanguageModel = class {
               requestUsageMetadata = void 0;
             }
             if (candidate == null) {
-              const promptBlockReason2 = (_g = value.promptFeedback) == null ? void 0 : _g.blockReason;
+              const promptBlockReason2 = (_h = value.promptFeedback) == null ? void 0 : _h.blockReason;
               if (promptBlockReason2 != null) {
                 finishReason = "content-filter";
                 providerMetadata = {
                   google: {
-                    promptFeedback: (_h = value.promptFeedback) != null ? _h : null,
+                    promptFeedback: (_i = value.promptFeedback) != null ? _i : null,
                     groundingMetadata: lastGroundingMetadata,
                     urlContextMetadata: lastUrlContextMetadata,
                     safetyRatings: null,
@@ -1593,9 +1692,9 @@ var GoogleGenerativeAILanguageModel = class {
               }
             }
             if (content != null) {
-              const parts = (_i = content.parts) != null ? _i : [];
+              const parts = (_j = content.parts) != null ? _j : [];
               for (const part of parts) {
-                if ("executableCode" in part && ((_j = part.executableCode) == null ? void 0 : _j.code)) {
+                if ("executableCode" in part && ((_k = part.executableCode) == null ? void 0 : _k.code)) {
                   const toolCallId = generateId2();
                   lastCodeExecutionToolCallId = toolCallId;
                   controller.enqueue({
@@ -1722,9 +1821,9 @@ var GoogleGenerativeAILanguageModel = class {
                 }
               }
             }
-            const promptBlockReason = (_k = value.promptFeedback) == null ? void 0 : _k.blockReason;
+            const promptBlockReason = (_l = value.promptFeedback) == null ? void 0 : _l.blockReason;
             const isPromptBlocked = candidate.finishReason == null && promptBlockReason != null;
-            const rawFinishReason = (_m = (_l = candidate.finishReason) != null ? _l : promptBlockReason) != null ? _m : void 0;
+            const rawFinishReason = (_n = (_m = candidate.finishReason) != null ? _m : promptBlockReason) != null ? _n : void 0;
             if (rawFinishReason != null) {
               finishReason = isPromptBlocked ? "content-filter" : mapGoogleGenerativeAIFinishReason({
                 finishReason: rawFinishReason,
@@ -1732,10 +1831,10 @@ var GoogleGenerativeAILanguageModel = class {
               });
               providerMetadata = {
                 google: {
-                  promptFeedback: (_n = value.promptFeedback) != null ? _n : null,
+                  promptFeedback: (_o = value.promptFeedback) != null ? _o : null,
                   groundingMetadata: lastGroundingMetadata,
                   urlContextMetadata: lastUrlContextMetadata,
-                  safetyRatings: (_o = candidate.safetyRatings) != null ? _o : null,
+                  safetyRatings: (_p = candidate.safetyRatings) != null ? _p : null,
                   serviceTier,
                   // Only surface the token when the response stopped early
                   // (budget used up, or continuing failed), not for the
@@ -2042,6 +2141,7 @@ var responseSchema = lazySchema3(
 var chunkSchema = lazySchema3(
   () => zodSchema3(
     z3.object({
+      error: getGoogleErrorSchema().nullish(),
       candidates: z3.array(
         z3.object({
           content: getContentSchema().nullish(),