@botpress/zai 2.5.5 → 2.5.6

package/dist/operations/extract.js

@@ -22,7 +22,12 @@ const NO_MORE = "\u25A0NO_MORE_ELEMENT\u25A0";
 const ZERO_ELEMENTS = "\u25A0ZERO_ELEMENTS\u25A0";
 const extract = async (input, _schema, _options, ctx) => {
   ctx.controller.signal.throwIfAborted();
-  let schema = transforms.fromJSONSchema(transforms.toJSONSchema(_schema));
+  let schema;
+  try {
+    schema = transforms.fromJSONSchema(transforms.toJSONSchema(_schema));
+  } catch {
+    schema = _schema;
+  }
   const options = Options.parse(_options ?? {});
   const tokenizer = await getTokenizer();
   const model = await ctx.getModel();

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@botpress/zai",
   "description": "Zui AI (zai) – An LLM utility library written on top of Zui and the Botpress API",
-  "version": "2.5.5",
+  "version": "2.5.6",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
   "exports": {

package/src/operations/extract.ts

@@ -125,7 +125,13 @@ const extract = async <S extends OfType<AnyObjectOrArray>>(
 ): Promise<S['_output']> => {
   ctx.controller.signal.throwIfAborted()
 
-  let schema = transforms.fromJSONSchema(transforms.toJSONSchema(_schema as any as z.ZodType))
+  let schema: z.ZodType
+  try {
+    schema = transforms.fromJSONSchema(transforms.toJSONSchema(_schema as any as z.ZodType))
+  } catch {
+    // The above transformers arent the legacy once. They are very strict and might fail on some schema types.
+    schema = _schema as any as z.ZodType
+  }
 
   const options = Options.parse(_options ?? {})
   const tokenizer = await getTokenizer()