From e2ddc87f61f73085539b798623ad18d537ade22b Mon Sep 17 00:00:00 2001 From: Christian Tzolov Date: Thu, 12 Dec 2024 11:05:18 +0100 Subject: [PATCH] Simplify the MCP Brave code --- .../ai/mcp/samples/brave/Application.java | 17 ++++++----------- 1 file changed, 6 insertions(+), 11 deletions(-) diff --git a/model-context-protocol/brave/src/main/java/org/springframework/ai/mcp/samples/brave/Application.java b/model-context-protocol/brave/src/main/java/org/springframework/ai/mcp/samples/brave/Application.java index 2465008..2e97fbe 100644 --- a/model-context-protocol/brave/src/main/java/org/springframework/ai/mcp/samples/brave/Application.java +++ b/model-context-protocol/brave/src/main/java/org/springframework/ai/mcp/samples/brave/Application.java @@ -27,20 +27,15 @@ public class Application { return args -> { - List functionCallbacks = mcpClient.listTools(null) - .tools() - .stream() - .map(tool -> new McpFunctionCallback(mcpClient, tool)) - .toList(); - - System.out.println("Available tools:"); - functionCallbacks.stream().map(fc -> fc.getName()).forEach(System.out::println); - var chatClient = chatClientBuilder - .defaultFunctions(functionCallbacks.toArray(new McpFunctionCallback[0])) + .defaultFunctions(mcpClient.listTools(null) + .tools() + .stream() + .map(tool -> new McpFunctionCallback(mcpClient, tool)) + .toArray(McpFunctionCallback[]::new)) .build(); - String question = "Can you explain what is Spring AI and if it supports the Model Context Protocol?"; + String question = "Does Spring AI supports the Model Context Protocol? Please provide some references."; System.out.println("QUESTION: " + question); System.out.println("ASSISTANT: " + chatClient.prompt(question).call().content());