Http流式响应接口封装

1. 背景

需要对接某个大模型的流式http对话接口,因为该项目中还需要对接另外的大模型服务但是另一个请求方式为websocket,为了实现模型服务切换以及对客户端做统一的流式响应,需要对大模型服务的调用做统一的流式响应封装。

环境:JDK11

2. 代码

2.1 回调函数

java 复制代码
@FunctionalInterface
public interface LLMChatCallback {

    void callback(String message);

}

2.2 统一对话接口

java 复制代码
public interface LLMService {

    /**
     * 模型对话
     *
     * @param payload 用于请求大模型的参数
     * @param callback 响应回调 调用者通过实现callback函数接收消息
     **/
    void dialogue(String payload, LLMChatCallback callback);

}

2.3 StringSubscriber

java 复制代码
import lombok.extern.slf4j.Slf4j;
import java.util.concurrent.Flow;

/**
 * http响应式流订阅
 **/
@Slf4j
public class StringSubscriber implements Flow.Subscriber<String> {

    Flow.Subscription subscription;

    /**
     * 消息回调接口
     */
    private final LLMChatCallback callback;

    public StringSubscriber(LLMChatCallback callback) {
        this.callback = callback;
    }

    @Override
    public void onSubscribe(Flow.Subscription subscription) {
        this.subscription = subscription;
        subscription.request(1);
    }

    @Override
    public void onNext(String item) {
        log.debug("onNext: {}", item);
        if (callback != null) {
            // 可做业务处理
            callback.callback(item);
        }
        subscription.request(1);
    }

    @Override
    public void onError(Throwable throwable) {
        log.error("StringSubscriber error", throwable);
    }

    @Override
    public void onComplete() {
        log.debug("StringSubscriber complete");
    }

}

2.4 大模型Service实现

java 复制代码
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import java.net.URI;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.time.Duration;

/**
 * B大模型服务
 **/
@Slf4j
@Service
public class BLMService implements LLMService {

    @Override
    public void dialogue(String payload, LLMChatCallback callback) {
        HttpRequest httpRequest = HttpRequest.newBuilder(URI.create("this is url")).POST(HttpRequest.BodyPublishers.ofString("this is payload")).build();
        HttpClient httpClient = HttpClient.newBuilder().connectTimeout(Duration.ofMinutes(5)).version(HttpClient.Version.HTTP_2).build();
        HttpResponse.BodySubscriber<Void> bodySubscriber = HttpResponse.BodySubscribers.fromLineSubscriber(new StringSubscriber(callback));
        httpClient.sendAsync(httpRequest, responseInfo -> bodySubscriber).join();
    }

}

2.5 使用

java 复制代码
@Resource
private BLMService bLMService;

public void void dialogue() {
    // 构建请求负载
    String payload = "this is payload";
    // 可以扩展使用工厂及策略模式获取对应的模型对话service
    bLMService.dialogue(payload, message -> {
        // 处理业务接收返回的信息 比如通过grpc的流式响应 onNext(message) 同步返回
        System.out.println(message);
    });
}
相关推荐
Hellyc9 分钟前
SpringMVC响应数据:页面跳转与回写数据
java·前端·学习
嘵奇23 分钟前
深入解析 Spring Boot 测试核心注解
java·spring boot·后端
癞皮狗不赖皮27 分钟前
Java安全基础-反射机制
java·反射机制·java安全基础
鹅肝手握高V五色27 分钟前
Wireshark入门教程:如何抓取和过滤网络数据包
websocket·网络协议·tcp/ip·http·网络安全·https·udp
郑梓妍31 分钟前
ubuntu改用户权限
服务器·网络·数据库
别惊鹊33 分钟前
(三)安装和使用Maven
java·maven
兢兢业业的小白鼠1 小时前
Java高级JVM知识点记录,内存结构,垃圾回收,类文件结构,类加载器
java·开发语言·jvm·tomcat
老六ip加速器1 小时前
如何改电脑网络ip地址:一步步指导
网络·网络协议·tcp/ip
落榜程序员1 小时前
Java 基础-29-final关键字-详解
java·开发语言
别说我什么都不会1 小时前
ohos.net.http请求HttpResponse header中set-ccokie值被转成array类型
网络协议·harmonyos