diff --git a/Partner-Core/src/main/java/work/slhaf/partner/core/cognation/CognationCapability.java b/Partner-Core/src/main/java/work/slhaf/partner/core/cognation/CognationCapability.java index 68bc5982..02999ae8 100644 --- a/Partner-Core/src/main/java/work/slhaf/partner/core/cognation/CognationCapability.java +++ b/Partner-Core/src/main/java/work/slhaf/partner/core/cognation/CognationCapability.java @@ -11,6 +11,8 @@ import java.util.concurrent.locks.Lock; @Capability("cognation") public interface CognationCapability { + String initiateTurn(String input); + List getChatMessages(); void cleanMessage(List messages); diff --git a/Partner-Core/src/main/java/work/slhaf/partner/core/cognation/CognationCore.java b/Partner-Core/src/main/java/work/slhaf/partner/core/cognation/CognationCore.java index 15d54248..85b55d05 100644 --- a/Partner-Core/src/main/java/work/slhaf/partner/core/cognation/CognationCore.java +++ b/Partner-Core/src/main/java/work/slhaf/partner/core/cognation/CognationCore.java @@ -7,9 +7,11 @@ import lombok.Setter; import lombok.extern.slf4j.Slf4j; import work.slhaf.partner.api.agent.factory.capability.annotation.CapabilityCore; import work.slhaf.partner.api.agent.factory.capability.annotation.CapabilityMethod; +import work.slhaf.partner.api.agent.runtime.interaction.AgentRuntime; import work.slhaf.partner.api.chat.pojo.Message; import work.slhaf.partner.api.chat.pojo.MetaMessage; import work.slhaf.partner.core.PartnerCore; +import work.slhaf.partner.runtime.interaction.data.context.PartnerRunningFlowContext; import java.io.IOException; import java.io.Serial; @@ -43,6 +45,13 @@ public class CognationCore extends PartnerCore { public CognationCore() throws IOException, ClassNotFoundException { } + @CapabilityMethod + public String initiateTurn(String input) { + PartnerRunningFlowContext primaryContext = PartnerRunningFlowContext.Companion.fromSelf(input); + PartnerRunningFlowContext executedContext = AgentRuntime.INSTANCE.submit(primaryContext); + return executedContext.getCoreResponse().getString("text"); + } + @CapabilityMethod public List getChatMessages() { return chatMessages;