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",