agent as tool

Author: alexcheng1982

examples/src/main/java/com/javaaidev/agenticpatterns/examples/agentastool/RestAPITestCaseGenerationAgent.java

@@ -0,0 +1,37 @@
+package com.javaaidev.agenticpatterns.examples.agentastool;
+
+import com.javaaidev.agenticpatterns.examples.agentastool.RestAPITestCaseGenerationAgent.RestAPITestCaseGenerationResponse;
+import com.javaaidev.agenticpatterns.taskexecution.TaskExecutionAgent;
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.ai.chat.client.ChatClient.ChatClientRequestSpec;
+
+public class RestAPITestCaseGenerationAgent extends
+    TaskExecutionAgent<Void, RestAPITestCaseGenerationResponse> {
+
+  protected RestAPITestCaseGenerationAgent(ChatClient chatClient) {
+    super(chatClient);
+  }
+
+  @Override
+  protected String getPromptTemplate() {
+    return """
+        Goal: Write Python code to test a REST API.
+        
+        Requirements:
+        - The API is responsible for creating a new user.
+        - Use a POST request with JSON content to create a new user.
+        - When creation is successful, the API returns a 201 code.
+        - The API url is http://localhost:8080/api/v1/user.
+        - User data has a defined structure, don't generate it yourself, use the tool to generate it.
+        """;
+  }
+
+  @Override
+  protected void updateRequest(ChatClientRequestSpec spec) {
+    spec.functions("userGenerationAgent");
+  }
+
+  public record RestAPITestCaseGenerationResponse(String code) {
+
+  }
+}

examples/src/main/java/com/javaaidev/agenticpatterns/examples/agentastool/RestAPITestCaseGenerationConfiguration.java

@@ -0,0 +1,18 @@
+package com.javaaidev.agenticpatterns.examples.agentastool;
+
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.ai.chat.client.advisor.SimpleLoggerAdvisor;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+public class RestAPITestCaseGenerationConfiguration {
+
+  @Bean
+  public RestAPITestCaseGenerationAgent restAPITestCaseGenerationAgent(
+      ChatClient.Builder chatClientBuilder,
+      SimpleLoggerAdvisor simpleLoggerAdvisor) {
+    return new RestAPITestCaseGenerationAgent(
+        chatClientBuilder.defaultAdvisors(simpleLoggerAdvisor).build());
+  }
+}

examples/src/main/java/com/javaaidev/agenticpatterns/examples/agentastool/RestAPITestCaseGenerationController.java

@@ -0,0 +1,22 @@
+package com.javaaidev.agenticpatterns.examples.agentastool;
+
+import com.javaaidev.agenticpatterns.examples.agentastool.RestAPITestCaseGenerationAgent.RestAPITestCaseGenerationResponse;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+@RequestMapping("/restapi_testcase")
+public class RestAPITestCaseGenerationController {
+
+  private final RestAPITestCaseGenerationAgent agent;
+
+  public RestAPITestCaseGenerationController(RestAPITestCaseGenerationAgent agent) {
+    this.agent = agent;
+  }
+
+  @PostMapping
+  public RestAPITestCaseGenerationResponse generateTestCase() {
+    return agent.call(null);
+  }
+}

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

@@ -5,10 +5,13 @@
 import com.javaaidev.agenticpatterns.taskexecution.TaskExecutionAgent;
 import java.util.List;
 import java.util.Map;
+import java.util.function.Function;
 import org.jspecify.annotations.Nullable;
 import org.springframework.ai.chat.client.ChatClient;
 
-public class UserGenerationAgent extends TaskExecutionAgent<UserGenerationRequest, List<User>> {
+public class UserGenerationAgent extends
+    TaskExecutionAgent<UserGenerationRequest, List<User>> implements
+    Function<UserGenerationRequest, List<User>> {
 
   public UserGenerationAgent(ChatClient chatClient) {
     super(chatClient);

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

@@ -4,11 +4,13 @@
 import org.springframework.ai.chat.client.advisor.SimpleLoggerAdvisor;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Description;
 
 @Configuration
 public class UserGenerationConfiguration {
 
   @Bean
+  @Description("Generate test user")
   public UserGenerationAgent userGenerationAgent(
       ChatClient.Builder chatClientBuilder,
       SimpleLoggerAdvisor simpleLoggerAdvisor) {

examples/src/main/resources/application.yaml

@@ -7,8 +7,8 @@ spring:
       api-key: ${OPENAI_API_KEY:}
       chat:
         options:
-          model: gpt-3.5-turbo
-          temperature: 0.
+          model: gpt-4o
+          temperature: 0.0
 logging:
   level:
     org.springframework.ai.chat.client.advisor.SimpleLoggerAdvisor: DEBUG

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

@@ -8,6 +8,7 @@
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Optional;
+import java.util.function.Function;
 import org.jspecify.annotations.Nullable;
 import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.ai.chat.client.ChatClient.ChatClientRequestSpec;
@@ -20,7 +21,8 @@
  * @param <Request>  Type of task input
  * @param <Response> Type of task output
  */
-public abstract class TaskExecutionAgent<Request, Response> extends Agent {
+public abstract class TaskExecutionAgent<Request, Response> extends Agent implements
+    Function<Request, Response> {
 
   /**
    * Get the prompt template
@@ -61,6 +63,11 @@ protected Map<String, Object> getPromptContext(@Nullable Request request) {
   protected void updateRequest(ChatClientRequestSpec spec) {
   }
 
+  @Override
+  public Response apply(Request request) {
+    return call(request);
+  }
+
   public Response call(@Nullable Request request) {
     var template = getPromptTemplate();
     if (template.isBlank()) {