本文主要是介绍使用 OKhttp3 实现 智普AI ChatGLM HTTP 调用(SSE、异步、同步),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
SSE 调用
SSE(Sever-Sent Event),就是浏览器向服务器发送一个HTTP请求,保持长连接,服务器不断单向地向浏览器推送“信息”(message),这么做是为了节约网络资源,不用一直发请求,建立新连接。
// 创建请求对象Request request = new Request.Builder().url(String.format(sseApi, seeId))
// .post(requestBody) // 请求体
// .addHeader("Authorization", "Bearer " + token).addHeader("Accept", "text/event-stream")
// .addHeader("Content-Type", "text/event-stream;charset=UTF-8").addHeader("Connection", "keep-alive").build();OkHttpClient okHttpClient = new OkHttpClient.Builder().connectTimeout(10, TimeUnit.SECONDS) // 建立连接的超时时间.readTimeout(30, TimeUnit.SECONDS) // 建立连接后读取数据的超时时间.build();// 创建一个 CountDownLatch 对象,其初始计数为1,表示需要等待一个事件发生后才能继续执行。CountDownLatch eventLatch = new CountDownLatch(1);// 实例化EventSource,注册EventSource监听器 -- 创建一个用于处理服务器发送事件的实例,并定义处理事件的回调逻辑final String[] finalMessage = {""};RealEventSource realEventSource = new RealEventSource(request, new EventSourceListener() {@Overridepublic void onEvent(EventSource eventSource, String id, String type, String data) {if ("finish".equals(type)) { // 消息类型,add 增量,finish 结束,error 错误,interrupted 中断eventLatch.countDown();finalMessage[0] = data;
// log.info(data); // 请求到的数据}}@Overridepublic void onFailure(EventSource eventSource, Throwable t, Response response) {t.printStackTrace();}});// 与服务器建立连接realEventSource.connect(okHttpClient);// await() 方法被调用来阻塞当前线程,直到 CountDownLatch 的计数变为0。eventLatch.await();return finalMessage[0];
异步调用
根据文档描述,首先得通过异步 POST 请求获得 task_id ,再根据 task_id 发送 GET 请求获得最终结果
// TODO 设置请求参数,同 SSE 调用// 开启 Http 客户端
OkHttpClient okHttpClient = new OkHttpClient();// 创建请求体
MediaType json = MediaType.parse("application/json; charset=utf-8");
RequestBody requestBody = RequestBody.create(json, requestParam.toString());// 第一步:发送异步请求(POST)获取 task_id,并存放到 taskIdFuture 中
CompletableFuture<String> taskIdFuture = new CompletableFuture<>();Request requestForTaskId = new Request.Builder().url("https://open.bigmodel.cn/api/paas/v3/model-api/chatglm_turbo/async-invoke").post(requestBody).addHeader("Authorization", "Bearer " + token).build();// 创建一个新的异步 HTTP 请求,并指定请求的回调函数
okHttpClient.newCall(requestForTaskId).enqueue(new Callback() {// 在请求成功并返回响应时被调用@Overridepublic void onResponse(Call call, Response response) throws IOException {if (response.isSuccessful()) {String responseBody = response.body().string();System.out.println("requestForTaskId: " + responseBody);// 解析 JSON 响应获取 task_idJSONObject jsonObject = JSON.parseObject(responseBody);String taskId = jsonObject.getJSONObject("data").getString("task_id");// 将结果设置到 CompletableFuturetaskIdFuture.complete(taskId);} else {taskIdFuture.completeExceptionally(new Exception("Request for task_id failed"));}}// 在请求失败时被调用@Overridepublic void onFailure(Call call, IOException e) {taskIdFuture.completeExceptionally(e);}
});// 阻塞主线程,等待 CompletableFuture 的结果,设置了最大等待时间
String taskId = taskIdFuture.get(10, TimeUnit.SECONDS);
System.out.println("Task ID: " + taskId);// TODO 第二步,使用 task_id 发送同步请求(GET)获取最终响应结果(和第四节基本一样)
同步调用
// TODO 设置请求参数,同 SSE 调用// 开启 Http 客户端
OkHttpClient client = new OkHttpClient();// 创建请求体
MediaType json = MediaType.parse("application/json; charset=utf-8");
RequestBody requestBody = RequestBody.create(json, requestParam.toString());// 创建请求对象
Request request = new Request.Builder().url("https://open.bigmodel.cn/api/paas/v3/model-api/chatglm_turbo/invoke").post(requestBody) .addHeader("Authorization", "Bearer " + token).build();// 发送请求
Response response = client.newCall(request).execute();// 处理响应
if (response.isSuccessful()) {String responseBody = response.body().string();System.out.println("Response: " + responseBody);
} else {System.out.println("Request failed: " + response.code() + " " + response.message());
}
这篇关于使用 OKhttp3 实现 智普AI ChatGLM HTTP 调用(SSE、异步、同步)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!