@botpress/zai 1.0.0-beta.2 → 1.0.0-beta.4

package/dist/src/operations/rewrite.test.js

@@ -1,96 +0,0 @@
-import { describe, it, expect, beforeEach, afterEach, afterAll } from 'vitest';
-import { check } from '@botpress/vai';
-import { getClient, getZai, metadata, tokenizer } from './__tests';
-import { TableAdapter } from '../adapters/botpress-table';
-const Zoe = `
-Part 1. Zoe walks to the park.
-Part 2. She meets her friend.
-Part 3. They play together.
-Part 4. They have a picnic.
-Part 5. They go home.
-`.trim();
-describe('zai.rewrite', { timeout: 60_000 }, () => {
-    const zai = getZai();
-    it('transforms text to all caps', async () => {
-        const result = await zai.rewrite(`Hello, what is the time today?`, 'write in all caps');
-        expect(result).toBe(`HELLO, WHAT IS THE TIME TODAY?`);
-    });
-    it('transforms text to all caps and respects tokens restrictions', async () => {
-        const result = await zai.rewrite(Zoe, 'write in all caps', { length: 15 });
-        expect(tokenizer.count(result)).toBeLessThanOrEqual(20);
-        expect(result).toContain(`PART 1. ZOE WALKS TO THE PARK`);
-        expect(result).not.toContain(`PART 3`);
-    });
-    it('french translation of the story', async () => {
-        const result = await zai.rewrite(Zoe, 'translate to french');
-        check(result, 'is a french story about Zeo and with 5 parts').toBe(true);
-    });
-    it('Throws if input is bigger than the model max tokens', async () => {
-        await expect(zai.rewrite(Zoe.repeat(100_000), 'translate to french')).rejects.toThrow(/tokens/i);
-    });
-});
-describe('zai.learn.rewrite', { timeout: 60_000 }, () => {
-    const client = getClient();
-    let tableName = 'ZaiTestRewriteInternalTable';
-    let taskId = 'rewrite';
-    let zai = getZai();
-    beforeEach(async () => {
-        zai = getZai().with({
-            activeLearning: {
-                enable: true,
-                taskId,
-                tableName
-            }
-        });
-    });
-    afterEach(async () => {
-        try {
-            await client.deleteTableRows({ table: tableName, deleteAllRows: true });
-        }
-        catch (err) { }
-    });
-    afterAll(async () => {
-        try {
-            await client.deleteTable({ table: tableName });
-        }
-        catch (err) { }
-    });
-    it('learns rewrite rules from examples', async () => {
-        const adapter = new TableAdapter({
-            client,
-            tableName
-        });
-        const value = await zai.learn(taskId).rewrite(`Botpress is awesome`, 'write it like we want it');
-        check(value, `The text means more or less the same as "Botpress is awesome" but slightly different`).toBe(true);
-        let rows = await client.findTableRows({ table: tableName });
-        expect(rows.rows.length).toBe(1);
-        expect(rows.rows[0].output.value).toBe(value);
-        await adapter.saveExample({
-            key: 't1',
-            taskId: `zai/${taskId}`,
-            taskType: 'zai.rewrite',
-            instructions: 'write it like we want it',
-            input: 'Microsoft is a big company',
-            output: `# MICROSOFT IS A BIG COMPANY`,
-            metadata,
-            status: 'approved'
-        });
-        await adapter.saveExample({
-            key: 't2',
-            taskId: `zai/${taskId}`,
-            taskType: 'zai.rewrite',
-            instructions: 'write it like we want it',
-            input: 'Google is an evil company',
-            output: `# GOOGLE IS AN EVIL COMPANY`,
-            metadata,
-            status: 'approved'
-        });
-        const second = await zai.learn(taskId).rewrite(`Botpress is awesome`, 'write it like we want it');
-        rows = await client.findTableRows({ table: tableName });
-        expect(rows.rows.length).toBe(3);
-        check(second, `The text is "BOTPRESS IS AWESOME" and starts with a hashtag`).toBe(true);
-        expect(rows.rows.length).toBe(3);
-        expect(rows.rows[0].output.value).toBe(second);
-    });
-});
-//# sourceMappingURL=rewrite.test.js.map

