From 5ffe456cdf1c3e2e5f53f47afd94c1951121f9e8 Mon Sep 17 00:00:00 2001
From: Miura-KR <ovis_w6v6_grappler95@yahoo.co.jp>
Date: Thu, 6 Jul 2023 22:06:16 +0900
Subject: [PATCH] Make the ExecutorService.shutdownNow method available to
 interrupt streaming response

---
 .../java/com/theokanning/openai/service/OpenAiService.java   | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/service/src/main/java/com/theokanning/openai/service/OpenAiService.java b/service/src/main/java/com/theokanning/openai/service/OpenAiService.java
index e1cb856f..30bbe892 100644
--- a/service/src/main/java/com/theokanning/openai/service/OpenAiService.java
+++ b/service/src/main/java/com/theokanning/openai/service/OpenAiService.java
@@ -373,6 +373,11 @@ public void shutdownExecutor() {
         this.executorService.shutdown();
     }
 
+    public List<Runnable> shutdownExecutorNow() {
+        Objects.requireNonNull(this.executorService, "executorService must be set in order to shut down");
+        return this.executorService.shutdownNow();
+    }
+
     public static OpenAiApi buildApi(String token, Duration timeout) {
         ObjectMapper mapper = defaultObjectMapper();
         OkHttpClient client = defaultClient(token, timeout);