add back beam search

cosmicoptima · Nov 5, 2023 · beafcdd · beafcdd
1 parent 1a2c1e2
commit beafcdd
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 3 deletions.
diff --git a/common.ts b/common.ts
@@ -26,6 +26,7 @@ export interface LoomSettings {
   topP: number;
   frequencyPenalty: number;
   presencePenalty: number;
+  bestOf: number;
   n: number;
 
   showSettings: boolean;

diff --git a/main.ts b/main.ts
@@ -73,6 +73,7 @@ const DEFAULT_SETTINGS: LoomSettings = {
   topP: 1,
   frequencyPenalty: 0,
   presencePenalty: 0,
+  bestOf: 0,
   n: 5,
 
   showSettings: false,
@@ -1274,7 +1275,8 @@ export default class LoomPlugin extends Plugin {
       body: JSON.stringify({
         prompt,
         max_tokens: this.settings.maxTokens,
-        n: this.settings.n,
+        best_of: this.settings.bestOf,
+		n: this.settings.n,
         temperature: this.settings.temperature,
         top_p: this.settings.topP,
 	    frequency_penalty: this.settings.frequencyPenalty,

diff --git a/manifest.json b/manifest.json
@@ -1,7 +1,7 @@
 {
 	"id": "loom",
 	"name": "Loom",
-	"version": "1.16.4",
+	"version": "1.17.0",
 	"minAppVersion": "0.15.0",
 	"description": "Loom in Obsidian",
 	"author": "celeste",

diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
 	"name": "obsidian-loom",
-	"version": "1.16.4",
+	"version": "1.17.0",
 	"description": "Loom in Obsidian",
 	"main": "main.js",
 	"scripts": {

diff --git a/views.ts b/views.ts
@@ -322,6 +322,7 @@ export class LoomView extends ItemView {
 	setting("Context length", "contextLength", String(settings.contextLength), "int");
 	setting("Length (in tokens)", "maxTokens", String(settings.maxTokens), "int");
 	setting("Number of completions", "n", String(settings.n), "int");
+    setting("Best of (0 = no beam search)", "bestOf", String(settings.bestOf), "int");
 	setting("Temperature", "temperature", String(settings.temperature), "float");
 	setting("Top p", "topP", String(settings.topP), "float");
 	setting("Frequency penalty", "frequencyPenalty", String(settings.frequencyPenalty), "float");