@ai-sdk/provider-utils 2.2.8 → 3.0.0-alpha.1

package/dist/index.mjs

@@ -146,14 +146,20 @@ function extractResponseHeaders(response) {
 
 // src/generate-id.ts
 import { InvalidArgumentError } from "@ai-sdk/provider";
-import { customAlphabet } from "nanoid/non-secure";
 var createIdGenerator = ({
   prefix,
-  size: defaultSize = 16,
+  size = 16,
   alphabet = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
   separator = "-"
 } = {}) => {
-  const generator = customAlphabet(alphabet, defaultSize);
+  const generator = () => {
+    const alphabetLength = alphabet.length;
+    const chars = new Array(size);
+    for (let i = 0; i < size; i++) {
+      chars[i] = alphabet[Math.random() * alphabetLength | 0];
+    }
+    return chars.join("");
+  };
   if (prefix == null) {
     return generator;
   }
@@ -163,7 +169,7 @@ var createIdGenerator = ({
       message: `The separator "${separator}" must not be part of the alphabet "${alphabet}".`
     });
   }
-  return (size) => `${prefix}${separator}${generator(size)}`;
+  return () => `${prefix}${separator}${generator()}`;
 };
 var generateId = createIdGenerator();
 
@@ -277,6 +283,20 @@ var getFromApi = async ({
   }
 };
 
