From 13cc77150aee852d02ac377910e491cde1c1d84b Mon Sep 17 00:00:00 2001 From: narawat Date: Mon, 9 Oct 2023 14:28:51 +0000 Subject: [PATCH] add removeLatestMsg --- src/interface.jl | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/interface.jl b/src/interface.jl index d29e392..6b7f715 100644 --- a/src/interface.jl +++ b/src/interface.jl @@ -72,6 +72,12 @@ function clearMessage(a::agent) end end +function removeLatestMsg(a::agent) + if length(a.messages) > 1 + pop!(a.messages) + end +end + function generatePromt(a::agent, userToken::String=" [/INST]", assistantToken=" [INST]", systemToken="[INST]<> content <>")