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

fix(Communication): fix Message appending logic and add refactor todo

Browse Source
This commit is contained in:
slhafzjw
2026-03-07 14:01:33 +08:00
parent 0b63ec8523
commit 1ce2038ab8
1 changed files with 9 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -79,6 +79,7 @@ public class CommunicationProducer extends AbstractAgentModule.Running<PartnerRu
setMessage(runningFlowContext.getCoreContext().toString()); setMessage(runningFlowContext.getCoreContext().toString());
} }
// TODO need to update message appending logic
private void setAppendedPromptMessage(PartnerRunningFlowContext runningFlowContext) { private void setAppendedPromptMessage(PartnerRunningFlowContext runningFlowContext) {
List<AppendPromptData> appendedPrompt = runningFlowContext.getModuleContext().getAppendedPrompt(); List<AppendPromptData> appendedPrompt = runningFlowContext.getModuleContext().getAppendedPrompt();
int appendedPromptSize = getAppendedPromptSize(appendedPrompt); int appendedPromptSize = getAppendedPromptSize(appendedPrompt);
@@ -179,7 +180,7 @@ public class CommunicationProducer extends AbstractAgentModule.Running<PartnerRu
runningFlowContext.getModuleContext().getExtraContext().put("total_token", chatResponse.getUsageBean().getTotal_tokens()); runningFlowContext.getModuleContext().getExtraContext().put("total_token", chatResponse.getUsageBean().getTotal_tokens());
//区分单人聊天场景 //区分单人聊天场景
// if (runningFlowContext.isSingle()) { // if (runningFlowContext.isSingle()) {
MetaMessage metaMessage = new MetaMessage(primaryUserMessage, assistantMessage); MetaMessage metaMessage = new MetaMessage(primaryUserMessage, assistantMessage);
cognationCapability.addMetaMessage(runningFlowContext.getSource(), metaMessage); cognationCapability.addMetaMessage(runningFlowContext.getSource(), metaMessage);
// } // }
} }
@@ -199,10 +200,7 @@ public class CommunicationProducer extends AbstractAgentModule.Running<PartnerRu
} }
private void setAppendedPromptMessage(List<AppendPromptData> appendPrompt) { private void setAppendedPromptMessage(List<AppendPromptData> appendPrompt) {
Message appendDeclareMessage = Message.builder() Message appendDeclareMessage = new Message(ChatConstant.Character.USER, ModelConstant.CharacterPrefix.SYSTEM + "认知补充开始");
.role(ChatConstant.Character.USER)
.content(ModelConstant.CharacterPrefix.SYSTEM + "认知补充开始")
.build();
this.appendedMessages.add(appendDeclareMessage); this.appendedMessages.add(appendDeclareMessage);
for (AppendPromptData data : appendPrompt) { for (AppendPromptData data : appendPrompt) {
setStartMessage(data); setStartMessage(data);
@@ -210,43 +208,29 @@ public class CommunicationProducer extends AbstractAgentModule.Running<PartnerRu
setEndMessage(data); setEndMessage(data);
setAssistantMessage(); setAssistantMessage();
} }
Message appendEndMessage = Message.builder() Message appendEndMessage = new Message(ChatConstant.Character.USER, ModelConstant.CharacterPrefix.SYSTEM + "认知补充结束");
.role(ChatConstant.Character.USER)
.content(ModelConstant.CharacterPrefix.SYSTEM + "认知补充结束")
.build();
this.appendedMessages.add(appendEndMessage); this.appendedMessages.add(appendEndMessage);
} }
private void setAssistantMessage() { private void setAssistantMessage() {
appendedMessages.add(Message.builder() Message message = new Message(ChatConstant.Character.ASSISTANT, "嗯，明白了");
.role(ChatConstant.Character.ASSISTANT) appendedMessages.add(message);
.content("嗯，明白了")
.build());
} }
private void setEndMessage(AppendPromptData data) { private void setEndMessage(AppendPromptData data) {
Message endMessage = Message.builder() Message endMessage = new Message(ChatConstant.Character.USER, ModelConstant.CharacterPrefix.SYSTEM + data.getModuleName() + "认知补充结束.");
.role(ChatConstant.Character.USER)
.content(ModelConstant.CharacterPrefix.SYSTEM + data.getModuleName() + "认知补充结束.")
.build();
appendedMessages.add(endMessage); appendedMessages.add(endMessage);
} }
private void setContentMessage(AppendPromptData data) { private void setContentMessage(AppendPromptData data) {
data.getAppendedPrompt().forEach((k, v) -> { data.getAppendedPrompt().forEach((k, v) -> {
Message contentMessage = Message.builder() Message contentMessage = new Message(ChatConstant.Character.USER, ModelConstant.CharacterPrefix.SYSTEM + k + v + "\r\n");
.role(ChatConstant.Character.USER)
.content(ModelConstant.CharacterPrefix.SYSTEM + k + v + "\r\n")
.build();
appendedMessages.add(contentMessage); appendedMessages.add(contentMessage);
}); });
} }
private void setStartMessage(AppendPromptData data) { private void setStartMessage(AppendPromptData data) {
Message startMessage = Message.builder() Message startMessage = new Message(ChatConstant.Character.USER, ModelConstant.CharacterPrefix.SYSTEM + data.getModuleName() + "以下为" + data.getModuleName() + "相关认知.");
.role(ChatConstant.Character.USER)
.content(ModelConstant.CharacterPrefix.SYSTEM + data.getModuleName() + "以下为" + data.getModuleName() + "相关认知.")
.build();
appendedMessages.add(startMessage); appendedMessages.add(startMessage);
} }