Skip to main content

​
AbstractAgent

The AbstractAgent class provides the foundation for agent implementations. It handles event stream processing, state management, and message history. Extend it and implement protected void run(RunAgentInput input, IEventStream<BaseEvent> stream).

​
Example Implementation

public class MyAgent extends AbstractAgent {
  public MyAgent() {
    super("my-agent", "A custom agent", "thread-123", List.of(), new State(), false);
  }

  @Override
  protected void run(RunAgentInput input, IEventStream<BaseEvent> stream) {
    // Emit events using the stream
    stream.next(new RunStartedEvent(input.getThreadId(), input.getRunId()));
    
    // Your agent logic here...
    
    stream.next(new RunFinishedEvent(input.getThreadId(), input.getRunId()));
  }
}

​
Configuration

Constructors take: agentId, description, threadId, initialMessages, state, debug.

​
Core Methods

  • CompletableFuture<Void> runAgent(RunAgentParameters parameters, AgentSubscriber subscriber) — orchestrates an async run and event distribution
  • Subscription subscribe(AgentSubscriber subscriber) — persistent subscriber registration
  • void addMessage(BaseMessage message) / void addMessages(List<BaseMessage>)
  • void setMessages(List<BaseMessage>)
  • void setState(State state) / State getState()

​
Protected Methods

  • protected abstract void run(RunAgentInput input, IEventStream<BaseEvent> stream) — emit events into stream
  • protected RunAgentInput prepareRunAgentInput(RunAgentParameters parameters)
  • protected void onInitialize(RunAgentInput input, List<AgentSubscriber> subscribers)
  • protected void handleEvent(BaseEvent event, List<AgentSubscriber> subscribers, AtomicReference<IEventStream<BaseEvent>> streamRef)
  • protected void handleComplete(List<AgentSubscriber> subscribers, RunAgentInput input, CompletableFuture<Void> future)
  • protected void handleError(Throwable error, List<AgentSubscriber> subscribers, CompletableFuture<Void> future)

​
Text Message Helpers

handleTextMessageStart, handleTextMessageContent, handleTextMessageChunk, handleTextMessageEnd integrate with MessageFactory to assemble streamed messages and notify subscribers.