From 19c0cddbcc8bd2e243ccfbe8c3caef8823495b90 Mon Sep 17 00:00:00 2001
From: ton <narawat@gmail.com>
Date: Tue, 10 Oct 2023 09:11:01 +0700
Subject: [PATCH] update

---
 src/interface.jl | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/interface.jl b/src/interface.jl
index a1afdd5..9e63739 100644
--- a/src/interface.jl
+++ b/src/interface.jl
@@ -87,7 +87,7 @@ function generatePrompt_tokenSuffix(a::agent;
     content = msg[:content]
 
     if role == "system"
-      prompt = replace(systemToken, "content" => content)
+      prompt = replace(systemToken, "content" => content) * " "
     elseif role == "user"
       prompt *= " " * content * userToken
     elseif role == "assistant"
@@ -101,7 +101,7 @@ function generatePrompt_tokenSuffix(a::agent;
 end
 
 function generatePrompt_tokenPrefix(a::agent; 
-                      userToken::String=" [/INST]", assistantToken=" [INST]", 
+                      userToken::String="[/INST]", assistantToken="[INST]", 
                       systemToken="[INST]<<SYS>> content <</SYS>>")
   prompt = nothing
   for msg in a.messages
@@ -109,7 +109,7 @@ function generatePrompt_tokenPrefix(a::agent;
     content = msg[:content]
 
     if role == "system"
-      prompt = replace(systemToken, "content" => content)
+      prompt = replace(systemToken, "content" => content) * " "
     elseif role == "user"
       prompt *= userToken * content * " "
     elseif role == "assistant"