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 = { "prefix": "", "stop": [""], }, - "amazon": { - "id": "amazon", - "name": "Amazon", - "user": "<|prompter|>{prompt}", - "assistant": "<|assistant|>", - "afterShot": "\n", - "stop": [""], - }, "mistral": { "id": "mistral", "name": "Mistral", @@ -183,6 +175,24 @@ const templates: Record = { "user": 2, "assistant": 1 }, + }, + "minichat": { + "id": "minichat", + "name": "Minichat", + "user": " [|User|] {prompt} ", + "assistant": "[|Assistant|]", + "stop": ["", "[|User|]"], + "afterShot": "\n", + }, + "phi": { + "id": "phi", + "name": "Phi", + "user": "Instruct: {prompt}", + "assistant": "Output:", + "linebreaks": { + "user": 1 + }, + "stop": ["", "Instruct:"] } };