fix: sensitive check

client/src/pages/api/chat/chat.ts

@@ -10,7 +10,7 @@ import { resStreamResponse } from '@/service/utils/chat';
 import { appKbSearch } from '../openapi/kb/appKbSearch';
 import { appKbSearch } from '../openapi/kb/appKbSearch';
 import { ChatRoleEnum, QUOTE_LEN_HEADER, GUIDE_PROMPT_HEADER } from '@/constants/chat';
 import { ChatRoleEnum, QUOTE_LEN_HEADER, GUIDE_PROMPT_HEADER } from '@/constants/chat';
 import { BillTypeEnum } from '@/constants/user';
 import { BillTypeEnum } from '@/constants/user';
-import { sensitiveCheck } from '@/service/api/text';
+import { sensitiveCheck } from '../openapi/text/sensitiveCheck';
 import { NEW_CHATID_HEADER } from '@/constants/chat';
 import { NEW_CHATID_HEADER } from '@/constants/chat';
 import { saveChat } from './saveChat';
 import { saveChat } from './saveChat';
 import { Types } from 'mongoose';
 import { Types } from 'mongoose';

client/src/pages/api/chat/shareChat/chat.ts

@@ -9,7 +9,7 @@ import { pushChatBill, updateShareChatBill } from '@/service/events/pushBill';
 import { resStreamResponse } from '@/service/utils/chat';
 import { resStreamResponse } from '@/service/utils/chat';
 import { ChatRoleEnum } from '@/constants/chat';
 import { ChatRoleEnum } from '@/constants/chat';
 import { BillTypeEnum } from '@/constants/user';
 import { BillTypeEnum } from '@/constants/user';
-import { sensitiveCheck } from '@/service/api/text';
+import { sensitiveCheck } from '../../openapi/text/sensitiveCheck';
 import { appKbSearch } from '../../openapi/kb/appKbSearch';
 import { appKbSearch } from '../../openapi/kb/appKbSearch';
 
 
 /* 发送提示词 */
 /* 发送提示词 */

client/src/pages/api/openapi/chat/chat.ts

@@ -9,7 +9,7 @@ import { pushChatBill } from '@/service/events/pushBill';
 import { ChatRoleEnum } from '@/constants/chat';
 import { ChatRoleEnum } from '@/constants/chat';
 import { withNextCors } from '@/service/utils/tools';
 import { withNextCors } from '@/service/utils/tools';
 import { BillTypeEnum } from '@/constants/user';
 import { BillTypeEnum } from '@/constants/user';
-import { sensitiveCheck } from '@/service/api/text';
+import { sensitiveCheck } from '../../openapi/text/sensitiveCheck';
 import { NEW_CHATID_HEADER } from '@/constants/chat';
 import { NEW_CHATID_HEADER } from '@/constants/chat';
 import { Types } from 'mongoose';
 import { Types } from 'mongoose';
 import { appKbSearch } from '../kb/appKbSearch';
 import { appKbSearch } from '../kb/appKbSearch';

client/src/pages/api/openapi/text/sensitiveCheck.ts

@@ -2,19 +2,35 @@
 import type { NextApiRequest, NextApiResponse } from 'next';
 import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { jsonRes } from '@/service/response';
 import { authUser, getSystemOpenAiKey } from '@/service/utils/auth';
 import { authUser, getSystemOpenAiKey } from '@/service/utils/auth';
-import type { TextPluginRequestParams } from '@/types/plugin';
 import axios from 'axios';
 import axios from 'axios';
 import { axiosConfig } from '@/service/utils/tools';
 import { axiosConfig } from '@/service/utils/tools';
 
 
+export type Props = {
+  input: string;
+};
+
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   try {
   try {
-    if (!global.systemEnv.sensitiveCheck) {
-      return jsonRes(res);
-    }
-
     await authUser({ req });
     await authUser({ req });
 
 
-    const { input } = req.body as TextPluginRequestParams;
+    const result = await sensitiveCheck(req.body);
+
+    jsonRes(res, {
+      data: result,
+      message: result
+    });
+  } catch (err) {
+    jsonRes(res, {
+      code: 500,
+      error: err
+    });
+  }
+}
+
+export async function sensitiveCheck({ input }: Props) {
+  if (!global.systemEnv.sensitiveCheck) {
+    return Promise.resolve('');
+  }
 
 
   const response = await axios({
   const response = await axios({
     ...axiosConfig(getSystemOpenAiKey('chat')),
     ...axiosConfig(getSystemOpenAiKey('chat')),
@@ -31,18 +47,9 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
 
 
   for (const val of values) {
   for (const val of values) {
     if (val > 0.2) {
     if (val > 0.2) {
-        return jsonRes(res, {
-          code: 500,
-          message: '您的内容不合规'
-        });
+      return Promise.reject('您的内容不合规');
     }
     }
   }
   }
 
 
-    jsonRes(res);
-  } catch (err) {
-    jsonRes(res, {
-      code: 500,
-      error: err
-    });
-  }
+  return '';
 }
 }

client/src/service/api/text.ts

-import { POST } from './request';
-import type { TextPluginRequestParams } from '@/types/plugin';
-
-export const sensitiveCheck = (data: TextPluginRequestParams) =>
-  POST('/openapi/text/sensitiveCheck', data);

client/src/types/plugin.d.ts

@@ -19,7 +19,3 @@ export type KbTestItemType = {
   time: Date;
   time: Date;
   results: (KbDataItemType & { score: number })[];
   results: (KbDataItemType & { score: number })[];
 };
 };
-
-export type TextPluginRequestParams = {
-  input: string;
-};