From 83d569df83ce277bb47adc81557735e39c1fed33 Mon Sep 17 00:00:00 2001
From: archer <545436317@qq.com>
Date: Wed, 2 Aug 2023 14:56:48 +0800
Subject: [PATCH] perf: auth event model

---
 client/src/pages/api/openapi/kb/pushData.ts | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/client/src/pages/api/openapi/kb/pushData.ts b/client/src/pages/api/openapi/kb/pushData.ts
index bd8729105..90835a3bc 100644
--- a/client/src/pages/api/openapi/kb/pushData.ts
+++ b/client/src/pages/api/openapi/kb/pushData.ts
@@ -35,6 +35,18 @@ export default withNextCors(async function handler(req: NextApiRequest, res: Nex
     if (!kbId || !Array.isArray(data) || !model) {
       throw new Error('缺少参数');
     }
+
+    // auth model
+    if (mode === TrainingModeEnum.qa && !global.qaModels.find((item) => item.model === model)) {
+      throw new Error('不支持的 QA 拆分模型');
+    }
+    if (
+      mode === TrainingModeEnum.index &&
+      !global.vectorModels.find((item) => item.model === model)
+    ) {
+      throw new Error('不支持的向量生成模型');
+    }
+
     await connectToDatabase();
 
     // 凭证校验