refactor: use string block for prompt template

bsorrentino · Mar 27, 2024 · ef5df2e · ef5df2e
1 parent 7dd851c
commit ef5df2e
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/agents/src/main/java/dev/langchain4j/Agent.java b/agents/src/main/java/dev/langchain4j/Agent.java
@@ -35,19 +35,19 @@ public Response<AiMessage> execute( Map<String,Object> inputs ) {
     }
 
     private PromptTemplate getToolResponseTemplate( ) {
-        var TEMPLATE_TOOL_RESPONSE = new StringBuilder()
-                .append("TOOL RESPONSE:").append('\n')
-                .append("---------------------").append('\n')
-                .append("{{observation}}").append('\n')
-                .append( "--------------------" ).append('\n')
-                .append('\n')
-                .toString();
+        var TEMPLATE_TOOL_RESPONSE = """
+                TOOL RESPONSE:
+                ---------------------
+                {{observation}}
+                --------------------
+                """;
         return PromptTemplate.from(TEMPLATE_TOOL_RESPONSE);
     }
 
     public Response<AiMessage> execute( String input, List<AgentExecutor.IntermediateStep> intermediateSteps ) {
         var agentScratchpadTemplate = getToolResponseTemplate();
-        var userMessageTemplate = PromptTemplate.from( "USER'S INPUT: {{input}}" ).apply( Map.of( "input", input));
+        var userMessageTemplate = PromptTemplate.from( "USER'S INPUT: {{input}}" )
+                                                    .apply( Map.of( "input", input));
 
         var messages = new ArrayList<ChatMessage>();
 
@@ -58,9 +58,9 @@ public Response<AiMessage> execute( String input, List<AgentExecutor.Intermediat
         }
 
         for( AgentExecutor.IntermediateStep step: intermediateSteps ) {
-            var agentScratchpad = agentScratchpadTemplate.apply( Map.of("observation", step.observation()) );
+            var agentScratchpad = agentScratchpadTemplate
+                                        .apply( Map.of("observation", step.observation()) );
             messages.add(new UserMessage(agentScratchpad.text()));
-                            ;
         }
 
         return chatLanguageModel.generate( messages, tools );