Return HTTP errors immediately to avoid AI character hanging

uezo · uezo · commit e604b844e59b · 2025-04-28T23:52:21.000+09:00
- When the server responds with a non-2xx HTTP status, send an immediate error response containing the status code.
- This prevents the AI character from hanging until the network timeout on HTTP errors.
diff --git a/Plugins/AIAvatarKitServiceWebGL.jslib b/Plugins/AIAvatarKitServiceWebGL.jslib
@@ -21,10 +21,21 @@ mergeInto(LibraryManager.library, {
             body: aakStreamRequest,
             signal: document.aakAbortController.signal
         })
-        .then((response) => response.body.getReader())
+        .then(response => {
+            if (!response.ok) {
+                SendMessage(
+                    targetObjectName,
+                    "SetAIAvatarKitMessageStreamChunk",
+                    sessionId + "::Error: " + response.status
+                );
+            }
+            return response.body.getReader();
+        })
         .then((reader) => {
             const readChunk = function({done, value}) {
                 if(done) {
+                    // Send empty message to ensure to stop stream handling
+                    SendMessage(targetObjectName, "SetAIAvatarKitMessageStreamChunk", sessionId + "::");
                     reader.releaseLock();
                     return;
                 }
diff --git a/Plugins/ChatGPTServiceWebGL.jslib b/Plugins/ChatGPTServiceWebGL.jslib
@@ -21,10 +21,21 @@ mergeInto(LibraryManager.library, {
             body: chatCompletionRequest,
             signal: document.chatGPTAbortController.signal
         })
-        .then((response) => response.body.getReader())
+        .then(response => {
+            if (!response.ok) {
+                SendMessage(
+                    targetObjectName,
+                    "SetChatCompletionStreamChunk",
+                    sessionId + "::Error: " + response.status
+                );
+            }
+            return response.body.getReader();
+        })
         .then((reader) => {
             const readChunk = function({done, value}) {
                 if(done) {
+                    // Send empty message to ensure to stop stream handling
+                    SendMessage(targetObjectName, "SetChatCompletionStreamChunk", sessionId + "::");
                     reader.releaseLock();
                     return;
                 }
diff --git a/Plugins/ClaudeServiceWebGL.jslib b/Plugins/ClaudeServiceWebGL.jslib
@@ -23,10 +23,21 @@ mergeInto(LibraryManager.library, {
             body: claudeStreamRequest,
             signal: document.claudeAbortController.signal
         })
-        .then((response) => response.body.getReader())
+        .then(response => {
+            if (!response.ok) {
+                SendMessage(
+                    targetObjectName,
+                    "SetClaudeMessageStreamChunk",
+                    sessionId + "::Error: " + response.status
+                );
+            }
+            return response.body.getReader();
+        })
         .then((reader) => {
             const readChunk = function({done, value}) {
                 if(done) {
+                    // Send empty message to ensure to stop stream handling
+                    SendMessage(targetObjectName, "SetClaudeMessageStreamChunk", sessionId + "::");
                     reader.releaseLock();
                     return;
                 }
diff --git a/Plugins/DifyServiceWebGL.jslib b/Plugins/DifyServiceWebGL.jslib
@@ -21,10 +21,21 @@ mergeInto(LibraryManager.library, {
             body: difyStreamRequest,
             signal: document.difyAbortController.signal
         })
-        .then((response) => response.body.getReader())
+        .then(response => {
+            if (!response.ok) {
+                SendMessage(
+                    targetObjectName,
+                    "SetDifyMessageStreamChunk",
+                    sessionId + "::Error: " + response.status
+                );
+            }
+            return response.body.getReader();
+        })
         .then((reader) => {
             const readChunk = function({done, value}) {
                 if(done) {
+                    // Send empty message to ensure to stop stream handling
+                    SendMessage(targetObjectName, "SetDifyMessageStreamChunk", sessionId + "::");
                     reader.releaseLock();
                     return;
                 }
diff --git a/Plugins/GeminiServiceWebGL.jslib b/Plugins/GeminiServiceWebGL.jslib
@@ -19,10 +19,21 @@ mergeInto(LibraryManager.library, {
             body: geminiStreamRequest,
             signal: document.geminiAbortController.signal
         })
-        .then((response) => response.body.getReader())
+        .then(response => {
+            if (!response.ok) {
+                SendMessage(
+                    targetObjectName,
+                    "SetGeminiMessageStreamChunk",
+                    sessionId + "::Error: " + response.status
+                );
+            }
+            return response.body.getReader();
+        })
         .then((reader) => {
             const readChunk = function({done, value}) {
                 if(done) {
+                    // Send empty message to ensure to stop stream handling
+                    SendMessage(targetObjectName, "SetGeminiMessageStreamChunk", sessionId + "::");
                     reader.releaseLock();
                     return;
                 }
