feat: model related kb

src/pages/api/chat/chat.ts

@@ -54,7 +54,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     const prompts = [...content, prompt];
 
     // 使用了知识库搜索
-    if (model.chat.useKb) {
+    if (model.chat.relatedKbs.length > 0) {
       const { code, searchPrompts } = await searchKb({
         userOpenAiKey,
         prompts,

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

@@ -50,7 +50,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     const modelConstantsData = ChatModelMap[model.chat.chatModel];
 
     // 使用了知识库搜索
-    if (model.chat.useKb) {
+    if (model.chat.relatedKbs.length > 0) {
       const { code, searchPrompts } = await searchKb({
         userOpenAiKey,
         prompts,

src/pages/api/model/update.ts

@@ -9,10 +9,10 @@ import { authModel } from '@/service/utils/auth';
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { name, avatar, chat, share, security } = req.body as ModelUpdateParams;
+    const { name, avatar, chat, share } = req.body as ModelUpdateParams;
     const { modelId } = req.query as { modelId: string };
 
-    if (!name || !chat || !security || !modelId) {
+    if (!name || !chat || !modelId) {
       throw new Error('参数错误');
     }
 
@@ -38,8 +38,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
         chat,
         'share.isShare': share.isShare,
         'share.isShareDetail': share.isShareDetail,
-        'share.intro': share.intro,
-        security
+        'share.intro': share.intro
       }
     );
 

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

@@ -70,7 +70,7 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse)
     const modelConstantsData = ChatModelMap[model.chat.chatModel];
 
     // 使用了知识库搜索
-    if (model.chat.useKb) {
+    if (model.chat.relatedKbs.length > 0) {
       const similarity = ModelVectorSearchModeMap[model.chat.searchMode]?.similarity || 0.22;
 
       const { code, searchPrompts } = await searchKb({