fix(zai): try catch extract schema transform (botpress#14714)

Author: franklevasseur

packages/zai/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": {

packages/zai/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()