From 238381256dbfb4ddf6a3917e21ad731160fe880e Mon Sep 17 00:00:00 2001
From: Finley Ge <32237950+FinleyGe@users.noreply.github.com>
Date: Tue, 20 Aug 2024 10:05:19 +0800
Subject: [PATCH] feat: custom-censor-check (#2437)

* feat: custom-censor-check

* chore: adjust
---
 packages/service/common/api/requestPlusApi.ts          | 2 +-
 packages/service/core/workflow/dispatch/chat/oneapi.ts | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/packages/service/common/api/requestPlusApi.ts b/packages/service/common/api/requestPlusApi.ts
index b9d6b7d9c..100d6cd56 100644
--- a/packages/service/common/api/requestPlusApi.ts
+++ b/packages/service/common/api/requestPlusApi.ts
@@ -1,7 +1,7 @@
 import { POST } from './plusRequest';
 
 export const postTextCensor = (data: { text: string }) =>
-  POST<{ code?: number; message: string }>('/common/censor/text_baidu', data)
+  POST<{ code?: number; message: string }>('/common/censor/check', data)
     .then((res) => {
       if (res?.code === 5000) {
         return Promise.reject(res);
diff --git a/packages/service/core/workflow/dispatch/chat/oneapi.ts b/packages/service/core/workflow/dispatch/chat/oneapi.ts
index f0291bc5c..a6d5392e9 100644
--- a/packages/service/core/workflow/dispatch/chat/oneapi.ts
+++ b/packages/service/core/workflow/dispatch/chat/oneapi.ts
@@ -112,17 +112,17 @@ export const dispatchChatCompletion = async (props: ChatProps): Promise<ChatResp
       systemPrompt,
       stringQuoteText
     }),
-    async () => {
+    (async () => {
       // censor model and system key
       if (modelConstantsData.censor && !user.openaiAccount?.key) {
-        await postTextCensor({
+        return postTextCensor({
           text: `${systemPrompt}
             ${datasetQuoteText}
             ${userChatInput}
           `
         });
       }
-    }
+    })()
   ]);
 
   // Get the request messages