Spring AI 智能体通过 MCP 集成本地文件数据

2024年12月28日 · 阅读需 6 分钟

PMC Member

MCP 简介

模型上下文协议（即 Model Context Protocol，MCP）是一个开放协议，它规范了应用程序如何向大型语言模型（LLM）提供上下文。MCP 提供了一种统一的方式将 AI 模型连接到不同的数据源和工具，它定义了统一的集成方式。在开发智能体（Agent）的过程中，我们经常需要将将智能体与数据和工具集成，MCP 以标准的方式规范了智能体与数据及工具的集成方式，可以帮助您在LLM之上构建智能体（Agent）和复杂的工作流。目前已经有大量的服务接入并提供了 MCP server 实现，当前这个生态正在以非常快的速度不断的丰富中，具体可参见：https://github.com/modelcontextprotocol/servers。

Spring AI MCP

Spring AI MCP 为模型上下文协议提供 Java 和 Spring 框架集成。它使 Spring AI 应用程序能够通过标准化的接口与不同的数据源和工具进行交互，支持同步和异步通信模式。

spring-ai-mcp-architecture

Spring AI MCP 采用模块化架构，包括以下组件：

Spring AI应用程序：使用Spring AI框架构建想要通过MCP访问数据的生成式AI应用程序
Spring MCP Client：MCP协议的Spring AI实现，与服务器保持1:1连接
MCP Server：轻量级程序，每个程序都通过标准化的模型上下文协议公开特定的功能
本地数据源：MCP Server 可以安全访问的计算机文件、数据库和服务
远程服务：MCP Server 可以通过互联网（例如，通过API）连接到的外部系统

通过一个示例快速体验 Spring AI MCP

这里我们提供一个示例智能体应用，这个智能体可以通过 MCP 查询或更新本地文件系统，并以文件系统中的数据作为上下文与模型交互。次示例演示如何使用模型上下文协议（MCP）将 Spring AI 与本地文件系统进行集成。

可在此查看示例完整源码。

运行示例

前提条件

安装 npx (Node Package eXecute): 首先确保本地机器安装了 npm，然后运行如下命令：
```
# 安装 npx
npm install -g npx
```

下载示例源码

git clone https://github.com/springaialibaba/spring-ai-alibaba-examples.git
cd spring-ai-alibaba-examples/spring-ai-alibaba-mcp-example/filesystem

设置环境变量

# 通义大模型 Dashscope API-KEY
export AI_DASHSCOPE_API_KEY=${your-api-key-here}

构建示例
```
./mvnw clean install
```

运行示例应用

运行示例，智能体将向模型发起提问（源码中包含预置问题，可通过源码查看），可通过控制台查看输出结果。

./mvnw spring-boot:run

如果您是在 IDE 中运行示例，并且遇到 filesystem mcp server 返回的文件访问权限问题，请确保指定当前进程工作目录为 spring-ai-alibaba-mcp-example/filesystem 目录。

示例架构（源码说明）

前文中我们讲解了 Spring AI 与 MCP 集成的基础架构，在接下来的示例中，我们将用到以下关键组件：

MCP Client，与 MCP 集成的关键，提供了与本地文件系统进行交互的能力。
Function Callbacks，Spring AI MCP 的 function calling 声明方式。
Chat Client，Spring AI 关键组件，用于LLM模型交互、智能体代理。

声明 ChatClient

// List<McpFunctionCallback> functionCallbacks;
var chatClient = chatClientBuilder.defaultFunctions(functionCallbacks).build();

和开发之前的 Spring AI 应用一样，我们先定义一个 ChatClient Bean，用于与大模型交互的代理。需要注意的是，我们为 ChatClient 注入的 functions 是通过 MCP 组件（McpFunctionCallback）创建的。

接下来让我们具体看一下 McpFunctionCallback 是怎么使用的。

声明 MCP Function Callbacks

以下代码段通过 mcpClient与 MCP server 交互，将 MCP 工具通过 McpFunctionCallback 适配为标准的 Spring AI function。

发现 MCP server 中可用的工具 tool（Spring AI 中叫做 function）列表
依次将每个 tool 转换成 Spring AI function callback
最终我们会将这些 McpFunctionCallback 注册到 ChatClient 使用

@Bean
public List<McpFunctionCallback> functionCallbacks(McpSyncClient mcpClient) {
    return mcpClient.listTools(null)
            .tools()
            .stream()
            .map(tool -> new McpFunctionCallback(mcpClient, tool))
            .toList();
}

可以看出，ChatClient 与模型交互的过程是没有变化的，模型在需要的时候告知 ChatClient 去做函数调用，只不过 Spring AI 通过 McpFunctionCallback 将实际的函数调用过程委托给了 MCP，通过标准的 MCP 协议与本地文件系统交互:

在与大模交互的过程中，ChatClient 处理相关的 function calls 请求
ChatClient 调用 MCP 工具（通过 McpClient）
McpClient 与 MCP server（即 filesystem）交互

初始化 McpClient

该智能体应用使用同步 MCP 客户端与本地运行的文件系统 MCP server 通信：

@Bean(destroyMethod = "close")
public McpSyncClient mcpClient() {
    var stdioParams = ServerParameters.builder("npx")
            .args("-y", "@modelcontextprotocol/server-filesystem", "path))
            .build(); // 1

    var mcpClient = McpClient.sync(new StdioServerTransport(stdioParams),
            Duration.ofSeconds(10), new ObjectMapper()); //2

    var init = mcpClient.initialize(); // 3
    System.out.println("MCP Initialized: " + init);

    return mcpClient;
}

在以上代码中：

配置 MCP server 启动命令与参数
初始化 McpClient：关联 MCP server、指定超时时间等
Spring AI 会使用 npx -y @modelcontextprotocol/server-filesystem "/path/to/file"在本地机器创建一个独立的子进程（代表本地 Mcp server），Spring AI 与 McpClient 通信，McpClient 进而通过与 Mcp server 的连接操作本地文件。

总结

MCP 统一，当前 Spring Ai ，随着生态中越来越多的系统与服务集成。在 Java 生态中，我们有非常多的基础服务与业务应用，如何将这些使用 Spring Boot、Spring Cloud、Dubbo 等开发的单体或微服务应用发布为 MCP server

MCP 简介​

Spring AI MCP​

通过一个示例快速体验 Spring AI MCP​

运行示例​

前提条件​

运行示例应用​

示例架构（源码说明）​

声明 ChatClient​

声明 MCP Function Callbacks​

初始化 McpClient​

总结​