codebase/exploring-model-context-protocol-with-spring-ai [BAEL-9132] (#18316)

* add MCP server

* add MCP client

* add test cases

* update chatbot configuration

* remove base API path

* start server before client test cases

* replace openai with anthropic model

Author: hardikSinghBehl

spring-ai-2/pom.xml

@@ -47,6 +47,14 @@
            <groupId>org.springframework.ai</groupId>
            <artifactId>spring-ai-markdown-document-reader</artifactId>
        </dependency>
+       <dependency>
+           <groupId>org.springframework.ai</groupId>
+           <artifactId>spring-ai-mcp-client-spring-boot-starter</artifactId>
+       </dependency>
+       <dependency>
+           <groupId>org.springframework.ai</groupId>
+           <artifactId>spring-ai-mcp-server-webmvc-spring-boot-starter</artifactId>
+       </dependency>
        <dependency>
            <groupId>org.springframework.ai</groupId>
            <artifactId>spring-ai-ollama-spring-boot-starter</artifactId>

spring-ai-2/src/main/java/com/baeldung/springai/mcp/client/ChatbotConfiguration.java

@@ -0,0 +1,22 @@
+package com.baeldung.springai.mcp.client;
+
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.ai.chat.model.ChatModel;
+import org.springframework.ai.mcp.SyncMcpToolCallbackProvider;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+import java.util.List;
+
+@Configuration
+class ChatbotConfiguration {
+
+    @Bean
+    ChatClient chatClient(ChatModel chatModel, SyncMcpToolCallbackProvider toolCallbackProvider) {
+        return ChatClient
+            .builder(chatModel)
+            .defaultTools(toolCallbackProvider.getToolCallbacks())
+            .build();
+    }
+
+}

spring-ai-2/src/main/java/com/baeldung/springai/mcp/client/ChatbotController.java

@@ -0,0 +1,29 @@
+package com.baeldung.springai.mcp.client;
+
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+class ChatbotController {
+
+    private final ChatbotService chatbotService;
+
+    ChatbotController(ChatbotService chatbotService) {
+        this.chatbotService = chatbotService;
+    }
+
+    @PostMapping("/chat")
+    ResponseEntity<ChatResponse> chat(@RequestBody ChatRequest chatRequest) {
+        String answer = chatbotService.chat(chatRequest.question());
+        return ResponseEntity.ok(new ChatResponse(answer));
+    }
+
+    record ChatRequest(String question) {
+    }
+
+    record ChatResponse(String answer) {
+    }
+
+}

spring-ai-2/src/main/java/com/baeldung/springai/mcp/client/ChatbotService.java

@@ -0,0 +1,23 @@
+package com.baeldung.springai.mcp.client;
+
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.stereotype.Service;
+
+@Service
+class ChatbotService {
+
+    private final ChatClient chatClient;
+
+    ChatbotService(ChatClient chatClient) {
+        this.chatClient = chatClient;
+    }
+
+    String chat(String question) {
+        return chatClient
+            .prompt()
+            .user(question)
+            .call()
+            .content();
+    }
+
+}

spring-ai-2/src/main/java/com/baeldung/springai/mcp/client/ClientApplication.java

@@ -0,0 +1,29 @@
+package com.baeldung.springai.mcp.client;
+
+import org.springframework.ai.autoconfigure.bedrock.converse.BedrockConverseProxyChatAutoConfiguration;
+import org.springframework.ai.autoconfigure.ollama.OllamaAutoConfiguration;
+import org.springframework.ai.autoconfigure.openai.OpenAiAutoConfiguration;
+import org.springframework.ai.autoconfigure.vectorstore.chroma.ChromaVectorStoreAutoConfiguration;
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.annotation.PropertySource;
+
+/**
+ * Excluding the below auto-configurations to avoid start up
+ * failure. Their corresponding starters are present on the classpath but are
+ * only needed by other articles in the shared codebase.
+ */
+@SpringBootApplication(exclude = {
+    OllamaAutoConfiguration.class,
+    OpenAiAutoConfiguration.class,
+    ChromaVectorStoreAutoConfiguration.class,
+    BedrockConverseProxyChatAutoConfiguration.class
+})
+@PropertySource("classpath:application-mcp-client.properties")
+class ClientApplication {
+
+    public static void main(String[] args) {
+        SpringApplication.run(ClientApplication.class, args);
+    }
+
+}

spring-ai-2/src/main/java/com/baeldung/springai/mcp/server/AuthorRepository.java

@@ -0,0 +1,25 @@
+package com.baeldung.springai.mcp.server;
+
+import org.springframework.ai.tool.annotation.Tool;
+
+import java.util.List;
+
+class AuthorRepository {
+
+    @Tool(description = "Get Baeldung author details using an article title")
+    Author getAuthorByArticleTitle(String articleTitle) {
+        return new Author("John Doe", "[email protected]");
+    }
+
+    @Tool(description = "Get highest rated Baeldung authors")
+    List<Author> getTopAuthors() {
+        return List.of(
+            new Author("John Doe", "[email protected]"),
+            new Author("Jane Doe", "[email protected]")
+        );
+    }
+
+    record Author(String name, String email) {
+    }
+
+}

spring-ai-2/src/main/java/com/baeldung/springai/mcp/server/MCPServerConfiguration.java

@@ -0,0 +1,19 @@
+package com.baeldung.springai.mcp.server;
+
+import org.springframework.ai.tool.ToolCallbackProvider;
+import org.springframework.ai.tool.method.MethodToolCallbackProvider;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+class MCPServerConfiguration {
+
+    @Bean
+    ToolCallbackProvider authorTools() {
+        return MethodToolCallbackProvider
+            .builder()
+            .toolObjects(new AuthorRepository())
+            .build();
+    }
+
+}

spring-ai-2/src/main/java/com/baeldung/springai/mcp/server/ServerApplication.java

@@ -0,0 +1,25 @@
+package com.baeldung.springai.mcp.server;
+
+import org.springframework.ai.autoconfigure.openai.OpenAiAutoConfiguration;
+import org.springframework.ai.autoconfigure.vectorstore.chroma.ChromaVectorStoreAutoConfiguration;
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.annotation.PropertySource;
+
+/**
+ * Excluding the below auto-configurations to avoid start up
+ * failure. Their corresponding starters are present on the classpath but are
+ * only needed by other articles in the shared codebase.
+ */
+@SpringBootApplication(exclude = {
+    OpenAiAutoConfiguration.class,
+    ChromaVectorStoreAutoConfiguration.class
+})
+@PropertySource("classpath:application-mcp-server.properties")
+public class ServerApplication {
+
+    public static void main(String[] args) {
+        SpringApplication.run(ServerApplication.class, args);
+    }
+
+}

spring-ai-2/src/main/resources/application-mcp-client.properties

@@ -0,0 +1,11 @@
+spring.ai.anthropic.api-key=${ANTHROPIC_API_KEY}
+spring.ai.anthropic.chat.options.model=claude-3-5-sonnet-20241022
+
+spring.ai.mcp.client.sse.connections.author-tools-server.url=http://localhost:8081
+
+spring.ai.mcp.client.stdio.connections.filesystem.command=npx
+spring.ai.mcp.client.stdio.connections.filesystem.args=-y, @modelcontextprotocol/server-filesystem, ./
+
+spring.ai.mcp.client.stdio.connections.brave-search.command=npx
+spring.ai.mcp.client.stdio.connections.brave-search.args=-y, @modelcontextprotocol/server-brave-search
+spring.ai.mcp.client.stdio.connections.brave-search.env.BRAVE_API_KEY=${BRAVE_API_KEY}

spring-ai-2/src/main/resources/application-mcp-server.properties

@@ -0,0 +1 @@
+server.port=8081