+// src/is-url-supported.ts
+function isUrlSupported({
+  mediaType,
+  url,
+  supportedUrls
+}) {
+  url = url.toLowerCase();
+  mediaType = mediaType.toLowerCase();
+  return Object.entries(supportedUrls).map(([key, value]) => {
+    const mediaType2 = key.toLowerCase();
+    return mediaType2 === "*" || mediaType2 === "*/*" ? { mediaTypePrefix: "", regexes: value } : { mediaTypePrefix: mediaType2.replace(/\*/, ""), regexes: value };
+  }).filter(({ mediaTypePrefix }) => mediaType.startsWith(mediaTypePrefix)).flatMap(({ regexes }) => regexes).some((pattern) => pattern.test(url));
+}
+
 // src/load-api-key.ts
 import { LoadAPIKeyError } from "@ai-sdk/provider";
 function loadApiKey({
@@ -368,14 +388,59 @@ function loadSetting({
 // src/parse-json.ts
 import {
   JSONParseError,
-  TypeValidationError as TypeValidationError2
+  TypeValidationError as TypeValidationError3
 } from "@ai-sdk/provider";
-import SecureJSON from "secure-json-parse";
+
+// src/secure-json-parse.ts
+var suspectProtoRx = /"__proto__"\s*:/;
+var suspectConstructorRx = /"constructor"\s*:/;
+function _parse(text) {
+  const obj = JSON.parse(text);
+  if (obj === null || typeof obj !== "object") {
+    return obj;
+  }
+  if (suspectProtoRx.test(text) === false && suspectConstructorRx.test(text) === false) {
+    return obj;
+  }
+  return filter(obj);
+}
+function filter(obj) {
+  let next = [obj];
+  while (next.length) {
+    const nodes = next;
+    next = [];
+    for (const node of nodes) {
+      if (Object.prototype.hasOwnProperty.call(node, "__proto__")) {
+        throw new SyntaxError("Object contains forbidden prototype property");
+      }
+      if (Object.prototype.hasOwnProperty.call(node, "constructor") && Object.prototype.hasOwnProperty.call(node.constructor, "prototype")) {
+        throw new SyntaxError("Object contains forbidden prototype property");
+      }
+      for (const key in node) {
+        const value = node[key];
+        if (value && typeof value === "object") {
+          next.push(value);
+        }
+      }
+    }
+  }
+  return obj;
+}
+function secureJsonParse(text) {
+  const { stackTraceLimit } = Error;
+  Error.stackTraceLimit = 0;
+  try {
+    return _parse(text);
+  } finally {
+    Error.stackTraceLimit = stackTraceLimit;
+  }
+}
 
 // src/validate-types.ts
-import { TypeValidationError } from "@ai-sdk/provider";
+import { TypeValidationError as TypeValidationError2 } from "@ai-sdk/provider";
 
 // src/validator.ts
+import { TypeValidationError } from "@ai-sdk/provider";
 var validatorSymbol = Symbol.for("vercel.ai.validator");
 function validator(validate) {
   return { [validatorSymbol]: true, validate };
@@ -384,99 +449,124 @@ function isValidator(value) {
   return typeof value === "object" && value !== null && validatorSymbol in value && value[validatorSymbol] === true && "validate" in value;
 }
 function asValidator(value) {
-  return isValidator(value) ? value : zodValidator(value);
+  return isValidator(value) ? value : standardSchemaValidator(value);
 }
-function zodValidator(zodSchema) {
-  return validator((value) => {
-    const result = zodSchema.safeParse(value);
-    return result.success ? { success: true, value: result.data } : { success: false, error: result.error };
+function standardSchemaValidator(standardSchema) {
+  return validator(async (value) => {
+    const result = await standardSchema["~standard"].validate(value);
+    return result.issues == null ? { success: true, value: result.value } : {
+      success: false,
+      error: new TypeValidationError({
+        value,
+        cause: result.issues
+      })
+    };
   });
 }
 
 // src/validate-types.ts
-function validateTypes({
+async function validateTypes({
   value,
-  schema: inputSchema
+  schema
 }) {
-  const result = safeValidateTypes({ value, schema: inputSchema });
+  const result = await safeValidateTypes({ value, schema });
   if (!result.success) {
-    throw TypeValidationError.wrap({ value, cause: result.error });
+    throw TypeValidationError2.wrap({ value, cause: result.error });
   }
   return result.value;
 }
-function safeValidateTypes({
+async function safeValidateTypes({
   value,
   schema
 }) {
   const validator2 = asValidator(schema);
   try {
     if (validator2.validate == null) {
-      return { success: true, value };
+      return { success: true, value, rawValue: value };
     }
-    const result = validator2.validate(value);
+    const result = await validator2.validate(value);
     if (result.success) {
-      return result;
+      return { success: true, value: result.value, rawValue: value };
     }
     return {
       success: false,
-      error: TypeValidationError.wrap({ value, cause: result.error })
+      error: TypeValidationError2.wrap({ value, cause: result.error }),
+      rawValue: value
     };
   } catch (error) {
     return {
       success: false,
-      error: TypeValidationError.wrap({ value, cause: error })
+      error: TypeValidationError2.wrap({ value, cause: error }),
+      rawValue: value
     };
   }
 }
 
 // src/parse-json.ts
-function parseJSON({
+async function parseJSON({
   text,
   schema
 }) {
   try {
-    const value = SecureJSON.parse(text);
+    const value = secureJsonParse(text);
     if (schema == null) {
       return value;
     }
     return validateTypes({ value, schema });
   } catch (error) {
-    if (JSONParseError.isInstance(error) || TypeValidationError2.isInstance(error)) {
+    if (JSONParseError.isInstance(error) || TypeValidationError3.isInstance(error)) {
       throw error;
     }
     throw new JSONParseError({ text, cause: error });
   }
 }
-function safeParseJSON({
+async function safeParseJSON({
   text,
   schema
 }) {
   try {
-    const value = SecureJSON.parse(text);
+    const value = secureJsonParse(text);
     if (schema == null) {
       return { success: true, value, rawValue: value };
     }
-    const validationResult = safeValidateTypes({ value, schema });
-    return validationResult.success ? { ...validationResult, rawValue: value } : validationResult;
+    return await safeValidateTypes({ value, schema });
   } catch (error) {
     return {
       success: false,
-      error: JSONParseError.isInstance(error) ? error : new JSONParseError({ text, cause: error })
+      error: JSONParseError.isInstance(error) ? error : new JSONParseError({ text, cause: error }),
+      rawValue: void 0
     };
   }
 }
 function isParsableJson(input) {
   try {
-    SecureJSON.parse(input);
+    secureJsonParse(input);
     return true;
   } catch (e) {
     return false;
   }
 }
 
+// src/parse-json-event-stream.ts
+function parseJsonEventStream({
+  stream,
+  schema
+}) {
+  return stream.pipeThrough(new TextDecoderStream()).pipeThrough(createEventSourceParserStream()).pipeThrough(
+    new TransformStream({
+      async transform({ data }, controller) {
+        if (data === "[DONE]") {
+          return;
+        }
+        controller.enqueue(await safeParseJSON({ text: data, schema }));
+      }
+    })
+  );
+}
+
 // src/parse-provider-options.ts
 import { InvalidArgumentError as InvalidArgumentError2 } from "@ai-sdk/provider";
-function parseProviderOptions({
+async function parseProviderOptions({
   provider,
   providerOptions,
   schema
@@ -484,7 +574,7 @@ function parseProviderOptions({
   if ((providerOptions == null ? void 0 : providerOptions[provider]) == null) {
     return void 0;
   }
-  const parsedProviderOptions = safeValidateTypes({
+  const parsedProviderOptions = await safeValidateTypes({
     value: providerOptions[provider],
     schema
   });
@@ -658,7 +748,7 @@ var createJsonErrorResponseHandler = ({
     };
   }
   try {
-    const parsedError = parseJSON({
+    const parsedError = await parseJSON({
       text: responseBody,
       schema: errorSchema
     });
@@ -697,21 +787,10 @@ var createEventSourceResponseHandler = (chunkSchema) => async ({ response }) =>
   }
   return {
     responseHeaders,
-    value: response.body.pipeThrough(new TextDecoderStream()).pipeThrough(createEventSourceParserStream()).pipeThrough(
-      new TransformStream({
-        transform({ data }, controller) {
-          if (data === "[DONE]") {
-            return;
-          }
-          controller.enqueue(
-            safeParseJSON({
-              text: data,
-              schema: chunkSchema
-            })
-          );
-        }
-      })
-    )
+    value: parseJsonEventStream({
+      stream: response.body,
+      schema: chunkSchema
+    })
   };
 };
 var createJsonStreamResponseHandler = (chunkSchema) => async ({ response }) => {
@@ -724,10 +803,10 @@ var createJsonStreamResponseHandler = (chunkSchema) => async ({ response }) => {
     responseHeaders,
     value: response.body.pipeThrough(new TextDecoderStream()).pipeThrough(
       new TransformStream({
-        transform(chunkText, controller) {
+        async transform(chunkText, controller) {
           if (chunkText.endsWith("\n")) {
             controller.enqueue(
-              safeParseJSON({
+              await safeParseJSON({
                 text: buffer + chunkText,
                 schema: chunkSchema
               })
@@ -743,7 +822,7 @@ var createJsonStreamResponseHandler = (chunkSchema) => async ({ response }) => {
 };
 var createJsonResponseHandler = (responseSchema) => async ({ response, url, requestBodyValues }) => {
   const responseBody = await response.text();
-  const parsedResult = safeParseJSON({
+  const parsedResult = await safeParseJSON({
     text: responseBody,
     schema: responseSchema
   });
@@ -811,6 +890,50 @@ var createStatusCodeErrorResponseHandler = () => async ({ response, url, request
   };
 };
 
+// src/zod-schema.ts
+import zodToJsonSchema from "zod-to-json-schema";
+function zodSchema(zodSchema2, options) {
+  var _a;
+  const useReferences = (_a = options == null ? void 0 : options.useReferences) != null ? _a : false;
+  return jsonSchema(
+    zodToJsonSchema(zodSchema2, {
+      $refStrategy: useReferences ? "root" : "none",
+      target: "jsonSchema7"
+      // note: openai mode breaks various gemini conversions
+    }),
+    {
+      validate: (value) => {
+        const result = zodSchema2.safeParse(value);
+        return result.success ? { success: true, value: result.data } : { success: false, error: result.error };
+      }
+    }
+  );
+}
+
+// src/schema.ts
+var schemaSymbol = Symbol.for("vercel.ai.schema");
+function jsonSchema(jsonSchema2, {
+  validate
+} = {}) {
+  return {
+    [schemaSymbol]: true,
+    _type: void 0,
+    // should never be used directly
+    [validatorSymbol]: true,
+    jsonSchema: jsonSchema2,
+    validate
+  };
+}
+function isSchema(value) {
+  return typeof value === "object" && value !== null && schemaSymbol in value && value[schemaSymbol] === true && "jsonSchema" in value && "validate" in value;
+}
+function asSchema(schema) {
+  return schema == null ? jsonSchema({
+    properties: {},
+    additionalProperties: false
+  }) : isSchema(schema) ? schema : zodSchema(schema);
+}
+
 // src/uint8-utils.ts
 var { btoa, atob } = globalThis;
 function convertBase64ToUint8Array(base64String) {
@@ -825,16 +948,24 @@ function convertUint8ArrayToBase64(array) {
   }
   return btoa(latin1string);
 }
+function convertToBase64(value) {
+  return value instanceof Uint8Array ? convertUint8ArrayToBase64(value) : value;
+}
 
 // src/without-trailing-slash.ts
 function withoutTrailingSlash(url) {
   return url == null ? void 0 : url.replace(/\/$/, "");
 }
+
+// src/index.ts
+export * from "@standard-schema/spec";
 export {
+  asSchema,
   asValidator,
   combineHeaders,
   convertAsyncIteratorToReadableStream,
   convertBase64ToUint8Array,
+  convertToBase64,
   convertUint8ArrayToBase64,
   createBinaryResponseHandler,
   createEventSourceParserStream,
@@ -851,11 +982,14 @@ export {
   getFromApi,
   isAbortError,
   isParsableJson,
+  isUrlSupported,
   isValidator,
+  jsonSchema,
   loadApiKey,
   loadOptionalSetting,
   loadSetting,
   parseJSON,
+  parseJsonEventStream,
   parseProviderOptions,
   postFormDataToApi,
   postJsonToApi,
@@ -864,10 +998,11 @@ export {
   resolve,
   safeParseJSON,
   safeValidateTypes,
+  standardSchemaValidator,
   validateTypes,
   validator,
   validatorSymbol,
   withoutTrailingSlash,
-  zodValidator
+  zodSchema
 };
 //# sourceMappingURL=index.mjs.map