@botpress/zai 1.0.0-beta.5 → 1.0.0-beta.7

package/dist/index.d.cts

@@ -1,6 +1,6 @@
 import { Client } from '@botpress/client';
 import sdk from '@botpress/sdk';
-import { TextTokenizer } from '@botpress/wasm';
+import * as wasm from '@botpress/wasm';
 
 type GenerationMetadata = sdk.z.input<typeof GenerationMetadata>;
 declare const GenerationMetadata: sdk.ZodObject<{
@@ -626,7 +626,7 @@ declare const ZaiConfig: sdk.ZodObject<{
     namespace?: string;
 }>;
 declare class Zai {
-    protected static tokenizer: TextTokenizer;
+    protected static tokenizer: wasm.TextTokenizer;
     protected client: Client;
     private originalConfig;
     private userId;
@@ -644,7 +644,7 @@ declare class Zai {
     }>;
     /** @internal */
     private _callModel;
-    protected getTokenizer(): Promise<TextTokenizer>;
+    protected getTokenizer(): Promise<wasm.TextTokenizer>;
     protected get taskId(): string;
     with(options: Partial<ZaiConfig>): Zai;
     learn(taskId: string): Zai;
@@ -658,7 +658,7 @@ declare const Options$6: sdk.ZodObject<{
 }, {
     length?: number;
 }>;
-declare module '../zai' {
+declare module '@botpress/zai' {
     interface Zai {
         /** Generates a text of the desired length according to the prompt */
         text(prompt: string, options?: Options$6): Promise<string>;
@@ -691,7 +691,7 @@ declare const Options$5: sdk.ZodObject<{
         output?: string;
     }[];
 }>;
-declare module '../zai' {
+declare module '@botpress/zai' {
     interface Zai {
         /** Rewrites a string according to match the prompt */
         rewrite(original: string, prompt: string, options?: Options$5): Promise<string>;
@@ -736,7 +736,7 @@ declare const Options$4: sdk.ZodObject<{
         overlap?: number;
     };
 }>;
-declare module '../zai' {
+declare module '@botpress/zai' {
     interface Zai {
         /** Summarizes a text of any length to a summary of the desired length */
         summarize(original: string, options?: Options$4): Promise<string>;
@@ -771,7 +771,7 @@ declare const Options$3: sdk.ZodObject<{
         reason?: string;
     }[];
 }>;
-declare module '../zai' {
+declare module '@botpress/zai' {
     interface Zai {
         /** Checks wether a condition is true or not */
         check(input: unknown, condition: string, options?: Options$3): Promise<boolean>;
@@ -809,7 +809,7 @@ declare const Options$2: sdk.ZodObject<{
     }[];
     tokensPerItem?: number;
 }>;
-declare module '../zai' {
+declare module '@botpress/zai' {
     interface Zai {
         /** Filters elements of an array against a condition */
         filter<T>(input: Array<T>, condition: string, options?: Options$2): Promise<Array<T>>;
@@ -827,7 +827,7 @@ declare const Options$1: sdk.ZodObject<{
     instructions?: string;
     chunkLength?: number;
 }>;
-declare module '../zai' {
+declare module '@botpress/zai' {
     interface Zai {
         /** Extracts one or many elements from an arbitrary input */
         extract<S extends sdk.z.AnyZodObject>(input: unknown, schema: S, options?: Options$1): Promise<sdk.z.infer<S>>;
@@ -904,7 +904,7 @@ declare const Options: sdk.ZodObject<{
 }>;
 type Labels<T extends string> = Record<T, string>;
 declare const Labels: sdk.ZodEffects<sdk.ZodRecord<sdk.ZodString, sdk.ZodString>, Record<string, string>, Record<string, string>>;
-declare module '../zai' {
+declare module '@botpress/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<{

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 import { Client } from '@botpress/client';
 import sdk from '@botpress/sdk';
-import { TextTokenizer } from '@botpress/wasm';
+import * as wasm from '@botpress/wasm';
 
 type GenerationMetadata = sdk.z.input<typeof GenerationMetadata>;
 declare const GenerationMetadata: sdk.ZodObject<{
@@ -626,7 +626,7 @@ declare const ZaiConfig: sdk.ZodObject<{
     namespace?: string;
 }>;
 declare class Zai {
-    protected static tokenizer: TextTokenizer;
+    protected static tokenizer: wasm.TextTokenizer;
     protected client: Client;
     private originalConfig;
     private userId;
@@ -644,7 +644,7 @@ declare class Zai {
     }>;
     /** @internal */
     private _callModel;
-    protected getTokenizer(): Promise<TextTokenizer>;
+    protected getTokenizer(): Promise<wasm.TextTokenizer>;
     protected get taskId(): string;
     with(options: Partial<ZaiConfig>): Zai;
     learn(taskId: string): Zai;
@@ -658,7 +658,7 @@ declare const Options$6: sdk.ZodObject<{
 }, {
     length?: number;
 }>;
-declare module '../zai' {
+declare module '@botpress/zai' {
     interface Zai {
         /** Generates a text of the desired length according to the prompt */
         text(prompt: string, options?: Options$6): Promise<string>;
@@ -691,7 +691,7 @@ declare const Options$5: sdk.ZodObject<{
         output?: string;
     }[];
 }>;
-declare module '../zai' {
+declare module '@botpress/zai' {
     interface Zai {
         /** Rewrites a string according to match the prompt */
         rewrite(original: string, prompt: string, options?: Options$5): Promise<string>;
@@ -736,7 +736,7 @@ declare const Options$4: sdk.ZodObject<{
         overlap?: number;
     };
 }>;
-declare module '../zai' {
+declare module '@botpress/zai' {
     interface Zai {
         /** Summarizes a text of any length to a summary of the desired length */
         summarize(original: string, options?: Options$4): Promise<string>;
@@ -771,7 +771,7 @@ declare const Options$3: sdk.ZodObject<{
         reason?: string;
     }[];
 }>;
-declare module '../zai' {
+declare module '@botpress/zai' {
     interface Zai {
         /** Checks wether a condition is true or not */
         check(input: unknown, condition: string, options?: Options$3): Promise<boolean>;
@@ -809,7 +809,7 @@ declare const Options$2: sdk.ZodObject<{
     }[];
     tokensPerItem?: number;
 }>;
-declare module '../zai' {
+declare module '@botpress/zai' {
     interface Zai {
         /** Filters elements of an array against a condition */
         filter<T>(input: Array<T>, condition: string, options?: Options$2): Promise<Array<T>>;
@@ -827,7 +827,7 @@ declare const Options$1: sdk.ZodObject<{
     instructions?: string;
     chunkLength?: number;
 }>;
-declare module '../zai' {
+declare module '@botpress/zai' {
     interface Zai {
         /** Extracts one or many elements from an arbitrary input */
         extract<S extends sdk.z.AnyZodObject>(input: unknown, schema: S, options?: Options$1): Promise<sdk.z.infer<S>>;
@@ -904,7 +904,7 @@ declare const Options: sdk.ZodObject<{
 }>;
 type Labels<T extends string> = Record<T, string>;
 declare const Labels: sdk.ZodEffects<sdk.ZodRecord<sdk.ZodString, sdk.ZodString>, Record<string, string>, Record<string, string>>;
-declare module '../zai' {
+declare module '@botpress/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<{

package/dist/index.js

@@ -3,7 +3,7 @@ var __name = (target, value) => __defProp(target, "name", { value, configurable:
 
 // src/zai.ts
 import sdk3 from "@botpress/sdk";
-import { getWasmTokenizer } from "@botpress/wasm";
+import * as wasm from "@botpress/wasm";
 
 // src/adapters/botpress-table.ts
 import sdk2 from "@botpress/sdk";
@@ -753,7 +753,7 @@ var Zai = class _Zai {
     } while (--retries > 0);
   }
   async getTokenizer() {
-    _Zai.tokenizer ??= await getWasmTokenizer();
+    _Zai.tokenizer ??= await wasm.getWasmTokenizer();
     return _Zai.tokenizer;
   }
   get taskId() {