@ai-sdk/google 3.0.142 → 3.0.144

package/dist/internal/index.js

@@ -178,25 +178,54 @@ function addTokenDetails(a, b) {
 
 // src/convert-json-schema-to-openapi-schema.ts
 var import_provider = require("@ai-sdk/provider");
+var maxSchemaDepth = 100;
+var maxSchemaNodes = 1e4;
+var maxSchemaSize = 1e6;
 var recursiveReferenceFunctionalityPrefix = "recursive JSON Schema reference:";
 function isRecursiveJSONSchemaReferenceError(error) {
   return import_provider.UnsupportedFunctionalityError.isInstance(error) && error.functionality.startsWith(recursiveReferenceFunctionalityPrefix);
 }
 function convertJSONSchemaToOpenAPISchema(jsonSchema, isRoot = true) {
+  consumeSchemaBudget(jsonSchema, {
+    nodes: maxSchemaNodes,
+    size: maxSchemaSize
+  });
   const rootSchema = typeof jsonSchema === "object" ? jsonSchema : void 0;
   return convertJSONSchemaDefinition(jsonSchema, isRoot, {
     definitions: rootSchema == null ? void 0 : rootSchema.definitions,
     dollarDefinitions: rootSchema == null ? void 0 : rootSchema.$defs,
-    resolvingReferences: /* @__PURE__ */ new Set()
+    resolvingReferences: /* @__PURE__ */ new Set(),
+    budget: { nodes: maxSchemaNodes, size: maxSchemaSize },
+    depth: 0
   });
 }
 function convertJSONSchemaDefinition(jsonSchema, isRoot, referenceContext) {
   if (jsonSchema == null) {
     return void 0;
   }
+  referenceContext.budget.nodes--;
+  referenceContext.budget.size -= 128;
+  checkSchemaBudget(referenceContext.budget, referenceContext.depth);
+  referenceContext = {
+    ...referenceContext,
+    depth: referenceContext.depth + 1
+  };
   if (typeof jsonSchema === "boolean") {
     return { type: "boolean", properties: {} };
   }
+  consumeSchemaBudget(
+    [
+      jsonSchema.$ref,
+      jsonSchema.type,
+      jsonSchema.description,
+      jsonSchema.required,
+      jsonSchema.format,
+      jsonSchema.const,
+      jsonSchema.enum,
+      jsonSchema.minLength
+    ],
+    referenceContext.budget
+  );
   if (jsonSchema.$ref != null) {
     return convertJSONSchemaReference({
       jsonSchema,
@@ -260,6 +289,7 @@ function convertJSONSchemaDefinition(jsonSchema, isRoot, referenceContext) {
   if (properties != null) {
     result.properties = Object.entries(properties).reduce(
       (acc, [key, value]) => {
+        consumeSchemaBudget(key, referenceContext.budget);
         acc[key] = convertJSONSchemaDefinition(value, false, referenceContext);
         return acc;
       },
@@ -315,6 +345,38 @@ function convertJSONSchemaDefinition(jsonSchema, isRoot, referenceContext) {
   }
   return result;
 }
+function checkSchemaBudget(budget, depth) {
+  if (budget.nodes < 0 || budget.size < 0 || depth > maxSchemaDepth) {
+    throw new import_provider.UnsupportedFunctionalityError({
+      functionality: "JSON Schema conversion exceeding resource limits",
+      message: "Google schema conversion exceeded its node, size, or depth limit."
+    });
+  }
+}
+function consumeSchemaBudget(value, budget, depth = 0) {
+  if (value === void 0) {
+    return;
+  }
+  budget.nodes--;
+  budget.size -= typeof value === "string" ? value.length * 6 + 2 : 32;
+  checkSchemaBudget(budget, depth);
+  if (Array.isArray(value)) {
+    for (const item of value) {
+      consumeSchemaBudget(item, budget, depth + 1);
+    }
+  } else if (value !== null && typeof value === "object") {
+    for (const key in value) {
+      if (Object.prototype.hasOwnProperty.call(value, key)) {
+        consumeSchemaBudget(key, budget, depth + 1);
+        consumeSchemaBudget(
+          value[key],
+          budget,
+          depth + 1
+        );
+      }
+    }
+  }
+}
 function addEnumToSchema({
   values,
   type,
@@ -1522,6 +1584,7 @@ var GoogleJSONAccumulator = class {
       const rawPath = jsonPath.replace(/^\$\./, "");
       if (!rawPath || rawPath === "$") continue;
       const segments = parsePath(rawPath);
+      if (!normalizeAndCheckIndexes(this.accumulatedArgs, segments)) continue;
       const existingValue = getNestedValue(this.accumulatedArgs, segments);
       const isStringContinuation = arg.stringValue != null && existingValue !== void 0;
       if (isStringContinuation) {
@@ -1742,6 +1805,25 @@ function getNestedValue(obj, segments) {
   }
   return current;
 }
+function normalizeAndCheckIndexes(obj, segments) {
+  let current = obj;
+  for (let i = 0; i < segments.length; i++) {
+    let pathSegment = segments[i];
+    if (Array.isArray(current) && typeof pathSegment === "string") {
+      const index = Number(pathSegment);
+      if (Number.isInteger(index) && index >= 0 && index < 2 ** 32 - 1 && String(index) === pathSegment) {
+        pathSegment = index;
+        segments[i] = index;
+      }
+    }
+    if (typeof pathSegment === "number") {
+      const length = Array.isArray(current) ? current.length : current == null ? 0 : void 0;
+      if (length !== void 0 && pathSegment > length) return false;
+    }
+    current = current != null && typeof current === "object" && hasOwnProperty(current, pathSegment) ? current[pathSegment] : void 0;
+  }
+  return true;
+}
 function setNestedValue(obj, segments, value) {
   let current = obj;
   for (let i = 0; i < segments.length - 1; i++) {
@@ -2016,7 +2098,7 @@ var GoogleGenerativeAILanguageModel = class {
     };
   }
   async doGenerate(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, _y, _z, _A, _B;
+    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, _A, _B, _C;
     const { args, warnings, providerOptionsName, extraHeaders } = await this.getArgs(options);
     const mergedHeaders = (0, import_provider_utils5.combineHeaders)(
       await (0, import_provider_utils5.resolve)(this.config.headers),
@@ -2218,7 +2300,7 @@ var GoogleGenerativeAILanguageModel = class {
           safetyRatings: (_y = candidate == null ? void 0 : candidate.safetyRatings) != null ? _y : null,
           usageMetadata: usageMetadata != null ? usageMetadata : null,
           finishMessage: (_z = candidate == null ? void 0 : candidate.finishMessage) != null ? _z : null,
-          serviceTier: (_A = usageMetadata == null ? void 0 : usageMetadata.serviceTier) != null ? _A : null,
+          serviceTier: (_B = (_A = usageMetadata == null ? void 0 : usageMetadata.serviceTier) != null ? _A : responseHeaders == null ? void 0 : responseHeaders["x-gemini-service-tier"]) != null ? _B : null,
           ...(candidate == null ? void 0 : candidate.continuationToken) != null && rawFinishReason === "MAX_TOKENS" ? { continuationToken: candidate.continuationToken } : {},
           ...continuedUsageMetadata != null && firstPromptTokenCount != null ? { initialPromptTokenCount: firstPromptTokenCount } : {}
         }
@@ -2226,7 +2308,7 @@ var GoogleGenerativeAILanguageModel = class {
       request: { body: args },
       response: {
         // TODO timestamp, model id
-        id: (_B = response.responseId) != null ? _B : void 0,
+        id: (_C = response.responseId) != null ? _C : void 0,
         headers: responseHeaders,
         body: rawResponse
       }
@@ -2702,7 +2784,7 @@ var GoogleGenerativeAILanguageModel = class {
             }
           },
           flush(controller) {
-            var _a;
+            var _a, _b;
             usage = addGoogleGenerativeAIUsageMetadata(continuedUsage, usage);
             if (currentTextBlockId !== null) {
               controller.enqueue({
@@ -2734,7 +2816,7 @@ var GoogleGenerativeAILanguageModel = class {
                   safetyRatings: lastSafetyRatings,
                   usageMetadata: usage != null ? usage : null,
                   finishMessage: lastFinishMessage,
-                  serviceTier: (_a = usage == null ? void 0 : usage.serviceTier) != null ? _a : null,
+                  serviceTier: (_b = (_a = usage == null ? void 0 : usage.serviceTier) != null ? _a : responseHeaders == null ? void 0 : responseHeaders["x-gemini-service-tier"]) != null ? _b : null,
                   // Only surface the token when the response stopped early
                   // (budget used up, or continuing failed), not for the
                   // checkpoint tokens of a stream that finished normally.
@@ -4718,6 +4800,8 @@ var import_v413 = require("zod/v4");
 var googleInteractionsLanguageModelOptions = (0, import_provider_utils15.lazySchema)(
   () => (0, import_provider_utils15.zodSchema)(
     import_v413.z.object({
+      /** User-defined request metadata, including `safety_identifier`. */
+      labels: import_v413.z.record(import_v413.z.string(), import_v413.z.string()).nullish(),
       previousInteractionId: import_v413.z.string().nullish(),
       store: import_v413.z.boolean().nullish(),
       agent: import_v413.z.string().nullish(),
@@ -5668,7 +5752,7 @@ var GoogleInteractionsLanguageModel = class {
     };
   }
   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, _y, _z, _A, _B, _C, _D, _E, _F;
+    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, _A, _B, _C, _D, _E, _F, _G;
     const warnings = [];
     const opts = await (0, import_provider_utils19.parseProviderOptions)({
       provider: "google",
@@ -5903,24 +5987,25 @@ var GoogleInteractionsLanguageModel = class {
     const args = pruneUndefined({
       ...isAgent ? { agent: this.agent } : { model: this.modelId },
       input,
+      labels: (_B = opts == null ? void 0 : opts.labels) != null ? _B : void 0,
       system_instruction: systemInstruction,
       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: (_B = opts == null ? void 0 : opts.previousInteractionId) != null ? _B : void 0,
-      service_tier: (_C = opts == null ? void 0 : opts.serviceTier) != null ? _C : void 0,
-      store: (_D = opts == null ? void 0 : opts.store) != null ? _D : void 0,
+      previous_interaction_id: (_C = opts == null ? void 0 : opts.previousInteractionId) != null ? _C : void 0,
+      service_tier: (_D = opts == null ? void 0 : opts.serviceTier) != null ? _D : void 0,
+      store: (_E = opts == null ? void 0 : opts.store) != null ? _E : void 0,
       generation_config: generationConfig != null && Object.keys(generationConfig).length > 0 ? generationConfig : void 0,
       agent_config: agentConfig,
       environment,
-      background: (_E = opts == null ? void 0 : opts.background) != null ? _E : void 0
+      background: (_F = opts == null ? void 0 : opts.background) != null ? _F : void 0
     });
     return {
       args,
       warnings,
       isAgent,
       isBackground: (opts == null ? void 0 : opts.background) === true,
-      pollingTimeoutMs: (_F = opts == null ? void 0 : opts.pollingTimeoutMs) != null ? _F : void 0
+      pollingTimeoutMs: (_G = opts == null ? void 0 : opts.pollingTimeoutMs) != null ? _G : void 0
     };
   }
   async doGenerate(options) {