ChatGPT流式输出如何终止

[q2316367743]

在使用ChatGPT流式输出时：

openAiClient.streamChatCompletion(chatCompletion, new EventSourceListener() {});

此处没有返回值，该如何终止这个流式输出

看了下源码：

public <T extends BaseChatCompletion> void streamChatCompletion(T chatCompletion, EventSourceListener eventSourceListener) {
      if (Objects.isNull(eventSourceListener)) {
          log.error("参数异常：EventSourceListener不能为空，可以参考：com.unfbx.chatgpt.sse.ConsoleEventSourceListener");
          throw new BaseException(CommonError.PARAM_ERROR);
      }
      if (!chatCompletion.isStream()) {
          chatCompletion.setStream(true);
      }
      try {
          EventSource.Factory factory = EventSources.createFactory(this.okHttpClient);
          ObjectMapper mapper = new ObjectMapper();
          String requestBody = mapper.writeValueAsString(chatCompletion);
          Request request = new Request.Builder()
                  .url(this.apiHost + "v1/chat/completions")
                  .post(RequestBody.create(MediaType.parse(ContentType.JSON.getValue()), requestBody))
                  .build();
          //创建事件
          EventSource eventSource = factory.newEventSource(request, eventSourceListener);
      } catch (JsonProcessingException e) {
          log.error("请求参数解析异常：{}", e);
          e.printStackTrace();
      } catch (Exception e) {
          log.error("请求参数解析异常：{}", e);
          e.printStackTrace();
      }
  }

最后的EventSource是存在cancel()方法的，但是没返回

public interface EventSource {
    Request request();

    void cancel();

    public interface Factory {
        EventSource newEventSource(Request var1, EventSourceListener var2);
    }
}

[LighthouseInTheSea]

        if (StringUtils.equals(data, "[DONE]")) {
            //OpenAI返回数据结束了
            sseEmitter.send(SseEmitter.event()
                    .id("[DONE]")
                    .data("[DONE]")
                    .reconnectTime(3000));
            // 传输完成后自动关闭sse
            sseEmitter.complete();
            return;
        }

[q2316367743]

我并不是要判断是否终止，我是希望在输出的过程中直接终止，就是类似于在nodejs中的AbortController

[Disdude]

外部给一个终止的信号，在每次获取chunk之前判断信号，如果需要终止则直接调用cancel，cancel之后会触发onFail