@ai-sdk/google 3.0.141 → 3.0.143

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,
@@ -1241,6 +1303,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);
@@ -1519,6 +1584,7 @@ var GoogleJSONAccumulator = class {
       const rawPath = jsonPath.replace(/^\$\./, "");
       if (!rawPath || rawPath === "$") continue;
       const segments = parsePath(rawPath);
+      if (!hasContiguousIndexes(this.accumulatedArgs, segments)) continue;
       const existingValue = getNestedValue(this.accumulatedArgs, segments);
       const isStringContinuation = arg.stringValue != null && existingValue !== void 0;
       if (isStringContinuation) {
@@ -1739,6 +1805,17 @@ function getNestedValue(obj, segments) {
   }
   return current;
 }
+function hasContiguousIndexes(obj, segments) {
+  let current = obj;
+  for (const pathSegment of segments) {
+    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++) {
@@ -4715,6 +4792,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(),
@@ -5665,7 +5744,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",
@@ -5900,24 +5979,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) {