From bc65e880be6da8c169cd7424c04b7d0048b4d422 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 9 May 2024 11:24:02 +0800
Subject: feat: enhence roles with messages (#495)

---
 src/client/prompt_format.rs | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

(limited to 'src/client/prompt_format.rs')

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 {
-- 
cgit v1.2.3