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

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

@@ -1,163 +0,0 @@
-import { describe, it, expect, beforeEach, afterEach, afterAll } from 'vitest';
-import { BotpressDocumentation, getClient, getZai, metadata } from './__tests';
-import { z } from '@bpinternal/zui';
-import { check } from '@botpress/vai';
-import { TableAdapter } from '../adapters/botpress-table';
-describe('zai.extract', () => {
-    const zai = getZai();
-    it('extract simple object from paragraph', async () => {
-        const person = await zai.extract('My name is John Doe, I am 30 years old and I live in Quebec', z.object({
-            name: z.string(),
-            age: z.number(),
-            location: z.string()
-        }));
-        expect(person).toMatchInlineSnapshot(`
-      {
-        "age": 30,
-        "location": "Quebec",
-        "name": "John Doe",
-      }
-    `);
-    });
-    it('extract an array of objects from paragraph', async () => {
-        const people = await zai.extract(`
-      My name is John Doe, I am 30 years old and I live in Quebec.
-      My name is Jane Doe, I am 25 years old and I live in Montreal.
-      His name is Jack Doe, he is 35 years old and he lives in Toronto.
-      Her name is Jill Doe, she is 40 years old and she lives in Vancouver.`, z.array(z.object({
-            name: z.string(),
-            age: z.number(),
-            location: z.string()
-        })));
-        expect(people).toMatchInlineSnapshot(`
-      [
-        {
-          "age": 30,
-          "location": "Quebec",
-          "name": "John Doe",
-        },
-        {
-          "age": 25,
-          "location": "Montreal",
-          "name": "Jane Doe",
-        },
-        {
-          "age": 35,
-          "location": "Toronto",
-          "name": "Jack Doe",
-        },
-        {
-          "age": 40,
-          "location": "Vancouver",
-          "name": "Jill Doe",
-        },
-      ]
-    `);
-    });
-    it('extract an object from anything as input', async () => {
-        const person = await zai.extract({
-            person: { first: 'John', last: 'Doe', age: 30 }
-        }, z.object({
-            a: z.string().describe('The full name of the person in the text'),
-            b: z.number().describe('The age of the person in the text')
-        }));
-        expect(person).toMatchInlineSnapshot(`
-      {
-        "a": "John Doe",
-        "b": 30,
-      }
-    `);
-    });
-    it('extract an array of objects from a super long text', async () => {
-        const features = await zai.extract(BotpressDocumentation, z.array(z
-            .object({
-            feature: z.string().describe('The name of the feature'),
-            parent: z.string().optional().describe('The parent feature').nullable(),
-            description: z.string().describe('The description of the feature')
-        })
-            .describe('A feature of Botpress')), {
-            instructions: 'Extract all things that looks like a Botpress feature in the provided input. You must extract a minimum of one element.'
-        });
-        check(features, 'Contains an element about flows / workflows').toBe(true);
-        check(features, 'Contains an element about tables').toBe(true);
-        check(features, 'Contains an element about webchat').toBe(true);
-        check(features, 'Contains an element about integrations').toBe(true);
-        check(features, 'Contains an element about HITL (human in the loop)').toBe(true);
-        check(features, 'Contains an element about knowledge bases (KB) or Files API').toBe(true);
-    });
-});
-describe('zai.learn.extract', () => {
-    const client = getClient();
-    let tableName = 'ZaiTestExtractInternalTable';
-    let taskId = 'extract';
-    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 a extraction format from examples', async () => {
-        const adapter = new TableAdapter({
-            client,
-            tableName
-        });
-        const value = await zai.learn(taskId).extract(`I really liked Casino Royale`, z.object({
-            name: z.string(),
-            movie: z.string()
-        }), { instructions: 'extract the name of the movie and name of the main character' });
-        check(value, 'extracted james bond and casino royale').toBe(true);
-        check(value, 'the values are NOT IN ALL CAPS').toBe(true);
-        let rows = await client.findTableRows({ table: tableName });
-        expect(rows.rows.length).toBe(1);
-        await adapter.saveExample({
-            key: 't1',
-            taskId: `zai/${taskId}`,
-            taskType: 'zai.extract',
-            instructions: 'extract name of movie and main character',
-            input: `I went to see the Titanic yesterday and I fell asleep`,
-            output: { name: 'JACK DAWSON', movie: 'TITANIC' },
-            metadata,
-            status: 'approved'
-        });
-        await adapter.saveExample({
-            key: 't2',
-            taskId: `zai/${taskId}`,
-            taskType: 'zai.extract',
-            instructions: 'extract name of movie and main character',
-            input: `Did you know that the gladiator movie has a lot of fighting scenes?`,
-            output: { name: 'MAXIMUS DECIMUS MERIDIUS', movie: 'GLADIATOR' },
-            metadata,
-            status: 'approved'
-        });
-        const second = await zai.learn(taskId).extract(`I really liked Casino Royale`, z.object({
-            name: z.string(),
-            movie: z.string()
-        }), { instructions: 'extract the name of the movie and name of the main character' });
-        expect(second).toMatchInlineSnapshot(`
-      {
-        "movie": "CASINO ROYALE",
-        "name": "JAMES BOND",
-      }
-    `);
-        rows = await client.findTableRows({ table: tableName });
-        expect(rows.rows.length).toBe(3);
-        expect(rows.rows[0].output.value).toMatchObject(second);
-    });
-});
-//# sourceMappingURL=extract.test.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"extract.test.js","sourceRoot":"","sources":["../../../src/operations/extract.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAA;AAE9E,OAAO,EAAE,qBAAqB,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAA;AAC9E,OAAO,EAAE,CAAC,EAAE,MAAM,iBAAiB,CAAA;AACnC,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAA;AACrC,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAA;AAEzD,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,MAAM,GAAG,GAAG,MAAM,EAAE,CAAA;IAEpB,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,OAAO,CAC9B,6DAA6D,EAC7D,CAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;YAChB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;YACf,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;SACrB,CAAC,CACH,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC;;;;;;KAMpC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;QAC1D,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,OAAO,CAC9B;;;;4EAIsE,EACtE,CAAC,CAAC,KAAK,CACL,CAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;YAChB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;YACf,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;SACrB,CAAC,CACH,CACF,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC;;;;;;;;;;;;;;;;;;;;;;;KAuBpC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,OAAO,CAC9B;YACE,MAAM,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,EAAE;SAChD,EACD,CAAC,CAAC,MAAM,CAAC;YACP,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;YACjE,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,mCAAmC,CAAC;SAC5D,CAAC,CACH,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC;;;;;KAKpC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,OAAO,CAChC,qBAAqB,EACrB,CAAC,CAAC,KAAK,CACL,CAAC;aACE,MAAM,CAAC;YACN,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;YACvD,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CAAC,QAAQ,EAAE;YACvE,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gCAAgC,CAAC;SACnE,CAAC;aACD,QAAQ,CAAC,uBAAuB,CAAC,CACrC,EACD;YACE,YAAY,EACV,yHAAyH;SAC5H,CACF,CAAA;QAED,KAAK,CAAC,QAAQ,EAAE,6CAA6C,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACzE,KAAK,CAAC,QAAQ,EAAE,kCAAkC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC9D,KAAK,CAAC,QAAQ,EAAE,mCAAmC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC/D,KAAK,CAAC,QAAQ,EAAE,wCAAwC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACpE,KAAK,CAAC,QAAQ,EAAE,oDAAoD,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAChF,KAAK,CAAC,QAAQ,EAAE,6DAA6D,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC3F,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,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,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,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,CAC3C,8BAA8B,EAC9B,CAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;YAChB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;SAClB,CAAC,EACF,EAAE,YAAY,EAAE,8DAA8D,EAAE,CACjF,CAAA;QAED,KAAK,CAAC,KAAK,EAAE,wCAAwC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACjE,KAAK,CAAC,KAAK,EAAE,gCAAgC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAEzD,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;QAEhC,MAAM,OAAO,CAAC,WAAW,CAAC;YACxB,GAAG,EAAE,IAAI;YACT,MAAM,EAAE,OAAO,MAAM,EAAE;YACvB,QAAQ,EAAE,aAAa;YACvB,YAAY,EAAE,0CAA0C;YACxD,KAAK,EAAE,uDAAuD;YAC9D,MAAM,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,SAAS,EAAE;YACjD,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,0CAA0C;YACxD,KAAK,EAAE,qEAAqE;YAC5E,MAAM,EAAE,EAAE,IAAI,EAAE,0BAA0B,EAAE,KAAK,EAAE,WAAW,EAAE;YAChE,QAAQ;YACR,MAAM,EAAE,UAAU;SACnB,CAAC,CAAA;QAEF,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAC5C,8BAA8B,EAC9B,CAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;YAChB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;SAClB,CAAC,EACF,EAAE,YAAY,EAAE,8DAA8D,EAAE,CACjF,CAAA;QAED,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC;;;;;KAKpC,CAAC,CAAA;QAEF,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;QAChC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC,CAAA;IACzD,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

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

