@trymirai/uzu 0.1.45 → 0.1.47

package/README.md

@@ -26,7 +26,7 @@ Add the `uzu` dependency to your project's `package.json`:
 
 ```json
 "dependencies": {
-    "@trymirai/uzu": "0.1.45"
+    "@trymirai/uzu": "0.1.47"
 }
 ```
 

package/bridging/chatSession.d.mts

@@ -6,5 +6,6 @@ export declare class ChatSession {
     private readonly napiChatSession;
     constructor(napiChatSession: NapiChatSession);
     run(input: Input, config?: RunConfig, progress?: (partialOutput: Output) => boolean): Output;
+    reset(): void;
 }
 //# sourceMappingURL=chatSession.d.mts.map

package/bridging/chatSession.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"chatSession.d.mts","sourceRoot":"","sources":["../src/bridging/chatSession.ts"],"names":[],"mappings":"OAAO,EAAE,WAAW,IAAI,eAAe,EAAwB;OACxD,EAAE,KAAK,EAAE;OACT,EAAE,MAAM,EAAE;OACV,EAAE,SAAS,EAAE;AAEpB,qBAAa,WAAW;IACpB,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAkB;gBAEtC,eAAe,EAAE,eAAe;IAI5C,GAAG,CACC,KAAK,EAAE,KAAK,EACZ,MAAM,GAAE,SAA+B,EACvC,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAC1D,MAAM;CAOZ"}
+{"version":3,"file":"chatSession.d.mts","sourceRoot":"","sources":["../src/bridging/chatSession.ts"],"names":[],"mappings":"OAAO,EAAE,WAAW,IAAI,eAAe,EAAwB;OACxD,EAAE,KAAK,EAAE;OACT,EAAE,MAAM,EAAE;OACV,EAAE,SAAS,EAAE;AAEpB,qBAAa,WAAW;IACpB,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAkB;gBAEtC,eAAe,EAAE,eAAe;IAI5C,GAAG,CACC,KAAK,EAAE,KAAK,EACZ,MAAM,GAAE,SAA+B,EACvC,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAC1D,MAAM;IAQT,KAAK,IAAI,IAAI;CAGhB"}

package/bridging/chatSession.d.ts

@@ -6,5 +6,6 @@ export declare class ChatSession {
     private readonly napiChatSession;
     constructor(napiChatSession: NapiChatSession);
     run(input: Input, config?: RunConfig, progress?: (partialOutput: Output) => boolean): Output;
+    reset(): void;
 }
 //# sourceMappingURL=chatSession.d.ts.map

package/bridging/chatSession.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chatSession.d.ts","sourceRoot":"","sources":["../src/bridging/chatSession.ts"],"names":[],"mappings":"OAAO,EAAE,WAAW,IAAI,eAAe,EAAwB;OACxD,EAAE,KAAK,EAAE;OACT,EAAE,MAAM,EAAE;OACV,EAAE,SAAS,EAAE;AAEpB,qBAAa,WAAW;IACpB,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAkB;gBAEtC,eAAe,EAAE,eAAe;IAI5C,GAAG,CACC,KAAK,EAAE,KAAK,EACZ,MAAM,GAAE,SAA+B,EACvC,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAC1D,MAAM;CAOZ"}
+{"version":3,"file":"chatSession.d.ts","sourceRoot":"","sources":["../src/bridging/chatSession.ts"],"names":[],"mappings":"OAAO,EAAE,WAAW,IAAI,eAAe,EAAwB;OACxD,EAAE,KAAK,EAAE;OACT,EAAE,MAAM,EAAE;OACV,EAAE,SAAS,EAAE;AAEpB,qBAAa,WAAW;IACpB,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAkB;gBAEtC,eAAe,EAAE,eAAe;IAI5C,GAAG,CACC,KAAK,EAAE,KAAK,EACZ,MAAM,GAAE,SAA+B,EACvC,QAAQ,GAAE,CAAC,aAAa,EAAE,MAAM,KAAK,OAAoB,GAC1D,MAAM;IAQT,KAAK,IAAI,IAAI;CAGhB"}

package/bridging/chatSession.js

@@ -14,6 +14,9 @@ class ChatSession {
         const napiOutput = this.napiChatSession.run(napiInput, napiConfig, napiProgress);
         return output_1.Output.fromNapi(napiOutput);
     }
+    reset() {
+        this.napiChatSession.reset();
+    }
 }
 exports.ChatSession = ChatSession;
 //# sourceMappingURL=chatSession.js.map

