feat(cognation): add initiateTurn to execute input via AgentRuntime.submit

This commit is contained in:
2026-03-05 16:49:17 +08:00
parent 839f19f15b
commit 40bd2deeba
2 changed files with 11 additions and 0 deletions

View File

@@ -11,6 +11,8 @@ import java.util.concurrent.locks.Lock;
@Capability("cognation")
public interface CognationCapability {
String initiateTurn(String input);
List<Message> getChatMessages();
void cleanMessage(List<Message> messages);

View File

@@ -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<CognationCore> {
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<Message> getChatMessages() {
return chatMessages;