diff --git a/src/interface.jl b/src/interface.jl index 2415b16..f664e30 100644 --- a/src/interface.jl +++ b/src/interface.jl @@ -4,7 +4,7 @@ module interface export agent, addNewMessage, clearMessage, removeLatestMsg, generatePrompt_tokenPrefix, generatePrompt_tokenSuffix -using JSON3, DataStructures +using JSON3, DataStructures, Dates #------------------------------------------------------------------------------------------------100 @@ -25,7 +25,7 @@ using JSON3, DataStructures """ # Ref: https://huggingface.co/TheBloke/Llama-2-7B-Chat-GGML/discussions/3 # - messages=[Dict(:role=>"system", :content=> "You are a helpful assistant."),] + messages=[Dict(:role=>"system", :content=> "You are a helpful assistant.", :timestamp=> Dates.now()),] end """ @@ -54,7 +54,7 @@ function addNewMessage(a::agent, role::String, content::String) messageleft = a.maxUserMsg else userMsg += 1 - d = Dict(:role=> role, :content=> content) + d = Dict(:role=> role, :content=> content, :timestamp=> Dates.now()) push!(a.messages, d) messageleft = a.maxUserMsg - userMsg end