package/dist/src/operations/rewrite.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"rewrite.test.js","sourceRoot":"","sources":["../../../src/operations/rewrite.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAA;AAC9E,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAA;AAErC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,WAAW,CAAA;AAClE,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAA;AAEzD,MAAM,GAAG,GAAG;;;;;;CAMX,CAAC,IAAI,EAAE,CAAA;AAER,QAAQ,CAAC,aAAa,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE;IAChD,MAAM,GAAG,GAAG,MAAM,EAAE,CAAA;IAEpB,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,gCAAgC,EAAE,mBAAmB,CAAC,CAAA;QACvF,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,gCAAgC,CAAC,CAAA;IACvD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,GAAG,EAAE,mBAAmB,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAA;QAC1E,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAA;QACvD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,+BAA+B,CAAC,CAAA;QACzD,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAA;IACxC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,OAAO,CAAC,GAAG,EAAE,qBAAqB,CAAC,CAAA;QAC5D,KAAK,CAAC,MAAM,EAAE,8CAA8C,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC1E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,qBAAqB,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;IAClG,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,mBAAmB,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE;IACtD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAA;IAC1B,IAAI,SAAS,GAAG,6BAA6B,CAAA;IAC7C,IAAI,MAAM,GAAG,SAAS,CAAA;IACtB,IAAI,GAAG,GAAG,MAAM,EAAE,CAAA;IAElB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,GAAG,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC;YAClB,cAAc,EAAE;gBACd,MAAM,EAAE,IAAI;gBACZ,MAAM;gBACN,SAAS;aACV;SACF,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,eAAe,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAA;QACzE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC,CAAA,CAAC;IAClB,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,WAAW,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAChD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC,CAAA,CAAC;IAClB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC;YAC/B,MAAM;YACN,SAAS;SACV,CAAC,CAAA;QAEF,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,qBAAqB,EAAE,0BAA0B,CAAC,CAAA;QAEhG,KAAK,CAAC,KAAK,EAAE,sFAAsF,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAE/G,IAAI,IAAI,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAC3D,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAChC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAE7C,MAAM,OAAO,CAAC,WAAW,CAAC;YACxB,GAAG,EAAE,IAAI;YACT,MAAM,EAAE,OAAO,MAAM,EAAE;YACvB,QAAQ,EAAE,aAAa;YACvB,YAAY,EAAE,0BAA0B;YACxC,KAAK,EAAE,4BAA4B;YACnC,MAAM,EAAE,8BAA8B;YACtC,QAAQ;YACR,MAAM,EAAE,UAAU;SACnB,CAAC,CAAA;QAEF,MAAM,OAAO,CAAC,WAAW,CAAC;YACxB,GAAG,EAAE,IAAI;YACT,MAAM,EAAE,OAAO,MAAM,EAAE;YACvB,QAAQ,EAAE,aAAa;YACvB,YAAY,EAAE,0BAA0B;YACxC,KAAK,EAAE,2BAA2B;YAClC,MAAM,EAAE,6BAA6B;YACrC,QAAQ;YACR,MAAM,EAAE,UAAU;SACnB,CAAC,CAAA;QAEF,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,qBAAqB,EAAE,0BAA0B,CAAC,CAAA;QAEjG,IAAI,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QACvD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAEhC,KAAK,CAAC,MAAM,EAAE,6DAA6D,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAEvF,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAChC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;IAChD,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/src/operations/summarize.d.ts

@@ -1,46 +0,0 @@
-import { z } from '@bpinternal/zui';
-export type Options = z.input<typeof Options>;
-declare const Options: import("@bpinternal/zui").ZodObject<{
-    prompt: import("@bpinternal/zui").ZodDefault<import("@bpinternal/zui").ZodString>;
-    format: import("@bpinternal/zui").ZodDefault<import("@bpinternal/zui").ZodString>;
-    length: import("@bpinternal/zui").ZodDefault<import("@bpinternal/zui").ZodNumber>;
-    intermediateFactor: import("@bpinternal/zui").ZodDefault<import("@bpinternal/zui").ZodNumber>;
-    maxIterations: import("@bpinternal/zui").ZodDefault<import("@bpinternal/zui").ZodNumber>;
-    sliding: import("@bpinternal/zui").ZodDefault<import("@bpinternal/zui").ZodObject<{
-        window: import("@bpinternal/zui").ZodNumber;
-        overlap: import("@bpinternal/zui").ZodNumber;
-    }, "strip", import("@bpinternal/zui").ZodTypeAny, {
-        window: number;
-        overlap: number;
-    }, {
-        window: number;
-        overlap: number;
-    }>>;
-}, "strip", import("@bpinternal/zui").ZodTypeAny, {
-    length: number;
-    prompt: string;
-    format: string;
-    intermediateFactor: number;
-    maxIterations: number;
-    sliding: {
-        window: number;
-        overlap: number;
-    };
-}, {
-    length?: number | undefined;
-    prompt?: string | undefined;
-    format?: string | undefined;
-    intermediateFactor?: number | undefined;
-    maxIterations?: number | undefined;
-    sliding?: {
-        window: number;
-        overlap: number;
-    } | undefined;
-}>;
-declare module '../zai' {
-    interface Zai {
-        /** Summarizes a text of any length to a summary of the desired length */
-        summarize(original: string, options?: Options): Promise<string>;
-    }
-}
-export {};

package/dist/src/operations/summarize.js

@@ -1,139 +0,0 @@
-import { z } from '@bpinternal/zui';
-import _ from 'lodash';
-import { Zai } from '../zai';
-import { PROMPT_INPUT_BUFFER, PROMPT_OUTPUT_BUFFER } from './constants';
-const Options = z.object({
-    prompt: z
-        .string()
-        .describe('What should the text be summarized to?')
-        .default('New information, concepts and ideas that are deemed important'),
-    format: z
-        .string()
-        .describe('How to format the example text')
-        .default('A normal text with multiple sentences and paragraphs. Use markdown to format the text into sections. Use headings, lists, and other markdown features to make the text more readable. Do not include links, images, or other non-text elements.'),
-    length: z.number().min(10).max(100_000).describe('The length of the summary in tokens').default(250),
-    intermediateFactor: z
-        .number()
-        .min(1)
-        .max(10)
-        .describe('How many times longer (than final length) are the intermediate summaries generated')
-        .default(4),
-    maxIterations: z.number().min(1).default(100),
-    sliding: z
-        .object({
-        window: z.number().min(10).max(100_000),
-        overlap: z.number().min(0).max(100_000)
-    })
-        .describe('Sliding window options')
-        .default({ window: 50_000, overlap: 250 })
-});
-const START = '■START■';
-const END = '■END■';
-Zai.prototype.summarize = async function (original, _options) {
-    const options = Options.parse(_options ?? {});
-    const tokenizer = await this.getTokenizer();
-    const INPUT_COMPONENT_SIZE = Math.max(100, (this.Model.input.maxTokens - PROMPT_INPUT_BUFFER) / 4);
-    options.prompt = tokenizer.truncate(options.prompt, INPUT_COMPONENT_SIZE);
-    options.format = tokenizer.truncate(options.format, INPUT_COMPONENT_SIZE);
-    const maxOutputSize = this.Model.output.maxTokens - PROMPT_OUTPUT_BUFFER;
-    if (options.length > maxOutputSize) {
-        throw new Error(`The desired output length is ${maxOutputSize} tokens long, which is more than the maximum of ${this.Model.output.maxTokens} tokens for this model (${this.Model.name})`);
-    }
-    // Ensure the sliding window is not bigger than the model input size
-    options.sliding.window = Math.min(options.sliding.window, this.Model.input.maxTokens - PROMPT_INPUT_BUFFER);
-    // Ensure the overlap is not bigger than the window
-    // Most extreme case possible (all 3 same size)
-    // |ooooooooooooooooooo|wwwwwwwwwwwwwww|ooooooooooooooooooo|
-    // |<---- overlap ---->|<--  window -->|<---- overlap ---->|
-    options.sliding.overlap = Math.min(options.sliding.overlap, options.sliding.window - 3 * options.sliding.overlap);
-    const format = (summary, newText) => {
-        return `
-${START}
-${summary.length ? summary : '<summary still empty>'}
-${END}
-
-Please amend the summary between the ${START} and ${END} tags to accurately reflect the prompt and the additional text below.
-
-<|start_new_information|>
-${newText}
-<|new_information|>`.trim();
-    };
-    const tokens = tokenizer.split(original);
-    const parts = Math.ceil(tokens.length / (options.sliding.window - options.sliding.overlap));
-    let iteration = 0;
-    // We split it recursively into smaller parts until we're at less than 4 window slides per part
-    // Then we use a merge strategy to combine the sub-chunks summaries
-    // This is basically a merge sort algorithm (but summary instead of sorting)
-    const N = 2; // This is the merge sort exponent
-    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)));
-        return this.summarize(allSummaries.join('\n\n============\n\n'), options);
-    }
-    const summaries = [];
-    let currentSummary = '';
-    for (let i = 0; i < tokens.length; i += options.sliding.window) {
-        const from = Math.max(0, i - options.sliding.overlap);
-        const to = Math.min(tokens.length, i + options.sliding.window + options.sliding.overlap);
-        const isFirst = i === 0;
-        const isLast = to >= tokens.length;
-        const slice = tokens.slice(from, to).join('');
-        if (iteration++ >= options.maxIterations) {
-            break;
-        }
-        const instructions = [
-            `At each step, you will receive a part of the text to summarize. Make sure to reply with the new summary in the tags ${START} and ${END}.`,
-            'Summarize the text and make sure that the main points are included.',
-            'Ignore any unnecessary details and focus on the main points.',
-            'Use short and concise sentences to increase readability and information density.',
-            'When looking at the new information, focus on: ' + options.prompt
-        ];
-        if (isFirst) {
-            instructions.push('The current summary is empty. You need to generate a summary that covers the main points of the text.');
-        }
-        let generationLength = options.length;
-        if (!isLast) {
-            generationLength = Math.min(tokenizer.count(currentSummary) + options.length * options.intermediateFactor, maxOutputSize);
-            instructions.push('You need to amend the summary to include the new information. Make sure the summary is complete and covers all the main points.');
-            instructions.push(`The current summary is ${currentSummary.length} tokens long.`);
-            instructions.push(`You can amend the summary to be up to ${generationLength} tokens long.`);
-        }
-        if (isLast) {
-            instructions.push('This is the last part you will have to summarize. Make sure the summary is complete and covers all the main points.');
-            instructions.push(`The current summary is ${currentSummary.length} tokens long. You need to make sure it is ${options.length} tokens or less.`);
-            if (currentSummary.length > options.length) {
-                instructions.push(`The current summary is already too long, so you need to shorten it to ${options.length} tokens while also including the new information.`);
-            }
-        }
-        const output = await this.callModel({
-            systemPrompt: `
-You are summarizing a text. The text is split into ${parts} parts, and you are currently working on part ${iteration}.
-At every step, you will receive the current summary and a new part of the text. You need to amend the summary to include the new information (if needed).
-The summary needs to cover the main points of the text and must be concise.
-
-IMPORTANT INSTRUCTIONS:
-${instructions.map((x) => `- ${x.trim()}`).join('\n')}
-
-FORMAT OF THE SUMMARY:
-${options.format}
-`.trim(),
-            messages: [{ type: 'text', content: format(currentSummary, slice), role: 'user' }],
-            maxTokens: generationLength,
-            stopSequences: [END]
-        });
-        let result = output.choices[0].content;
-        if (result.includes(START)) {
-            result = result.slice(result.indexOf(START) + START.length);
-        }
-        if (result.includes('■')) {
-            // can happen if the model truncates the text before the entire END tag is written
-            result = result.slice(0, result.indexOf('■'));
-        }
-        summaries.push(result);
-        currentSummary = result;
-    }
-    return currentSummary.trim();
-};
-//# sourceMappingURL=summarize.js.map

