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

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

@@ -1,36 +0,0 @@
-import { z } from '@bpinternal/zui';
-export type Options = z.input<typeof Options>;
-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;
-        check: import("@bpinternal/zui").ZodBoolean;
-        reason: import("@bpinternal/zui").ZodOptional<import("@bpinternal/zui").ZodString>;
-    }, "strip", import("@bpinternal/zui").ZodTypeAny, {
-        check: boolean;
-        input?: any;
-        reason?: string | undefined;
-    }, {
-        check: boolean;
-        input?: any;
-        reason?: string | undefined;
-    }>, "many">>;
-}, "strip", import("@bpinternal/zui").ZodTypeAny, {
-    examples: {
-        check: boolean;
-        input?: any;
-        reason?: string | undefined;
-    }[];
-}, {
-    examples?: {
-        check: boolean;
-        input?: any;
-        reason?: string | undefined;
-    }[] | undefined;
-}>;
-declare module '../zai' {
-    interface Zai {
-        /** Checks wether a condition is true or not */
-        check(input: unknown, condition: string, options?: Options): Promise<boolean>;
-    }
-}
-export {};

package/dist/src/operations/check.js

@@ -1,146 +0,0 @@
-import { z } from '@bpinternal/zui';
-import { fastHash, stringify, takeUntilTokens } from '../utils';
-import { Zai } from '../zai';
-import { PROMPT_INPUT_BUFFER } from './constants';
-const Example = z.object({
-    input: z.any(),
-    check: z.boolean(),
-    reason: z.string().optional()
-});
-const Options = z.object({
-    examples: z.array(Example).describe('Examples to check the condition against').default([])
-});
-const TRUE = '■TRUE■';
-const FALSE = '■FALSE■';
-const END = '■END■';
-Zai.prototype.check = async function (input, condition, _options) {
-    const options = Options.parse(_options ?? {});
-    const tokenizer = await this.getTokenizer();
-    const PROMPT_COMPONENT = Math.max(this.Model.input.maxTokens - PROMPT_INPUT_BUFFER, 100);
-    const taskId = this.taskId;
-    const taskType = 'zai.check';
-    const PROMPT_TOKENS = {
-        INPUT: Math.floor(0.5 * PROMPT_COMPONENT),
-        CONDITION: Math.floor(0.2 * PROMPT_COMPONENT)
-    };
-    // Truncate the input to fit the model's input size
-    const inputAsString = tokenizer.truncate(stringify(input), PROMPT_TOKENS.INPUT);
-    condition = tokenizer.truncate(condition, PROMPT_TOKENS.CONDITION);
-    // All tokens remaining after the input and condition are accounted can be used for examples
-    const EXAMPLES_TOKENS = PROMPT_COMPONENT - tokenizer.count(inputAsString) - tokenizer.count(condition);
-    const Key = fastHash(JSON.stringify({
-        taskType,
-        taskId,
-        input: inputAsString,
-        condition
-    }));
-    const examples = taskId
-        ? await this.adapter.getExamples({
-            input: inputAsString,
-            taskType,
-            taskId
-        })
-        : [];
-    const exactMatch = examples.find((x) => x.key === Key);
-    if (exactMatch) {
-        return exactMatch.output;
-    }
-    const defaultExamples = [
-        { input: '50 Cent', check: true, reason: '50 Cent is widely recognized as a public personality.' },
-        {
-            input: ['apple', 'banana', 'carrot', 'house'],
-            check: false,
-            reason: 'The list contains a house, which is not a fruit. Also, the list contains a carrot, which is a vegetable.'
-        }
-    ];
-    const userExamples = [
-        ...examples.map((e) => ({ input: e.input, check: e.output, reason: e.explanation })),
-        ...options.examples
-    ];
-    let exampleId = 1;
-    const formatInput = (input, condition) => {
-        const header = userExamples.length ? `Expert Example #${exampleId++}` : `Example of condition: "${condition}"`;
-        return `
-${header}
-<|start_input|>
-${input.trim()}
-<|end_input|>
-`.trim();
-    };
-    const formatOutput = (answer, justification) => {
-        return `
-Analysis: ${justification}
-Final Answer: ${answer ? TRUE : FALSE}
-${END}
-`.trim();
-    };
-    const formatExample = (example) => [
-        { type: 'text', content: formatInput(stringify(example.input ?? null), condition), role: 'user' },
-        {
-            type: 'text',
-            content: formatOutput(example.check, example.reason ?? ''),
-            role: 'assistant'
-        }
-    ];
-    const allExamples = takeUntilTokens(userExamples.length ? userExamples : defaultExamples, EXAMPLES_TOKENS, (el) => tokenizer.count(stringify(el.input)) + tokenizer.count(el.reason ?? ''))
-        .map(formatExample)
-        .flat();
-    const specialInstructions = userExamples.length
-        ? `
-- You have been provided with examples from previous experts. Make sure to read them carefully before making your decision.
-- Make sure to refer to the examples provided by the experts to justify your decision (when applicable).
-- When in doubt, ground your decision on the examples provided by the experts instead of your own intuition.
-- When no example is similar to the input, make sure to provide a clear justification for your decision while inferring the decision-making process from the examples provided by the experts.
-`.trim()
-        : '';
-    const output = await this.callModel({
-        systemPrompt: `
-Check if the following condition is true or false for the given input. Before answering, make sure to read the input and the condition carefully.
-Justify your answer, then answer with either ${TRUE} or ${FALSE} at the very end, then add ${END} to finish the response.
-IMPORTANT: Make sure to answer with either ${TRUE} or ${FALSE} at the end of your response, but NOT both.
----
-Expert Examples (#1 to #${exampleId - 1}):
-${specialInstructions}
-`.trim(),
-        stopSequences: [END],
-        messages: [
-            ...allExamples,
-            {
-                type: 'text',
-                content: `
-Considering the below input and above examples, is the following condition true or false?
-${formatInput(inputAsString, condition)}
-In your "Analysis", please refer to the Expert Examples # to justify your decision.`.trim(),
-                role: 'user'
-            }
-        ]
-    });
-    const answer = output.choices[0].content;
-    const hasTrue = answer.includes(TRUE);
-    const hasFalse = answer.includes(FALSE);
-    if (!hasTrue && !hasFalse) {
-        throw new Error(`The model did not return a valid answer. The response was: ${answer}`);
-    }
-    let finalAnswer;
-    if (hasTrue && hasFalse) {
-        // If both TRUE and FALSE are present, we need to check which one was answered last
-        finalAnswer = answer.lastIndexOf(TRUE) > answer.lastIndexOf(FALSE);
-    }
-    else {
-        finalAnswer = hasTrue;
-    }
-    if (taskId) {
-        await this.adapter.saveExample({
-            key: Key,
-            taskType,
-            taskId,
-            input: inputAsString,
-            instructions: condition,
-            metadata: output.metadata,
-            output: finalAnswer,
-            explanation: answer.replace(TRUE, '').replace(FALSE, '').replace(END, '').replace('Final Answer:', '').trim()
-        });
-    }
-    return finalAnswer;
-};
-//# sourceMappingURL=check.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"check.js","sourceRoot":"","sources":["../../../src/operations/check.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,iBAAiB,CAAA;AAEnC,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,MAAM,aAAa,CAAA;AAEjD,MAAM,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC;IACvB,KAAK,EAAE,CAAC,CAAC,GAAG,EAAE;IACd,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE;IAClB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC9B,CAAC,CAAA;AAGF,MAAM,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC;IACvB,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,yCAAyC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;CAC3F,CAAC,CAAA;AASF,MAAM,IAAI,GAAG,QAAQ,CAAA;AACrB,MAAM,KAAK,GAAG,SAAS,CAAA;AACvB,MAAM,GAAG,GAAG,OAAO,CAAA;AAEnB,GAAG,CAAC,SAAS,CAAC,KAAK,GAAG,KAAK,WAAsB,KAAK,EAAE,SAAS,EAAE,QAAQ;IACzE,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAA;IAC7C,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAA;IAC3C,MAAM,gBAAgB,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,GAAG,mBAAmB,EAAE,GAAG,CAAC,CAAA;IAExF,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAA;IAC1B,MAAM,QAAQ,GAAG,WAAW,CAAA;IAE5B,MAAM,aAAa,GAAG;QACpB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,GAAG,gBAAgB,CAAC;QACzC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,GAAG,gBAAgB,CAAC;KAC9C,CAAA;IAED,mDAAmD;IACnD,MAAM,aAAa,GAAG,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,aAAa,CAAC,KAAK,CAAC,CAAA;IAC/E,SAAS,GAAG,SAAS,CAAC,QAAQ,CAAC,SAAS,EAAE,aAAa,CAAC,SAAS,CAAC,CAAA;IAElE,4FAA4F;IAC5F,MAAM,eAAe,GAAG,gBAAgB,GAAG,SAAS,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC,CAAA;IAEtG,MAAM,GAAG,GAAG,QAAQ,CAClB,IAAI,CAAC,SAAS,CAAC;QACb,QAAQ;QACR,MAAM;QACN,KAAK,EAAE,aAAa;QACpB,SAAS;KACV,CAAC,CACH,CAAA;IAED,MAAM,QAAQ,GAAG,MAAM;QACrB,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAkB;YAC9C,KAAK,EAAE,aAAa;YACpB,QAAQ;YACR,MAAM;SACP,CAAC;QACJ,CAAC,CAAC,EAAE,CAAA;IAEN,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAA;IACtD,IAAI,UAAU,EAAE,CAAC;QACf,OAAO,UAAU,CAAC,MAAM,CAAA;IAC1B,CAAC;IAED,MAAM,eAAe,GAAG;QACtB,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,uDAAuD,EAAE;QAClG;YACE,KAAK,EAAE,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE,OAAO,CAAC;YAC7C,KAAK,EAAE,KAAK;YACZ,MAAM,EAAE,0GAA0G;SACnH;KACF,CAAA;IAED,MAAM,YAAY,GAAG;QACnB,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;QACpF,GAAG,OAAO,CAAC,QAAQ;KACpB,CAAA;IAED,IAAI,SAAS,GAAG,CAAC,CAAA;IAEjB,MAAM,WAAW,GAAG,CAAC,KAAa,EAAE,SAAiB,EAAE,EAAE;QACvD,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,mBAAmB,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC,0BAA0B,SAAS,GAAG,CAAA;QAE9G,OAAO;EACT,MAAM;;EAEN,KAAK,CAAC,IAAI,EAAE;;CAEb,CAAC,IAAI,EAAE,CAAA;IACN,CAAC,CAAA;IAED,MAAM,YAAY,GAAG,CAAC,MAAe,EAAE,aAAqB,EAAE,EAAE;QAC9D,OAAO;YACC,aAAa;gBACT,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK;EACnC,GAAG;CACJ,CAAC,IAAI,EAAE,CAAA;IACN,CAAC,CAAA;IAED,MAAM,aAAa,GAAG,CAAC,OAAyD,EAAE,EAAE,CAAC;QACnF,EAAE,IAAI,EAAE,MAAe,EAAE,OAAO,EAAE,WAAW,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,EAAE,SAAS,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE;QACnH;YACE,IAAI,EAAE,MAAe;YACrB,OAAO,EAAE,YAAY,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,MAAM,IAAI,EAAE,CAAC;YAC1D,IAAI,EAAE,WAAoB;SAC3B;KACF,CAAA;IAED,MAAM,WAAW,GAAG,eAAe,CACjC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,eAAe,EACpD,eAAe,EACf,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,IAAI,EAAE,CAAC,CAChF;SACE,GAAG,CAAC,aAAa,CAAC;SAClB,IAAI,EAAE,CAAA;IAET,MAAM,mBAAmB,GAAG,YAAY,CAAC,MAAM;QAC7C,CAAC,CAAC;;;;;CAKL,CAAC,IAAI,EAAE;QACJ,CAAC,CAAC,EAAE,CAAA;IAEN,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC;QAClC,YAAY,EAAE;;+CAE6B,IAAI,OAAO,KAAK,8BAA8B,GAAG;6CACnD,IAAI,OAAO,KAAK;;0BAEnC,SAAS,GAAG,CAAC;EACrC,mBAAmB;CACpB,CAAC,IAAI,EAAE;QACJ,aAAa,EAAE,CAAC,GAAG,CAAC;QACpB,QAAQ,EAAE;YACR,GAAG,WAAW;YACd;gBACE,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE;;EAEf,WAAW,CAAC,aAAa,EAAE,SAAS,CAAC;oFAC6C,CAAC,IAAI,EAAE;gBACnF,IAAI,EAAE,MAAM;aACb;SACF;KACF,CAAC,CAAA;IAEF,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAiB,CAAA;IAElD,MAAM,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAA;IACrC,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAA;IAEvC,IAAI,CAAC,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,8DAA8D,MAAM,EAAE,CAAC,CAAA;IACzF,CAAC;IAED,IAAI,WAAoB,CAAA;IAExB,IAAI,OAAO,IAAI,QAAQ,EAAE,CAAC;QACxB,mFAAmF;QACnF,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAA;IACpE,CAAC;SAAM,CAAC;QACN,WAAW,GAAG,OAAO,CAAA;IACvB,CAAC;IAED,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC;YAC7B,GAAG,EAAE,GAAG;YACR,QAAQ;YACR,MAAM;YACN,KAAK,EAAE,aAAa;YACpB,YAAY,EAAE,SAAS;YACvB,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,MAAM,EAAE,WAAW;YACnB,WAAW,EAAE,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE;SAC9G,CAAC,CAAA;IACJ,CAAC;IAED,OAAO,WAAW,CAAA;AACpB,CAAC,CAAA"}

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

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

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

