Package com.javaaidev.agenticpatterns.parallelizationworkflow

Class ParallelizationWorkflowAgent<Request,Response>

java.lang.Object

com.javaaidev.agenticpatterns.core.Agent

com.javaaidev.agenticpatterns.taskexecution.TaskExecutionAgent<Request,Response>

com.javaaidev.agenticpatterns.parallelizationworkflow.ParallelizationWorkflowAgent<Request,Response>

Type Parameters:

Request -

Response -

All Implemented Interfaces:

Function<Request,Response>

Direct Known Subclasses:

DirectAssembling, PromptBasedAssembling

public abstract class ParallelizationWorkflowAgent<Request,Response>
extends TaskExecutionAgent<Request,Response>

Parallelization Workflow agent, refer to doc

Field Summary

Fields

Modifier and Type	Field	Description
protected final CopyOnWriteArrayList<SubtaskContext>	subtasks

Fields inherited from class com.javaaidev.agenticpatterns.taskexecution.TaskExecutionAgent

responseType

Fields inherited from class com.javaaidev.agenticpatterns.core.Agent

chatClient, observationRegistry

Constructor Summary

Constructors

Constructor	Description
ParallelizationWorkflowAgent(org.springframework.ai.chat.client.ChatClient chatClient, @Nullable io.micrometer.observation.ObservationRegistry observationRegistry)
ParallelizationWorkflowAgent(org.springframework.ai.chat.client.ChatClient chatClient, @Nullable Type responseType, @Nullable io.micrometer.observation.ObservationRegistry observationRegistry)

Method Summary

Modifier and Type	Method	Description
protected <TaskRequest, TaskResponse> void	addSubtask(String taskId, TaskExecutionAgent<TaskRequest,TaskResponse> subtask, Function<Request,TaskRequest> requestTransformer)	Add new subtask
protected @Nullable List<SubtaskCreationRequest<Request>>	createTasks(@Nullable Request request)	Create a list of subtasks from request.
protected Duration	getMaxTaskExecutionDuration()	Max duration of a subtask execution
protected ExecutorService	getTaskExecutorService()	Create the ExecutorService to execute subtasks.
protected TaskExecutionResults	runSubtasks(@Nullable Request request)

Methods inherited from class com.javaaidev.agenticpatterns.taskexecution.TaskExecutionAgent

apply, call, getPromptContext, getPromptTemplate, instrumentedCall, updateChatClientRequest

Methods inherited from class com.javaaidev.agenticpatterns.core.Agent

getName

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface java.util.function.Function

andThen, compose

Field Details

subtasks

protected final CopyOnWriteArrayList<SubtaskContext> subtasks

Constructor Details

ParallelizationWorkflowAgent

public ParallelizationWorkflowAgent(org.springframework.ai.chat.client.ChatClient chatClient,
 @Nullable io.micrometer.observation.ObservationRegistry observationRegistry)

ParallelizationWorkflowAgent

public ParallelizationWorkflowAgent(org.springframework.ai.chat.client.ChatClient chatClient,
 @Nullable Type responseType,
 @Nullable io.micrometer.observation.ObservationRegistry observationRegistry)

Method Details

addSubtask

protected <TaskRequest,
TaskResponse> void addSubtask(String taskId,
 TaskExecutionAgent<TaskRequest,TaskResponse> subtask,
 Function<Request,TaskRequest> requestTransformer)

Add new subtask

Type Parameters:

TaskRequest - Task input type

TaskResponse - Task output type

Parameters:

taskId - Task id

subtask - Subtask implemented as TaskExecutionAgent

requestTransformer - Transform request to task's input

getMaxTaskExecutionDuration

protected Duration getMaxTaskExecutionDuration()

Max duration of a subtask execution

Returns:

Max duration

createTasks

protected @Nullable List<SubtaskCreationRequest<Request>> createTasks(@Nullable Request request)

Create a list of subtasks from request. Subtasks added by addSubtask(String, TaskExecutionAgent, Function) will be merged into this list.

Parameters:

request - Request

Returns:

Subtasks

getTaskExecutorService

protected ExecutorService getTaskExecutorService()

Create the ExecutorService to execute subtasks.

The default executor service uses virtual threads.

Returns:

executor service

runSubtasks

protected TaskExecutionResults runSubtasks(@Nullable Request request)