Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-18 06:47:02 +08:00
Code Issues 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, {

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

@@ -514,24 +514,24 @@ 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 }) => this.prompt
this.prompt .get(promptName)
.get(promptName) .then(prompt => ({ id, action: !!prompt?.action }))
.then(prompt => ({ id, action: !!prompt?.action }))
)
) )
) )
.filter(({ action }) => action) )
.map(({ id }) => id); .filter(({ action }) => action)
.map(({ id }) => id);
await tx.aiSession.updateMany({ await tx.aiSession.updateMany({
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!