@@ -1 +0,0 @@
-{"version":3,"file":"check.test.js","sourceRoot":"","sources":["../../../src/operations/check.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAA;AAE9E,OAAO,EAAE,qBAAqB,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAA;AAE9E,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAA;AAEzD,QAAQ,CAAC,WAAW,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE;IAC9C,MAAM,GAAG,GAAG,MAAM,EAAE,CAAA;IAEpB,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,+CAA+C,EAAE,wBAAwB,CAAC,CAAA;QACxG,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAC1B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,uBAAuB,GAAG,MAAM,GAAG,CAAC,KAAK,CAC7C,qBAAqB,EACrB,gDAAgD,CACjD,CAAA;QAED,MAAM,YAAY,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,qBAAqB,EAAE,yCAAyC,CAAC,CAAA;QAEtG,MAAM,CAAC,uBAAuB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC1C,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IAClC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,GAAG,GAAG,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAA;QACpG,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,+BAA+B,CAAC,CAAA;QACtE,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,GAAG,EAAE,wBAAwB,CAAC,CAAA;QAE/D,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC3B,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IAC9B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;QACnC,MAAM,QAAQ,GAAG;YACf;gBACE,KAAK,EAAE,kBAAkB;gBACzB,KAAK,EAAE,IAAI;gBACX,MAAM,EAAE,iEAAiE;aAC1E;YACD;gBACE,KAAK,EAAE,uBAAuB;gBAC9B,KAAK,EAAE,KAAK;gBACZ,MAAM,EACJ,qHAAqH;aACxH;YACD;gBACE,KAAK,EAAE,YAAY;gBACnB,KAAK,EAAE,IAAI;gBACX,MAAM,EAAE,iFAAiF;aAC1F;SACF,CAAA;QAED,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,kBAAkB,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAA;QAChF,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,kBAAkB,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAA;QACvE,MAAM,SAAS,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,kBAAkB,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAA;QAChF,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,MAAM,EAAE,kBAAkB,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAA;QACtE,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,QAAQ,EAAE,kBAAkB,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAA;QAE1E,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC5B,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACtB,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAE5B,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QACxB,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;IAC5B,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA;AAEF,QAAQ,CAAC,iBAAiB,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,GAAG,EAAE;IACpD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAA;IAC1B,IAAI,SAAS,GAAG,2BAA2B,CAAA;IAC3C,IAAI,MAAM,GAAG,OAAO,CAAA;IACpB,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,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,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,KAAK,CAAC,WAAW,EAAE,eAAe,CAAC,CAAA;QACzE,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAExB,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,OAAO,CAAC,KAAK,CAAC,CAAA;QAEhD,MAAM,OAAO,CAAC,WAAW,CAAC;YACxB,GAAG,EAAE,IAAI;YACT,MAAM,EAAE,OAAO,MAAM,EAAE;YACvB,QAAQ,EAAE,WAAW;YACrB,YAAY,EAAE,eAAe;YAC7B,KAAK,EAAE,YAAY;YACnB,MAAM,EAAE,KAAK;YACb,WAAW,EAAE,8EAA8E;YAC3F,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,WAAW;YACrB,YAAY,EAAE,eAAe;YAC7B,KAAK,EAAE,qBAAqB;YAC5B,MAAM,EAAE,IAAI;YACZ,WAAW,EAAE,2CAA2C;YACxD,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,WAAW;YACrB,YAAY,EAAE,eAAe;YAC7B,KAAK,EAAE,QAAQ;YACf,MAAM,EAAE,KAAK;YACb,WAAW,EAAE,2GAA2G;YACxH,QAAQ;YACR,MAAM,EAAE,UAAU;SACnB,CAAC,CAAA;QAEF,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,WAAW,EAAE,eAAe,CAAC,CAAA;QAC1E,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAE1B,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,OAAO,CAAC,MAAM,CAAC,CAAA;IACnD,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

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

@@ -1,2 +0,0 @@
-export declare const PROMPT_INPUT_BUFFER = 1048;
-export declare const PROMPT_OUTPUT_BUFFER = 512;

package/dist/src/operations/constants.js

@@ -1,3 +0,0 @@
-export const PROMPT_INPUT_BUFFER = 1048;
-export const PROMPT_OUTPUT_BUFFER = 512;
-//# sourceMappingURL=constants.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../../src/operations/constants.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,mBAAmB,GAAG,IAAI,CAAA;AACvC,MAAM,CAAC,MAAM,oBAAoB,GAAG,GAAG,CAAA"}

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

@@ -1,5 +0,0 @@
-export declare class JsonParsingError extends Error {
-    json: unknown;
-    error: Error;
-    constructor(json: unknown, error: Error);
-}

package/dist/src/operations/errors.js

@@ -1,11 +0,0 @@
-export class JsonParsingError extends Error {
-    json;
-    error;
-    constructor(json, error) {
-        const message = `Error parsing JSON:\n\n---JSON---\n${json}\n\n---Error---\n\n ${error}`;
-        super(message);
-        this.json = json;
-        this.error = error;
-    }
-}
-//# sourceMappingURL=errors.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"errors.js","sourceRoot":"","sources":["../../../src/operations/errors.ts"],"names":[],"mappings":"AAAA,MAAM,OAAO,gBAAiB,SAAQ,KAAK;IAEhC;IACA;IAFT,YACS,IAAa,EACb,KAAY;QAEnB,MAAM,OAAO,GAAG,sCAAsC,IAAI,uBAAuB,KAAK,EAAE,CAAA;QACxF,KAAK,CAAC,OAAO,CAAC,CAAA;QAJP,SAAI,GAAJ,IAAI,CAAS;QACb,UAAK,GAAL,KAAK,CAAO;IAIrB,CAAC;CACF"}

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

@@ -1,20 +0,0 @@
-import { ZodObject, z } from '@bpinternal/zui';
-export type Options = z.input<typeof Options>;
-declare const Options: ZodObject<{
-    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, {
-    chunkLength: number;
-    instructions?: string | undefined;
-}, {
-    instructions?: string | undefined;
-    chunkLength?: number | undefined;
-}>;
-declare module '../zai' {
-    interface Zai {
-        /** Extracts one or many elements from an arbitrary input */
-        extract<S extends z.AnyZodObject>(input: unknown, schema: S, options?: Options): Promise<z.infer<S>>;
-        extract<S extends z.AnyZodObject>(input: unknown, schema: z.ZodArray<S>, options?: Options): Promise<Array<z.infer<S>>>;
-    }
-}
-export {};

package/dist/src/operations/extract.js

@@ -1,235 +0,0 @@
-import { ZodArray, ZodObject, z } from '@bpinternal/zui';
-import JSON5 from 'json5';
-import { jsonrepair } from 'jsonrepair';
-import _ from 'lodash';
-import { fastHash, stringify, takeUntilTokens } from '../utils';
-import { Zai } from '../zai';
-import { PROMPT_INPUT_BUFFER } from './constants';
-import { JsonParsingError } from './errors';
-const Options = z.object({
-    instructions: z.string().optional().describe('Instructions to guide the user on how to extract the data'),
-    chunkLength: z
-        .number()
-        .min(100)
-        .max(100_000)
-        .optional()
-        .describe('The maximum number of tokens per chunk')
-        .default(16_000)
-});
-const START = '■json_start■';
-const END = '■json_end■';
-const NO_MORE = '■NO_MORE_ELEMENT■';
-Zai.prototype.extract = async function (input, schema, _options) {
-    const options = Options.parse(_options ?? {});
-    const tokenizer = await this.getTokenizer();
-    const taskId = this.taskId;
-    const taskType = 'zai.extract';
-    const PROMPT_COMPONENT = Math.max(this.Model.input.maxTokens - PROMPT_INPUT_BUFFER, 100);
-    let isArrayOfObjects = false;
-    const originalSchema = schema;
-    if (schema instanceof ZodObject) {
-        // Do nothing
-    }
-    else if (schema instanceof ZodArray) {
-        if (schema._def.type instanceof ZodObject) {
-            isArrayOfObjects = true;
-            schema = schema._def.type;
-        }
-        else {
-            throw new Error('Schema must be a ZodObject or a ZodArray<ZodObject>');
-        }
-    }
-    else {
-        throw new Error('Schema must be either a ZuiObject or a ZuiArray<ZuiObject>');
-    }
-    const schemaTypescript = schema.toTypescript({ declaration: false });
-    const schemaLength = tokenizer.count(schemaTypescript);
-    options.chunkLength = Math.min(options.chunkLength, this.Model.input.maxTokens - PROMPT_INPUT_BUFFER - schemaLength);
-    const keys = Object.keys(schema.shape);
-    let inputAsString = stringify(input);
-    if (tokenizer.count(inputAsString) > options.chunkLength) {
-        // 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)));
-            return all.flat();
-        }
-        else {
-            // Truncate the input to fit the model's input size
-            inputAsString = tokenizer.truncate(stringify(input), options.chunkLength);
-        }
-    }
-    const instructions = [];
-    if (options.instructions) {
-        instructions.push(options.instructions);
-    }
-    const shape = `{ ${keys.map((key) => `"${key}": ...`).join(', ')} }`;
-    const abbv = '{ ... }';
-    if (isArrayOfObjects) {
-        instructions.push('You may have multiple elements, or zero elements in the input.');
-        instructions.push('You must extract each element separately.');
-        instructions.push(`Each element must be a JSON object with exactly the format: ${START}${shape}${END}`);
-        instructions.push(`When you are done extracting all elements, type "${NO_MORE}" to finish.`);
-        instructions.push(`For example, if you have zero elements, the output should look like this: ${NO_MORE}`);
-        instructions.push(`For example, if you have two elements, the output should look like this: ${START}${abbv}${END}${START}${abbv}${END}${NO_MORE}`);
-    }
-    else {
-        instructions.push('You may have exactly one element in the input.');
-        instructions.push(`The element must be a JSON object with exactly the format: ${START}${shape}${END}`);
-    }
-    // All tokens remaining after the input and condition are accounted can be used for examples
-    const EXAMPLES_TOKENS = PROMPT_COMPONENT - tokenizer.count(inputAsString) - tokenizer.count(instructions.join('\n'));
-    const Key = fastHash(JSON.stringify({
-        taskType,
-        taskId,
-        input: inputAsString,
-        instructions: options.instructions
-    }));
-    const examples = taskId
-        ? await this.adapter.getExamples({
-            input: inputAsString,
-            taskType,
-            taskId
-        })
-        : [];
-    const exactMatch = examples.find((x) => x.key === Key);
-    if (exactMatch) {
-        return exactMatch.output;
-    }
-    const defaultExample = isArrayOfObjects
-        ? {
-            input: `The story goes as follow.
-Once upon a time, there was a person named Alice who was 30 years old.
-Then, there was a person named Bob who was 25 years old.
-The end.`,
-            schema: 'Array<{ name: string, age: number }>',
-            instructions: 'Extract all people',
-            extracted: [
-                {
-                    name: 'Alice',
-                    age: 30
-                },
-                {
-                    name: 'Bob',
-                    age: 25
-                }
-            ]
-        }
-        : {
-            input: `The story goes as follow.
-Once upon a time, there was a person named Alice who was 30 years old.
-The end.`,
-            schema: '{ name: string, age: number }',
-            instructions: 'Extract the person',
-            extracted: { name: 'Alice', age: 30 }
-        };
-    const userExamples = examples.map((e) => ({
-        input: e.input,
-        extracted: e.output,
-        schema: schemaTypescript,
-        instructions: options.instructions
-    }));
-    let exampleId = 1;
-    const formatInput = (input, schema, instructions) => {
-        const header = userExamples.length
-            ? `Expert Example #${exampleId++}`
-            : "Here's an example to help you understand the format:";
-        return `
-${header}
-
-<|start_schema|>
-${schema}
-<|end_schema|>
-
-<|start_instructions|>
-${instructions ?? 'No specific instructions, just follow the schema above.'}
-<|end_instructions|>
-
-<|start_input|>
-${input.trim()}
-<|end_input|>
-  `.trim();
-    };
-    const formatOutput = (extracted) => {
-        extracted = _.isArray(extracted) ? extracted : [extracted];
-        return (extracted
-            .map((x) => `
-${START}
-${JSON.stringify(x, null, 2)}
-${END}`.trim())
-            .join('\n') + NO_MORE);
-    };
-    const formatExample = (example) => [
-        {
-            type: 'text',
-            content: formatInput(stringify(example.input ?? null), example.schema, example.instructions),
-            role: 'user'
-        },
-        {
-            type: 'text',
-            content: formatOutput(example.extracted),
-            role: 'assistant'
-        }
-    ];
-    const allExamples = takeUntilTokens(userExamples.length ? userExamples : [defaultExample], EXAMPLES_TOKENS, (el) => tokenizer.count(stringify(el.input)) + tokenizer.count(stringify(el.extracted)))
-        .map(formatExample)
-        .flat();
-    const output = await this.callModel({
-        systemPrompt: `
-Extract the following information from the input:
-${schemaTypescript}
-====
-
-${instructions.map((x) => `• ${x}`).join('\n')}
-`.trim(),
-        stopSequences: [isArrayOfObjects ? NO_MORE : END],
-        messages: [
-            ...allExamples,
-            {
-                role: 'user',
-                type: 'text',
-                content: formatInput(inputAsString, schemaTypescript, options.instructions ?? '')
-            }
-        ]
-    });
-    const answer = output.choices[0].content;
-    const elements = answer
-        .split(START)
-        .filter((x) => x.trim().length > 0)
-        .map((x) => {
-        try {
-            const json = x.slice(0, x.indexOf(END)).trim();
-            const repairedJson = jsonrepair(json);
-            const parsedJson = JSON5.parse(repairedJson);
-            return schema.parse(parsedJson);
-        }
-        catch (error) {
-            throw new JsonParsingError(x, error);
-        }
-    })
-        .filter((x) => x !== null);
-    let final;
-    if (isArrayOfObjects) {
-        final = elements;
-    }
-    else if (elements.length === 0) {
-        final = schema.parse({});
-    }
-    else {
-        final = elements[0];
-    }
-    if (taskId) {
-        await this.adapter.saveExample({
-            key: Key,
-            taskId: `zai/${taskId}`,
-            taskType,
-            instructions: options.instructions ?? 'No specific instructions',
-            input: inputAsString,
-            output: final,
-            metadata: output.metadata
-        });
-    }
-    return final;
-};
-//# sourceMappingURL=extract.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"extract.js","sourceRoot":"","sources":["../../../src/operations/extract.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,EAAE,MAAM,iBAAiB,CAAA;AAExD,OAAO,KAAK,MAAM,OAAO,CAAA;AACzB,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAA;AAEvC,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,MAAM,aAAa,CAAA;AACjD,OAAO,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAA;AAG3C,MAAM,OAAO,GAAG,CAAC,CAAC,MAAM,CAAC;IACvB,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,2DAA2D,CAAC;IACzG,WAAW,EAAE,CAAC;SACX,MAAM,EAAE;SACR,GAAG,CAAC,GAAG,CAAC;SACR,GAAG,CAAC,OAAO,CAAC;SACZ,QAAQ,EAAE;SACV,QAAQ,CAAC,wCAAwC,CAAC;SAClD,OAAO,CAAC,MAAM,CAAC;CACnB,CAAC,CAAA;AAcF,MAAM,KAAK,GAAG,cAAc,CAAA;AAC5B,MAAM,GAAG,GAAG,YAAY,CAAA;AACxB,MAAM,OAAO,GAAG,mBAAmB,CAAA;AAEnC,GAAG,CAAC,SAAS,CAAC,OAAO,GAAG,KAAK,WAAsB,KAAK,EAAE,MAAM,EAAE,QAAQ;IACxE,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,aAAa,CAAA;IAE9B,MAAM,gBAAgB,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,GAAG,mBAAmB,EAAE,GAAG,CAAC,CAAA;IAExF,IAAI,gBAAgB,GAAG,KAAK,CAAA;IAC5B,MAAM,cAAc,GAAG,MAAM,CAAA;IAE7B,IAAI,MAAM,YAAY,SAAS,EAAE,CAAC;QAChC,aAAa;IACf,CAAC;SAAM,IAAI,MAAM,YAAY,QAAQ,EAAE,CAAC;QACtC,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,YAAY,SAAS,EAAE,CAAC;YAC1C,gBAAgB,GAAG,IAAI,CAAA;YACvB,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAA;QAC3B,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAA;QACxE,CAAC;IACH,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAA;IAC/E,CAAC;IAED,MAAM,gBAAgB,GAAG,MAAM,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAA;IACpE,MAAM,YAAY,GAAG,SAAS,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAA;IAEtD,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,GAAG,mBAAmB,GAAG,YAAY,CAAC,CAAA;IAEpH,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IAEtC,IAAI,aAAa,GAAG,SAAS,CAAC,KAAK,CAAC,CAAA;IAEpC,IAAI,SAAS,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACzD,mFAAmF;QACnF,IAAI,gBAAgB,EAAE,CAAC;YACrB,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,aAAa,CAAC,CAAA;YAC7C,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAA;YAC1E,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC,CAAC,CAAA;YAEzF,OAAO,GAAG,CAAC,IAAI,EAAE,CAAA;QACnB,CAAC;aAAM,CAAC;YACN,mDAAmD;YACnD,aAAa,GAAG,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,CAAA;QAC3E,CAAC;IACH,CAAC;IAED,MAAM,YAAY,GAAa,EAAE,CAAA;IAEjC,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;QACzB,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAA;IACzC,CAAC;IAED,MAAM,KAAK,GAAG,KAAK,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,GAAG,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAA;IACpE,MAAM,IAAI,GAAG,SAAS,CAAA;IAEtB,IAAI,gBAAgB,EAAE,CAAC;QACrB,YAAY,CAAC,IAAI,CAAC,gEAAgE,CAAC,CAAA;QACnF,YAAY,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAA;QAC9D,YAAY,CAAC,IAAI,CAAC,+DAA+D,KAAK,GAAG,KAAK,GAAG,GAAG,EAAE,CAAC,CAAA;QACvG,YAAY,CAAC,IAAI,CAAC,oDAAoD,OAAO,cAAc,CAAC,CAAA;QAC5F,YAAY,CAAC,IAAI,CAAC,6EAA6E,OAAO,EAAE,CAAC,CAAA;QACzG,YAAY,CAAC,IAAI,CACf,4EAA4E,KAAK,GAAG,IAAI,GAAG,GAAG,GAAG,KAAK,GAAG,IAAI,GAAG,GAAG,GAAG,OAAO,EAAE,CAChI,CAAA;IACH,CAAC;SAAM,CAAC;QACN,YAAY,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAA;QACnE,YAAY,CAAC,IAAI,CAAC,8DAA8D,KAAK,GAAG,KAAK,GAAG,GAAG,EAAE,CAAC,CAAA;IACxG,CAAC;IAED,4FAA4F;IAC5F,MAAM,eAAe,GAAG,gBAAgB,GAAG,SAAS,CAAC,KAAK,CAAC,aAAa,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAA;IAEpH,MAAM,GAAG,GAAG,QAAQ,CAClB,IAAI,CAAC,SAAS,CAAC;QACb,QAAQ;QACR,MAAM;QACN,KAAK,EAAE,aAAa;QACpB,YAAY,EAAE,OAAO,CAAC,YAAY;KACnC,CAAC,CACH,CAAA;IAED,MAAM,QAAQ,GAAG,MAAM;QACrB,CAAC,CAAC,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAkB;YAC9C,KAAK,EAAE,aAAa;YACpB,QAAQ;YACR,MAAM;SACP,CAAC;QACJ,CAAC,CAAC,EAAE,CAAA;IAEN,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAA;IACtD,IAAI,UAAU,EAAE,CAAC;QACf,OAAO,UAAU,CAAC,MAAM,CAAA;IAC1B,CAAC;IAED,MAAM,cAAc,GAAG,gBAAgB;QACrC,CAAC,CAAC;YACE,KAAK,EAAE;;;SAGN;YACD,MAAM,EAAE,sCAAsC;YAC9C,YAAY,EAAE,oBAAoB;YAClC,SAAS,EAAE;gBACT;oBACE,IAAI,EAAE,OAAO;oBACb,GAAG,EAAE,EAAE;iBACR;gBACD;oBACE,IAAI,EAAE,KAAK;oBACX,GAAG,EAAE,EAAE;iBACR;aACF;SACF;QACH,CAAC,CAAC;YACE,KAAK,EAAE;;SAEN;YACD,MAAM,EAAE,+BAA+B;YACvC,YAAY,EAAE,oBAAoB;YAClC,SAAS,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE;SACtC,CAAA;IAEL,MAAM,YAAY,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACxC,KAAK,EAAE,CAAC,CAAC,KAAK;QACd,SAAS,EAAE,CAAC,CAAC,MAAM;QACnB,MAAM,EAAE,gBAAgB;QACxB,YAAY,EAAE,OAAO,CAAC,YAAY;KACnC,CAAC,CAAC,CAAA;IAEH,IAAI,SAAS,GAAG,CAAC,CAAA;IAEjB,MAAM,WAAW,GAAG,CAAC,KAAa,EAAE,MAAc,EAAE,YAAqB,EAAE,EAAE;QAC3E,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM;YAChC,CAAC,CAAC,mBAAmB,SAAS,EAAE,EAAE;YAClC,CAAC,CAAC,sDAAsD,CAAA;QAE1D,OAAO;EACT,MAAM;;;EAGN,MAAM;;;;EAIN,YAAY,IAAI,yDAAyD;;;;EAIzE,KAAK,CAAC,IAAI,EAAE;;GAEX,CAAC,IAAI,EAAE,CAAA;IACR,CAAC,CAAA;IAED,MAAM,YAAY,GAAG,CAAC,SAAc,EAAE,EAAE;QACtC,SAAS,GAAG,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAA;QAE1D,OAAO,CACL,SAAS;aACN,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CACT;EACR,KAAK;EACL,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;EAC1B,GAAG,EAAE,CAAC,IAAI,EAAE,CACL;aACA,IAAI,CAAC,IAAI,CAAC,GAAG,OAAO,CACxB,CAAA;IACH,CAAC,CAAA;IAED,MAAM,aAAa,GAAG,CAAC,OAA+E,EAAE,EAAE,CAAC;QACzG;YACE,IAAI,EAAE,MAAe;YACrB,OAAO,EAAE,WAAW,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,YAAY,CAAC;YAC5F,IAAI,EAAE,MAAe;SACtB;QACD;YACE,IAAI,EAAE,MAAe;YACrB,OAAO,EAAE,YAAY,CAAC,OAAO,CAAC,SAAS,CAAC;YACxC,IAAI,EAAE,WAAoB;SAC3B;KACF,CAAA;IAED,MAAM,WAAW,GAAG,eAAe,CACjC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,EACrD,eAAe,EACf,CAAC,EAAE,EAAE,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CACxF;SACE,GAAG,CAAC,aAAa,CAAC;SAClB,IAAI,EAAE,CAAA;IAET,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC;QAClC,YAAY,EAAE;;EAEhB,gBAAgB;;;EAGhB,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;CAC7C,CAAC,IAAI,EAAE;QACJ,aAAa,EAAE,CAAC,gBAAgB,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;QACjD,QAAQ,EAAE;YACR,GAAG,WAAW;YACd;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,WAAW,CAAC,aAAa,EAAE,gBAAgB,EAAE,OAAO,CAAC,YAAY,IAAI,EAAE,CAAC;aAClF;SACF;KACF,CAAC,CAAA;IAEF,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAiB,CAAA;IAElD,MAAM,QAAQ,GAAG,MAAM;SACpB,KAAK,CAAC,KAAK,CAAC;SACZ,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;SAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACT,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAA;YAC9C,MAAM,YAAY,GAAG,UAAU,CAAC,IAAI,CAAC,CAAA;YACrC,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,YAAY,CAAC,CAAA;YAE5C,OAAO,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAA;QACjC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,gBAAgB,CAAC,CAAC,EAAE,KAAK,CAAC,CAAA;QACtC,CAAC;IACH,CAAC,CAAC;SACD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAA;IAE5B,IAAI,KAAU,CAAA;IAEd,IAAI,gBAAgB,EAAE,CAAC;QACrB,KAAK,GAAG,QAAQ,CAAA;IAClB,CAAC;SAAM,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACjC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;IAC1B,CAAC;SAAM,CAAC;QACN,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAA;IACrB,CAAC;IAED,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC;YAC7B,GAAG,EAAE,GAAG;YACR,MAAM,EAAE,OAAO,MAAM,EAAE;YACvB,QAAQ;YACR,YAAY,EAAE,OAAO,CAAC,YAAY,IAAI,0BAA0B;YAChE,KAAK,EAAE,aAAa;YACpB,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE,MAAM,CAAC,QAAQ;SAC1B,CAAC,CAAA;IACJ,CAAC;IAED,OAAO,KAAK,CAAA;AACd,CAAC,CAAA"}

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

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

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 {};