Version 0.1.2: fixes

Author: synw

package.json

@@ -1,6 +1,6 @@
 {
   "name": "modprompt",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "Prompt templates for language models",
   "license": "MIT",
   "scripts": {

src/cls.ts

@@ -82,17 +82,23 @@ class PromptTemplate {
       user: this.user,
       assistant: this.assistant,
     }
+    if (this?.prefix) {
+      res.prefix = this.prefix
+    }
     if (this?.system) {
-      this.system = this.system
+      res.system = this.system
     }
     if (this?.shots) {
-      this.shots = this.shots
+      res.shots = this.shots
+    }
+    if (this?.afterShot) {
+      res.afterShot = this.afterShot
     }
     if (this?.stop) {
-      this.stop = this.stop
+      res.stop = this.stop
     }
     if (this?.linebreaks) {
-      this.linebreaks = this.linebreaks
+      res.linebreaks = this.linebreaks
     }
     return res
   }

src/db.ts

@@ -35,15 +35,7 @@ const templates: Record<string, LmTemplate> = {
       "user": 0
     },
     "prefix": "<s>",
-  },
-  "llama_instruct": {
-    "id": "llama_instruct",
-    "name": "Llama instruct",
-    "user": "[INST] {prompt}",
-    "assistant": " [/INST]",
-    "linebreaks": {
-      "user": 1
-    },
+    "stop": ["</s>"],
   },
   "amazon": {
     "id": "amazon",