LangChain4j项目中MCP客户端JSON错误处理的优化实践

在LangChain4j项目的开发过程中，我们发现了一个关于MCP（Model Control Protocol）客户端错误处理的重要问题。当MCP服务器返回错误响应时，特别是'invalid_request'这类常见错误，DefaultMcpClient组件会尝试将这些错误信息作为JSON格式进行解析，这在某些情况下会导致额外的解析错误。

问题背景

MCP作为模型控制协议，在LangChain4j项目中承担着重要角色。在实际应用中，由于LLM（大语言模型）自动生成的请求参数可能存在各种问题，MCP服务器返回错误响应的情况相当常见。然而当前实现中，客户端对所有响应都强制进行JSON解析，这显然不够健壮。

问题表现

当出现以下情况时，系统会出现异常：

1. 用户提供明显不合法的文件路径参数
2. MCP服务器返回非JSON格式的错误信息
3. 错误响应中包含特殊字符导致JSON解析失败

这些问题在McpToolsExampleOverStdio示例程序中可以稳定复现，在实际生产环境中也偶有发生。

技术分析

DefaultMcpClient当前的错误处理机制存在两个主要缺陷：

1. 过度严格的格式假设：假设所有服务器响应都是合法的JSON，这在错误场景下并不成立
2. 错误信息丢失：当JSON解析失败时，原始错误信息被丢弃，导致LLM无法根据错误进行调整

解决方案

我们建议采用以下改进措施：

1. 响应内容嗅探：在尝试JSON解析前，先检查响应内容是否符合JSON格式特征
2. 错误信息保留：当JSON解析失败时，将原始响应内容完整保留并封装
3. 分层错误处理：
   • 第一层：处理网络和协议级错误
   • 第二层：尝试解析业务错误
   • 第三层：保底处理原始错误信息

实现建议

try {
    // 尝试JSON解析
    ErrorResponse errorResponse = objectMapper.readValue(responseBody, ErrorResponse.class);
    return new McpError(errorResponse.getMessage());
} catch (JsonProcessingException e) {
    // JSON解析失败时保留原始信息
    return new McpError("原始错误信息: " + responseBody);
}

预期收益

改进后的错误处理机制将带来以下好处：

1. 提高系统健壮性：能够处理各种格式的错误响应
2. 改善用户体验：LLM能够获取完整的错误信息进行自我修正
3. 便于问题诊断：保留原始错误信息有助于问题排查

总结

在AI应用开发中，健壮的错误处理机制尤为重要。LangChain4j项目通过优化MCP客户端的错误处理逻辑，不仅解决了当前的问题，还为未来可能的扩展打下了良好基础。这种分层处理、渐进式解析的思路，也值得在其他类似场景中借鉴应用。