code format

Author: alexcheng1982

README.md

@@ -4,13 +4,15 @@
 
 > See JavaAIDev [article](https://javaaidev.com/docs/rag/samples/text-to-sql) for more details.
 
-Test data: [Netflix data](https://github.com/neondatabase/postgres-sample-dbs?tab=readme-ov-file#netflix-data)
+Test
+data: [Netflix data](https://github.com/neondatabase/postgres-sample-dbs?tab=readme-ov-file#netflix-data)
 
 How to run:
 
 1. Requires Java 21 to build and run.
 2. Start Postgres database using Docker Compose.
-3. Start the server and use [Chat Agent UI](http://localhost:8080/webjars/chat-agent-ui/index.html) to run query.
+3. Start the server and use [Chat Agent UI](http://localhost:8080/webjars/chat-agent-ui/index.html)
+   to run query.
 
 Sample query:
 
@@ -30,4 +32,5 @@ See the screenshot below.
 
 ## Full Text-to-SQL Implementation
 
-For a complete Text-to-SQL implementation, check out my [course](https://www.udemy.com/course/spring-ai-text-to-sql/?referralCode=6180D9A02FA8BA9D4F60).
+For a complete Text-to-SQL implementation, check out
+my [course](https://www.udemy.com/course/spring-ai-text-to-sql/?referralCode=6180D9A02FA8BA9D4F60).

src/main/java/com/javaaidev/text2sql/AppConfiguration.java

@@ -13,7 +13,8 @@
 public class AppConfiguration {
 
   @Bean
-  public DatabaseMetadataHelper databaseMetadataHelper(DataSource dataSource,
+  public DatabaseMetadataHelper databaseMetadataHelper(
+      DataSource dataSource,
       ObjectMapper objectMapper) {
     return new DatabaseMetadataHelper(dataSource, objectMapper);
   }

src/main/java/com/javaaidev/text2sql/DatabaseMetadataAdvisor.java

@@ -46,12 +46,15 @@ private String getDatabaseMetadata() {
   }
 
   @Override
-  public ChatClientRequest before(ChatClientRequest chatClientRequest, AdvisorChain advisorChain) {
+  public ChatClientRequest before(ChatClientRequest chatClientRequest,
+      AdvisorChain advisorChain) {
     var systemText = new PromptTemplate(DEFAULT_SYSTEM_TEXT).render(Map.of(
         "table_schemas", tableSchemas
     ));
     return chatClientRequest.mutate()
-        .prompt(chatClientRequest.prompt().augmentSystemMessage(systemText)).build();
+        .prompt(
+            chatClientRequest.prompt().augmentSystemMessage(systemText))
+        .build();
   }
 
   @Override

src/main/java/com/javaaidev/text2sql/controller/ChatController.java

@@ -23,10 +23,12 @@ public ChatController(ChatClient.Builder builder,
   }
 
   @PostMapping("/chat")
-  public Flux<ServerSentEvent<ChatAgentResponse>> chat(@RequestBody ChatAgentRequest request) {
+  public Flux<ServerSentEvent<ChatAgentResponse>> chat(
+      @RequestBody ChatAgentRequest request) {
     return ModelAdapter.toStreamingResponse(
         chatClient.prompt()
-            .messages(ModelAdapter.fromRequest(request).toArray(new Message[0]))
+            .messages(
+                ModelAdapter.fromRequest(request).toArray(new Message[0]))
             .toolNames("runSqlQuery")
             .stream()
             .chatResponse());

src/main/java/com/javaaidev/text2sql/tool/RunSqlQueryTool.java

@@ -10,7 +10,8 @@
 import org.springframework.util.CollectionUtils;
 
 /**
- * Use {@linkplain JdbcClient} to run SQL query and output result in CSV format
+ * Use {@linkplain JdbcClient} to run SQL query and output result in CSV
+ * format
  */
 public class RunSqlQueryTool implements
     Function<RunSqlQueryRequest, RunSqlQueryResponse> {
@@ -48,7 +49,8 @@ private String runQuery(String query) throws IOException {
         .build();
     var builder = new StringBuilder();
     for (Map<String, Object> row : rows) {
-      printer.printRecord(builder, fields.stream().map(row::get).toArray());
+      printer.printRecord(builder,
+          fields.stream().map(row::get).toArray());
     }
     return builder.toString();
   }

src/test/java/com/javaaidev/text2sql/tool/RunSqlQueryToolTest.java

@@ -1,25 +1,26 @@
 package com.javaaidev.text2sql.tool;
 
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.when;
+
 import java.util.List;
 import java.util.Map;
-import static org.junit.jupiter.api.Assertions.*;
 import org.junit.jupiter.api.Test;
-import static org.mockito.ArgumentMatchers.any;
-import static org.mockito.Mockito.when;
-import static org.mockito.Mockito.when;
 import org.mockito.Mockito;
 import org.springframework.jdbc.core.simple.JdbcClient;
 import org.springframework.jdbc.core.simple.JdbcClient.ResultQuerySpec;
 import org.springframework.jdbc.core.simple.JdbcClient.StatementSpec;
 
 class RunSqlQueryToolTest {
+
   @Test
   void testCsvFormat() {
     var mock = Mockito.mock(JdbcClient.class);
     var resultQuerySpecMock = Mockito.mock(ResultQuerySpec.class);
     when(resultQuerySpecMock.listOfRows()).thenReturn(List.of(
         Map.of("v1", "Hello", "v2", 123),
-        Map.of("v1", "World","v2", 456)
+        Map.of("v1", "World", "v2", 456)
     ));
     var statementSpecMock = Mockito.mock(StatementSpec.class);
     when(statementSpecMock.query()).thenReturn(resultQuerySpecMock);