refactor task execution agent example

Author: alexcheng1982

examples/src/main/java/com/javaaidev/agenticpatterns/examples/taskexecution/UserGenerationAgent.java

@@ -1,7 +1,6 @@
 package com.javaaidev.agenticpatterns.examples.taskexecution;
 
 import com.javaaidev.agenticpatterns.core.AgentUtils;
-import com.javaaidev.agenticpatterns.examples.taskexecution.UserGenerationAgent.UserGenerationRequest;
 import com.javaaidev.agenticpatterns.taskexecution.TaskExecutionAgent;
 import io.micrometer.observation.ObservationRegistry;
 import java.util.List;
@@ -23,7 +22,4 @@ protected String getPromptTemplate() {
     return AgentUtils.loadPromptTemplateFromClasspath("prompt_template/generate-user.st");
   }
 
-  public record UserGenerationRequest(int count) {
-
-  }
 }

examples/src/main/java/com/javaaidev/agenticpatterns/examples/taskexecution/UserGenerationAgentController.java

@@ -1,7 +1,6 @@
 package com.javaaidev.agenticpatterns.examples.taskexecution;
 
-import com.javaaidev.agenticpatterns.examples.taskexecution.UserGenerationAgent.UserGenerationRequest;
-import java.util.List;
+import com.javaaidev.agenticpatterns.taskexecution.TaskExecutionAgent;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -11,14 +10,15 @@
 @RequestMapping("/users_generation")
 public class UserGenerationAgentController {
 
-  private final UserGenerationAgent userGenerationAgent;
+  private final TaskExecutionAgent<UserGenerationRequest, UserGenerationResponse> userGenerationAgent;
 
-  public UserGenerationAgentController(UserGenerationAgent userGenerationAgent) {
+  public UserGenerationAgentController(
+      TaskExecutionAgent<UserGenerationRequest, UserGenerationResponse> userGenerationAgent) {
     this.userGenerationAgent = userGenerationAgent;
   }
-  
+
   @PostMapping
-  public List<User> generateUsers(@RequestBody UserGenerationRequest request) {
+  public UserGenerationResponse generateUsers(@RequestBody UserGenerationRequest request) {
     return userGenerationAgent.call(request);
   }
 }

examples/src/main/java/com/javaaidev/agenticpatterns/examples/taskexecution/UserGenerationConfiguration.java

@@ -1,8 +1,11 @@
 package com.javaaidev.agenticpatterns.examples.taskexecution;
 
+import com.javaaidev.agenticpatterns.core.AgentUtils;
+import com.javaaidev.agenticpatterns.taskexecution.TaskExecutionAgent;
 import io.micrometer.observation.ObservationRegistry;
 import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.ai.chat.client.advisor.SimpleLoggerAdvisor;
+import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.context.annotation.Description;
@@ -11,12 +14,21 @@
 public class UserGenerationConfiguration {
 
   @Bean
-  @Description("Generate test user")
-  public UserGenerationAgent userGenerationAgent(
+  @Description("Generate test users")
+  @Qualifier("userGenerationAgent")
+  public TaskExecutionAgent<UserGenerationRequest, UserGenerationResponse> userGenerationAgent(
       ChatClient.Builder chatClientBuilder,
       SimpleLoggerAdvisor simpleLoggerAdvisor,
-      ObservationRegistry observationRegistry) {
-    return new UserGenerationAgent(
-        chatClientBuilder.defaultAdvisors(simpleLoggerAdvisor).build(), observationRegistry);
+      ObservationRegistry observationRegistry
+  ) {
+    var chatClient = chatClientBuilder.defaultAdvisors(simpleLoggerAdvisor).build();
+    return TaskExecutionAgent.<UserGenerationRequest, UserGenerationResponse>defaultBuilder()
+        .chatClient(chatClient)
+        .responseType(UserGenerationResponse.class)
+        .promptTemplate(
+            AgentUtils.loadPromptTemplateFromClasspath("prompt_template/generate-user.st"))
+        .name("UserGeneration")
+        .observationRegistry(observationRegistry)
+        .build();
   }
 }

examples/src/main/java/com/javaaidev/agenticpatterns/examples/taskexecution/UserGenerationRequest.java

@@ -0,0 +1,5 @@
+package com.javaaidev.agenticpatterns.examples.taskexecution;
+
+public record UserGenerationRequest(int count) {
+
+}

examples/src/main/java/com/javaaidev/agenticpatterns/examples/taskexecution/UserGenerationResponse.java

@@ -0,0 +1,7 @@
+package com.javaaidev.agenticpatterns.examples.taskexecution;
+
+import java.util.List;
+
+public record UserGenerationResponse(List<User> users) {
+
+}

patterns/task-execution/src/main/java/com/javaaidev/agenticpatterns/taskexecution/TaskExecutionAgent.java

@@ -171,7 +171,7 @@ private Response doCall(@Nullable Request request) {
       throw new AgentExecutionException("Invalid type " + type);
     }
     if (output == null) {
-      throw new RuntimeException("Empty or bad response from LLM");
+      throw new AgentExecutionException("Empty or bad response from LLM");
     }
     return output;
   }