diff --git a/Partner-Core/src/main/java/work/slhaf/partner/module/modules/core/CommunicationProducer.java b/Partner-Core/src/main/java/work/slhaf/partner/module/modules/core/CommunicationProducer.java index f39738b4..76e8f536 100644 --- a/Partner-Core/src/main/java/work/slhaf/partner/module/modules/core/CommunicationProducer.java +++ b/Partner-Core/src/main/java/work/slhaf/partner/module/modules/core/CommunicationProducer.java @@ -79,6 +79,7 @@ public class CommunicationProducer extends AbstractAgentModule.Running appendedPrompt = runningFlowContext.getModuleContext().getAppendedPrompt(); int appendedPromptSize = getAppendedPromptSize(appendedPrompt); @@ -179,7 +180,7 @@ public class CommunicationProducer extends AbstractAgentModule.Running appendPrompt) { - Message appendDeclareMessage = Message.builder() - .role(ChatConstant.Character.USER) - .content(ModelConstant.CharacterPrefix.SYSTEM + "认知补充开始") - .build(); + Message appendDeclareMessage = new Message(ChatConstant.Character.USER, ModelConstant.CharacterPrefix.SYSTEM + "认知补充开始"); this.appendedMessages.add(appendDeclareMessage); for (AppendPromptData data : appendPrompt) { setStartMessage(data); @@ -210,43 +208,29 @@ public class CommunicationProducer extends AbstractAgentModule.Running { - Message contentMessage = Message.builder() - .role(ChatConstant.Character.USER) - .content(ModelConstant.CharacterPrefix.SYSTEM + k + v + "\r\n") - .build(); + Message contentMessage = new Message(ChatConstant.Character.USER, ModelConstant.CharacterPrefix.SYSTEM + k + v + "\r\n"); appendedMessages.add(contentMessage); }); } private void setStartMessage(AppendPromptData data) { - Message startMessage = Message.builder() - .role(ChatConstant.Character.USER) - .content(ModelConstant.CharacterPrefix.SYSTEM + data.getModuleName() + "以下为" + data.getModuleName() + "相关认知.") - .build(); + Message startMessage = new Message(ChatConstant.Character.USER, ModelConstant.CharacterPrefix.SYSTEM + data.getModuleName() + "以下为" + data.getModuleName() + "相关认知."); appendedMessages.add(startMessage); }