Spring/spring-ai-examples
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Simplify the MCP Brave code

Browse Source
This commit is contained in:
Christian Tzolov
2024-12-12 11:05:18 +01:00
parent 30b0a59b49
commit e2ddc87f61
1 changed files with 6 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
model-context-protocol/brave/src/main/java/org/springframework/ai/mcp/samples/brave/Application.java
View File

@@ -27,20 +27,15 @@ public class Application {
return args -> {
List<McpFunctionCallback> 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());