fix: gql result check for clean history (#7154)

packages/backend/server/src/plugins/copilot/resolver.ts

@@ -314,7 +314,7 @@ export class CopilotResolver {
     return session;
   }
 
-  @Mutation(() => String, {
+  @Mutation(() => [String], {
     description: 'Cleanup sessions',
   })
   async cleanupCopilotSession(
@@ -336,11 +336,10 @@ export class CopilotResolver {
       return new TooManyRequestsException('Server is busy');
     }
 
-    const ret = await this.chatSession.cleanup({
+    return await this.chatSession.cleanup({
       ...options,
       userId: user.id,
     });
-    return ret;
   }
 
   @Mutation(() => String, {

packages/backend/server/src/plugins/copilot/session.ts

@@ -514,24 +514,24 @@ export class ChatSessionService {
 
       // only mark action session as deleted
       // chat session always can be reuse
-      {
-        const actionIds = (
-          await Promise.all(
-            sessions.map(({ id, promptName }) =>
-              this.prompt
-                .get(promptName)
-                .then(prompt => ({ id, action: !!prompt?.action }))
-            )
+      const actionIds = (
+        await Promise.all(
+          sessions.map(({ id, promptName }) =>
+            this.prompt
+              .get(promptName)
+              .then(prompt => ({ id, action: !!prompt?.action }))
           )
         )
-          .filter(({ action }) => action)
-          .map(({ id }) => id);
+      )
+        .filter(({ action }) => action)
+        .map(({ id }) => id);
 
-        await tx.aiSession.updateMany({
-          where: { id: { in: actionIds } },
-          data: { deletedAt: new Date() },
-        });
-      }
+      await tx.aiSession.updateMany({
+        where: { id: { in: actionIds } },
+        data: { deletedAt: new Date() },
+      });
+
+      return [...sessionIds, ...actionIds];
     });
   }
 

packages/backend/server/src/schema.gql

@@ -254,7 +254,7 @@ type Mutation {
   changePassword(newPassword: String!, token: String!): UserType!
 
   """Cleanup sessions"""
-  cleanupCopilotSession(options: DeleteSessionInput!): String!
+  cleanupCopilotSession(options: DeleteSessionInput!): [String!]!
 
   """Create a subscription checkout link of stripe"""
   createCheckoutSession(input: CreateCheckoutSessionInput!): String!