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

refactor(CommunicationProducer): update datetime pattern and source in communicating

Browse Source
This commit is contained in:
slhafzjw
2026-03-10 15:01:07 +08:00
parent f5f64971f3
commit 331d415925
1 changed files with 5 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -23,13 +23,8 @@ import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource; import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult; import javax.xml.transform.stream.StreamResult;
import java.io.StringWriter; import java.io.StringWriter;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter; import java.time.format.DateTimeFormatter;
import java.util.ArrayList; import java.util.*;
import java.util.Comparator;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors; import java.util.stream.Collectors;
import static work.slhaf.partner.common.util.ExtractUtil.extractJson; import static work.slhaf.partner.common.util.ExtractUtil.extractJson;
@@ -76,7 +71,7 @@ public class CommunicationProducer extends AbstractAgentModule.Running<PartnerRu
@Override @Override
public void execute(PartnerRunningFlowContext runningFlowContext) { public void execute(PartnerRunningFlowContext runningFlowContext) {
log.debug("Communicating with: {}",runningFlowContext.getSource()); log.debug("Communicating with: {}", runningFlowContext.getSource());
executeChat(runningFlowContext); executeChat(runningFlowContext);
} }
@@ -124,11 +119,9 @@ public class CommunicationProducer extends AbstractAgentModule.Running<PartnerRu
cognationCapability.getMessageLock().lock(); cognationCapability.getMessageLock().lock();
try { try {
chatMessages.removeIf(this::isStructuredUserMessage); chatMessages.removeIf(this::isStructuredUserMessage);
// TODO 此处的时间标识应当采用 RunningFlowContext 携带时间
String dateTime = LocalDateTime.now().format(DateTimeFormatter.ofPattern("\r\n**[yyyy-MM-dd HH:mm:ss]"));
Message primaryUserMessage = new Message( Message primaryUserMessage = new Message(
Message.Character.USER, Message.Character.USER,
formatConversationUserMessage(runningFlowContext) + dateTime formatConversationUserMessage(runningFlowContext)
); );
chatMessages.add(primaryUserMessage); chatMessages.add(primaryUserMessage);
Message assistantMessage = new Message(Message.Character.ASSISTANT, response); Message assistantMessage = new Message(Message.Character.ASSISTANT, response);
@@ -218,7 +211,8 @@ public class CommunicationProducer extends AbstractAgentModule.Running<PartnerRu
} }
private String formatConversationUserMessage(PartnerRunningFlowContext runningFlowContext) { private String formatConversationUserMessage(PartnerRunningFlowContext runningFlowContext) {
return runningFlowContext.getSource() + ": " + runningFlowContext.getInput(); String datetime = runningFlowContext.getInfo().getDateTime().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
return "[" + runningFlowContext.getSource() + " " + "(" + datetime + ")" + "]" + ": " + runningFlowContext.getInput();
} }
private Document newDocument() throws Exception { private Document newDocument() throws Exception {