package/dist/src/operations/summarize.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"summarize.js","sourceRoot":"","sources":["../../../src/operations/summarize.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,iBAAiB,CAAA;AAEnC,OAAO,CAAC,MAAM,QAAQ,CAAA;AACtB,OAAO,EAAE,GAAG,EAAE,MAAM,QAAQ,CAAA;AAC5B,OAAO,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAA;AAGvE,MAAM,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC;IACvB,MAAM,EAAE,CAAC;SACN,MAAM,EAAE;SACR,QAAQ,CAAC,wCAAwC,CAAC;SAClD,OAAO,CAAC,+DAA+D,CAAC;IAC3E,MAAM,EAAE,CAAC;SACN,MAAM,EAAE;SACR,QAAQ,CAAC,gCAAgC,CAAC;SAC1C,OAAO,CACN,iPAAiP,CAClP;IACH,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,qCAAqC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC;IACpG,kBAAkB,EAAE,CAAC;SAClB,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,EAAE,CAAC;SACP,QAAQ,CAAC,oFAAoF,CAAC;SAC9F,OAAO,CAAC,CAAC,CAAC;IACb,aAAa,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC;IAC7C,OAAO,EAAE,CAAC;SACP,MAAM,CAAC;QACN,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC;QACvC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC;KACxC,CAAC;SACD,QAAQ,CAAC,wBAAwB,CAAC;SAClC,OAAO,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;CAC7C,CAAC,CAAA;AASF,MAAM,KAAK,GAAG,SAAS,CAAA;AACvB,MAAM,GAAG,GAAG,OAAO,CAAA;AAEnB,GAAG,CAAC,SAAS,CAAC,SAAS,GAAG,KAAK,WAAsB,QAAQ,EAAE,QAAQ;IACrE,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAA;IAC7C,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAA;IAE3C,MAAM,oBAAoB,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC,CAAA;IAClG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAA;IACzE,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,oBAAoB,CAAC,CAAA;IAEzE,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,GAAG,oBAAoB,CAAA;IACxE,IAAI,OAAO,CAAC,MAAM,GAAG,aAAa,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CACb,gCAAgC,aAAa,mDAAmD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,2BAA2B,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CACzK,CAAA;IACH,CAAC;IAED,oEAAoE;IACpE,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,GAAG,mBAAmB,CAAC,CAAA;IAE3G,mDAAmD;IACnD,+CAA+C;IAC/C,4DAA4D;IAC5D,4DAA4D;IAC5D,OAAO,CAAC,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;IAEjH,MAAM,MAAM,GAAG,CAAC,OAAe,EAAE,OAAe,EAAE,EAAE;QAClD,OAAO;EACT,KAAK;EACL,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,uBAAuB;EAClD,GAAG;;uCAEkC,KAAK,QAAQ,GAAG;;;EAGrD,OAAO;oBACW,CAAC,IAAI,EAAE,CAAA;IACzB,CAAC,CAAA;IAED,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAA;IACxC,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAA;IAC3F,IAAI,SAAS,GAAG,CAAC,CAAA;IAEjB,+FAA+F;IAC/F,mEAAmE;IACnE,4EAA4E;IAC5E,MAAM,CAAC,GAAG,CAAC,CAAA,CAAC,kCAAkC;IAC9C,MAAM,YAAY,GAAG,KAAK,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;IAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAA;IAExD,IAAI,YAAY,EAAE,CAAC;QACjB,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAA;QAChE,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,CAAA;QAC7F,OAAO,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,CAAC,sBAAsB,CAAC,EAAE,OAAO,CAAC,CAAA;IAC3E,CAAC;IAED,MAAM,SAAS,GAAa,EAAE,CAAA;IAC9B,IAAI,cAAc,GAAG,EAAE,CAAA;IAEvB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QAC/D,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;QACrD,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAA;QACxF,MAAM,OAAO,GAAG,CAAC,KAAK,CAAC,CAAA;QACvB,MAAM,MAAM,GAAG,EAAE,IAAI,MAAM,CAAC,MAAM,CAAA;QAElC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAA;QAE7C,IAAI,SAAS,EAAE,IAAI,OAAO,CAAC,aAAa,EAAE,CAAC;YACzC,MAAK;QACP,CAAC;QAED,MAAM,YAAY,GAAa;YAC7B,uHAAuH,KAAK,QAAQ,GAAG,GAAG;YAC1I,qEAAqE;YACrE,8DAA8D;YAC9D,kFAAkF;YAClF,iDAAiD,GAAG,OAAO,CAAC,MAAM;SACnE,CAAA;QAED,IAAI,OAAO,EAAE,CAAC;YACZ,YAAY,CAAC,IAAI,CACf,uGAAuG,CACxG,CAAA;QACH,CAAC;QAED,IAAI,gBAAgB,GAAG,OAAO,CAAC,MAAM,CAAA;QAErC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,gBAAgB,GAAG,IAAI,CAAC,GAAG,CACzB,SAAS,CAAC,KAAK,CAAC,cAAc,CAAC,GAAG,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,kBAAkB,EAC7E,aAAa,CACd,CAAA;YAED,YAAY,CAAC,IAAI,CACf,iIAAiI,CAClI,CAAA;YAED,YAAY,CAAC,IAAI,CAAC,0BAA0B,cAAc,CAAC,MAAM,eAAe,CAAC,CAAA;YACjF,YAAY,CAAC,IAAI,CAAC,yCAAyC,gBAAgB,eAAe,CAAC,CAAA;QAC7F,CAAC;QAED,IAAI,MAAM,EAAE,CAAC;YACX,YAAY,CAAC,IAAI,CACf,qHAAqH,CACtH,CAAA;YACD,YAAY,CAAC,IAAI,CACf,0BAA0B,cAAc,CAAC,MAAM,6CAA6C,OAAO,CAAC,MAAM,kBAAkB,CAC7H,CAAA;YAED,IAAI,cAAc,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;gBAC3C,YAAY,CAAC,IAAI,CACf,yEAAyE,OAAO,CAAC,MAAM,mDAAmD,CAC3I,CAAA;YACH,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC;YAClC,YAAY,EAAE;qDACiC,KAAK,iDAAiD,SAAS;;;;;EAKlH,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;;;EAGnD,OAAO,CAAC,MAAM;CACf,CAAC,IAAI,EAAE;YACF,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,cAAc,EAAE,KAAK,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;YAClF,SAAS,EAAE,gBAAgB;YAC3B,aAAa,EAAE,CAAC,GAAG,CAAC;SACrB,CAAC,CAAA;QAEF,IAAI,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAiB,CAAA;QAEhD,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3B,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC,CAAA;QAC7D,CAAC;QAED,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YACzB,kFAAkF;YAClF,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAA;QAC/C,CAAC;QAED,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QACtB,cAAc,GAAG,MAAM,CAAA;IACzB,CAAC;IAED,OAAO,cAAc,CAAC,IAAI,EAAE,CAAA;AAC9B,CAAC,CAAA"}

