响应为json时将错误输出

LLM-Red-Team · Apr 9, 2024 · 0dd4fa0 · 0dd4fa0
1 parent 26e1735
commit 0dd4fa0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/api/controllers/chat.ts b/src/api/controllers/chat.ts
@@ -215,12 +215,13 @@ async function createCompletion(
         responseType: "stream",
       }
     );
-
-    if (result.headers["content-type"].indexOf("text/event-stream") == -1)
+    if (result.headers["content-type"].indexOf("text/event-stream") == -1) {
+      result.data.on("data", buffer => logger.error(buffer.toString()));
       throw new APIException(
         EX.API_REQUEST_FAILED,
         `Stream response Content-Type invalid: ${result.headers["content-type"]}`
       );
+    }
 
     const streamStartTime = util.timestamp();
     // 接收流为输出文本
@@ -316,6 +317,7 @@ async function createCompletionStream(
         `Invalid response Content-Type:`,
         result.headers["content-type"]
       );
+      result.data.on("data", buffer => logger.error(buffer.toString()));
       const transStream = new PassThrough();
       transStream.end(
         `data: ${JSON.stringify({