@@ -1,39 +0,0 @@
-import { z } from '@bpinternal/zui';
-export type Options = z.input<typeof Options>;
-declare const Options: import("@bpinternal/zui").ZodObject<{
-    tokensPerItem: import("@bpinternal/zui").ZodDefault<import("@bpinternal/zui").ZodOptional<import("@bpinternal/zui").ZodNumber>>;
-    examples: import("@bpinternal/zui").ZodDefault<import("@bpinternal/zui").ZodArray<import("@bpinternal/zui").ZodObject<{
-        input: import("@bpinternal/zui").ZodAny;
-        filter: import("@bpinternal/zui").ZodBoolean;
-        reason: import("@bpinternal/zui").ZodOptional<import("@bpinternal/zui").ZodString>;
-    }, "strip", import("@bpinternal/zui").ZodTypeAny, {
-        filter: boolean;
-        input?: any;
-        reason?: string | undefined;
-    }, {
-        filter: boolean;
-        input?: any;
-        reason?: string | undefined;
-    }>, "many">>;
-}, "strip", import("@bpinternal/zui").ZodTypeAny, {
-    examples: {
-        filter: boolean;
-        input?: any;
-        reason?: string | undefined;
-    }[];
-    tokensPerItem: number;
-}, {
-    examples?: {
-        filter: boolean;
-        input?: any;
-        reason?: string | undefined;
-    }[] | undefined;
-    tokensPerItem?: number | undefined;
-}>;
-declare module '../zai' {
-    interface Zai {
-        /** Filters elements of an array against a condition */
-        filter<T>(input: Array<T>, condition: string, options?: Options): Promise<Array<T>>;
-    }
-}
-export {};

