@trymirai/uzu 0.1.50 → 0.2.0

package/interactors/chatSessionInteractor.d.ts

@@ -1,4 +1,5 @@
 import { ChatSession } from "../bridging/chatSession.js";
+import { GrammarConfig } from "../bridging/grammarConfig.js";
 import { Input } from "../bridging/input.js";
 import { Message } from "../bridging/message.js";
 import { Output } from "../bridging/output.js";
@@ -17,6 +18,7 @@ export declare class ChatSessionInteractor implements Interactor<ChatSession> {
     enableThinking(enableThinking: boolean): ChatSessionInteractor;
     samplingPolicy(samplingPolicy: SamplingPolicy): ChatSessionInteractor;
     samplingMethod(samplingMethod: SamplingMethod): ChatSessionInteractor;
+    grammarConfig(grammarConfig: GrammarConfig | null): ChatSessionInteractor;
     run(input: Input, progress?: (partialOutput: Output) => boolean): Promise<Output>;
     reply(text: string, progress?: (partialOutput: Output) => boolean): Promise<Output>;
     replyToMessages(messages: Message[], progress?: (partialOutput: Output) => boolean): Promise<Output>;

package/interactors/chatSessionInteractor.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chatSessionInteractor.d.ts","sourceRoot":"","sources":["../src/interactors/chatSessionInteractor.ts"],"names":[],"mappings":"OAAO,EAAE,WAAW,EAAE;OACf,EAAE,KAAK,EAAE;OACT,EAAE,OAAO,EAAE;OACX,EAAE,MAAM,EAAE;OACV,EAAE,SAAS,EAAE;OACb,EAAE,cAAc,EAAE;OAClB,EAAE,cAAc,EAAE;OAClB,EAAE,mBAAmB,EAAE;OACvB,EAAE,UAAU,EAAE,gBAAgB,EAAE;AAEvC,qBAAa,qBAAsB,YAAW,UAAU,CAAC,WAAW,CAAC;IACjE,QAAQ,CAAC,eAAe,EAAE,mBAAmB,CAAC;IAC9C,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC,WAAW,CAAC,CAAC;IAC/C,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC;gBAGvB,eAAe,EAAE,mBAAmB,EACpC,OAAO,EAAE,gBAAgB,CAAC,WAAW,CAAC,EACtC,MAAM,EAAE,SAAS;IAOf,QAAQ,IAAI,OAAO,CAAC,WAAW,CAAC;IAMtC,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,qBAAqB;IAKvD,cAAc,CAAC,cAAc,EAAE,OAAO,GAAG,qBAAqB;IAK9D,cAAc,CAAC,cAAc,EAAE,cAAc,GAAG,qBAAqB;IAKrE,cAAc,CAAC,cAAc,EAAE,cAAc,GAAG,qBAAqB;IAO/D,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAAG,OAAO,CAAC,MAAM,CAAC;IAU7F,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAAG,OAAO,CAAC,MAAM,CAAC;IAI/F,eAAe,CACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAC1D,OAAO,CAAC,MAAM,CAAC;CAGrB"}
+{"version":3,"file":"chatSessionInteractor.d.ts","sourceRoot":"","sources":["../src/interactors/chatSessionInteractor.ts"],"names":[],"mappings":"OAAO,EAAE,WAAW,EAAE;OACf,EAAE,aAAa,EAAE;OACjB,EAAE,KAAK,EAAE;OACT,EAAE,OAAO,EAAE;OACX,EAAE,MAAM,EAAE;OACV,EAAE,SAAS,EAAE;OACb,EAAE,cAAc,EAAE;OAClB,EAAE,cAAc,EAAE;OAClB,EAAE,mBAAmB,EAAE;OACvB,EAAE,UAAU,EAAE,gBAAgB,EAAE;AAEvC,qBAAa,qBAAsB,YAAW,UAAU,CAAC,WAAW,CAAC;IACjE,QAAQ,CAAC,eAAe,EAAE,mBAAmB,CAAC;IAC9C,QAAQ,CAAC,MAAM,EAAE,gBAAgB,CAAC,WAAW,CAAC,CAAC;IAC/C,QAAQ,CAAC,MAAM,EAAE,SAAS,CAAC;gBAGvB,eAAe,EAAE,mBAAmB,EACpC,OAAO,EAAE,gBAAgB,CAAC,WAAW,CAAC,EACtC,MAAM,EAAE,SAAS;IAOf,QAAQ,IAAI,OAAO,CAAC,WAAW,CAAC;IAMtC,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,qBAAqB;IAKvD,cAAc,CAAC,cAAc,EAAE,OAAO,GAAG,qBAAqB;IAK9D,cAAc,CAAC,cAAc,EAAE,cAAc,GAAG,qBAAqB;IAKrE,cAAc,CAAC,cAAc,EAAE,cAAc,GAAG,qBAAqB;IAKrE,aAAa,CAAC,aAAa,EAAE,aAAa,GAAG,IAAI,GAAG,qBAAqB;IAOnE,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAAG,OAAO,CAAC,MAAM,CAAC;IAU7F,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAAG,OAAO,CAAC,MAAM,CAAC;IAI/F,eAAe,CACjB,QAAQ,EAAE,OAAO,EAAE,EACnB,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAC1D,OAAO,CAAC,MAAM,CAAC;CAGrB"}

package/interactors/chatSessionInteractor.js

@@ -29,6 +29,10 @@ class ChatSessionInteractor {
         const config = this.config.withSamplingPolicy(samplingPolicy_1.SamplingPolicy.custom(samplingMethod));
         return new ChatSessionInteractor(this.modelInteractor, this.entity, config);
     }
+    grammarConfig(grammarConfig) {
+        const config = this.config.withGrammarConfig(grammarConfig);
+        return new ChatSessionInteractor(this.modelInteractor, this.entity, config);
+    }
     /* Run */
     async run(input, progress = () => true) {
         const outputPromise = (async () => {

package/interactors/chatSessionInteractor.js.map

@@ -1 +1 @@
-{"version":3,"file":"chatSessionInteractor.js","sourceRoot":"","sources":["../src/interactors/chatSessionInteractor.ts"],"names":[],"mappings":";;;AACA,gDAA0C;AAK1C,kEAA4D;AAI5D,MAAa,qBAAqB;IAK9B,YACI,eAAoC,EACpC,OAAsC,EACtC,MAAiB;QAEjB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC;QACtB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,YAAY;IAEZ,WAAW,CAAC,WAAmB;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QACxD,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,cAAc,CAAC,cAAuB;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,+BAAc,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC;QACrF,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,SAAS;IAET,KAAK,CAAC,GAAG,CAAC,KAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC7E,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YACzD,OAAO,MAAM,CAAC;QAClB,CAAC,CAAC,EAAE,CAAC;QACL,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;QACnC,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC/E,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,aAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,eAAe,CACjB,QAAmB,EACnB,WAA+C,GAAG,EAAE,CAAC,IAAI;QAEzD,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,aAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,CAAC;IAC9D,CAAC;CACJ;AA/DD,sDA+DC"}
+{"version":3,"file":"chatSessionInteractor.js","sourceRoot":"","sources":["../src/interactors/chatSessionInteractor.ts"],"names":[],"mappings":";;;AAEA,gDAA0C;AAK1C,kEAA4D;AAI5D,MAAa,qBAAqB;IAK9B,YACI,eAAoC,EACpC,OAAsC,EACtC,MAAiB;QAEjB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC;QACtB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,YAAY;IAEZ,WAAW,CAAC,WAAmB;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QACxD,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,cAAc,CAAC,cAAuB;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,+BAAc,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC;QACrF,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,aAAa,CAAC,aAAmC;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;QAC5D,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,SAAS;IAET,KAAK,CAAC,GAAG,CAAC,KAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC7E,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YACzD,OAAO,MAAM,CAAC;QAClB,CAAC,CAAC,EAAE,CAAC;QACL,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;QACnC,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC/E,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,aAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,eAAe,CACjB,QAAmB,EACnB,WAA+C,GAAG,EAAE,CAAC,IAAI;QAEzD,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,aAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,CAAC;IAC9D,CAAC;CACJ;AApED,sDAoEC"}

package/interactors/chatSessionInteractor.mjs

@@ -26,6 +26,10 @@ export class ChatSessionInteractor {
         const config = this.config.withSamplingPolicy(SamplingPolicy.custom(samplingMethod));
         return new ChatSessionInteractor(this.modelInteractor, this.entity, config);
     }
+    grammarConfig(grammarConfig) {
+        const config = this.config.withGrammarConfig(grammarConfig);
+        return new ChatSessionInteractor(this.modelInteractor, this.entity, config);
+    }
     /* Run */
     async run(input, progress = () => true) {
         const outputPromise = (async () => {

package/interactors/chatSessionInteractor.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"chatSessionInteractor.mjs","sourceRoot":"","sources":["../src/interactors/chatSessionInteractor.ts"],"names":[],"mappings":"OACO,EAAE,KAAK,EAAE;OAKT,EAAE,cAAc,EAAE;AAIzB,MAAM,OAAO,qBAAqB;IAK9B,YACI,eAAoC,EACpC,OAAsC,EACtC,MAAiB;QAEjB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC;QACtB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,YAAY;IAEZ,WAAW,CAAC,WAAmB;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QACxD,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,cAAc,CAAC,cAAuB;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC;QACrF,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,SAAS;IAET,KAAK,CAAC,GAAG,CAAC,KAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC7E,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YACzD,OAAO,MAAM,CAAC;QAClB,CAAC,CAAC,EAAE,CAAC;QACL,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;QACnC,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC/E,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,eAAe,CACjB,QAAmB,EACnB,WAA+C,GAAG,EAAE,CAAC,IAAI;QAEzD,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,CAAC;IAC9D,CAAC;CACJ"}
+{"version":3,"file":"chatSessionInteractor.mjs","sourceRoot":"","sources":["../src/interactors/chatSessionInteractor.ts"],"names":[],"mappings":"OAEO,EAAE,KAAK,EAAE;OAKT,EAAE,cAAc,EAAE;AAIzB,MAAM,OAAO,qBAAqB;IAK9B,YACI,eAAoC,EACpC,OAAsC,EACtC,MAAiB;QAEjB,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC;QACtB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,QAAQ;QACV,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,YAAY;IAEZ,WAAW,CAAC,WAAmB;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC;QACxD,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,cAAc,CAAC,cAAuB;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAC9D,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,cAAc,CAAC,cAA8B;QACzC,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC;QACrF,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,aAAa,CAAC,aAAmC;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,aAAa,CAAC,CAAC;QAC5D,OAAO,IAAI,qBAAqB,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAChF,CAAC;IAED,SAAS;IAET,KAAK,CAAC,GAAG,CAAC,KAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC7E,MAAM,aAAa,GAAG,CAAC,KAAK,IAAI,EAAE;YAC9B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YACzD,OAAO,MAAM,CAAC;QAClB,CAAC,CAAC,EAAE,CAAC;QACL,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC;QACnC,OAAO,MAAM,CAAC;IAClB,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY,EAAE,WAA+C,GAAG,EAAE,CAAC,IAAI;QAC/E,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,QAAQ,CAAC,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,eAAe,CACjB,QAAmB,EACnB,WAA+C,GAAG,EAAE,CAAC,IAAI;QAEzD,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,QAAQ,CAAC,CAAC;IAC9D,CAAC;CACJ"}

package/internal/tslib.js

@@ -1,2 +1,57 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.__setModuleDefault = exports.__createBinding = void 0;
+exports.__importStar = __importStar;
+var __createBinding = Object.create
+    ? function (o, m, k, k2) {
+        if (k2 === void 0)
+            k2 = k;
+        var desc = Object.getOwnPropertyDescriptor(m, k);
+        if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+            desc = {
+                enumerable: true,
+                get: function () {
+                    return m[k];
+                },
+            };
+        }
+        Object.defineProperty(o, k2, desc);
+    }
+    : function (o, m, k, k2) {
+        if (k2 === void 0)
+            k2 = k;
+        o[k2] = m[k];
+    };
+exports.__createBinding = __createBinding;
+var __setModuleDefault = Object.create
+    ? function (o, v) {
+        Object.defineProperty(o, "default", { enumerable: true, value: v });
+    }
+    : function (o, v) {
+        o["default"] = v;
+    };
+exports.__setModuleDefault = __setModuleDefault;
+var ownKeys = function (o) {
+    ownKeys =
+        Object.getOwnPropertyNames ||
+            function (o2) {
+                var ar = [];
+                for (var k in o2)
+                    if (Object.prototype.hasOwnProperty.call(o2, k))
+                        ar[ar.length] = k;
+                return ar;
+            };
+    return ownKeys(o);
+};
+function __importStar(mod) {
+    if (mod && mod.__esModule)
+        return mod;
+    var result = {};
+    if (mod != null) {
+        for (var k = ownKeys(mod), i = 0; i < k.length; i++)
+            if (k[i] !== "default")
+                __createBinding(result, mod, k[i]);
+    }
+    __setModuleDefault(result, mod);
+    return result;
+}

package/napi/uzu.d.ts

@@ -1,7 +1,7 @@
 /* auto-generated by NAPI-RS */
 /* eslint-disable */
 export declare class ChatSession {
-  run(input: Input, config: RunConfig, progress?: (arg: Output) => boolean | undefined | null): Output
+  run(input: Input, config: RunConfig, progress?: ((arg: Output) => boolean) | undefined | null): Output
   reset(): void
 }
 
@@ -12,9 +12,9 @@ export declare class Engine {
   createModelDownloadHandle(repoId: string): ModelDownloadHandle
   constructor()
   createChatSession(model: ChatModel, config: Config): ChatSession
-  registerLicenseStatusHandler(callback?: (arg: LicenseStatus) => void | undefined | null): void
-  registerModelDownloadStateHandler(callback?: (arg0: string, arg1: ModelDownloadState) => void | undefined | null): void
-  registerChatModelsHandler(callback?: (arg: Array<ChatModel>) => void | undefined | null): void
+  registerLicenseStatusHandler(callback?: ((arg: LicenseStatus) => void) | undefined | null): void
+  registerModelDownloadStateHandler(callback?: ((arg0: string, arg1: ModelDownloadState) => void) | undefined | null): void
+  registerChatModelsHandler(callback?: ((arg: Array<ChatModel>) => void) | undefined | null): void
 }
 
 export declare class ModelDownloadHandle {
@@ -83,6 +83,11 @@ export declare const enum FinishReason {
   ContextLimitReached = 3
 }
 
+export type GrammarConfig =
+  | { type: 'JsonSchema', schema: string }
+  | { type: 'Regex', pattern: string }
+  | { type: 'BuiltinJson' }
+
 export type Input =
   | { type: 'Text', text: string }
   | { type: 'Messages', messages: Array<Message> }
@@ -170,6 +175,7 @@ export interface RunConfig {
   tokensLimit: number
   enableThinking: boolean
   samplingPolicy: SamplingPolicy
+  grammarConfig?: GrammarConfig
 }
 
 export interface RunStats {
@@ -179,7 +185,7 @@ export interface RunStats {
 
 export type SamplingMethod =
   | { type: 'Greedy' }
-  | { type: 'Stochastic', temperature?: number, topK?: number, topP?: number }
+  | { type: 'Stochastic', temperature?: number, topK?: number, topP?: number, minP?: number }
 
 export type SamplingPolicy =
   | { type: 'Default' }

package/napi/uzu.js

@@ -47,6 +47,7 @@ const {
   AsyncBatchSize,
   ContextLength,
   ContextMode,
+  GrammarConfig,
   Input,
   LicenseStatus,
   ModelStorageError,
@@ -85,6 +86,7 @@ module.exports = {
   AsyncBatchSize,
   ContextLength,
   ContextMode,
+  GrammarConfig,
   Input,
   LicenseStatus,
   ModelStorageError,

package/napi/uzu.mjs

@@ -47,6 +47,7 @@ export const {
   AsyncBatchSize,
   ContextLength,
   ContextMode,
+  GrammarConfig,
   Input,
   LicenseStatus,
   ModelStorageError,

package/package.json

@@ -3,6 +3,9 @@
   "cpu": [
     "arm64"
   ],
+  "dependencies": {
+    "zod": "^4.2.1"
+  },
   "description": "A high-performance inference engine for AI models",
   "exports": {
     ".": {
@@ -98,5 +101,5 @@
   },
   "type": "commonjs",
   "types": "./index.d.ts",
-  "version": "0.1.50"
+  "version": "0.2.0"
 }

package/src/bridging/grammarConfig.ts

@@ -0,0 +1,40 @@
+import * as z from 'zod';
+import { GrammarConfig as NapiGrammarConfig } from '../napi/uzu';
+import { ToNapi } from './napi';
+
+export class GrammarConfig implements ToNapi<NapiGrammarConfig> {
+    private readonly napiGrammarConfig: NapiGrammarConfig;
+
+    private constructor(napiGrammarConfig: NapiGrammarConfig) {
+        this.napiGrammarConfig = napiGrammarConfig;
+    }
+
+    static jsonSchema(schema: string): GrammarConfig {
+        let a = z.object({
+            name: z.string(),
+            age: z.number(),
+        });
+        const napiGrammarConfig: NapiGrammarConfig = { type: 'JsonSchema', schema: schema };
+        return new GrammarConfig(napiGrammarConfig);
+    }
+
+    static regex(pattern: string): GrammarConfig {
+        const napiGrammarConfig: NapiGrammarConfig = { type: 'Regex', pattern: pattern };
+        return new GrammarConfig(napiGrammarConfig);
+    }
+
+    static builtinJson(): GrammarConfig {
+        const napiGrammarConfig: NapiGrammarConfig = { type: 'BuiltinJson' };
+        return new GrammarConfig(napiGrammarConfig);
+    }
+
+    static fromType(type: z.ZodType): GrammarConfig {
+        let schema = z.toJSONSchema(type);
+        let schemaString = JSON.stringify(schema);
+        return GrammarConfig.jsonSchema(schemaString);
+    }
+
+    toNapi(): NapiGrammarConfig {
+        return this.napiGrammarConfig;
+    }
+}

package/src/bridging/parsedText.ts

@@ -1,3 +1,4 @@
+import * as z from 'zod';
 import { ParsedText as NapiParsedText } from '../napi/uzu';
 
 export class ParsedText {
@@ -14,4 +15,13 @@ export class ParsedText {
         const response = napiEntity.response ?? null;
         return new ParsedText(chainOfThought, response);
     }
+
+    structuredResponse<T extends z.ZodType>(type: T): z.infer<T> | null {
+        if (this.response === null) {
+            return null;
+        }
+        const data = JSON.parse(this.response);
+        const result = type.parse(data);
+        return result;
+    }
 }

package/src/bridging/runConfig.ts

@@ -1,4 +1,5 @@
 import { RunConfig as NapiRunConfig } from '../napi/uzu';
+import { GrammarConfig } from './grammarConfig';
 import { ToNapi } from './napi';
 import { SamplingPolicy } from './samplingPolicy';
 
@@ -6,36 +7,51 @@ export class RunConfig implements ToNapi<NapiRunConfig> {
     readonly tokensLimit: number;
     readonly enableThinking: boolean;
     readonly samplingPolicy: SamplingPolicy;
+    readonly grammarConfig: GrammarConfig | null;
 
-    private constructor(tokensLimit: number, enableThinking: boolean, samplingPolicy: SamplingPolicy) {
+    private constructor(
+        tokensLimit: number,
+        enableThinking: boolean,
+        samplingPolicy: SamplingPolicy,
+        grammarConfig: GrammarConfig | null,
+    ) {
         this.tokensLimit = tokensLimit;
         this.enableThinking = enableThinking;
         this.samplingPolicy = samplingPolicy;
+        this.grammarConfig = grammarConfig;
     }
 
     static default(): RunConfig {
-        return new RunConfig(1024, true, SamplingPolicy.default());
+        return new RunConfig(1024, true, SamplingPolicy.default(), null);
     }
 
     withTokensLimit(tokensLimit: number): RunConfig {
-        return new RunConfig(tokensLimit, this.enableThinking, this.samplingPolicy);
+        return new RunConfig(tokensLimit, this.enableThinking, this.samplingPolicy, this.grammarConfig);
     }
 
     withEnableThinking(enableThinking: boolean): RunConfig {
-        return new RunConfig(this.tokensLimit, enableThinking, this.samplingPolicy);
+        return new RunConfig(this.tokensLimit, enableThinking, this.samplingPolicy, this.grammarConfig);
     }
 
     withSamplingPolicy(samplingPolicy: SamplingPolicy): RunConfig {
-        return new RunConfig(this.tokensLimit, this.enableThinking, samplingPolicy);
+        return new RunConfig(this.tokensLimit, this.enableThinking, samplingPolicy, this.grammarConfig);
+    }
+
+    withGrammarConfig(grammarConfig: GrammarConfig | null): RunConfig {
+        return new RunConfig(this.tokensLimit, this.enableThinking, this.samplingPolicy, grammarConfig);
     }
 
     toNapi(): NapiRunConfig {
         const napiSamplingPolicy = this.samplingPolicy.toNapi();
-        const napiRunConfig: NapiRunConfig = {
+        let napiRunConfig: NapiRunConfig = {
             tokensLimit: this.tokensLimit,
             enableThinking: this.enableThinking,
             samplingPolicy: napiSamplingPolicy,
         };
+        if (this.grammarConfig !== null) {
+            let napiGrammarConfig = this.grammarConfig.toNapi();
+            napiRunConfig.grammarConfig = napiGrammarConfig;
+        }
         return napiRunConfig;
     }
 }

package/src/bridging/samplingMethod.ts

@@ -13,7 +13,12 @@ export class SamplingMethod implements ToNapi<NapiSamplingMethod> {
         return new SamplingMethod(napiSamplingMethod);
     }
 
-    static stochastic(temperature: number | null, topK: number | null, topP: number | null): SamplingMethod {
+    static stochastic(
+        temperature: number | null,
+        topK: number | null,
+        topP: number | null,
+        minP: number | null,
+    ): SamplingMethod {
         let napiSamplingMethod: NapiSamplingMethod = {
             type: 'Stochastic',
         };
@@ -26,6 +31,9 @@ export class SamplingMethod implements ToNapi<NapiSamplingMethod> {
         if (topP !== null) {
             napiSamplingMethod.topP = topP;
         }
+        if (minP !== null) {
+            napiSamplingMethod.minP = minP;
+        }
         return new SamplingMethod(napiSamplingMethod);
     }
 

package/src/index.ts

@@ -10,6 +10,7 @@ export { DownloadProgressUpdate } from './bridging/downloadProgressUpdate';
 export { DownloadPhase, DownloadState } from './bridging/downloadState';
 export { Engine as default } from './bridging/engine';
 export { FinishReason } from './bridging/finishReason';
+export { GrammarConfig } from './bridging/grammarConfig';
 export { Input } from './bridging/input';
 export { Message } from './bridging/message';
 export { ModelType } from './bridging/modelType';

package/src/interactors/chatSessionInteractor.ts

@@ -1,4 +1,5 @@
 import { ChatSession } from '../bridging/chatSession';
+import { GrammarConfig } from '../bridging/grammarConfig';
 import { Input } from '../bridging/input';
 import { Message } from '../bridging/message';
 import { Output } from '../bridging/output';
@@ -49,6 +50,11 @@ export class ChatSessionInteractor implements Interactor<ChatSession> {
         return new ChatSessionInteractor(this.modelInteractor, this.entity, config);
     }
 
+    grammarConfig(grammarConfig: GrammarConfig | null): ChatSessionInteractor {
+        const config = this.config.withGrammarConfig(grammarConfig);
+        return new ChatSessionInteractor(this.modelInteractor, this.entity, config);
+    }
+
     /* Run */
 
     async run(input: Input, progress: (partialOutput: Output) => boolean = () => true): Promise<Output> {

package/src/napi/uzu.d.ts

@@ -1,7 +1,7 @@
 /* auto-generated by NAPI-RS */
 /* eslint-disable */
 export declare class ChatSession {
-  run(input: Input, config: RunConfig, progress?: (arg: Output) => boolean | undefined | null): Output
+  run(input: Input, config: RunConfig, progress?: ((arg: Output) => boolean) | undefined | null): Output
   reset(): void
 }
 
@@ -12,9 +12,9 @@ export declare class Engine {
   createModelDownloadHandle(repoId: string): ModelDownloadHandle
   constructor()
   createChatSession(model: ChatModel, config: Config): ChatSession
-  registerLicenseStatusHandler(callback?: (arg: LicenseStatus) => void | undefined | null): void
-  registerModelDownloadStateHandler(callback?: (arg0: string, arg1: ModelDownloadState) => void | undefined | null): void
-  registerChatModelsHandler(callback?: (arg: Array<ChatModel>) => void | undefined | null): void
+  registerLicenseStatusHandler(callback?: ((arg: LicenseStatus) => void) | undefined | null): void
+  registerModelDownloadStateHandler(callback?: ((arg0: string, arg1: ModelDownloadState) => void) | undefined | null): void
+  registerChatModelsHandler(callback?: ((arg: Array<ChatModel>) => void) | undefined | null): void
 }
 
 export declare class ModelDownloadHandle {
@@ -83,6 +83,11 @@ export declare const enum FinishReason {
   ContextLimitReached = 3
 }
 
+export type GrammarConfig =
+  | { type: 'JsonSchema', schema: string }
+  | { type: 'Regex', pattern: string }
+  | { type: 'BuiltinJson' }
+
 export type Input =
   | { type: 'Text', text: string }
   | { type: 'Messages', messages: Array<Message> }
@@ -170,6 +175,7 @@ export interface RunConfig {
   tokensLimit: number
   enableThinking: boolean
   samplingPolicy: SamplingPolicy
+  grammarConfig?: GrammarConfig
 }
 
 export interface RunStats {
@@ -179,7 +185,7 @@ export interface RunStats {
 
 export type SamplingMethod =
   | { type: 'Greedy' }
-  | { type: 'Stochastic', temperature?: number, topK?: number, topP?: number }
+  | { type: 'Stochastic', temperature?: number, topK?: number, topP?: number, minP?: number }
 
 export type SamplingPolicy =
   | { type: 'Default' }

package/src/napi/uzu.js

@@ -47,6 +47,7 @@ const {
   AsyncBatchSize,
   ContextLength,
   ContextMode,
+  GrammarConfig,
   Input,
   LicenseStatus,
   ModelStorageError,
@@ -85,6 +86,7 @@ module.exports = {
   AsyncBatchSize,
   ContextLength,
   ContextMode,
+  GrammarConfig,
   Input,
   LicenseStatus,
   ModelStorageError,

package/src/napi/uzu.mjs

@@ -47,6 +47,7 @@ export const {
   AsyncBatchSize,
   ContextLength,
   ContextMode,
+  GrammarConfig,
   Input,
   LicenseStatus,
   ModelStorageError,

package/src/version.ts

@@ -1 +1 @@
-export const VERSION = '0.1.50';
+export const VERSION = '0.2.0';

package/version.d.mts

@@ -1,2 +1,2 @@
-export declare const VERSION = "0.1.50";
+export declare const VERSION = "0.2.0";
 //# sourceMappingURL=version.d.mts.map

package/version.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"version.d.mts","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,OAAO,WAAW,CAAC"}
+{"version":3,"file":"version.d.mts","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,OAAO,UAAU,CAAC"}

package/version.d.ts

@@ -1,2 +1,2 @@
-export declare const VERSION = "0.1.50";
+export declare const VERSION = "0.2.0";
 //# sourceMappingURL=version.d.ts.map

package/version.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"version.d.ts","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,OAAO,WAAW,CAAC"}
+{"version":3,"file":"version.d.ts","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,OAAO,UAAU,CAAC"}

package/version.js

@@ -1,5 +1,5 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.VERSION = void 0;
-exports.VERSION = '0.1.50';
+exports.VERSION = '0.2.0';
 //# sourceMappingURL=version.js.map

package/version.js.map

@@ -1 +1 @@
-{"version":3,"file":"version.js","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":";;;AAAa,QAAA,OAAO,GAAG,QAAQ,CAAC"}
+{"version":3,"file":"version.js","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":";;;AAAa,QAAA,OAAO,GAAG,OAAO,CAAC"}

package/version.mjs

@@ -1,2 +1,2 @@
-export const VERSION = '0.1.50';
+export const VERSION = '0.2.0';
 //# sourceMappingURL=version.mjs.map

package/version.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"version.mjs","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,OAAO,GAAG,QAAQ,CAAC"}
+{"version":3,"file":"version.mjs","sourceRoot":"","sources":["src/version.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,CAAC"}