diff options
| author | sigoden <sigoden@gmail.com> | 2024-05-09 11:24:02 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-09 11:24:02 +0800 |
| commit | bc65e880be6da8c169cd7424c04b7d0048b4d422 (patch) | |
| tree | 6a99dcf66a68813ae51358324d884e5d17d20a1b /src/client/prompt_format.rs | |
| parent | 5d73768accc6a7133094f7522862978d6f901158 (diff) | |
| download | aichat-bc65e880be6da8c169cd7424c04b7d0048b4d422.tar.gz | |
feat: enhence roles with messages (#495)
Diffstat (limited to 'src/client/prompt_format.rs')
| -rw-r--r-- | src/client/prompt_format.rs | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/client/prompt_format.rs b/src/client/prompt_format.rs index 61647a8..8a90924 100644 --- a/src/client/prompt_format.rs +++ b/src/client/prompt_format.rs @@ -13,13 +13,13 @@ pub struct PromptFormat<'a> { pub const GENERIC_PROMPT_FORMAT: PromptFormat<'static> = PromptFormat { begin: "", - system_pre_message: "### System\n", + system_pre_message: "", system_post_message: "\n", - user_pre_message: "### User\n", + user_pre_message: "### Instruction:\n", user_post_message: "\n", - assistant_pre_message: "### Assistant\n", + assistant_pre_message: "### Response:\n", assistant_post_message: "\n", - end: "### Assistant\n", + end: "### Response:\n", }; pub const MISTRAL_PROMPT_FORMAT: PromptFormat<'static> = PromptFormat { |