package/dist/src/operations/filter.js

@@ -1,184 +0,0 @@
-import { z } from '@bpinternal/zui';
-import _ from 'lodash';
-import { fastHash, stringify, takeUntilTokens } from '../utils';
-import { Zai } from '../zai';
-import { PROMPT_INPUT_BUFFER, PROMPT_OUTPUT_BUFFER } from './constants';
-const Example = z.object({
-    input: z.any(),
-    filter: z.boolean(),
-    reason: z.string().optional()
-});
-const Options = z.object({
-    tokensPerItem: z
-        .number()
-        .min(1)
-        .max(100_000)
-        .optional()
-        .describe('The maximum number of tokens per item')
-        .default(250),
-    examples: z.array(Example).describe('Examples to filter the condition against').default([])
-});
-const END = '■END■';
-Zai.prototype.filter = async function (input, condition, _options) {
-    const options = Options.parse(_options ?? {});
-    const tokenizer = await this.getTokenizer();
-    const taskId = this.taskId;
-    const taskType = 'zai.filter';
-    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_INPUT_ARRAY_MAX = TOKENS_TOTAL_MAX - TOKENS_EXAMPLES_MAX - TOKENS_CONDITION_MAX;
-    condition = tokenizer.truncate(condition, TOKENS_CONDITION_MAX);
-    let chunks = [];
-    let currentChunk = [];
-    let currentChunkTokens = 0;
-    for (const element of input) {
-        const elementAsString = tokenizer.truncate(stringify(element, false), options.tokensPerItem);
-        const elementTokens = tokenizer.count(elementAsString);
-        if (currentChunkTokens + elementTokens > TOKENS_INPUT_ARRAY_MAX || currentChunk.length >= MAX_ITEMS_PER_CHUNK) {
-            chunks.push(currentChunk);
-            currentChunk = [];
-            currentChunkTokens = 0;
-        }
-        currentChunk.push(element);
-        currentChunkTokens += elementTokens;
-    }
-    if (currentChunk.length > 0) {
-        chunks.push(currentChunk);
-    }
-    chunks = chunks.filter((x) => x.length > 0);
-    // ■1:true■2:true■3:true
-    const formatInput = (input, condition) => {
-        return `
-Condition to check:
-${condition}
-
-Items (from ■0 to ■${input.length - 1})
-==============================
-${input.map((x, idx) => `■${idx} = ${stringify(x.input ?? null, false)}`).join('\n')}
-`.trim();
-    };
-    const formatExamples = (examples) => {
-        return `
-${examples.map((x, idx) => `■${idx}:${!!x.filter ? 'true' : 'false'}`).join('')}
-${END}
-====
-Here's the reasoning behind each example:
-${examples.map((x, idx) => `■${idx}:${!!x.filter ? 'true' : 'false'}:${x.reason ?? 'No reason provided'}`).join('\n')}
-`.trim();
-    };
-    const genericExamples = [
-        {
-            input: 'apple',
-            filter: true,
-            reason: 'Apples are fruits'
-        },
-        {
-            input: 'Apple Inc.',
-            filter: false,
-            reason: 'Apple Inc. is a company, not a fruit'
-        },
-        {
-            input: 'banana',
-            filter: true,
-            reason: 'Bananas are fruits'
-        },
-        {
-            input: 'potato',
-            filter: false,
-            reason: 'Potatoes are vegetables'
-        }
-    ];
-    const genericExamplesMessages = [
-        {
-            type: 'text',
-            content: formatInput(genericExamples, 'is a fruit'),
-            role: 'user'
-        },
-        {
-            type: 'text',
-            content: formatExamples(genericExamples),
-            role: 'assistant'
-        }
-    ];
-    const filterChunk = async (chunk) => {
-        const examples = taskId
-            ? await this.adapter
-                .getExamples({
-                // The Table API can't search for a huge input string
-                input: JSON.stringify(chunk).slice(0, 1000),
-                taskType,
-                taskId
-            })
-                .then((x) => x.map((y) => ({ filter: y.output, input: y.input, reason: y.explanation })))
-            : [];
-        const allExamples = takeUntilTokens([...examples, ...(options.examples ?? [])], TOKENS_EXAMPLES_MAX, (el) => tokenizer.count(stringify(el.input)));
-        const exampleMessages = [
-            {
-                type: 'text',
-                content: formatInput(allExamples, condition),
-                role: 'user'
-            },
-            {
-                type: 'text',
-                content: formatExamples(allExamples),
-                role: 'assistant'
-            }
-        ];
-        const output = await this.callModel({
-            systemPrompt: `
-You are given a list of items. Your task is to filter out the items that meet the condition below.
-You need to return the full list of items with the format:
-■x:true■y:false■z:true (where x, y, z are the indices of the items in the list)
-You need to start with "■0" and go up to the last index "■${chunk.length - 1}".
-If an item meets the condition, you should return ":true", otherwise ":false".
-
-IMPORTANT: Make sure to read the condition and the examples carefully before making your decision.
-The condition is: "${condition}"
-`.trim(),
-            stopSequences: [END],
-            messages: [
-                ...(exampleMessages.length ? exampleMessages : genericExamplesMessages),
-                {
-                    type: 'text',
-                    content: formatInput(chunk.map((x) => ({ input: x })), condition),
-                    role: 'user'
-                }
-            ]
-        });
-        const answer = output.choices[0].content;
-        const indices = answer
-            .trim()
-            .split('■')
-            .filter((x) => x.length > 0)
-            .map((x) => {
-            const [idx, filter] = x.split(':');
-            return { idx: parseInt(idx.trim()), filter: filter.toLowerCase().trim() === 'true' };
-        });
-        const partial = chunk.filter((_, idx) => {
-            return indices.find((x) => x.idx === idx)?.filter ?? false;
-        });
-        if (taskId) {
-            const key = fastHash(stringify({
-                taskId,
-                taskType,
-                input: JSON.stringify(chunk),
-                condition
-            }));
-            await this.adapter.saveExample({
-                key,
-                taskType,
-                taskId,
-                input: JSON.stringify(chunk),
-                output: partial,
-                instructions: condition,
-                metadata: output.metadata
-            });
-        }
-        return partial;
-    };
-    const filteredChunks = await Promise.all(chunks.map(filterChunk));
-    return filteredChunks.flat();
-};
-//# sourceMappingURL=filter.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"filter.js","sourceRoot":"","sources":["../../../src/operations/filter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,iBAAiB,CAAA;AAEnC,OAAO,CAAC,MAAM,QAAQ,CAAA;AACtB,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,UAAU,CAAA;AAC/D,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,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE;IACd,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE;IACnB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC9B,CAAC,CAAA;AAGF,MAAM,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC;IACvB,aAAa,EAAE,CAAC;SACb,MAAM,EAAE;SACR,GAAG,CAAC,CAAC,CAAC;SACN,GAAG,CAAC,OAAO,CAAC;SACZ,QAAQ,EAAE;SACV,QAAQ,CAAC,uCAAuC,CAAC;SACjD,OAAO,CAAC,GAAG,CAAC;IACf,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,0CAA0C,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;CAC5F,CAAC,CAAA;AASF,MAAM,GAAG,GAAG,OAAO,CAAA;AAEnB,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,KAAK,WAAsB,KAAK,EAAE,SAAS,EAAE,QAAQ;IAC1E,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAA;IAC7C,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAA;IAE3C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;IAC1B,MAAM,QAAQ,GAAG,YAAY,CAAA;IAE7B,MAAM,mBAAmB,GAAG,EAAE,CAAA;IAC9B,MAAM,gBAAgB,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,GAAG,mBAAmB,GAAG,oBAAoB,CAAA;IAChG,MAAM,mBAAmB,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,gBAAgB,GAAG,GAAG,CAAC,CAAC,CAAA;IAC7E,MAAM,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,gBAAgB,GAAG,IAAI,EAAE,GAAG,EAAE,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAA;IAC9F,MAAM,sBAAsB,GAAG,gBAAgB,GAAG,mBAAmB,GAAG,oBAAoB,CAAA;IAE5F,SAAS,GAAG,SAAS,CAAC,QAAQ,CAAC,SAAS,EAAE,oBAAoB,CAAC,CAAA;IAE/D,IAAI,MAAM,GAAwB,EAAE,CAAA;IACpC,IAAI,YAAY,GAAiB,EAAE,CAAA;IACnC,IAAI,kBAAkB,GAAG,CAAC,CAAA;IAE1B,KAAK,MAAM,OAAO,IAAI,KAAK,EAAE,CAAC;QAC5B,MAAM,eAAe,GAAG,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,CAAA;QAC5F,MAAM,aAAa,GAAG,SAAS,CAAC,KAAK,CAAC,eAAe,CAAC,CAAA;QAEtD,IAAI,kBAAkB,GAAG,aAAa,GAAG,sBAAsB,IAAI,YAAY,CAAC,MAAM,IAAI,mBAAmB,EAAE,CAAC;YAC9G,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;YACzB,YAAY,GAAG,EAAE,CAAA;YACjB,kBAAkB,GAAG,CAAC,CAAA;QACxB,CAAC;QAED,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC1B,kBAAkB,IAAI,aAAa,CAAA;IACrC,CAAC;IAED,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5B,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAA;IAC3B,CAAC;IAED,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;IAE3C,wBAAwB;IAExB,MAAM,WAAW,GAAG,CAAC,KAAgB,EAAE,SAAiB,EAAE,EAAE;QAC1D,OAAO;;EAET,SAAS;;qBAEU,KAAK,CAAC,MAAM,GAAG,CAAC;;EAEnC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,GAAG,MAAM,SAAS,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;CACnF,CAAC,IAAI,EAAE,CAAA;IACN,CAAC,CAAA;IAED,MAAM,cAAc,GAAG,CAAC,QAAmB,EAAE,EAAE;QAC7C,OAAO;EACT,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC;EAC7E,GAAG;;;EAGH,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,CAAC,MAAM,IAAI,oBAAoB,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;CACpH,CAAC,IAAI,EAAE,CAAA;IACN,CAAC,CAAA;IAED,MAAM,eAAe,GAAc;QACjC;YACE,KAAK,EAAE,OAAO;YACd,MAAM,EAAE,IAAI;YACZ,MAAM,EAAE,mBAAmB;SAC5B;QACD;YACE,KAAK,EAAE,YAAY;YACnB,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,sCAAsC;SAC/C;QACD;YACE,KAAK,EAAE,QAAQ;YACf,MAAM,EAAE,IAAI;YACZ,MAAM,EAAE,oBAAoB;SAC7B;QACD;YACE,KAAK,EAAE,QAAQ;YACf,MAAM,EAAE,KAAK;YACb,MAAM,EAAE,yBAAyB;SAClC;KACF,CAAA;IAED,MAAM,uBAAuB,GAAG;QAC9B;YACE,IAAI,EAAE,MAAe;YACrB,OAAO,EAAE,WAAW,CAAC,eAAe,EAAE,YAAY,CAAC;YACnD,IAAI,EAAE,MAAe;SACtB;QACD;YACE,IAAI,EAAE,MAAe;YACrB,OAAO,EAAE,cAAc,CAAC,eAAe,CAAC;YACxC,IAAI,EAAE,WAAoB;SAC3B;KACF,CAAA;IAED,MAAM,WAAW,GAAG,KAAK,EAAE,KAAmB,EAAE,EAAE;QAChD,MAAM,QAAQ,GAAG,MAAM;YACrB,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO;iBACf,WAAW,CAAkB;gBAC5B,qDAAqD;gBACrD,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC;gBAC3C,QAAQ;gBACR,MAAM;aACP,CAAC;iBACD,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CACV,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAiB,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,WAAW,EAAE,CAAmB,CAAC,CACzG;YACL,CAAC,CAAC,EAAE,CAAA;QAEN,MAAM,WAAW,GAAG,eAAe,CAAC,CAAC,GAAG,QAAQ,EAAE,GAAG,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,EAAE,mBAAmB,EAAE,CAAC,EAAE,EAAE,EAAE,CAC1G,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CACrC,CAAA;QAED,MAAM,eAAe,GAAG;YACtB;gBACE,IAAI,EAAE,MAAe;gBACrB,OAAO,EAAE,WAAW,CAAC,WAAW,EAAE,SAAS,CAAC;gBAC5C,IAAI,EAAE,MAAe;aACtB;YACD;gBACE,IAAI,EAAE,MAAe;gBACrB,OAAO,EAAE,cAAc,CAAC,WAAW,CAAC;gBACpC,IAAI,EAAE,WAAoB;aAC3B;SACF,CAAA;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC;YAClC,YAAY,EAAE;;;;4DAIwC,KAAK,CAAC,MAAM,GAAG,CAAC;;;;qBAIvD,SAAS;CAC7B,CAAC,IAAI,EAAE;YACF,aAAa,EAAE,CAAC,GAAG,CAAC;YACpB,QAAQ,EAAE;gBACR,GAAG,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,uBAAuB,CAAC;gBACvE;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,WAAW,CAClB,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAY,CAAC,EAC3C,SAAS,CACV;oBACD,IAAI,EAAE,MAAM;iBACb;aACF;SACF,CAAC,CAAA;QAEF,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAiB,CAAA;QAClD,MAAM,OAAO,GAAG,MAAM;aACnB,IAAI,EAAE;aACN,KAAK,CAAC,GAAG,CAAC;aACV,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;aAC3B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACT,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;YAClC,OAAO,EAAE,GAAG,EAAE,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,KAAK,MAAM,EAAE,CAAA;QACtF,CAAC,CAAC,CAAA;QAEJ,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE;YACtC,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,EAAE,MAAM,IAAI,KAAK,CAAA;QAC5D,CAAC,CAAC,CAAA;QAEF,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,GAAG,GAAG,QAAQ,CAClB,SAAS,CAAC;gBACR,MAAM;gBACN,QAAQ;gBACR,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;gBAC5B,SAAS;aACV,CAAC,CACH,CAAA;YAED,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC;gBAC7B,GAAG;gBACH,QAAQ;gBACR,MAAM;gBACN,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC;gBAC5B,MAAM,EAAE,OAAO;gBACf,YAAY,EAAE,SAAS;gBACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ;aAC1B,CAAC,CAAA;QACJ,CAAC;QAED,OAAO,OAAO,CAAA;IAChB,CAAC,CAAA;IAED,MAAM,cAAc,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAA;IAEjE,OAAO,cAAc,CAAC,IAAI,EAAE,CAAA;AAC9B,CAAC,CAAA"}

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

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

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

