From dd793e0833ed49458212cd41b48a9bd89303b55b Mon Sep 17 00:00:00 2001
From: code-october <148516338+code-october@users.noreply.github.com>
Date: Sat, 12 Oct 2024 06:19:11 +0000
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84=E5=88=A4=E6=96=AD=E9=80=BB?=
 =?UTF-8?q?=E8=BE=91=EF=BC=8C=E6=B7=BB=E5=8A=A0jest=E6=B5=8B=E8=AF=95?=
 =?UTF-8?q?=E6=A0=B7=E4=BE=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/store/chat.ts           |  6 +--
 test/isValidMessage.test.ts | 96 +++++++++++++++++++++++++++++++++++++
 2 files changed, 99 insertions(+), 3 deletions(-)
 create mode 100644 test/isValidMessage.test.ts

diff --git a/app/store/chat.ts b/app/store/chat.ts
index f01dd7b41..197025853 100644
--- a/app/store/chat.ts
+++ b/app/store/chat.ts
@@ -743,15 +743,15 @@ export const useChatStore = createPersistStore(
             const codeBlockContent = message.slice(3, -3).trim();
             const jsonString = codeBlockContent.replace(/^json\s*/i, '').trim();
             try {
-              // 返回 json 格式消息，含 error.message 字段，判定为错误回复，否则为正常回复
+              // 返回 json 格式消息，error 字段为 true 或者包含 error.message 字段，判定为错误回复，否则为正常回复
               const jsonObject = JSON.parse(jsonString);
-              if (jsonObject?.error?.message) {
+              if (jsonObject?.error == true || jsonObject?.error?.message) {
                 return false;
               }
               return true;
             } catch (e) {
               console.log("Invalid JSON format.");
-              // 非 json 格式，大概率是正常回复
+              // 非 json 格式，通常可认为是正常回复
               return true;
             }
           }
diff --git a/test/isValidMessage.test.ts b/test/isValidMessage.test.ts
new file mode 100644
index 000000000..edea222f9
--- /dev/null
+++ b/test/isValidMessage.test.ts
@@ -0,0 +1,96 @@
+function isValidMessage(message: any): boolean {
+    if (typeof message !== "string") {
+        return false;
+    }
+    if (message.startsWith("```") && message.endsWith("```")) {
+        const codeBlockContent = message.slice(3, -3).trim();
+        const jsonString = codeBlockContent.replace(/^json\s*/i, '').trim();
+        try {
+            // 返回 json 格式消息，含 error.message 字段，判定为错误回复，否则为正常回复
+            const jsonObject = JSON.parse(jsonString);
+            if (jsonObject?.error == true || jsonObject?.error?.message) {
+                return false;
+            }
+            return true;
+        } catch (e) {
+            console.log("Invalid JSON format.");
+            // 非 json 格式，大概率是正常回复
+            return true;
+        }
+    }
+    return true;
+}
+
+describe("is valid message module", () => {
+    test("error msg no.0", () => {
+        const message = "Hello! How can I assist you today?";
+        expect(isValidMessage(message)).toBe(true);
+    });
+    test("error msg no.1", () => {
+        const message = `
+        \`\`\`json
+        {
+            "error": true,
+            "msg": "金额不足"
+        }
+        \`\`\`
+        `;
+        expect(isValidMessage(message)).toBe(false);
+    });
+    test("error msg no.2", () => {
+        const message = `
+        \`\`\`
+        {
+            "error": {
+                "message": "You didn't provide an API key. You need to provide your API key in an Authorization header using Bearer auth (i.e. Authorization: Bearer YOUR_KEY), or as the password field (with blank username) if you're accessing the API from your browser and are prompted for a username and password. You can obtain an API key from https://platform.openai.com/account/api-keys.",
+                "type": "invalid_request_error",
+                "param": null,
+                "code": null
+            }
+        }
+        \`\`\`
+        `;
+        expect(isValidMessage(message)).toBe(false);
+    });
+    test("error msg no.3", () => {
+        const message = `
+        \`\`\`
+        {
+            "error": {
+                "message": "Incorrect API key provided: 123456. You can find your API key at https://platform.openai.com/account/api-keys.",
+                "type": "invalid_request_error",
+                "param": null,
+                "code": "invalid_api_key"
+            }
+        }
+        \`\`\`
+        `;
+        expect(isValidMessage(message)).toBe(false);
+    });
+    test("error msg no.4", () => {
+        const message = `
+        \`\`\`
+        {
+            "error": {
+                "message": "当前分组 default 下对于模型 gpt-4 无可用渠道 (request id: 2024101214105418395279367750613)",
+                "type": "one_api_error"
+            }
+        }
+        \`\`\`
+        `;
+        expect(isValidMessage(message)).toBe(false);
+    });
+    test("error msg no.5", () => {
+        const message = `
+        \`\`\`
+        {
+            "error": {
+                "message": "该令牌状态不可用 (request id: 2024101214105418395279367750613)",
+                "type": "one_api_error"
+            }
+        }
+        \`\`\`
+        `;
+        expect(isValidMessage(message)).toBe(false);
+    });
+});
\ No newline at end of file