wangliwen
/
langchain4j-aideepin
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: chatMemory store system message

This commit is contained in:
moyangzhan 2024-04-12 12:24:10 +08:00
parent 031b7f66e2
commit 131e8e38a0
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
adi-common/src/main/java/com/moyz/adi/common/util/MapDBChatMemoryStore.java
View File

@ -2,11 +2,13 @@ package com.moyz.adi.common.util;
import dev.langchain4j.data.message.AiMessage; import dev.langchain4j.data.message.AiMessage;
import dev.langchain4j.data.message.ChatMessage; import dev.langchain4j.data.message.ChatMessage;
import dev.langchain4j.data.message.SystemMessage;
import dev.langchain4j.store.memory.chat.ChatMemoryStore; import dev.langchain4j.store.memory.chat.ChatMemoryStore;
import lombok.extern.slf4j.Slf4j; import lombok.extern.slf4j.Slf4j;
import org.mapdb.DB; import org.mapdb.DB;
import org.mapdb.DBMaker; import org.mapdb.DBMaker;
import java.util.ArrayList;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
@ -31,10 +33,18 @@ public class MapDBChatMemoryStore implements ChatMemoryStore {
@Override @Override
public void updateMessages(Object memoryId, List<ChatMessage> messages) { public void updateMessages(Object memoryId, List<ChatMessage> messages) {
if(messages.size() > 0 && messages.get(0) instanceof AiMessage){ //AiMessage in first position is not allow
if (messages.size() > 0 && messages.get(0) instanceof AiMessage) {
messages.remove(0); messages.remove(0);
} }
String json = messagesToJson(messages); //Filter out the available messages.(UserMessage,AiMessage)
List<ChatMessage> availableMessage = new ArrayList<>();
for (ChatMessage chatMessage : messages) {
if (!(chatMessage instanceof SystemMessage)) {
availableMessage.add(chatMessage);
}
}
String json = messagesToJson(availableMessage);
log.info("updateMessages,{}", json); log.info("updateMessages,{}", json);
map.put((String) memoryId, json); map.put((String) memoryId, json);
db.commit(); db.commit();