@@ -1,151 +0,0 @@
-import { describe, it, expect, beforeEach, afterEach, afterAll, vi } from 'vitest';
-import { getClient, getZai, metadata } from './__tests';
-import { TableAdapter } from '../adapters/botpress-table';
-describe('zai.filter', { timeout: 60_000 }, () => {
-    let zai = getZai();
-    beforeEach(async () => {
-        zai = getZai();
-    });
-    it('basic filter with small items', async () => {
-        const value = await zai.filter([
-            { name: 'John', description: 'is a bad person' },
-            { name: 'Alice', description: 'is a good person' },
-            { name: 'Bob', description: 'is a good person' },
-            { name: 'Eve', description: 'is a bad person' },
-            { name: 'Alex', description: 'is a good person' },
-            { name: 'Sara', description: 'donates to charity every month' },
-            { name: 'Tom', description: 'commits crimes and is in jail' }
-        ], 'generally good people');
-        const names = value.map((v) => v.name);
-        expect(names).toMatchInlineSnapshot(`
-      [
-        "Alice",
-        "Bob",
-        "Alex",
-        "Sara",
-      ]
-    `);
-    });
-    it('filtering huge array chunks it up', async () => {
-        const callAction = vi.fn();
-        const client = { ...getClient(), callAction };
-        zai = getZai().with({
-            client
-        });
-        const hugeArray = Array.from({ length: 100 }, (_, i) => ({
-            name: `Person #${i}#`,
-            description: 'blah blah '.repeat(50_000)
-        }));
-        try {
-            await zai.filter(hugeArray, 'generally good people', { tokensPerItem: 100_000 });
-        }
-        catch (err) { }
-        expect(callAction.mock.calls.length).toBeGreaterThan(20);
-        expect(JSON.stringify(callAction.mock.calls.at(0))).toContain('Person #0#');
-        expect(JSON.stringify(callAction.mock.calls.at(0))).not.toContain('Person #99#');
-        expect(JSON.stringify(callAction.mock.calls.at(-1))).not.toContain('Person #0#');
-        expect(JSON.stringify(callAction.mock.calls.at(-1))).toContain('Person #99#');
-        callAction.mockReset();
-        try {
-            await zai.filter(hugeArray, 'generally good people', { tokensPerItem: 100 });
-        }
-        catch (err) { }
-        expect(callAction.mock.calls.length).toBe(2);
-    });
-    it('filter with examples', async () => {
-        const examples = [
-            {
-                input: 'Rasa (framework)',
-                filter: true,
-                reason: 'Rasa is a chatbot framework, so it competes with us (Botpress).'
-            },
-            {
-                input: 'Rasa (coffee company)',
-                filter: false,
-                reason: 'Rasa (coffee company) is not in the chatbot or AI agent industry, therefore it does not compete with us (Botpress).'
-            },
-            {
-                input: 'Dialogflow',
-                filter: true,
-                reason: 'Dialogflow is a chatbot development product, so it competes with us (Botpress).'
-            }
-        ];
-        const value = await zai.filter([{ name: 'Moveworks' }, { name: 'Ada.cx' }, { name: 'Nike' }, { name: 'Voiceflow' }, { name: 'Adidas' }], 'competes with us', { examples });
-        const names = value.map((v) => v.name);
-        expect(names).toMatchInlineSnapshot(`
-      [
-        "Moveworks",
-        "Ada.cx",
-        "Voiceflow",
-      ]
-    `);
-    });
-});
-describe('zai.learn.filter', { timeout: 60_000 }, () => {
-    const client = getClient();
-    let tableName = 'ZaiTestFilterInternalTable';
-    let taskId = 'filter';
-    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 a filtering rule from examples', async () => {
-        const adapter = new TableAdapter({
-            client,
-            tableName
-        });
-        await adapter.saveExample({
-            key: 't1',
-            taskId: `zai/${taskId}`,
-            taskType: 'zai.filter',
-            instructions: 'competes with us?',
-            input: ['Rasa (framework)', 'Rasa (coffee company)'],
-            output: ['Rasa (framework)'],
-            explanation: `Rasa is a chatbot framework, so it competes with us (Botpress). We should keep it. Rasa (coffee company) is not in the chatbot or AI agent industry, therefore it does not compete with us (Botpress). We should filter it out.`,
-            metadata,
-            status: 'approved'
-        });
-        await adapter.saveExample({
-            key: 't2',
-            taskId: `zai/${taskId}`,
-            taskType: 'zai.filter',
-            instructions: 'competes with us?',
-            input: ['Voiceflow', 'Dialogflow'],
-            output: ['Voiceflow', 'Dialogflow'],
-            explanation: `Voiceflow is a chatbot development product, so it competes with us (Botpress). We should keep it. Dialogflow is a chatbot development product, so it competes with us (Botpress). We should keep it.`,
-            metadata,
-            status: 'approved'
-        });
-        const second = await zai
-            .learn(taskId)
-            .filter(['Nike', 'Ada.cx', 'Adidas', 'Moveworks', 'Lululemon'], 'competes with us? (botpress)');
-        expect(second).toMatchInlineSnapshot(`
-      [
-        "Ada.cx",
-        "Moveworks",
-      ]
-    `);
-        const rows = await client.findTableRows({ table: tableName });
-        expect(rows.rows.length).toBe(3);
-        expect(rows.rows.at(-1).output.value).toEqual(second);
-    });
-});
-//# sourceMappingURL=filter.test.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"filter.test.js","sourceRoot":"","sources":["../../../src/operations/filter.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAElF,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAA;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAA;AAGzD,QAAQ,CAAC,YAAY,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE;IAC/C,IAAI,GAAG,GAAG,MAAM,EAAE,CAAA;IAElB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,GAAG,GAAG,MAAM,EAAE,CAAA;IAChB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;QAC7C,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,MAAM,CAC5B;YACE,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,iBAAiB,EAAE;YAChD,EAAE,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,kBAAkB,EAAE;YAClD,EAAE,IAAI,EAAE,KAAK,EAAE,WAAW,EAAE,kBAAkB,EAAE;YAChD,EAAE,IAAI,EAAE,KAAK,EAAE,WAAW,EAAE,iBAAiB,EAAE;YAC/C,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,kBAAkB,EAAE;YACjD,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,gCAAgC,EAAE;YAC/D,EAAE,IAAI,EAAE,KAAK,EAAE,WAAW,EAAE,+BAA+B,EAAE;SAC9D,EACD,uBAAuB,CACxB,CAAA;QAED,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAA;QACtC,MAAM,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC;;;;;;;KAOnC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,EAAE,CAAA;QAC1B,MAAM,MAAM,GAAG,EAAE,GAAG,SAAS,EAAE,EAAE,UAAU,EAAuB,CAAA;QAElE,GAAG,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC;YAClB,MAAM;SACP,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACvD,IAAI,EAAE,WAAW,CAAC,GAAG;YACrB,WAAW,EAAE,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC;SACzC,CAAC,CAAC,CAAA;QAEH,IAAI,CAAC;YACH,MAAM,GAAG,CAAC,MAAM,CAAC,SAAS,EAAE,uBAAuB,EAAE,EAAE,aAAa,EAAE,OAAO,EAAE,CAAC,CAAA;QAClF,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC,CAAA,CAAC;QAEhB,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,EAAE,CAAC,CAAA;QACxD,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,YAAY,CAAC,CAAA;QAC3E,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,aAAa,CAAC,CAAA;QAEhF,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,YAAY,CAAC,CAAA;QAChF,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAA;QAE7E,UAAU,CAAC,SAAS,EAAE,CAAA;QAEtB,IAAI,CAAC;YACH,MAAM,GAAG,CAAC,MAAM,CAAC,SAAS,EAAE,uBAAuB,EAAE,EAAE,aAAa,EAAE,GAAG,EAAE,CAAC,CAAA;QAC9E,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC,CAAA,CAAC;QAEhB,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;IAC9C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACpC,MAAM,QAAQ,GAAG;YACf;gBACE,KAAK,EAAE,kBAAkB;gBACzB,MAAM,EAAE,IAAI;gBACZ,MAAM,EAAE,iEAAiE;aAC1E;YACD;gBACE,KAAK,EAAE,uBAAuB;gBAC9B,MAAM,EAAE,KAAK;gBACb,MAAM,EACJ,qHAAqH;aACxH;YACD;gBACE,KAAK,EAAE,YAAY;gBACnB,MAAM,EAAE,IAAI;gBACZ,MAAM,EAAE,iFAAiF;aAC1F;SACF,CAAA;QAED,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,MAAM,CAC5B,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EACxG,kBAAkB,EAClB,EAAE,QAAQ,EAAE,CACb,CAAA;QAED,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAA;QACtC,MAAM,CAAC,KAAK,CAAC,CAAC,qBAAqB,CAAC;;;;;;KAMnC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,kBAAkB,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE;IACrD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAA;IAC1B,IAAI,SAAS,GAAG,4BAA4B,CAAA;IAC5C,IAAI,MAAM,GAAG,QAAQ,CAAA;IACrB,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,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC;YAC/B,MAAM;YACN,SAAS;SACV,CAAC,CAAA;QAEF,MAAM,OAAO,CAAC,WAAW,CAAC;YACxB,GAAG,EAAE,IAAI;YACT,MAAM,EAAE,OAAO,MAAM,EAAE;YACvB,QAAQ,EAAE,YAAY;YACtB,YAAY,EAAE,mBAAmB;YACjC,KAAK,EAAE,CAAC,kBAAkB,EAAE,uBAAuB,CAAC;YACpD,MAAM,EAAE,CAAC,kBAAkB,CAAC;YAC5B,WAAW,EAAE,iOAAiO;YAC9O,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,YAAY;YACtB,YAAY,EAAE,mBAAmB;YACjC,KAAK,EAAE,CAAC,WAAW,EAAE,YAAY,CAAC;YAClC,MAAM,EAAE,CAAC,WAAW,EAAE,YAAY,CAAC;YACnC,WAAW,EAAE,sMAAsM;YACnN,QAAQ;YACR,MAAM,EAAE,UAAU;SACnB,CAAC,CAAA;QAEF,MAAM,MAAM,GAAG,MAAM,GAAG;aACrB,KAAK,CAAC,MAAM,CAAC;aACb,MAAM,CAAC,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,WAAW,EAAE,WAAW,CAAC,EAAE,8BAA8B,CAAC,CAAA;QAEjG,MAAM,CAAC,MAAM,CAAC,CAAC,qBAAqB,CAAC;;;;;KAKpC,CAAC,CAAA;QAEF,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAA;QAC7D,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAChC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAA;IACxD,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

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

