From b12261e412e90dc4acb50d4d5b313fe0fc27d90d Mon Sep 17 00:00:00 2001
From: synw <synwx@protonmail.com>
Date: Wed, 27 Dec 2023 12:29:31 +0100
Subject: [PATCH] Version 0.3.6: add Phi and Minichat prompt templates

---
 package.json |  2 +-
 src/db.ts    | 26 ++++++++++++++++++--------
 2 files changed, 19 insertions(+), 9 deletions(-)

diff --git a/package.json b/package.json
index c010151..5aa14a3 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "modprompt",
-  "version": "0.3.5",
+  "version": "0.3.6",
   "description": "Prompt templates for language models",
   "license": "MIT",
   "scripts": {
diff --git a/src/db.ts b/src/db.ts
index fd2e665..574c497 100644
--- a/src/db.ts
+++ b/src/db.ts
@@ -37,14 +37,6 @@ const templates: Record<string, LmTemplate> = {
     "prefix": "<s>",
     "stop": ["</s>"],
   },
-  "amazon": {
-    "id": "amazon",
-    "name": "Amazon",
-    "user": "<|prompter|>{prompt}</s>",
-    "assistant": "<|assistant|>",
-    "afterShot": "\n",
-    "stop": ["</s>"],
-  },
   "mistral": {
     "id": "mistral",
     "name": "Mistral",
@@ -183,6 +175,24 @@ const templates: Record<string, LmTemplate> = {
       "user": 2,
       "assistant": 1
     },
+  },
+  "minichat": {
+    "id": "minichat",
+    "name": "Minichat",
+    "user": "<s> [|User|] {prompt} </s>",
+    "assistant": "[|Assistant|]",
+    "stop": ["</s>", "[|User|]"],
+    "afterShot": "\n",
+  },
+  "phi": {
+    "id": "phi",
+    "name": "Phi",
+    "user": "Instruct: {prompt}",
+    "assistant": "Output:",
+    "linebreaks": {
+      "user": 1
+    },
+    "stop": ["</s>", "Instruct:"]
   }
 };