package/dist/src/operations/summarize.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/src/operations/summarize.test.js

@@ -1,22 +0,0 @@
-import { check } from '@botpress/vai';
-import { describe, it } from 'vitest';
-import { BotpressDocumentation, getZai } from './__tests';
-describe('zai.summarize', () => {
-    const zai = getZai();
-    it.skip('summarize long document to a concise 2000 token summary', async () => {
-        const result = await zai.summarize(BotpressDocumentation, {
-            length: 2000,
-            prompt: `Extract the Table of Contents for the Botpress Documentation. Pay special attention to all the different features. Focus on horizontal coverage of features rather than going in depth into one feature. The goal is to have a complete overview of what the documentation covers.`
-        });
-        check(result, 'The text is a summary of the Botpress documentation').toBe(true);
-        check(result, 'The text explains shortly what botpress is').toBe(true);
-        check(result, 'The text uses markdown format').toBe(true);
-        check(result, 'The text has some information about integrations').toBe(true);
-        check(result, 'The text has a section about Flows (or Workflows)').toBe(true);
-        check(result, 'The text has a section about the Botpress API').toBe(true);
-        check(result, 'The text mentions the notion of workspaces').toBe(true);
-        check(result, 'The text has some information about the Webchat').toBe(true);
-        check(result, 'The text has some information about HITL (human in the loop)').toBe(true);
-    });
-});
-//# sourceMappingURL=summarize.test.js.map

