add removeLatestMsg

2023-10-09 14:28:51 +00:00
parent 7187c03692
commit 13cc77150a
1 changed files with 6 additions and 0 deletions
--- 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]<<SYS>> content <</SYS>>")