package/bridging/chatSession.js.map

@@ -1 +1 @@
-{"version":3,"file":"chatSession.js","sourceRoot":"","sources":["../src/bridging/chatSession.ts"],"names":[],"mappings":";;;AAEA,wCAAkC;AAClC,8CAAwC;AAExC,MAAa,WAAW;IAGpB,YAAY,eAAgC;QACxC,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IAC3C,CAAC;IAED,GAAG,CACC,KAAY,EACZ,SAAoB,qBAAS,CAAC,OAAO,EAAE,EACvC,WAA+C,GAAG,EAAE,CAAC,IAAI;QAEzD,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACjC,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QACnC,MAAM,YAAY,GAAG,CAAC,aAAyB,EAAE,EAAE,CAAC,QAAQ,CAAC,eAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC;QAC7F,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,EAAE,YAAY,CAAC,CAAC;QACjF,OAAO,eAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IACvC,CAAC;CACJ;AAlBD,kCAkBC"}
+{"version":3,"file":"chatSession.js","sourceRoot":"","sources":["../src/bridging/chatSession.ts"],"names":[],"mappings":";;;AAEA,wCAAkC;AAClC,8CAAwC;AAExC,MAAa,WAAW;IAGpB,YAAY,eAAgC;QACxC,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IAC3C,CAAC;IAED,GAAG,CACC,KAAY,EACZ,SAAoB,qBAAS,CAAC,OAAO,EAAE,EACvC,WAA+C,GAAG,EAAE,CAAC,IAAI;QAEzD,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACjC,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QACnC,MAAM,YAAY,GAAG,CAAC,aAAyB,EAAE,EAAE,CAAC,QAAQ,CAAC,eAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC;QAC7F,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,EAAE,YAAY,CAAC,CAAC;QACjF,OAAO,eAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IACvC,CAAC;IAED,KAAK;QACD,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;IACjC,CAAC;CACJ;AAtBD,kCAsBC"}

package/bridging/chatSession.mjs

@@ -11,5 +11,8 @@ export class ChatSession {
         const napiOutput = this.napiChatSession.run(napiInput, napiConfig, napiProgress);
         return Output.fromNapi(napiOutput);
     }
+    reset() {
+        this.napiChatSession.reset();
+    }
 }
 //# sourceMappingURL=chatSession.mjs.map

package/bridging/chatSession.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"chatSession.mjs","sourceRoot":"","sources":["../src/bridging/chatSession.ts"],"names":[],"mappings":"OAEO,EAAE,MAAM,EAAE;OACV,EAAE,SAAS,EAAE;AAEpB,MAAM,OAAO,WAAW;IAGpB,YAAY,eAAgC;QACxC,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IAC3C,CAAC;IAED,GAAG,CACC,KAAY,EACZ,SAAoB,SAAS,CAAC,OAAO,EAAE,EACvC,WAA+C,GAAG,EAAE,CAAC,IAAI;QAEzD,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACjC,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QACnC,MAAM,YAAY,GAAG,CAAC,aAAyB,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC;QAC7F,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,EAAE,YAAY,CAAC,CAAC;QACjF,OAAO,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IACvC,CAAC;CACJ"}
+{"version":3,"file":"chatSession.mjs","sourceRoot":"","sources":["../src/bridging/chatSession.ts"],"names":[],"mappings":"OAEO,EAAE,MAAM,EAAE;OACV,EAAE,SAAS,EAAE;AAEpB,MAAM,OAAO,WAAW;IAGpB,YAAY,eAAgC;QACxC,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;IAC3C,CAAC;IAED,GAAG,CACC,KAAY,EACZ,SAAoB,SAAS,CAAC,OAAO,EAAE,EACvC,WAA+C,GAAG,EAAE,CAAC,IAAI;QAEzD,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;QACjC,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;QACnC,MAAM,YAAY,GAAG,CAAC,aAAyB,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC;QAC7F,MAAM,UAAU,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,EAAE,YAAY,CAAC,CAAC;QACjF,OAAO,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;IACvC,CAAC;IAED,KAAK;QACD,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;IACjC,CAAC;CACJ"}

package/bridging/samplingMethod.d.mts