@@ -1,79 +0,0 @@
-import { z } from '@bpinternal/zui';
-type Label = keyof typeof LABELS;
-declare const LABELS: {
-    readonly ABSOLUTELY_NOT: "ABSOLUTELY_NOT";
-    readonly PROBABLY_NOT: "PROBABLY_NOT";
-    readonly AMBIGUOUS: "AMBIGUOUS";
-    readonly PROBABLY_YES: "PROBABLY_YES";
-    readonly ABSOLUTELY_YES: "ABSOLUTELY_YES";
-};
-type Example<T extends string> = {
-    input: unknown;
-    labels: Partial<Record<T, {
-        label: Label;
-        explanation?: string;
-    }>>;
-};
-export type Options<T extends string> = Omit<z.input<typeof Options>, 'examples'> & {
-    examples?: Array<Partial<Example<T>>>;
-};
-declare const Options: import("@bpinternal/zui").ZodObject<{
-    examples: import("@bpinternal/zui").ZodDefault<import("@bpinternal/zui").ZodArray<import("@bpinternal/zui").ZodObject<{
-        input: import("@bpinternal/zui").ZodAny;
-        labels: import("@bpinternal/zui").ZodRecord<import("@bpinternal/zui").ZodString, import("@bpinternal/zui").ZodObject<{
-            label: import("@bpinternal/zui").ZodEnum<never>;
-            explanation: import("@bpinternal/zui").ZodOptional<import("@bpinternal/zui").ZodString>;
-        }, "strip", import("@bpinternal/zui").ZodTypeAny, {
-            label: never;
-            explanation?: string | undefined;
-        }, {
-            label: never;
-            explanation?: string | undefined;
-        }>>;
-    }, "strip", import("@bpinternal/zui").ZodTypeAny, {
-        labels: Record<string, {
-            label: never;
-            explanation?: string | undefined;
-        }>;
-        input?: any;
-    }, {
-        labels: Record<string, {
-            label: never;
-            explanation?: string | undefined;
-        }>;
-        input?: any;
-    }>, "many">>;
-    instructions: import("@bpinternal/zui").ZodOptional<import("@bpinternal/zui").ZodString>;
-    chunkLength: import("@bpinternal/zui").ZodDefault<import("@bpinternal/zui").ZodOptional<import("@bpinternal/zui").ZodNumber>>;
-}, "strip", import("@bpinternal/zui").ZodTypeAny, {
-    examples: {
-        labels: Record<string, {
-            label: never;
-            explanation?: string | undefined;
-        }>;
-        input?: any;
-    }[];
-    chunkLength: number;
-    instructions?: string | undefined;
-}, {
-    instructions?: string | undefined;
-    examples?: {
-        labels: Record<string, {
-            label: never;
-            explanation?: string | undefined;
-        }>;
-        input?: any;
-    }[] | undefined;
-    chunkLength?: number | undefined;
-}>;
-type Labels<T extends string> = Record<T, string>;
-declare const Labels: z.ZodTransformer<import("@bpinternal/zui").ZodRecord<import("@bpinternal/zui").ZodString, import("@bpinternal/zui").ZodString>, Record<string, string>, Record<string, string>>;
-declare module '../zai' {
-    interface Zai {
-        /** Tags the provided input with a list of predefined labels */
-        label<T extends string>(input: unknown, labels: Labels<T>, options?: Options<T>): Promise<{
-            [K in T]: boolean;
-        }>;
-    }
-}
-export {};