diff --git a/.gitignore b/.gitignore index 280cb2e..7c3252c 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,4 @@ models/ data/ -example/webui/pool \ No newline at end of file +example/webui/pool +uploads/ \ No newline at end of file diff --git a/agent/agent.go b/agent/agent.go index ca09245..9b920ad 100644 --- a/agent/agent.go +++ b/agent/agent.go @@ -154,6 +154,18 @@ func (a *Agent) SetConversation(conv []openai.ChatCompletionMessage) { func (a *Agent) ResetConversation() { a.Lock() defer a.Unlock() + + // store into memory the conversation before pruning it + if a.options.enableKB { + for _, message := range a.currentConversation { + if message.Role == "user" { + if err := a.options.ragdb.Store(message.Content); err != nil { + xlog.Error("Error storing into memory", "error", err) + } + } + } + } + a.currentConversation = []openai.ChatCompletionMessage{} }