@@ -4,8 +4,7 @@ export declare class SamplingMethod implements ToNapi<NapiSamplingMethod> {
     private readonly napiSamplingMethod;
     private constructor();
     static greedy(): SamplingMethod;
-    static temperature(temperature: number): SamplingMethod;
-    static topP(topP: number): SamplingMethod;
+    static stochastic(temperature: number | null, topK: number | null, topP: number | null): SamplingMethod;
     toNapi(): NapiSamplingMethod;
 }
 //# sourceMappingURL=samplingMethod.d.mts.map

package/bridging/samplingMethod.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"samplingMethod.d.mts","sourceRoot":"","sources":["../src/bridging/samplingMethod.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,IAAI,kBAAkB,EAAE;OACxC,EAAE,MAAM,EAAE;AAEjB,qBAAa,cAAe,YAAW,MAAM,CAAC,kBAAkB,CAAC;IAC7D,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAqB;IAExD,OAAO;IAIP,MAAM,CAAC,MAAM,IAAI,cAAc;IAK/B,MAAM,CAAC,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,cAAc;IAQvD,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;IAKzC,MAAM,IAAI,kBAAkB;CAG/B"}
+{"version":3,"file":"samplingMethod.d.mts","sourceRoot":"","sources":["../src/bridging/samplingMethod.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,IAAI,kBAAkB,EAAE;OACxC,EAAE,MAAM,EAAE;AAEjB,qBAAa,cAAe,YAAW,MAAM,CAAC,kBAAkB,CAAC;IAC7D,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAqB;IAExD,OAAO;IAIP,MAAM,CAAC,MAAM,IAAI,cAAc;IAK/B,MAAM,CAAC,UAAU,CAAC,WAAW,EAAE,MAAM,GAAG,IAAI,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI,GAAG,cAAc;IAgBvG,MAAM,IAAI,kBAAkB;CAG/B"}

package/bridging/samplingMethod.d.ts

@@ -4,8 +4,7 @@ export declare class SamplingMethod implements ToNapi<NapiSamplingMethod> {
     private readonly napiSamplingMethod;
     private constructor();
     static greedy(): SamplingMethod;
-    static temperature(temperature: number): SamplingMethod;
-    static topP(topP: number): SamplingMethod;
+    static stochastic(temperature: number | null, topK: number | null, topP: number | null): SamplingMethod;
     toNapi(): NapiSamplingMethod;
 }
 //# sourceMappingURL=samplingMethod.d.ts.map

package/bridging/samplingMethod.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"samplingMethod.d.ts","sourceRoot":"","sources":["../src/bridging/samplingMethod.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,IAAI,kBAAkB,EAAE;OACxC,EAAE,MAAM,EAAE;AAEjB,qBAAa,cAAe,YAAW,MAAM,CAAC,kBAAkB,CAAC;IAC7D,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAqB;IAExD,OAAO;IAIP,MAAM,CAAC,MAAM,IAAI,cAAc;IAK/B,MAAM,CAAC,WAAW,CAAC,WAAW,EAAE,MAAM,GAAG,cAAc;IAQvD,MAAM,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;IAKzC,MAAM,IAAI,kBAAkB;CAG/B"}
+{"version":3,"file":"samplingMethod.d.ts","sourceRoot":"","sources":["../src/bridging/samplingMethod.ts"],"names":[],"mappings":"OAAO,EAAE,cAAc,IAAI,kBAAkB,EAAE;OACxC,EAAE,MAAM,EAAE;AAEjB,qBAAa,cAAe,YAAW,MAAM,CAAC,kBAAkB,CAAC;IAC7D,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAqB;IAExD,OAAO;IAIP,MAAM,CAAC,MAAM,IAAI,cAAc;IAK/B,MAAM,CAAC,UAAU,CAAC,WAAW,EAAE,MAAM,GAAG,IAAI,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI,GAAG,cAAc;IAgBvG,MAAM,IAAI,kBAAkB;CAG/B"}

package/bridging/samplingMethod.js

@@ -9,15 +9,19 @@ class SamplingMethod {
         const napiSamplingMethod = { type: 'Greedy' };
         return new SamplingMethod(napiSamplingMethod);
     }
-    static temperature(temperature) {
-        const napiSamplingMethod = {
-            type: 'Temperature',
-            temperature,
+    static stochastic(temperature, topK, topP) {
+        let napiSamplingMethod = {
+            type: 'Stochastic',
         };
-        return new SamplingMethod(napiSamplingMethod);
-    }
-    static topP(topP) {
-        const napiSamplingMethod = { type: 'TopP', topP };
+        if (temperature !== null) {
+            napiSamplingMethod.temperature = temperature;
+        }
+        if (topK !== null) {
+            napiSamplingMethod.topK = topK;
+        }
+        if (topP !== null) {
+            napiSamplingMethod.topP = topP;
+        }
         return new SamplingMethod(napiSamplingMethod);
     }
     toNapi() {

package/bridging/samplingMethod.js.map

@@ -1 +1 @@
-{"version":3,"file":"samplingMethod.js","sourceRoot":"","sources":["../src/bridging/samplingMethod.ts"],"names":[],"mappings":";;;AAGA,MAAa,cAAc;IAGvB,YAAoB,kBAAsC;QACtD,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;IACjD,CAAC;IAED,MAAM,CAAC,MAAM;QACT,MAAM,kBAAkB,GAAuB,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QAClE,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,CAAC,WAAW,CAAC,WAAmB;QAClC,MAAM,kBAAkB,GAAuB;YAC3C,IAAI,EAAE,aAAa;YACnB,WAAW;SACd,CAAC;QACF,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,IAAY;QACpB,MAAM,kBAAkB,GAAuB,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QACtE,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM;QACF,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACnC,CAAC;CACJ;AA5BD,wCA4BC"}
+{"version":3,"file":"samplingMethod.js","sourceRoot":"","sources":["../src/bridging/samplingMethod.ts"],"names":[],"mappings":";;;AAGA,MAAa,cAAc;IAGvB,YAAoB,kBAAsC;QACtD,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;IACjD,CAAC;IAED,MAAM,CAAC,MAAM;QACT,MAAM,kBAAkB,GAAuB,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QAClE,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,CAAC,UAAU,CAAC,WAA0B,EAAE,IAAmB,EAAE,IAAmB;QAClF,IAAI,kBAAkB,GAAuB;YACzC,IAAI,EAAE,YAAY;SACrB,CAAC;QACF,IAAI,WAAW,KAAK,IAAI,EAAE,CAAC;YACvB,kBAAkB,CAAC,WAAW,GAAG,WAAW,CAAC;QACjD,CAAC;QACD,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAChB,kBAAkB,CAAC,IAAI,GAAG,IAAI,CAAC;QACnC,CAAC;QACD,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAChB,kBAAkB,CAAC,IAAI,GAAG,IAAI,CAAC;QACnC,CAAC;QACD,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM;QACF,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACnC,CAAC;CACJ;AA/BD,wCA+BC"}

package/bridging/samplingMethod.mjs

@@ -6,15 +6,19 @@ export class SamplingMethod {
         const napiSamplingMethod = { type: 'Greedy' };
         return new SamplingMethod(napiSamplingMethod);
     }
-    static temperature(temperature) {
-        const napiSamplingMethod = {
-            type: 'Temperature',
-            temperature,
+    static stochastic(temperature, topK, topP) {
+        let napiSamplingMethod = {
+            type: 'Stochastic',
         };
-        return new SamplingMethod(napiSamplingMethod);
-    }
-    static topP(topP) {
-        const napiSamplingMethod = { type: 'TopP', topP };
+        if (temperature !== null) {
+            napiSamplingMethod.temperature = temperature;
+        }
+        if (topK !== null) {
+            napiSamplingMethod.topK = topK;
+        }
+        if (topP !== null) {
+            napiSamplingMethod.topP = topP;
+        }
         return new SamplingMethod(napiSamplingMethod);
     }
     toNapi() {

package/bridging/samplingMethod.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"samplingMethod.mjs","sourceRoot":"","sources":["../src/bridging/samplingMethod.ts"],"names":[],"mappings":"AAGA,MAAM,OAAO,cAAc;IAGvB,YAAoB,kBAAsC;QACtD,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;IACjD,CAAC;IAED,MAAM,CAAC,MAAM;QACT,MAAM,kBAAkB,GAAuB,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QAClE,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,CAAC,WAAW,CAAC,WAAmB;QAClC,MAAM,kBAAkB,GAAuB;YAC3C,IAAI,EAAE,aAAa;YACnB,WAAW;SACd,CAAC;QACF,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,IAAY;QACpB,MAAM,kBAAkB,GAAuB,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;QACtE,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM;QACF,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACnC,CAAC;CACJ"}
+{"version":3,"file":"samplingMethod.mjs","sourceRoot":"","sources":["../src/bridging/samplingMethod.ts"],"names":[],"mappings":"AAGA,MAAM,OAAO,cAAc;IAGvB,YAAoB,kBAAsC;QACtD,IAAI,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;IACjD,CAAC;IAED,MAAM,CAAC,MAAM;QACT,MAAM,kBAAkB,GAAuB,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;QAClE,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,CAAC,UAAU,CAAC,WAA0B,EAAE,IAAmB,EAAE,IAAmB;QAClF,IAAI,kBAAkB,GAAuB;YACzC,IAAI,EAAE,YAAY;SACrB,CAAC;QACF,IAAI,WAAW,KAAK,IAAI,EAAE,CAAC;YACvB,kBAAkB,CAAC,WAAW,GAAG,WAAW,CAAC;QACjD,CAAC;QACD,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAChB,kBAAkB,CAAC,IAAI,GAAG,IAAI,CAAC;QACnC,CAAC;QACD,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAChB,kBAAkB,CAAC,IAAI,GAAG,IAAI,CAAC;QACnC,CAAC;QACD,OAAO,IAAI,cAAc,CAAC,kBAAkB,CAAC,CAAC;IAClD,CAAC;IAED,MAAM;QACF,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACnC,CAAC;CACJ"}

package/napi/uzu.d.ts

@@ -2,6 +2,7 @@
 /* eslint-disable */
 export declare class ChatSession {
   run(input: Input, config: RunConfig, progress?: (arg: Output) => boolean | undefined | null): Output
+  reset(): void
 }
 
 export declare class Engine {
@@ -173,8 +174,7 @@ export interface RunStats {
 
 export type SamplingMethod =
   | { type: 'Greedy' }
-  | { type: 'Temperature', temperature: number }
-  | { type: 'TopP', topP: number }
+  | { type: 'Stochastic', temperature?: number, topK?: number, topP?: number }
 
 export type SamplingPolicy =
   | { type: 'Default' }

package/package.json

@@ -98,5 +98,5 @@
   },
   "type": "commonjs",
   "types": "./index.d.ts",
-  "version": "0.1.45"
+  "version": "0.1.47"
 }

package/src/bridging/chatSession.ts

@@ -21,4 +21,8 @@ export class ChatSession {
         const napiOutput = this.napiChatSession.run(napiInput, napiConfig, napiProgress);
         return Output.fromNapi(napiOutput);
     }
+
+    reset(): void {
+        this.napiChatSession.reset();
+    }
 }

package/src/bridging/samplingMethod.ts

@@ -13,16 +13,19 @@ export class SamplingMethod implements ToNapi<NapiSamplingMethod> {
         return new SamplingMethod(napiSamplingMethod);
     }
 
-    static temperature(temperature: number): SamplingMethod {
-        const napiSamplingMethod: NapiSamplingMethod = {
-            type: 'Temperature',
-            temperature,
+    static stochastic(temperature: number | null, topK: number | null, topP: number | null): SamplingMethod {
+        let napiSamplingMethod: NapiSamplingMethod = {
+            type: 'Stochastic',
         };
-        return new SamplingMethod(napiSamplingMethod);
-    }
-
-    static topP(topP: number): SamplingMethod {
-        const napiSamplingMethod: NapiSamplingMethod = { type: 'TopP', topP };
+        if (temperature !== null) {
+            napiSamplingMethod.temperature = temperature;
+        }
+        if (topK !== null) {
+            napiSamplingMethod.topK = topK;
+        }
+        if (topP !== null) {
+            napiSamplingMethod.topP = topP;
+        }
         return new SamplingMethod(napiSamplingMethod);
     }
 

package/src/napi/uzu.d.ts

@@ -2,6 +2,7 @@
 /* eslint-disable */
 export declare class ChatSession {
   run(input: Input, config: RunConfig, progress?: (arg: Output) => boolean | undefined | null): Output
+  reset(): void
 }
 
 export declare class Engine {
@@ -173,8 +174,7 @@ export interface RunStats {
 
 export type SamplingMethod =
   | { type: 'Greedy' }
-  | { type: 'Temperature', temperature: number }
-  | { type: 'TopP', topP: number }
+  | { type: 'Stochastic', temperature?: number, topK?: number, topP?: number }
 
 export type SamplingPolicy =
   | { type: 'Default' }

package/src/version.ts

@@ -1 +1 @@
-export const VERSION = '0.1.45';
+export const VERSION = '0.1.47';

package/version.d.mts

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

package/version.d.ts

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

package/version.js

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

package/version.mjs

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