diff --git a/Partner-Core/src/main/java/work/slhaf/partner/core/cognition/ContextWorkspace.kt b/Partner-Core/src/main/java/work/slhaf/partner/core/cognition/ContextWorkspace.kt index 00c12f13..0bfe550a 100644 --- a/Partner-Core/src/main/java/work/slhaf/partner/core/cognition/ContextWorkspace.kt +++ b/Partner-Core/src/main/java/work/slhaf/partner/core/cognition/ContextWorkspace.kt @@ -265,6 +265,7 @@ abstract class BlockContent protected constructor( return element } + @JvmOverloads protected fun appendListElement( document: Document, parent: Element, @@ -287,6 +288,7 @@ abstract class BlockContent protected constructor( return wrapper } + @JvmOverloads protected fun appendRepeatedElements( document: Document, parent: Element, 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 6b5399ac..08637c6f 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 @@ -21,10 +21,7 @@ import javax.xml.transform.TransformerFactory; import javax.xml.transform.dom.DOMSource; import javax.xml.transform.stream.StreamResult; import java.io.StringWriter; -import java.util.ArrayList; -import java.util.LinkedHashMap; -import java.util.List; -import java.util.Map; +import java.util.*; import java.util.stream.Collectors; import static work.slhaf.partner.common.util.ExtractUtil.extractJson; @@ -82,7 +79,8 @@ public class CommunicationProducer extends AbstractAgentModule.Running chatMessages = cognitionCapability.getChatMessages(); + appendRepeatedElements(document, root, "chat_message", List.of(chatMessages.subList(chatMessages.size() - 5, chatMessages.size() - 1))); + } + }, + Set.of(ContextBlock.VisibleDomain.COGNITION), + 100, + 5, + 4 + ); + cognitionCapability.contextWorkspace().register(block); + } + private void updateCoreResponse(PartnerRunningFlowContext runningFlowContext, String responseText) { runningFlowContext.getCoreResponse().put("text", responseText); } @@ -113,7 +128,7 @@ public class CommunicationProducer extends AbstractAgentModule.Running chatMessages = cognitionCapability.getChatMessages();