diff --git a/codegen/db.yml b/codegen/db.yml
index a6dbb50..9240f29 100644
--- a/codegen/db.yml
+++ b/codegen/db.yml
@@ -9,19 +9,6 @@ alpaca:
linebreaks:
system: 2
user: 2
-cerebrum:
- id: cerebrum
- name: Cerebrum
- system:
- schema: "{system}"
- message: "A chat between a user and a thinking artificial intelligence assistant. The assistant describes its thought process and gives helpful and detailed answers to the user's questions."
- user: "User: {prompt}"
- assistant: "Ai:"
- linebreaks:
- user: 1
- prefix: ""
- stop:
- - ""
chatml:
id: "chatml"
name: "ChatMl"
@@ -110,13 +97,6 @@ gemma:
stop:
- ""
afterShot: "\n"
-guanaco:
- id: guanaco
- name: Guanaco
- user: "### Human: {prompt}"
- assistant: "### Assistant:"
- linebreaks:
- user: 1
human_response:
id: human_response
name: Human response
@@ -175,6 +155,16 @@ mistral:
stop:
- ""
afterShot: "\n"
+mistral-system:
+ id: mistral-system
+ name: Mistral system
+ user: "[INST]{prompt}"
+ assistant: "[/INST]"
+ system:
+ schema: "[SYSTEM_PROMPT]{system_prompt}[/SYSTEM_PROMPT]"
+ stop:
+ - ""
+ afterShot: "\n"
nemotron:
id: nemotron
name: Nemotron
@@ -237,16 +227,6 @@ orca:
linebreaks:
system: 2
user: 2
-phi:
- id: phi
- name: Phi
- user: "Instruct: {prompt}"
- assistant: "Output:"
- linebreaks:
- user: 1
- stop:
- - ""
- - "Instruct:"
phi3:
id: phi3
name: Phi 3
@@ -269,17 +249,6 @@ phi4:
- "<|im_end|>"
- "<|im_sep|>"
afterShot: "<|im_end|>\n"
-synthia-cot:
- id: synthia-cot
- name: Synthia CoT
- system:
- schema: "SYSTEM: {system}"
- message: "Elaborate on the topic using a Tree of Thoughts and backtrack when necessary to construct a clear, cohesive Chain of Thought reasoning. Always answer without hesitation."
- user: "USER: {prompt}"
- assistant: "ASSISTANT:"
- linebreaks:
- system: 1
- user: 1
vicuna:
id: vicuna
name: Vicuna
diff --git a/package.json b/package.json
index e7e6396..5acc07d 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
{
"name": "modprompt",
- "version": "0.9.4",
+ "version": "0.9.5",
"description": "Prompt templates for language models",
"license": "MIT",
"scripts": {
diff --git a/src/db.ts b/src/db.ts
index 48c2e87..1d1b9bd 100644
--- a/src/db.ts
+++ b/src/db.ts
@@ -17,23 +17,6 @@ const templates: Record = {
},
"user": "### Instruction:\n{prompt}"
},
- "cerebrum": {
- "assistant": "Ai:",
- "id": "cerebrum",
- "linebreaks": {
- "user": 1
- },
- "name": "Cerebrum",
- "prefix": "",
- "stop": [
- ""
- ],
- "system": {
- "message": "A chat between a user and a thinking artificial intelligence assistant. The assistant describes its thought process and gives helpful and detailed answers to the user's questions.",
- "schema": "{system}"
- },
- "user": "User: {prompt}"
- },
"chatml": {
"afterShot": " <|im_end|>\n",
"assistant": "<|im_start|>assistant",
@@ -148,15 +131,6 @@ const templates: Record = {
],
"user": "user\n{prompt}"
},
- "guanaco": {
- "assistant": "### Assistant:",
- "id": "guanaco",
- "linebreaks": {
- "user": 1
- },
- "name": "Guanaco",
- "user": "### Human: {prompt}"
- },
"human_response": {
"assistant": "### RESPONSE:",
"id": "human_response",
@@ -230,6 +204,19 @@ const templates: Record = {
],
"user": "[INST] {prompt}"
},
+ "mistral-system": {
+ "afterShot": "\n",
+ "assistant": "[/INST]",
+ "id": "mistral-system",
+ "name": "Mistral system",
+ "stop": [
+ ""
+ ],
+ "system": {
+ "schema": "[SYSTEM_PROMPT]{system_prompt}[/SYSTEM_PROMPT]"
+ },
+ "user": "[INST]{prompt}"
+ },
"nemotron": {
"afterShot": "\n\n",
"assistant": "Assistant",
@@ -309,19 +296,6 @@ const templates: Record = {
},
"user": "### User:\n{prompt}"
},
- "phi": {
- "assistant": "Output:",
- "id": "phi",
- "linebreaks": {
- "user": 1
- },
- "name": "Phi",
- "stop": [
- "",
- "Instruct:"
- ],
- "user": "Instruct: {prompt}"
- },
"phi3": {
"afterShot": "<|end|>\n",
"assistant": "<|assistant|>",
@@ -350,20 +324,6 @@ const templates: Record = {
},
"user": "<|im_start|>user<|im_sep|>{prompt}<|im_end|>"
},
- "synthia-cot": {
- "assistant": "ASSISTANT:",
- "id": "synthia-cot",
- "linebreaks": {
- "system": 1,
- "user": 1
- },
- "name": "Synthia CoT",
- "system": {
- "message": "Elaborate on the topic using a Tree of Thoughts and backtrack when necessary to construct a clear, cohesive Chain of Thought reasoning. Always answer without hesitation.",
- "schema": "SYSTEM: {system}"
- },
- "user": "USER: {prompt}"
- },
"vicuna": {
"assistant": "### ASSISTANT:",
"id": "vicuna",