luulam/AFFiNE
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
darkskygit
2024-06-05 07:49:40 +00:00
parent ecff441516
commit a5f9bfdda9
3 changed files with 18 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
packages/backend/server/src/plugins/copilot/resolver.ts
View File

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

4
packages/backend/server/src/plugins/copilot/session.ts
View File

@@ -514,7 +514,6 @@ export class ChatSessionService {
// only mark action session as deleted // only mark action session as deleted
// chat session always can be reuse // chat session always can be reuse
{
const actionIds = ( const actionIds = (
await Promise.all( await Promise.all(
sessions.map(({ id, promptName }) => sessions.map(({ id, promptName }) =>
@@ -531,7 +530,8 @@ export class ChatSessionService {
where: { id: { in: actionIds } }, where: { id: { in: actionIds } },
data: { deletedAt: new Date() }, data: { deletedAt: new Date() },
}); });
}
return [...sessionIds, ...actionIds];
}); });
} }

2
packages/backend/server/src/schema.gql
View File

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