improve the MCP sqlite demos

model-context-protocol/sqlite/chatbot/src/main/java/org/springframework/ai/mcp/samples/sqlite/Application.java

@@ -0,0 +1,100 @@
+package org.springframework.ai.mcp.samples.sqlite;
+
+import java.nio.file.Paths;
+import java.time.Duration;
+import java.util.List;
+import java.util.Scanner;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;
+import org.springframework.ai.chat.memory.InMemoryChatMemory;
+import org.springframework.ai.mcp.client.McpClient;
+import org.springframework.ai.mcp.client.McpSyncClient;
+import org.springframework.ai.mcp.client.stdio.ServerParameters;
+import org.springframework.ai.mcp.client.stdio.StdioServerTransport;
+import org.springframework.ai.mcp.spring.McpFunctionCallback;
+import org.springframework.boot.CommandLineRunner;
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.ConfigurableApplicationContext;
+import org.springframework.context.annotation.Bean;
+
+@SpringBootApplication
+public class Application {
+
+	public static void main(String[] args) {
+		SpringApplication.run(Application.class, args);
+	}
+
+	@Bean
+	public CommandLineRunner interactiveChat(ChatClient.Builder chatClientBuilder,
+			List<McpFunctionCallback> functionCallbacks,
+			ConfigurableApplicationContext context) {
+		return args -> {
+
+			var chatClient = chatClientBuilder
+					.defaultFunctions(functionCallbacks.toArray(new McpFunctionCallback[0]))
+					.defaultAdvisors(new MessageChatMemoryAdvisor(new InMemoryChatMemory()))
+					.build();
+
+			var scanner = new Scanner(System.in);
+			System.out.println("\nStarting interactive chat session. Type 'exit' to quit.");
+
+			try {
+				while (true) {
+					System.out.print("\nUSER: ");
+					String input = scanner.nextLine();
+
+					if (input.equalsIgnoreCase("exit")) {
+						System.out.println("Ending chat session.");
+						break;
+					}
+
+					System.out.print("ASSISTANT: ");
+					System.out.println(chatClient.prompt(input).call().content());
+				}
+			} finally {
+				scanner.close();
+				context.close();
+			}
+
+		};
+	}
+
+	@Bean
+	public List<McpFunctionCallback> functionCallbacks(McpSyncClient mcpClient) {
+
+		var callbacks = mcpClient.listTools(null)
+				.tools()
+				.stream()
+				.map(tool -> new McpFunctionCallback(mcpClient, tool))
+				.toList();
+		return callbacks;
+	}
+
+	@Bean(destroyMethod = "close")
+	public McpSyncClient mcpClient() {
+
+		var stdioParams = ServerParameters.builder("uvx")
+				.args("mcp-server-sqlite", "--db-path",
+						getDbPath())
+				.build();
+
+		var mcpClient = McpClient.sync(new StdioServerTransport(stdioParams),
+				Duration.ofSeconds(10), new ObjectMapper());
+
+		var init = mcpClient.initialize();
+
+		System.out.println("MCP Initialized: " + init);
+
+		return mcpClient;
+
+	}
+
+	private static String getDbPath() {
+		return Paths.get(System.getProperty("user.dir"), "test.db").toString();
+	}
+
+}

model-context-protocol/sqlite/chatbot/src/main/resources/application.properties

@@ -0,0 +1,4 @@
+spring.application.name=mcp
+spring.main.web-application-type=none
+
+spring.ai.openai.api-key=${OPENAI_API_KEY}