slhaf/Partner
1
0
Fork 0
mirror of https://github.com/slhaf/Partner.git synced 2026-05-12 08:43:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(CommunicationProducer): snapshot chat history when assembling conversation

Browse Source
This commit is contained in:
slhafzjw
2026-03-10 19:53:20 +08:00
parent 5ad80d8b86
commit f51401e2f2
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
Partner-Core/src/main/java/work/slhaf/partner/module/modules/core/CommunicationProducer.java
View File

@@ -44,12 +44,9 @@ public class CommunicationProducer extends AbstractAgentModule.Running<PartnerRu
@InjectCapability
private CognationCapability cognationCapability;
private final List<Message> chatMessages = new ArrayList<>();
@Init
public void init() {
this.chatMessages.clear();
this.chatMessages.addAll(this.cognationCapability.getChatMessages());
log.info("CommunicationProducer 注册完毕...");
}
@@ -104,12 +101,13 @@ public class CommunicationProducer extends AbstractAgentModule.Running<PartnerRu
}
private List<Message> buildChatMessages(PartnerRunningFlowContext runningFlowContext) {
List<Message> temp = new ArrayList<>(chatMessages.size() + 2);
List<Message> historyMessages = snapshotConversationMessages();
List<Message> temp = new ArrayList<>(historyMessages.size() + 2);
Message contextMessage = buildContextMessage(runningFlowContext);
if (contextMessage != null) {
temp.add(contextMessage);
}
temp.addAll(chatMessages);
temp.addAll(historyMessages);
temp.add(buildInputMessage(runningFlowContext));
return temp;
}
@@ -117,6 +115,7 @@ public class CommunicationProducer extends AbstractAgentModule.Running<PartnerRu
private void updateModuleContextAndChatMessages(PartnerRunningFlowContext runningFlowContext, String response) {
cognationCapability.getMessageLock().lock();
try {
List<Message> chatMessages = cognationCapability.getChatMessages();
chatMessages.removeIf(this::isStructuredUserMessage);
Message primaryUserMessage = new Message(
Message.Character.USER,
@@ -130,6 +129,12 @@ public class CommunicationProducer extends AbstractAgentModule.Running<PartnerRu
}
}
private List<Message> snapshotConversationMessages() {
List<Message> snapshot = new ArrayList<>(cognationCapability.snapshotChatMessages());
snapshot.removeIf(this::isStructuredUserMessage);
return snapshot;
}
private Message buildContextMessage(PartnerRunningFlowContext runningFlowContext) {
List<ContextBlock> contextBlocks = filterContextBlocks(
runningFlowContext.getContextBlocks(),