code refactoring

Author: alexcheng1982

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

@@ -27,7 +27,7 @@ protected String getPromptTemplate() {
   }
 
   @Override
-  protected void updateRequest(ChatClientRequestSpec spec) {
+  protected void updateChatClientRequest(ChatClientRequestSpec spec) {
     spec.functions("userGenerationAgent");
   }
 

examples/src/main/java/com/javaaidev/agenticpatterns/examples/routingworkflow/CustomerSupportAgent.java

@@ -72,7 +72,7 @@ protected String getPromptTemplate() {
     }
 
     @Override
-    protected void updateRequest(ChatClientRequestSpec spec) {
+    protected void updateChatClientRequest(ChatClientRequestSpec spec) {
       spec.system("You are a customer support agent for payment, be polite and helper");
     }
   }
@@ -100,7 +100,7 @@ protected String getPromptTemplate() {
     }
 
     @Override
-    protected void updateRequest(ChatClientRequestSpec spec) {
+    protected void updateChatClientRequest(ChatClientRequestSpec spec) {
       spec.system("You are a customer support agent for shipping, be polite and helper");
     }
   }
@@ -128,7 +128,7 @@ protected String getPromptTemplate() {
     }
 
     @Override
-    protected void updateRequest(ChatClientRequestSpec spec) {
+    protected void updateChatClientRequest(ChatClientRequestSpec spec) {
       spec.system(
           "You are a customer support agent for general questions, be polite and helper");
     }

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

@@ -25,7 +25,7 @@ protected String getPromptTemplate() {
   }
 
   @Override
-  protected @Nullable Map<String, Object> getPromptContext(
+  protected Map<String, Object> getPromptContext(
       @Nullable UserGenerationRequest userGenerationRequest) {
     var count = AgentUtils.safeGet(userGenerationRequest, UserGenerationRequest::count,
         1);

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

@@ -28,13 +28,6 @@
 public abstract class TaskExecutionAgent<Request, Response> extends Agent implements
     Function<Request, Response> {
 
-  /**
-   * Get the prompt template
-   *
-   * @return prompt template
-   */
-  protected abstract String getPromptTemplate();
-
   @Nullable
   protected Type responseType;
 
@@ -49,7 +42,7 @@ protected TaskExecutionAgent(ChatClient chatClient, @Nullable Type responseType)
   protected TaskExecutionAgent(ChatClient chatClient,
       @Nullable ObservationRegistry observationRegistry) {
     super(chatClient, observationRegistry);
-    responseType = TypeResolver.resolveType(this.getClass(), TaskExecutionAgent.class, 1);
+    this.responseType = TypeResolver.resolveType(this.getClass(), TaskExecutionAgent.class, 1);
   }
 
   protected TaskExecutionAgent(ChatClient chatClient, @Nullable Type responseType,
@@ -58,6 +51,13 @@ protected TaskExecutionAgent(ChatClient chatClient, @Nullable Type responseType,
     this.responseType = responseType;
   }
 
+  /**
+   * Get the prompt template
+   *
+   * @return prompt template
+   */
+  protected abstract String getPromptTemplate();
+
   /**
    * Prepare for the values of variables in the prompt template
    *
@@ -74,7 +74,7 @@ protected Map<String, Object> getPromptContext(@Nullable Request request) {
    *
    * @param spec {@linkplain ChatClientRequestSpec} from Spring AI
    */
-  protected void updateRequest(ChatClientRequestSpec spec) {
+  protected void updateChatClientRequest(ChatClientRequestSpec spec) {
   }
 
   @Override
@@ -120,7 +120,7 @@ private Response doCall(@Nullable Request request) {
         HashMap::new);
     var requestSpec = chatClient.prompt().user(userSpec -> userSpec.text(template)
         .params(context));
-    updateRequest(requestSpec);
+    updateChatClientRequest(requestSpec);
     var responseSpec = requestSpec.call();
     Response output;
     if (type instanceof Class<?> clazz) {