feat: 删除模型数据

src/pages/api/model/data/delModelDataById.ts

@@ -2,6 +2,8 @@ import type { NextApiRequest, NextApiResponse } from 'next';
 import { jsonRes } from '@/service/response';
 import { connectToDatabase, ModelData } from '@/service/mongo';
 import { authToken } from '@/service/utils/tools';
+import { connectRedis } from '@/service/redis';
+import { VecModelDataIndex } from '@/constants/redis';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
@@ -22,14 +24,27 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     const userId = await authToken(authorization);
 
     await connectToDatabase();
+    const redis = await connectRedis();
+
+    const data = await ModelData.findById(dataId);
 
     await ModelData.deleteOne({
       _id: dataId,
       userId
     });
 
+    // 删除 redis 数据
+    data?.q.forEach(async (item) => {
+      try {
+        await redis.json.del(`${VecModelDataIndex}:${item.id}`);
+      } catch (error) {
+        console.log(error);
+      }
+    });
+
     jsonRes(res);
   } catch (err) {
+    console.log(err);
     jsonRes(res, {
       code: 500,
       error: err

src/pages/api/model/del.ts

@@ -6,6 +6,8 @@ import { TrainingStatusEnum } from '@/constants/model';
 import { getOpenAIApi } from '@/service/utils/chat';
 import { TrainingItemType } from '@/types/training';
 import { httpsAgent } from '@/service/utils/tools';
+import { connectRedis } from '@/service/redis';
+import { VecModelDataIndex } from '@/constants/redis';
 
 /* 获取我的模型 */
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
@@ -25,6 +27,22 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
     const userId = await authToken(authorization);
 
     await connectToDatabase();
+    const redis = await connectRedis();
+
+    const modelDataList = await ModelData.find({
+      modelId
+    });
+
+    // 删除 redis
+    modelDataList?.forEach((modelData) =>
+      modelData.q.forEach(async (item) => {
+        try {
+          await redis.json.del(`${VecModelDataIndex}:${item.id}`);
+        } catch (error) {
+          console.log(error);
+        }
+      })
+    );
 
     let requestQueue: any[] = [];
     // 删除对应的聊天
@@ -73,7 +91,8 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
         userId
       })
     );
-    await requestQueue;
+
+    await Promise.all(requestQueue);
 
     jsonRes(res);
   } catch (err) {