Package com.javaaidev.agenticpatterns.taskexecution

Class TaskExecutionAgent<Request,Response>

java.lang.Object

com.javaaidev.agenticpatterns.core.Agent

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

Type Parameters:

Request - Type of task input

Response - Type of task output

All Implemented Interfaces:

Function<Request,Response>

Direct Known Subclasses:

NoLLMTaskExecutionAgent

public abstract class TaskExecutionAgent<Request,Response>
extends Agent
implements Function<Request,Response>

Task Execution Agent, refer to the pattern

Field Summary

Fields

Modifier and Type	Field	Description
protected @Nullable Type	responseType

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

chatClient, observationRegistry

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method	Description
Response	apply(Request request)
Response	call(@Nullable Request request)
protected @Nullable Map<String,Object>	getPromptContext(@Nullable Request request)	Prepare for the values of variables in the prompt template
protected abstract String	getPromptTemplate()	Get the prompt template
protected Response	instrumentedCall(Request request, Function<Request,Response> action)
protected void	updateChatClientRequest(org.springframework.ai.chat.client.ChatClient.ChatClientRequestSpec spec)	Customize request sent to LLM

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

responseType

protected @Nullable Type responseType

Constructor Details

TaskExecutionAgent

protected TaskExecutionAgent(org.springframework.ai.chat.client.ChatClient chatClient)

TaskExecutionAgent

protected TaskExecutionAgent(org.springframework.ai.chat.client.ChatClient chatClient,
 @Nullable Type responseType)

TaskExecutionAgent

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

TaskExecutionAgent

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

Method Details

getPromptTemplate

protected abstract String getPromptTemplate()

Get the prompt template

Returns:

prompt template

getPromptContext

protected @Nullable Map<String,Object> getPromptContext(@Nullable Request request)

Prepare for the values of variables in the prompt template

Parameters:

request - Task input

Returns:

Values of values

updateChatClientRequest

protected void updateChatClientRequest(org.springframework.ai.chat.client.ChatClient.ChatClientRequestSpec spec)

Customize request sent to LLM

Parameters:

spec - ChatClient.ChatClientRequestSpec from Spring AI

apply

public Response apply(Request request)

Specified by:

apply in interface Function<Request,Response>

call

public Response call(@Nullable Request request)

instrumentedCall

protected Response instrumentedCall(Request request,
 Function<Request,Response> action)