Minor fix and new template

synw · Sep 19, 2023 · c69eeb6 · c69eeb6
1 parent c061e58
commit c69eeb6
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "modprompt",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "description": "Prompt templates for language models",
   "license": "MIT",
   "scripts": {

diff --git a/src/cls.ts b/src/cls.ts
@@ -37,6 +37,9 @@ class ModTemplate implements LmTemplate {
     this.name = tpl.name;
     this.user = tpl.user;
     this.assistant = tpl.assistant;
+    if (tpl?.system) {
+      this.system = tpl.system
+    }
     if (tpl?.shots) {
       this.shots = tpl.shots
     }

diff --git a/src/db.ts b/src/db.ts
@@ -116,14 +116,27 @@ const templates: Record<string, LmTemplate> = {
     },
   },
   "human_response": {
-    "name": "Guanaco",
+    "name": "Human response",
     "user": "### HUMAN:\n{prompt}",
     "assistant": "### RESPONSE:",
     "linebreaks": {
       "user": 2,
       "assistant": 1
     },
   },
+  "coding_assistant": {
+    "name": "Coding assistant",
+    "system": {
+      "schema": "{system}",
+      "message": "You are a coding assistant that will help the user to resolve the following instruction:"
+    },
+    "user": "### Instruction: {prompt}",
+    "assistant": "### Solution:",
+    "linebreaks": {
+      "user": 2,
+      "system": 1,
+    },
+  }
 };
 
 export { templates }