From ef5df2e9ae9d202bf7abb211d8041f9aabf80e6e Mon Sep 17 00:00:00 2001
From: bsorrentino <bartolomeo.sorrentino@gmail.com>
Date: Thu, 28 Mar 2024 00:20:12 +0100
Subject: [PATCH] refactor: use string block for prompt template

---
 .../src/main/java/dev/langchain4j/Agent.java  | 20 +++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/agents/src/main/java/dev/langchain4j/Agent.java b/agents/src/main/java/dev/langchain4j/Agent.java
index 2eaf56b..d8d26a8 100644
--- 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 );