@ai-sdk/google 4.0.47 → 4.0.48

package/dist/internal/index.js

@@ -50,21 +50,37 @@ import {
   UnsupportedFunctionalityError
 } from "@ai-sdk/provider";
 function convertJSONSchemaToOpenAPISchema(jsonSchema, isRoot = true) {
+  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()
+  });
+}
+function convertJSONSchemaDefinition(jsonSchema, isRoot, referenceContext) {
   if (jsonSchema == null) {
     return void 0;
   }
+  if (typeof jsonSchema === "boolean") {
+    return { type: "boolean", properties: {} };
+  }
+  if (jsonSchema.$ref != null) {
+    return convertJSONSchemaReference({
+      jsonSchema,
+      reference: jsonSchema.$ref,
+      isRoot,
+      referenceContext
+    });
+  }
   if (isEmptyObjectSchema(jsonSchema)) {
     if (isRoot) {
       return void 0;
     }
-    if (typeof jsonSchema === "object" && jsonSchema.description) {
+    if (jsonSchema.description) {
       return { type: "object", description: jsonSchema.description };
     }
     return { type: "object" };
   }
-  if (typeof jsonSchema === "boolean") {
-    return { type: "boolean", properties: {} };
-  }
   const {
     type,
     description,
@@ -106,18 +122,20 @@ function convertJSONSchemaToOpenAPISchema(jsonSchema, isRoot = true) {
   if (properties != null) {
     result.properties = Object.entries(properties).reduce(
       (acc, [key, value]) => {
-        acc[key] = convertJSONSchemaToOpenAPISchema(value, false);
+        acc[key] = convertJSONSchemaDefinition(value, false, referenceContext);
         return acc;
       },
       {}
     );
   }
   if (items) {
-    result.items = Array.isArray(items) ? items.map((item) => convertJSONSchemaToOpenAPISchema(item, false)) : convertJSONSchemaToOpenAPISchema(items, false);
+    result.items = Array.isArray(items) ? items.map(
+      (item) => convertJSONSchemaDefinition(item, false, referenceContext)
+    ) : convertJSONSchemaDefinition(items, false, referenceContext);
   }
   if (allOf) {
     result.allOf = allOf.map(
-      (item) => convertJSONSchemaToOpenAPISchema(item, false)
+      (item) => convertJSONSchemaDefinition(item, false, referenceContext)
     );
   }
   if (anyOf) {
@@ -128,9 +146,10 @@ function convertJSONSchemaToOpenAPISchema(jsonSchema, isRoot = true) {
         (schema) => !(typeof schema === "object" && (schema == null ? void 0 : schema.type) === "null")
       );
       if (nonNullSchemas.length === 1) {
-        const converted = convertJSONSchemaToOpenAPISchema(
+        const converted = convertJSONSchemaDefinition(
           nonNullSchemas[0],
-          false
+          false,
+          referenceContext
         );
         if (typeof converted === "object") {
           result.nullable = true;
@@ -138,19 +157,19 @@ function convertJSONSchemaToOpenAPISchema(jsonSchema, isRoot = true) {
         }
       } else {
         result.anyOf = nonNullSchemas.map(
-          (item) => convertJSONSchemaToOpenAPISchema(item, false)
+          (item) => convertJSONSchemaDefinition(item, false, referenceContext)
         );
         result.nullable = true;
       }
     } else {
       result.anyOf = anyOf.map(
-        (item) => convertJSONSchemaToOpenAPISchema(item, false)
+        (item) => convertJSONSchemaDefinition(item, false, referenceContext)
       );
     }
   }
   if (oneOf) {
     result.oneOf = oneOf.map(
-      (item) => convertJSONSchemaToOpenAPISchema(item, false)
+      (item) => convertJSONSchemaDefinition(item, false, referenceContext)
     );
   }
   if (minLength !== void 0) {
@@ -158,6 +177,76 @@ function convertJSONSchemaToOpenAPISchema(jsonSchema, isRoot = true) {
   }
   return result;
 }
+function convertJSONSchemaReference({
+  jsonSchema,
+  reference,
+  isRoot,
+  referenceContext
+}) {
+  const { definition, referenceKey } = getReferencedDefinition(
+    reference,
+    referenceContext
+  );
+  if (referenceContext.resolvingReferences.has(referenceKey)) {
+    throw new UnsupportedFunctionalityError({
+      functionality: `recursive JSON Schema reference: ${reference}`,
+      message: "Google schema conversion does not support recursive JSON Schema references."
+    });
+  }
+  const resolvingReferences = new Set(referenceContext.resolvingReferences);
+  resolvingReferences.add(referenceKey);
+  const { $ref: _reference, ...siblingSchema } = jsonSchema;
+  const resolvedSchema = typeof definition === "boolean" ? definition ? siblingSchema : false : { ...definition, ...siblingSchema };
+  return convertJSONSchemaDefinition(resolvedSchema, isRoot, {
+    ...referenceContext,
+    resolvingReferences
+  });
+}
+function getReferencedDefinition(reference, referenceContext) {
+  const definitionSources = [
+    {
+      prefix: "#/$defs/",
+      definitions: referenceContext.dollarDefinitions
+    },
+    {
+      prefix: "#/definitions/",
+      definitions: referenceContext.definitions
+    }
+  ];
+  const source = definitionSources.find(
+    ({ prefix }) => reference.startsWith(prefix)
+  );
+  const encodedDefinitionName = source ? reference.slice(source.prefix.length) : void 0;
+  if (source == null || encodedDefinitionName == null || encodedDefinitionName.length === 0 || encodedDefinitionName.includes("/")) {
+    throwUnsupportedReference(reference);
+  }
+  let decodedDefinitionName;
+  try {
+    decodedDefinitionName = decodeURIComponent(encodedDefinitionName);
+  } catch (e) {
+    throwUnsupportedReference(reference);
+  }
+  if (decodedDefinitionName.includes("/") || /~(?![01])/u.test(decodedDefinitionName) || source.definitions == null) {
+    throwUnsupportedReference(reference);
+  }
+  const definitionName = decodedDefinitionName.replace(
+    /~[01]/g,
+    (match) => match === "~1" ? "/" : "~"
+  );
+  if (!Object.prototype.hasOwnProperty.call(source.definitions, definitionName)) {
+    throwUnsupportedReference(reference);
+  }
+  return {
+    definition: source.definitions[definitionName],
+    referenceKey: `${source.prefix}${definitionName}`
+  };
+}
+function throwUnsupportedReference(reference) {
+  throw new UnsupportedFunctionalityError({
+    functionality: `JSON Schema reference: ${reference}`,
+    message: "Google schema conversion only supports references to direct children of root-level $defs or definitions."
+  });
+}
 function addEnumToSchema({
   values,
   type,