Spring AI调用VLLM服务时HTTP协议版本问题的解决方案

在使用Spring AI框架调用VLLM大模型服务时，开发者可能会遇到400 Bad Request错误。这个问题通常是由于HTTP协议版本不兼容导致的。本文将深入分析问题原因并提供完整的解决方案。

问题现象

当开发者通过Spring AI框架调用VLLM服务时，虽然API POST请求在直接调用时能够正常工作，但在Spring AI框架中却会出现400错误。从日志中可以看到，错误发生在HTTP请求阶段，提示"400 Bad Request from POST"。

问题根源

经过分析，这个问题的主要原因是Spring AI框架默认使用的HTTP客户端可能尝试使用HTTP/2协议与VLLM服务通信。然而，VLLM服务的API端点可能不完全支持HTTP/2协议，导致协议协商失败。

解决方案

要解决这个问题，我们需要强制HTTP客户端使用HTTP/1.1协议。以下是完整的配置方案：

import java.net.http.HttpClient;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.http.client.JdkClientHttpRequestFactory;
import org.springframework.http.client.reactive.JdkClientHttpConnector;
import org.springframework.web.client.RestClient;
import org.springframework.web.reactive.function.client.WebClient;

@Configuration
public class HttpClientConfiguration {
  
  private HttpClient client() {
    return HttpClient.newBuilder()
        .version(HttpClient.Version.HTTP_1_1)  // 强制使用HTTP/1.1协议
        .build();
  }

  @Bean
  @Primary
  public RestClient.Builder restClientBuilder() {
    return RestClient.builder()
        .requestFactory(new JdkClientHttpRequestFactory(client()));
  }

  @Bean
  @Primary
  public WebClient.Builder webClientBuilder() {
    return WebClient.builder()
        .clientConnector(new JdkClientHttpConnector(client()));
  }
}

实现原理

1. HTTP客户端配置：通过HttpClient.newBuilder()创建客户端实例时，明确指定使用HTTP/1.1版本。

2. 双重适配：配置同时支持RestClient和WebClient两种HTTP客户端，确保Spring AI框架中各种调用场景都能正常工作。

3. 主Bean标记：使用@Primary注解确保自定义配置优先于框架默认配置。

注意事项

1. 此解决方案适用于Spring AI 1.0.0-M8及以上版本。

2. 如果项目中同时使用其他HTTP客户端，可能需要调整配置以避免冲突。

3. 在生产环境中，建议进一步配置连接池和超时参数以获得更好的性能。

扩展知识

HTTP/1.1和HTTP/2的主要区别在于：

• HTTP/1.1使用文本格式传输数据，而HTTP/2使用二进制格式
• HTTP/2支持多路复用，可以在一个连接上并行传输多个请求
• HTTP/2支持头部压缩，减少数据传输量

虽然HTTP/2在性能上有优势，但在某些特定场景下，强制使用HTTP/1.1仍然是必要的兼容性解决方案。

通过以上配置，开发者可以顺利解决Spring AI调用VLLM服务时的协议兼容性问题，确保大模型服务能够正常工作。