package/dist/src/operations/summarize.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"summarize.test.js","sourceRoot":"","sources":["../../../src/operations/summarize.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAA;AACrC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAErC,OAAO,EAAE,qBAAqB,EAAE,MAAM,EAAE,MAAM,WAAW,CAAA;AAEzD,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,MAAM,GAAG,GAAG,MAAM,EAAE,CAAA;IAEpB,EAAE,CAAC,IAAI,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,SAAS,CAAC,qBAAqB,EAAE;YACxD,MAAM,EAAE,IAAI;YACZ,MAAM,EAAE,oRAAoR;SAC7R,CAAC,CAAA;QAEF,KAAK,CAAC,MAAM,EAAE,qDAAqD,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC/E,KAAK,CAAC,MAAM,EAAE,4CAA4C,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACtE,KAAK,CAAC,MAAM,EAAE,+BAA+B,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACzD,KAAK,CAAC,MAAM,EAAE,kDAAkD,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC5E,KAAK,CAAC,MAAM,EAAE,mDAAmD,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC7E,KAAK,CAAC,MAAM,EAAE,+CAA+C,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACzE,KAAK,CAAC,MAAM,EAAE,4CAA4C,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACtE,KAAK,CAAC,MAAM,EAAE,iDAAiD,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC3E,KAAK,CAAC,MAAM,EAAE,8DAA8D,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC1F,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/src/operations/text.d.ts

@@ -1,16 +0,0 @@
-import { z } from '@bpinternal/zui';
-export type Options = z.input<typeof Options>;
-declare const Options: import("@bpinternal/zui").ZodObject<{
-    length: import("@bpinternal/zui").ZodOptional<import("@bpinternal/zui").ZodNumber>;
-}, "strip", import("@bpinternal/zui").ZodTypeAny, {
-    length?: number | undefined;
-}, {
-    length?: number | undefined;
-}>;
-declare module '../zai' {
-    interface Zai {
-        /** Generates a text of the desired length according to the prompt */
-        text(prompt: string, options?: Options): Promise<string>;
-    }
-}
-export {};

package/dist/src/operations/text.js

@@ -1,47 +0,0 @@
-import { z } from '@bpinternal/zui';
-import _ from 'lodash';
-import { Zai } from '../zai';
-import { PROMPT_INPUT_BUFFER, PROMPT_OUTPUT_BUFFER } from './constants';
-const Options = z.object({
-    length: z.number().min(1).max(100_000).optional().describe('The maximum number of tokens to generate')
-});
-Zai.prototype.text = async function (prompt, _options) {
-    const options = Options.parse(_options ?? {});
-    const tokenizer = await this.getTokenizer();
-    prompt = tokenizer.truncate(prompt, Math.max(this.Model.input.maxTokens - PROMPT_INPUT_BUFFER, 100));
-    if (options.length) {
-        options.length = Math.min(this.Model.output.maxTokens - PROMPT_OUTPUT_BUFFER, options.length);
-    }
-    const instructions = [];
-    let chart = '';
-    if (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`);
-    }
-    if (options.length && options.length <= 500) {
-        chart = `
-| Tokens      | Text Length (approximate)            |
-|-------------|--------------------------------------|
-| < 5 tokens  | 1-3 words                            |
-| 5-10 tokens | 3-6 words                            |
-| 10-20 tokens| 6-15 words                           |
-| 20-50 tokens| A short sentence (15-30 words)       |
-| 50-100 tokens| A medium sentence (30-70 words)     |
-| 100-200 tokens| A short paragraph (70-150 words)   |
-| 200-300 tokens| A medium paragraph (150-200 words) |
-| 300-500 tokens| A long paragraph (200-300 words)   |`.trim();
-    }
-    const output = await this.callModel({
-        systemPrompt: `
-Generate a text that fulfills the user prompt below. Answer directly to the prompt, without any acknowledgements or fluff. Also, make sure the text is standalone and complete.
-${instructions.map((x) => `- ${x}`).join('\n')}
-${chart}
-`.trim(),
-        temperature: 0.7,
-        messages: [{ type: 'text', content: prompt, role: 'user' }],
-        maxTokens: options.length
-    });
-    return output.choices[0].content;
-};
-//# sourceMappingURL=text.js.map

package/dist/src/operations/text.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"text.js","sourceRoot":"","sources":["../../../src/operations/text.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,iBAAiB,CAAA;AAEnC,OAAO,CAAC,MAAM,QAAQ,CAAA;AACtB,OAAO,EAAE,GAAG,EAAE,MAAM,QAAQ,CAAA;AAC5B,OAAO,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAA;AAGvE,MAAM,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC;IACvB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,0CAA0C,CAAC;CACvG,CAAC,CAAA;AASF,GAAG,CAAC,SAAS,CAAC,IAAI,GAAG,KAAK,WAAsB,MAAM,EAAE,QAAQ;IAC9D,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAA;IAC7C,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAA;IAE3C,MAAM,GAAG,SAAS,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,GAAG,mBAAmB,EAAE,GAAG,CAAC,CAAC,CAAA;IAEpG,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACnB,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,GAAG,oBAAoB,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;IAC/F,CAAC;IAED,MAAM,YAAY,GAAa,EAAE,CAAA;IACjC,IAAI,KAAK,GAAG,EAAE,CAAA;IAEd,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACnB,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,IAAI,EAAE,CAAC,EAAE,OAAO,CAAC,MAAM,CAAC,CAAA;QAChE,YAAY,CAAC,IAAI,CAAC,iCAAiC,MAAM,eAAe,CAAC,CAAA;QACzE,YAAY,CAAC,IAAI,CAAC,yDAAyD,MAAM,eAAe,CAAC,CAAA;IACnG,CAAC;IAED,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,IAAI,GAAG,EAAE,CAAC;QAC5C,KAAK,GAAG;;;;;;;;;;uDAU2C,CAAC,IAAI,EAAE,CAAA;IAC5D,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC;QAClC,YAAY,EAAE;;EAEhB,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;EAC5C,KAAK;CACN,CAAC,IAAI,EAAE;QACJ,WAAW,EAAE,GAAG;QAChB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;QAC3D,SAAS,EAAE,OAAO,CAAC,MAAM;KAC1B,CAAC,CAAA;IACF,OAAO,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAiB,CAAA;AAC5C,CAAC,CAAA"}

package/dist/src/operations/text.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/src/operations/text.test.js

@@ -1,51 +0,0 @@
-import { describe, it, expect } from 'vitest';
-import { check } from '@botpress/vai';
-import { getZai, tokenizer } from './__tests';
-describe('zai.text', { timeout: 60_000 }, () => {
-    const zai = getZai();
-    it('generate a horror novel with no params', async () => {
-        const story = await zai.text('write a short horror novel');
-        check(story, 'is a short horror story').toBe(true);
-    });
-    it('No fluffy text at the beginning', async () => {
-        const story = await zai.text('write a short horror novel');
-        check(story, 'There is no LLM fluff at the beginning', {
-            examples: [
-                {
-                    value: 'Title: A horror story\nChapter 1: The woods\nOnce upen a time, ...',
-                    expected: true,
-                    reason: 'It begins straight with a story, no fluff at the beginning'
-                },
-                { value: 'Once upon a time, a ...', expected: true, reason: 'The story starts directly' },
-                {
-                    value: 'Sure, I will generate a story.\nOnce upen a time, a...',
-                    expected: false,
-                    reason: 'There is some fluff at the beginning'
-                }
-            ]
-        }).toBe(true);
-    });
-    it('No fluffy text at the end', async () => {
-        const story = await zai.text('write a short horror novel');
-        check(story, 'There is no LLM fluff at the end', {
-            examples: [
-                {
-                    value: 'Title: A horror story\nChapter 1: The woods\nOnce upen a time, ... The End.',
-                    expected: true,
-                    reason: 'The end is clear and direct, no fluff at the end'
-                },
-                {
-                    value: 'Sure, I will generate a story.\nOnce upen a time, a... The End.\nLet me know if you want more or if you are happy with this.',
-                    expected: false,
-                    reason: 'There is some fluff from the assistant at the end.'
-                }
-            ]
-        }).toBe(true);
-    });
-    it('length/max tokens param', async () => {
-        const story = await zai.text('write a short but complete horror story (with conclusion)', { length: 100 });
-        expect(tokenizer.count(story)).toBeLessThanOrEqual(110);
-        check(story, 'could be the beginning of a horror story').toBe(true);
-    });
-});
-//# sourceMappingURL=text.test.js.map

package/dist/src/operations/text.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"text.test.js","sourceRoot":"","sources":["../../../src/operations/text.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAA;AAC7C,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAA;AAErC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,WAAW,CAAA;AAE7C,QAAQ,CAAC,UAAU,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE;IAC7C,MAAM,GAAG,GAAG,MAAM,EAAE,CAAA;IAEpB,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;QACtD,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAA;QAC1D,KAAK,CAAC,KAAK,EAAE,yBAAyB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACpD,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;QAC/C,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAA;QAE1D,KAAK,CAAC,KAAK,EAAE,wCAAwC,EAAE;YACrD,QAAQ,EAAE;gBACR;oBACE,KAAK,EAAE,oEAAoE;oBAC3E,QAAQ,EAAE,IAAI;oBACd,MAAM,EAAE,4DAA4D;iBACrE;gBACD,EAAE,KAAK,EAAE,yBAAyB,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,2BAA2B,EAAE;gBACzF;oBACE,KAAK,EAAE,wDAAwD;oBAC/D,QAAQ,EAAE,KAAK;oBACf,MAAM,EAAE,sCAAsC;iBAC/C;aACF;SACF,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACf,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAA;QAE1D,KAAK,CAAC,KAAK,EAAE,kCAAkC,EAAE;YAC/C,QAAQ,EAAE;gBACR;oBACE,KAAK,EAAE,6EAA6E;oBACpF,QAAQ,EAAE,IAAI;oBACd,MAAM,EAAE,kDAAkD;iBAC3D;gBACD;oBACE,KAAK,EACH,8HAA8H;oBAChI,QAAQ,EAAE,KAAK;oBACf,MAAM,EAAE,oDAAoD;iBAC7D;aACF;SACF,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACf,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,2DAA2D,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAA;QAC1G,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAA;QAEvD,KAAK,CAAC,KAAK,EAAE,0CAA0C,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACrE,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/src/operations/zai-learn.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/src/operations/zai-learn.test.js

@@ -1,71 +0,0 @@
-import { describe, it, expect, afterAll, beforeEach, afterEach, vi } from 'vitest';
-import { getClient, getZai } from './__tests';
-import { check } from '@botpress/vai';
-describe('zai.learn / generic', { timeout: 60_000 }, () => {
-    const client = getClient();
-    let tableName = 'ZaiTestInternalTable';
-    let taskId = 'test';
-    let zai = getZai();
-    beforeEach(async () => {
-        zai = getZai().with({
-            activeLearning: {
-                enable: true,
-                taskId,
-                tableName
-            }
-        });
-    });
-    afterEach(async () => {
-        try {
-            await client.deleteTableRows({ table: tableName, deleteAllRows: true });
-        }
-        catch (err) { }
-    });
-    afterAll(async () => {
-        try {
-            await client.deleteTable({ table: tableName });
-        }
-        catch (err) { }
-    });
-    it('saves examples to tables', async () => {
-        const value = await zai
-            .learn(taskId)
-            .check('This text is very clearly written in English.', 'is an english sentence');
-        const { rows } = await client.findTableRows({ table: tableName });
-        expect(value).toBe(true);
-        expect(rows.length).toBe(1);
-        check(rows[0].explanation, 'is an explanation sentence');
-        expect(rows[0].explanation).not.toContain('Final Answer:');
-        expect(rows[0].output).toMatchObject({ value: true });
-        expect(rows[0].input).toMatchInlineSnapshot(`
-      {
-        "value": "This text is very clearly written in English.",
-      }
-    `);
-        expect(rows[0].taskId).toEqual('zai/test');
-        expect(rows[0].taskType).toBe('zai.check');
-    });
-    it('works even if tables are down', async () => {
-        const upsertTableRows = vi.fn(async () => {
-            throw new Error('Table is down');
-        });
-        const findTableRows = vi.fn(async () => {
-            throw new Error('Table is down');
-        });
-        const client = {
-            ...getClient(),
-            findTableRows,
-            upsertTableRows
-        };
-        const value = await zai
-            .with({ client })
-            .learn(taskId)
-            .check('This text is very clearly written in English.', 'is an english sentence');
-        const { rows } = await getClient().findTableRows({ table: tableName });
-        expect(value).toBe(true);
-        expect(rows.length).toBe(0);
-        expect(upsertTableRows).toHaveBeenCalledTimes(1);
-        expect(findTableRows).toHaveBeenCalledTimes(1);
-    });
-});
-//# sourceMappingURL=zai-learn.test.js.map

package/dist/src/operations/zai-learn.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"zai-learn.test.js","sourceRoot":"","sources":["../../../src/operations/zai-learn.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAElF,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,WAAW,CAAA;AAE7C,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAA;AAGrC,QAAQ,CAAC,qBAAqB,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE;IACxD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAA;IAC1B,IAAI,SAAS,GAAG,sBAAsB,CAAA;IACtC,IAAI,MAAM,GAAG,MAAM,CAAA;IACnB,IAAI,GAAG,GAAG,MAAM,EAAE,CAAA;IAElB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,GAAG,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC;YAClB,cAAc,EAAE;gBACd,MAAM,EAAE,IAAI;gBACZ,MAAM;gBACN,SAAS;aACV;SACF,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,eAAe,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAA;QACzE,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC,CAAA,CAAC;IAClB,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,WAAW,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAChD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC,CAAA,CAAC;IAClB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,KAAK,GAAG,MAAM,GAAG;aACpB,KAAK,CAAC,MAAM,CAAC;aACb,KAAK,CAAC,+CAA+C,EAAE,wBAAwB,CAAC,CAAA;QAEnF,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAEjE,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACxB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAE3B,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,4BAA4B,CAAC,CAAA;QACxD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,eAAe,CAAC,CAAA;QAC1D,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,aAAa,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAA;QACrD,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC;;;;KAI3C,CAAC,CAAA;QACF,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAA;QAC1C,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;YACvC,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAA;QAClC,CAAC,CAAC,CAAA;QAEF,MAAM,aAAa,GAAG,EAAE,CAAC,EAAE,CAAC,KAAK,IAAI,EAAE;YACrC,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAA;QAClC,CAAC,CAAC,CAAA;QAEF,MAAM,MAAM,GAAG;YACb,GAAG,SAAS,EAAE;YACd,aAAa;YACb,eAAe;SACK,CAAA;QAEtB,MAAM,KAAK,GAAG,MAAM,GAAG;aACpB,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;aAChB,KAAK,CAAC,MAAM,CAAC;aACb,KAAK,CAAC,+CAA+C,EAAE,wBAAwB,CAAC,CAAA;QAEnF,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,SAAS,EAAE,CAAC,aAAa,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAEtE,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACxB,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAC3B,MAAM,CAAC,eAAe,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;QAChD,MAAM,CAAC,aAAa,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAChD,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/src/operations/zai-retry.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/src/operations/zai-retry.test.js

@@ -1,50 +0,0 @@
-import { describe, it, expect, vi } from 'vitest';
-import { getClient, getZai } from './__tests';
-describe('zai retry', { timeout: 60_000 }, () => {
-    const client = getClient();
-    let zai = getZai().with({ retry: { maxRetries: 3 } });
-    it('retries 3 times and succeeds', async () => {
-        let retryCount = 0;
-        const throwingClient = {
-            ...client,
-            callAction: vi.fn((input) => {
-                if (++retryCount < 3) {
-                    throw new Error('Failed to call model');
-                }
-                return client.callAction(input);
-            })
-        };
-        const value = await zai
-            .with({ client: throwingClient })
-            .check('This text is very clearly written in English.', 'is an english sentence');
-        expect(value).toBe(true);
-        expect(retryCount).toBe(3);
-    });
-    it('retries 0 times when success', async () => {
-        const fn = vi.fn((input) => client.callAction(input));
-        const throwingClient = {
-            ...client,
-            callAction: fn
-        };
-        const value = await zai
-            .with({ client: throwingClient })
-            .check('This text is very clearly written in English.', 'is an english sentence');
-        expect(value).toBe(true);
-        expect(fn).toHaveBeenCalledOnce();
-    });
-    it('fails when exceeded max', async () => {
-        const fn = vi.fn(() => {
-            throw new Error('Failed to call model');
-        });
-        const throwingClient = {
-            ...client,
-            callAction: fn
-        };
-        const value = zai
-            .with({ client: throwingClient, retry: { maxRetries: 1 } })
-            .check('This text is very clearly written in English.', 'is an english sentence');
-        await expect(value).rejects.toThrowError(/retries/);
-        expect(fn).toHaveBeenCalledTimes(2);
-    });
-});
-//# sourceMappingURL=zai-retry.test.js.map

package/dist/src/operations/zai-retry.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"zai-retry.test.js","sourceRoot":"","sources":["../../../src/operations/zai-retry.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAEjD,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,WAAW,CAAA;AAI7C,QAAQ,CAAC,WAAW,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE;IAC9C,MAAM,MAAM,GAAG,SAAS,EAAE,CAAA;IAC1B,IAAI,GAAG,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC,CAAA;IAErD,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,IAAI,UAAU,GAAG,CAAC,CAAA;QAClB,MAAM,cAAc,GAAG;YACrB,GAAG,MAAM;YACT,UAAU,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC1B,IAAI,EAAE,UAAU,GAAG,CAAC,EAAE,CAAC;oBACrB,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAA;gBACzC,CAAC;gBACD,OAAO,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAA;YACjC,CAAC,CAAC;SACkB,CAAA;QAEtB,MAAM,KAAK,GAAG,MAAM,GAAG;aACpB,IAAI,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;aAChC,KAAK,CAAC,+CAA+C,EAAE,wBAAwB,CAAC,CAAA;QAEnF,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACxB,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IAC5B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAA;QAErD,MAAM,cAAc,GAAG;YACrB,GAAG,MAAM;YACT,UAAU,EAAE,EAAE;SACM,CAAA;QAEtB,MAAM,KAAK,GAAG,MAAM,GAAG;aACpB,IAAI,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;aAChC,KAAK,CAAC,+CAA+C,EAAE,wBAAwB,CAAC,CAAA;QAEnF,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACxB,MAAM,CAAC,EAAE,CAAC,CAAC,oBAAoB,EAAE,CAAA;IACnC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE;YACpB,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAA;QACzC,CAAC,CAAC,CAAA;QAEF,MAAM,cAAc,GAAG;YACrB,GAAG,MAAM;YACT,UAAU,EAAE,EAAE;SACM,CAAA;QAEtB,MAAM,KAAK,GAAG,GAAG;aACd,IAAI,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,KAAK,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;aAC1D,KAAK,CAAC,+CAA+C,EAAE,wBAAwB,CAAC,CAAA;QAEnF,MAAM,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAA;QACnD,MAAM,CAAC,EAAE,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IACrC,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}