Store into memory conversation before resetting it

Signed-off-by: mudler <mudler@localai.io>
This commit is contained in:
mudler
2024-12-15 18:13:54 +01:00
parent a204b3677b
commit 6318f1aa97
2 changed files with 14 additions and 1 deletions

3
.gitignore vendored
View File

@@ -1,3 +1,4 @@
models/ models/
data/ data/
example/webui/pool example/webui/pool
uploads/

View File

@@ -154,6 +154,18 @@ func (a *Agent) SetConversation(conv []openai.ChatCompletionMessage) {
func (a *Agent) ResetConversation() { func (a *Agent) ResetConversation() {
a.Lock() a.Lock()
defer a.Unlock() 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{} a.currentConversation = []openai.ChatCompletionMessage{}
} }