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