@botpress/zai 2.5.4 → 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.4",
+  "version": "2.5.6",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
   "exports": {
@@ -32,7 +32,7 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
-    "@botpress/cognitive": "0.3.2",
+    "@botpress/cognitive": "0.3.3",
     "json5": "^2.2.3",
     "jsonrepair": "^3.10.0",
     "lodash-es": "^4.17.21",

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()