Skip to content

Commit 55451e4

Browse files
committed
feat: refresh method
Signed-off-by: potato <[email protected]>
1 parent 5f187bc commit 55451e4

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

spring-ai-model/src/main/java/org/springframework/ai/chat/memory/InMemoryChatMemoryRepository.java

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,4 +60,17 @@ public void deleteByConversationId(String conversationId) {
6060
this.chatMemoryStore.remove(conversationId);
6161
}
6262

63+
@Override
64+
public void refresh(String conversationId, List<Message> deletes, List<Message> adds) {
65+
this.chatMemoryStore.compute(conversationId, (key, currentMessages) -> {
66+
if (currentMessages == null) {
67+
return new ArrayList<>(adds);
68+
}
69+
List<Message> updatedMessages = new ArrayList<>(currentMessages);
70+
updatedMessages.removeAll(deletes);
71+
updatedMessages.addAll(adds);
72+
return updatedMessages;
73+
});
74+
}
75+
6376
}

0 commit comments

Comments
 (0)