diff --git a/src/interface.jl b/src/interface.jl index 89859f8..e51ba7a 100644 --- a/src/interface.jl +++ b/src/interface.jl @@ -1,7 +1,7 @@ module interface -export agent, addNewMessage, clearMessage, removeLatestMsg, generatePromt +export agent, addNewMessage, clearMessage, removeLatestMsg, generatePrompt using JSON3, DataStructures @@ -78,7 +78,7 @@ function removeLatestMsg(a::agent) end end -function generatePromt(a::agent, +function generatePrompt(a::agent, userToken::String=" [/INST]", assistantToken=" [INST]", systemToken="[INST]<> content <>") prompt = nothing