feat: move dataset (#277)

client/src/pages/api/plugins/kb/list.ts

@@ -4,10 +4,11 @@ import { connectToDatabase, KB } from '@/service/mongo';
 import { authUser } from '@/service/utils/auth';
 import { getVectorModel } from '@/service/utils/data';
 import { KbListItemType } from '@/types/plugin';
+import { KbTypeEnum } from '@/constants/kb';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { parentId } = req.query as { parentId: string };
+    const { parentId, type } = req.query as { parentId?: string; type?: `${KbTypeEnum}` };
     // 凭证校验
     const { userId } = await authUser({ req, authToken: true });
 
@@ -15,7 +16,8 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
 
     const kbList = await KB.find({
       userId,
-      parentId: parentId || null
+      ...(parentId !== undefined && { parentId: parentId || null }),
+      ...(type && { type })
     }).sort({ updateTime: -1 });
 
     const data = await Promise.all(

client/src/pages/api/plugins/kb/update.ts

@@ -6,9 +6,9 @@ import type { KbUpdateParams } from '@/api/request/kb';
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse<any>) {
   try {
-    const { id, name, avatar, tags = '' } = req.body as KbUpdateParams;
+    const { id, parentId, name, avatar, tags } = req.body as KbUpdateParams;
 
-    if (!id || !name) {
+    if (!id) {
       throw new Error('缺少参数');
     }
 
@@ -23,9 +23,12 @@ export default async function handler(req: NextApiRequest, res: NextApiResponse<
         userId
       },
       {
+        ...(parentId !== undefined && { parentId: parentId || null }),
         ...(name && { name }),
         ...(avatar && { avatar }),
-        tags: tags.split(' ').filter((item) => item)
+        ...(typeof tags === 'string' && {
+          tags: tags.split(' ').filter((item) => item)
+        })
       }
     );