@botpress/zai 1.0.1-beta.1 → 1.0.1-beta.3

package/dist/{csj → node}/adapters/botpress-table.js

@@ -1,5 +1,4 @@
-import sdk from "@botpress/sdk";
-const { z } = sdk;
+import { z } from "@bpinternal/zui";
 import { BotpressClient, GenerationMetadata } from "../utils";
 import { Adapter } from "./adapter";
 const CRITICAL_TAGS = {

package/dist/{csj → node}/operations/check.js

@@ -1,5 +1,4 @@
-import sdk from "@botpress/sdk";
-const { z } = sdk;
+import { z } from "@bpinternal/zui";
 import { fastHash, stringify, takeUntilTokens } from "../utils";
 import { Zai } from "../zai";
 import { PROMPT_INPUT_BUFFER } from "./constants";

package/dist/{csj → node}/operations/extract.js

@@ -1,8 +1,7 @@
-import sdk from "@botpress/sdk";
-const { z } = sdk;
+import { z } from "@bpinternal/zui";
 import JSON5 from "json5";
 import { jsonrepair } from "jsonrepair";
-import _ from "lodash";
+import { chunk, isArray } from "lodash-es";
 import { fastHash, stringify, takeUntilTokens } from "../utils";
 import { Zai } from "../zai";
 import { PROMPT_INPUT_BUFFER } from "./constants";
@@ -23,9 +22,9 @@ Zai.prototype.extract = async function(input, schema, _options) {
   const PROMPT_COMPONENT = Math.max(this.Model.input.maxTokens - PROMPT_INPUT_BUFFER, 100);
   let isArrayOfObjects = false;
   const originalSchema = schema;
-  if (schema instanceof sdk.ZodObject) {
-  } else if (schema instanceof sdk.ZodArray) {
-    if (schema._def.type instanceof sdk.ZodObject) {
+  if (schema instanceof z.ZodObject) {
+  } else if (schema instanceof z.ZodArray) {
+    if (schema._def.type instanceof z.ZodObject) {
       isArrayOfObjects = true;
       schema = schema._def.type;
     } else {
@@ -42,8 +41,8 @@ Zai.prototype.extract = async function(input, schema, _options) {
   if (tokenizer.count(inputAsString) > options.chunkLength) {
     if (isArrayOfObjects) {
       const tokens = tokenizer.split(inputAsString);
-      const chunks = _.chunk(tokens, options.chunkLength).map((x) => x.join(""));
-      const all = await Promise.all(chunks.map((chunk) => this.extract(chunk, originalSchema)));
+      const chunks = chunk(tokens, options.chunkLength).map((x) => x.join(""));
+      const all = await Promise.all(chunks.map((chunk2) => this.extract(chunk2, originalSchema)));
       return all.flat();
     } else {
       inputAsString = tokenizer.truncate(stringify(input), options.chunkLength);
@@ -137,7 +136,7 @@ ${input2.trim()}
   `.trim();
   };
   const formatOutput = (extracted) => {
-    extracted = _.isArray(extracted) ? extracted : [extracted];
+    extracted = isArray(extracted) ? extracted : [extracted];
     return extracted.map(
       (x) => `
 ${START}

package/dist/{csj → node}/operations/filter.js

@@ -1,6 +1,5 @@
-import sdk from "@botpress/sdk";
-const { z } = sdk;
-import _ from "lodash";
+import { z } from "@bpinternal/zui";
+import { clamp } from "lodash-es";
 import { fastHash, stringify, takeUntilTokens } from "../utils";
 import { Zai } from "../zai";
 import { PROMPT_INPUT_BUFFER, PROMPT_OUTPUT_BUFFER } from "./constants";
@@ -22,7 +21,7 @@ Zai.prototype.filter = async function(input, condition, _options) {
   const MAX_ITEMS_PER_CHUNK = 50;
   const TOKENS_TOTAL_MAX = this.Model.input.maxTokens - PROMPT_INPUT_BUFFER - PROMPT_OUTPUT_BUFFER;
   const TOKENS_EXAMPLES_MAX = Math.floor(Math.max(250, TOKENS_TOTAL_MAX * 0.5));
-  const TOKENS_CONDITION_MAX = _.clamp(TOKENS_TOTAL_MAX * 0.25, 250, tokenizer.count(condition));
+  const TOKENS_CONDITION_MAX = clamp(TOKENS_TOTAL_MAX * 0.25, 250, tokenizer.count(condition));
   const TOKENS_INPUT_ARRAY_MAX = TOKENS_TOTAL_MAX - TOKENS_EXAMPLES_MAX - TOKENS_CONDITION_MAX;
   condition = tokenizer.truncate(condition, TOKENS_CONDITION_MAX);
   let chunks = [];
@@ -152,7 +151,7 @@ The condition is: "${condition}"
       const [idx, filter] = x.split(":");
       return { idx: parseInt((idx == null ? void 0 : idx.trim()) ?? ""), filter: (filter == null ? void 0 : filter.toLowerCase().trim()) === "true" };
     });
-    const partial = chunk.filter((_2, idx) => {
+    const partial = chunk.filter((_, idx) => {
       var _a2;
       return ((_a2 = indices.find((x) => x.idx === idx)) == null ? void 0 : _a2.filter) ?? false;
     });

package/dist/{csj → node}/operations/label.js

@@ -1,6 +1,5 @@
-import sdk from "@botpress/sdk";
-const { z } = sdk;
-import _ from "lodash";
+import { z } from "@bpinternal/zui";
+import { clamp, chunk } from "lodash-es";
 import { fastHash, stringify, takeUntilTokens } from "../utils";
 import { Zai } from "../zai";
 import { PROMPT_INPUT_BUFFER } from "./constants";
@@ -62,9 +61,9 @@ Zai.prototype.label = async function(input, _labels, _options) {
   const tokenizer = await this.getTokenizer();
   const taskId = this.taskId;
   const taskType = "zai.label";
-  const TOTAL_MAX_TOKENS = _.clamp(options.chunkLength, 1e3, this.Model.input.maxTokens - PROMPT_INPUT_BUFFER);
-  const CHUNK_EXAMPLES_MAX_TOKENS = _.clamp(Math.floor(TOTAL_MAX_TOKENS * 0.5), 250, 1e4);
-  const CHUNK_INPUT_MAX_TOKENS = _.clamp(
+  const TOTAL_MAX_TOKENS = clamp(options.chunkLength, 1e3, this.Model.input.maxTokens - PROMPT_INPUT_BUFFER);
+  const CHUNK_EXAMPLES_MAX_TOKENS = clamp(Math.floor(TOTAL_MAX_TOKENS * 0.5), 250, 1e4);
+  const CHUNK_INPUT_MAX_TOKENS = clamp(
     TOTAL_MAX_TOKENS - CHUNK_EXAMPLES_MAX_TOKENS,
     TOTAL_MAX_TOKENS * 0.5,
     TOTAL_MAX_TOKENS
@@ -72,8 +71,8 @@ Zai.prototype.label = async function(input, _labels, _options) {
   const inputAsString = stringify(input);
   if (tokenizer.count(inputAsString) > CHUNK_INPUT_MAX_TOKENS) {
     const tokens = tokenizer.split(inputAsString);
-    const chunks = _.chunk(tokens, CHUNK_INPUT_MAX_TOKENS).map((x) => x.join(""));
-    const allLabels = await Promise.all(chunks.map((chunk) => this.label(chunk, _labels)));
+    const chunks = chunk(tokens, CHUNK_INPUT_MAX_TOKENS).map((x) => x.join(""));
+    const allLabels = await Promise.all(chunks.map((chunk2) => this.label(chunk2, _labels)));
     return allLabels.reduce((acc, x) => {
       Object.keys(x).forEach((key) => {
         if (acc[key] === true) {

package/dist/{csj → node}/operations/rewrite.js

@@ -1,5 +1,4 @@
-import sdk from "@botpress/sdk";
-const { z } = sdk;
+import { z } from "@bpinternal/zui";
 import { fastHash, stringify, takeUntilTokens } from "../utils";
 import { Zai } from "../zai";
 import { PROMPT_INPUT_BUFFER } from "./constants";

package/dist/{csj → node}/operations/summarize.js

@@ -1,6 +1,5 @@
-import sdk from "@botpress/sdk";
-const { z } = sdk;
-import _ from "lodash";
+import { z } from "@bpinternal/zui";
+import { chunk } from "lodash-es";
 import { Zai } from "../zai";
 import { PROMPT_INPUT_BUFFER, PROMPT_OUTPUT_BUFFER } from "./constants";
 const Options = z.object({
@@ -52,8 +51,8 @@ ${newText}
   const useMergeSort = parts >= Math.pow(2, N);
   const chunkSize = Math.ceil(tokens.length / (parts * N));
   if (useMergeSort) {
-    const chunks = _.chunk(tokens, chunkSize).map((x) => x.join(""));
-    const allSummaries = await Promise.all(chunks.map((chunk) => this.summarize(chunk, options)));
+    const chunks = chunk(tokens, chunkSize).map((x) => x.join(""));
+    const allSummaries = await Promise.all(chunks.map((chunk2) => this.summarize(chunk2, options)));
     return this.summarize(allSummaries.join("\n\n============\n\n"), options);
   }
   const summaries = [];

package/dist/{csj → node}/operations/text.js

@@ -1,6 +1,5 @@
-import sdk from "@botpress/sdk";
-const { z } = sdk;
-import _ from "lodash";
+import { z } from "@bpinternal/zui";
+import { clamp } from "lodash-es";
 import { Zai } from "../zai";
 import { PROMPT_INPUT_BUFFER, PROMPT_OUTPUT_BUFFER } from "./constants";
 const Options = z.object({
@@ -17,7 +16,7 @@ Zai.prototype.text = async function(prompt, _options) {
   const instructions = [];
   let chart = "";
   if (options.length) {
-    const length = _.clamp(options.length * 0.75, 5, options.length);
+    const length = clamp(options.length * 0.75, 5, options.length);
     instructions.push(`IMPORTANT: Length constraint: ${length} tokens/words`);
     instructions.push(`The text must be standalone and complete in less than ${length} tokens/words`);
   }

package/dist/{csj → node}/utils.js

@@ -1,5 +1,4 @@
-import sdk from "@botpress/sdk";
-const { z } = sdk;
+import { z } from "@bpinternal/zui";
 export const stringify = (input, beautify = true) => {
   return typeof input === "string" && !!input.length ? input : input ? JSON.stringify(input, beautify ? null : void 0, beautify ? 2 : void 0) : "<input is null, false, undefined or empty>";
 };

package/dist/{csj → node}/zai.js

@@ -1,10 +1,9 @@
-import sdk from "@botpress/sdk";
+import { z } from "@bpinternal/zui";
 import { getWasmTokenizer } from "@botpress/wasm";
 import { TableAdapter } from "./adapters/botpress-table";
 import { MemoryAdapter } from "./adapters/memory";
 import { Models } from "./models";
 import { BotpressClient } from "./utils";
-const { z } = sdk;
 const ActiveLearning = z.object({
   enable: z.boolean().describe("Whether to enable active learning").default(false),
   tableName: z.string().regex(

package/package.json

@@ -1,21 +1,25 @@
 {
   "name": "@botpress/zai",
   "description": "Zui AI (zai) – An LLM utility library written on top of Zui and the Botpress API",
-  "version": "1.0.1-beta.1",
+  "version": "1.0.1-beta.3",
   "exports": {
     ".": {
       "types": "./src/index.ts",
-      "import": "./dist/esm/index.js",
-      "require": "./dist/cjs/index.js"
+      "node": "./dist/node/index.js",
+      "default": "./dist/node/index.js"
+    },
+    "./browser": {
+      "types": "./src/index.ts",
+      "import": "./dist/browser/index.js"
     }
   },
-  "main": "./dist/cjs/index.js",
-  "browser": "./dist/esm/index.js",
+  "main": "./dist/node/index.js",
+  "browser": "./dist/browser/index.js",
   "types": "./src/index.ts",
   "scripts": {
     "build": "npm run build:nodejs && npm run build:browser",
-    "build:nodejs": "esbuild src/**/*.ts src/*.ts --platform=node --target=node16 --outdir=dist/csj",
-    "build:browser": "esbuild src/**/*.ts src/*.ts --platform=browser --target=es2017 --format=esm --outdir=dist/esm",
+    "build:nodejs": "esbuild src/**/*.ts src/*.ts --platform=node --target=node16 --outdir=dist/node",
+    "build:browser": "esbuild src/index.ts --bundle --external:@bpinternal/zui --external:@botpress/client --external:@botpress/wasm --platform=browser --target=chrome58,firefox57,safari11 --outdir=dist/browser",
     "watch": "tsup --watch",
     "test": "vitest run --config vitest.config.ts",
     "test:update": "vitest -u run --config vitest.config.ts",
@@ -29,22 +33,23 @@
   "license": "ISC",
   "dependencies": {
     "json5": "^2.2.1",
-    "jsonrepair": "^3.2.0"
+    "jsonrepair": "^3.2.0",
+    "lodash-es": "^4.17.21"
   },
   "devDependencies": {
     "@botpress/vai": "0.0.1-beta.7",
-    "@types/lodash": "^4.17.0",
+    "@types/lodash-es": "^4.17.0",
     "dotenv": "^16.3.1",
     "esbuild": "^0.24.2",
+    "lodash": "^4.17.21",
     "ts-node": "^10.9.2",
     "tsup": "^8.3.5",
     "typescript": "^5.7.2",
     "vitest": "^2.0.5"
   },
   "peerDependencies": {
-    "@botpress/client": "^0.30.0",
-    "@botpress/sdk": "^1.6.1",
+    "@botpress/client": "^0.40.0",
     "@botpress/wasm": "^1.0.0",
-    "lodash": "^4.17.21"
+    "@bpinternal/zui": "0.13.3-beta.1"
   }
 }

package/scripts/update-models.mts

@@ -1,6 +1,6 @@
 import { Client } from '@botpress/client'
 
-import _ from 'lodash'
+import { orderBy, isArray } from 'lodash-es'
 import fs from 'node:fs'
 
 const LLM_LIST_MODELS = 'listLanguageModels'
@@ -43,7 +43,7 @@ for (const integrationId in bot.integrations) {
         input: {}
       })
 
-      if (_.isArray(output?.models)) {
+      if (isArray(output?.models)) {
         for (const model of output.models) {
           models.push({
             id: `${integration.name}__${model.id}`,
@@ -60,7 +60,7 @@ for (const integrationId in bot.integrations) {
   }
 }
 
-const content = JSON.stringify(_.orderBy(models, ['integration', 'name']), null, 2)
+const content = JSON.stringify(orderBy(models, ['integration', 'name']), null, 2)
 
 fs.writeFileSync(
   './src/models.ts',

package/scripts/update-types.mts

@@ -1,12 +1,10 @@
 import { Client } from '@botpress/client'
-import sdk from '@botpress/sdk'
+import { z } from '@bpinternal/zui'
 
-import _ from 'lodash'
+import { maxBy } from 'lodash-es'
 import fs from 'node:fs'
 import path from 'node:path'
 
-const { z } = sdk
-
 const Interfaces = ['llm'] as const
 
 const client = new Client({
@@ -21,7 +19,7 @@ for (const name of Interfaces) {
   })
 
   const { interface: latest } = await client.getInterface({
-    id: _.maxBy(interfaces, 'version')!.id
+    id: maxBy(interfaces, 'version')!.id
   })
 
   for (const action of Object.keys(latest.actions)) {
@@ -40,16 +38,8 @@ for (const name of Interfaces) {
 
 export namespace ${name} {
     export namespace ${action} {
-      export ${sdk.z
-        .fromJsonSchema(input)
-        .title('Input')
-        .dereference(references)
-        .toTypescript({ declaration: 'type' })};
-      export ${sdk.z
-        .fromJsonSchema(output)
-        .title('Output')
-        .dereference(references)
-        .toTypescript({ declaration: 'type' })};
+      export ${z.fromJsonSchema(input).title('Input').dereference(references).toTypescript({ declaration: 'type' })};
+      export ${z.fromJsonSchema(output).title('Output').dereference(references).toTypescript({ declaration: 'type' })};
     }
 }`
 

package/src/adapters/botpress-table.ts

@@ -1,6 +1,5 @@
 import { type Client } from '@botpress/client'
-import sdk from '@botpress/sdk'
-const { z } = sdk
+import { z } from '@bpinternal/zui'
 
 import { BotpressClient, GenerationMetadata } from '../utils'
 import { Adapter, GetExamplesProps, SaveExampleProps } from './adapter'
@@ -31,7 +30,7 @@ const Props = z.object({
     )
 })
 
-export type TableSchema = sdk.z.input<typeof TableSchema>
+export type TableSchema = z.input<typeof TableSchema>
 const TableSchema = z.object({
   taskType: z.string().describe('The type of the task (filter, extract, etc.)'),
   taskId: z.string(),
@@ -67,7 +66,7 @@ export class TableAdapter extends Adapter {
   private status: 'initialized' | 'ready' | 'error'
   private errors = [] as string[]
 
-  constructor(props: sdk.z.input<typeof Props>) {
+  constructor(props: z.input<typeof Props>) {
     super()
     props = Props.parse(props)
     this.client = props.client

package/src/operations/__tests/index.ts

@@ -1,5 +1,6 @@
 import { Client } from '@botpress/client'
 import { type TextTokenizer, getWasmTokenizer } from '@botpress/wasm'
+
 import fs from 'node:fs'
 import path from 'node:path'
 import { beforeAll } from 'vitest'

package/src/operations/check.ts

@@ -1,5 +1,4 @@
-import sdk from '@botpress/sdk'
-const { z } = sdk
+import { z } from '@bpinternal/zui'
 
 import { fastHash, stringify, takeUntilTokens } from '../utils'
 import { Zai } from '../zai'
@@ -11,7 +10,7 @@ const Example = z.object({
   reason: z.string().optional()
 })
 
-export type Options = sdk.z.input<typeof Options>
+export type Options = z.input<typeof Options>
 const Options = z.object({
   examples: z.array(Example).describe('Examples to check the condition against').default([])
 })

package/src/operations/extract.ts

@@ -1,16 +1,15 @@
-import sdk from '@botpress/sdk'
-const { z } = sdk
+import { z } from '@bpinternal/zui'
 
 import JSON5 from 'json5'
 import { jsonrepair } from 'jsonrepair'
 
-import _ from 'lodash'
+import { chunk, isArray } from 'lodash-es'
 import { fastHash, stringify, takeUntilTokens } from '../utils'
 import { Zai } from '../zai'
 import { PROMPT_INPUT_BUFFER } from './constants'
 import { JsonParsingError } from './errors'
 
-export type Options = sdk.z.input<typeof Options>
+export type Options = z.input<typeof Options>
 const Options = z.object({
   instructions: z.string().optional().describe('Instructions to guide the user on how to extract the data'),
   chunkLength: z
@@ -25,12 +24,12 @@ const Options = z.object({
 declare module '@botpress/zai' {
   interface Zai {
     /** Extracts one or many elements from an arbitrary input */
-    extract<S extends sdk.z.AnyZodObject>(input: unknown, schema: S, options?: Options): Promise<sdk.z.infer<S>>
-    extract<S extends sdk.z.AnyZodObject>(
+    extract<S extends z.AnyZodObject>(input: unknown, schema: S, options?: Options): Promise<z.infer<S>>
+    extract<S extends z.AnyZodObject>(
       input: unknown,
-      schema: sdk.z.ZodArray<S>,
+      schema: z.ZodArray<S>,
       options?: Options
-    ): Promise<Array<sdk.z.infer<S>>>
+    ): Promise<Array<z.infer<S>>>
   }
 }
 
@@ -50,10 +49,10 @@ Zai.prototype.extract = async function (this: Zai, input, schema, _options) {
   let isArrayOfObjects = false
   const originalSchema = schema
 
-  if (schema instanceof sdk.ZodObject) {
+  if (schema instanceof z.ZodObject) {
     // Do nothing
-  } else if (schema instanceof sdk.ZodArray) {
-    if (schema._def.type instanceof sdk.ZodObject) {
+  } else if (schema instanceof z.ZodArray) {
+    if (schema._def.type instanceof z.ZodObject) {
       isArrayOfObjects = true
       schema = schema._def.type
     } else {
@@ -76,8 +75,8 @@ Zai.prototype.extract = async function (this: Zai, input, schema, _options) {
     // If we want to extract an array of objects, we will run this function recursively
     if (isArrayOfObjects) {
       const tokens = tokenizer.split(inputAsString)
-      const chunks = _.chunk(tokens, options.chunkLength).map((x) => x.join(''))
-      const all = await Promise.all(chunks.map((chunk) => this.extract(chunk, originalSchema as sdk.AnyZodObject)))
+      const chunks = chunk(tokens, options.chunkLength).map((x) => x.join(''))
+      const all = await Promise.all(chunks.map((chunk) => this.extract(chunk, originalSchema as z.AnyZodObject)))
 
       return all.flat()
     } else {
@@ -194,7 +193,7 @@ ${input.trim()}
   }
 
   const formatOutput = (extracted: any) => {
-    extracted = _.isArray(extracted) ? extracted : [extracted]
+    extracted = isArray(extracted) ? extracted : [extracted]
 
     return (
       extracted

package/src/operations/filter.ts

@@ -1,19 +1,18 @@
-import sdk from '@botpress/sdk'
-const { z } = sdk
+import { z } from '@bpinternal/zui'
 
-import _ from 'lodash'
+import { clamp } from 'lodash-es'
 import { fastHash, stringify, takeUntilTokens } from '../utils'
 import { Zai } from '../zai'
 import { PROMPT_INPUT_BUFFER, PROMPT_OUTPUT_BUFFER } from './constants'
 
-type Example = sdk.z.input<typeof Example>
+type Example = z.input<typeof Example>
 const Example = z.object({
   input: z.any(),
   filter: z.boolean(),
   reason: z.string().optional()
 })
 
-export type Options = sdk.z.input<typeof Options>
+export type Options = z.input<typeof Options>
 const Options = z.object({
   tokensPerItem: z
     .number()
@@ -44,7 +43,7 @@ Zai.prototype.filter = async function (this: Zai, input, condition, _options) {
   const MAX_ITEMS_PER_CHUNK = 50
   const TOKENS_TOTAL_MAX = this.Model.input.maxTokens - PROMPT_INPUT_BUFFER - PROMPT_OUTPUT_BUFFER
   const TOKENS_EXAMPLES_MAX = Math.floor(Math.max(250, TOKENS_TOTAL_MAX * 0.5))
-  const TOKENS_CONDITION_MAX = _.clamp(TOKENS_TOTAL_MAX * 0.25, 250, tokenizer.count(condition))
+  const TOKENS_CONDITION_MAX = clamp(TOKENS_TOTAL_MAX * 0.25, 250, tokenizer.count(condition))
   const TOKENS_INPUT_ARRAY_MAX = TOKENS_TOTAL_MAX - TOKENS_EXAMPLES_MAX - TOKENS_CONDITION_MAX
 
   condition = tokenizer.truncate(condition, TOKENS_CONDITION_MAX)

package/src/operations/label.ts

@@ -1,7 +1,6 @@
-import sdk from '@botpress/sdk'
-const { z } = sdk
+import { z } from '@bpinternal/zui'
 
-import _ from 'lodash'
+import { clamp, chunk } from 'lodash-es'
 import { fastHash, stringify, takeUntilTokens } from '../utils'
 import { Zai } from '../zai'
 import { PROMPT_INPUT_BUFFER } from './constants'
@@ -21,7 +20,7 @@ type Example<T extends string> = {
   labels: Partial<Record<T, { label: Label; explanation?: string }>>
 }
 
-export type Options<T extends string> = Omit<sdk.z.input<typeof Options>, 'examples'> & {
+export type Options<T extends string> = Omit<z.input<typeof Options>, 'examples'> & {
   examples?: Array<Partial<Example<T>>>
 }
 
@@ -108,9 +107,9 @@ Zai.prototype.label = async function <T extends string>(this: Zai, input, _label
   const taskId = this.taskId
   const taskType = 'zai.label'
 
-  const TOTAL_MAX_TOKENS = _.clamp(options.chunkLength, 1000, this.Model.input.maxTokens - PROMPT_INPUT_BUFFER)
-  const CHUNK_EXAMPLES_MAX_TOKENS = _.clamp(Math.floor(TOTAL_MAX_TOKENS * 0.5), 250, 10_000)
-  const CHUNK_INPUT_MAX_TOKENS = _.clamp(
+  const TOTAL_MAX_TOKENS = clamp(options.chunkLength, 1000, this.Model.input.maxTokens - PROMPT_INPUT_BUFFER)
+  const CHUNK_EXAMPLES_MAX_TOKENS = clamp(Math.floor(TOTAL_MAX_TOKENS * 0.5), 250, 10_000)
+  const CHUNK_INPUT_MAX_TOKENS = clamp(
     TOTAL_MAX_TOKENS - CHUNK_EXAMPLES_MAX_TOKENS,
     TOTAL_MAX_TOKENS * 0.5,
     TOTAL_MAX_TOKENS
@@ -120,7 +119,7 @@ Zai.prototype.label = async function <T extends string>(this: Zai, input, _label
 
   if (tokenizer.count(inputAsString) > CHUNK_INPUT_MAX_TOKENS) {
     const tokens = tokenizer.split(inputAsString)
-    const chunks = _.chunk(tokens, CHUNK_INPUT_MAX_TOKENS).map((x) => x.join(''))
+    const chunks = chunk(tokens, CHUNK_INPUT_MAX_TOKENS).map((x) => x.join(''))
     const allLabels = await Promise.all(chunks.map((chunk) => this.label(chunk, _labels)))
 
     // Merge all the labels together (those who are true will remain true)

package/src/operations/rewrite.ts

@@ -1,17 +1,16 @@
-import sdk from '@botpress/sdk'
-const { z } = sdk
+import { z } from '@bpinternal/zui'
 
 import { fastHash, stringify, takeUntilTokens } from '../utils'
 import { Zai } from '../zai'
 import { PROMPT_INPUT_BUFFER } from './constants'
 
-type Example = sdk.z.input<typeof Example> & { instructions?: string }
+type Example = z.input<typeof Example> & { instructions?: string }
 const Example = z.object({
   input: z.string(),
   output: z.string()
 })
 
-export type Options = sdk.z.input<typeof Options>
+export type Options = z.input<typeof Options>
 const Options = z.object({
   examples: z.array(Example).default([]),
   length: z.number().min(10).max(16_000).optional().describe('The maximum number of tokens to generate')

package/src/operations/summarize.ts

@@ -1,11 +1,10 @@
-import sdk from '@botpress/sdk'
-const { z } = sdk
+import { z } from '@bpinternal/zui'
 
-import _ from 'lodash'
+import { chunk } from 'lodash-es'
 import { Zai } from '../zai'
 import { PROMPT_INPUT_BUFFER, PROMPT_OUTPUT_BUFFER } from './constants'
 
-export type Options = sdk.z.input<typeof Options>
+export type Options = z.input<typeof Options>
 const Options = z.object({
   prompt: z
     .string()
@@ -93,7 +92,7 @@ ${newText}
   const chunkSize = Math.ceil(tokens.length / (parts * N))
 
   if (useMergeSort) {
-    const chunks = _.chunk(tokens, chunkSize).map((x) => x.join(''))
+    const chunks = chunk(tokens, chunkSize).map((x) => x.join(''))
     const allSummaries = await Promise.all(chunks.map((chunk) => this.summarize(chunk, options)))
     return this.summarize(allSummaries.join('\n\n============\n\n'), options)
   }

package/src/operations/text.ts

@@ -1,11 +1,10 @@
-import sdk from '@botpress/sdk'
-const { z } = sdk
+import { z } from '@bpinternal/zui'
 
-import _ from 'lodash'
+import { clamp } from 'lodash-es'
 import { Zai } from '../zai'
 import { PROMPT_INPUT_BUFFER, PROMPT_OUTPUT_BUFFER } from './constants'
 
-export type Options = sdk.z.input<typeof Options>
+export type Options = z.input<typeof Options>
 const Options = z.object({
   length: z.number().min(1).max(100_000).optional().describe('The maximum number of tokens to generate')
 })
@@ -31,7 +30,7 @@ Zai.prototype.text = async function (this: Zai, prompt, _options) {
   let chart = ''
 
   if (options.length) {
-    const length = _.clamp(options.length * 0.75, 5, options.length)
+    const length = clamp(options.length * 0.75, 5, options.length)
     instructions.push(`IMPORTANT: Length constraint: ${length} tokens/words`)
     instructions.push(`The text must be standalone and complete in less than ${length} tokens/words`)
   }

package/src/utils.ts

@@ -1,6 +1,5 @@
 import type { Client } from '@botpress/client'
-import sdk from '@botpress/sdk'
-const { z } = sdk
+import { z } from '@bpinternal/zui'
 
 export const stringify = (input: unknown, beautify = true) => {
   return typeof input === 'string' && !!input.length

package/src/zai.ts

@@ -1,5 +1,5 @@
 import { Client } from '@botpress/client'
-import sdk from '@botpress/sdk'
+import { z } from '@bpinternal/zui'
 import { type TextTokenizer, getWasmTokenizer } from '@botpress/wasm'
 
 import { Adapter } from './adapters/adapter'
@@ -7,12 +7,12 @@ import { TableAdapter } from './adapters/botpress-table'
 import { MemoryAdapter } from './adapters/memory'
 import { Models } from './models'
 import { llm } from './sdk-interfaces/llm/generateContent'
+
 import { BotpressClient, GenerationMetadata } from './utils'
 
-const { z } = sdk
 type ModelId = (typeof Models)[number]['id']
 
-type ActiveLearning = sdk.z.input<typeof ActiveLearning>
+type ActiveLearning = z.input<typeof ActiveLearning>
 const ActiveLearning = z.object({
   enable: z.boolean().describe('Whether to enable active learning').default(false),
   tableName: z
@@ -33,7 +33,7 @@ const ActiveLearning = z.object({
     .default('default')
 })
 
-type ZaiConfig = sdk.z.input<typeof ZaiConfig>
+type ZaiConfig = z.input<typeof ZaiConfig>
 const ZaiConfig = z.object({
   client: BotpressClient,
   userId: z.string().describe('The ID of the user consuming the API').optional(),

package/dist/esm/adapters/adapter.js

@@ -1,5 +0,0 @@
-class Adapter {
-}
-export {
-  Adapter
-};