Jayden/AFFiNE-Mirror
1
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-02-14 13:25:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add retry support for copilot (#6947)

Browse Source
This commit is contained in:
darkskygit
2024-05-15 11:02:37 +00:00
parent 7e7a4120aa
commit 0076359d6a
5 changed files with 161 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
packages/backend/server/src/plugins/copilot/controller.ts
View File

@@ -82,14 +82,21 @@ export class CopilotController {
private async appendSessionMessage(
sessionId: string,
messageId: string
messageId?: string
): Promise<ChatSession> {
const session = await this.chatSession.get(sessionId);
if (!session) {
throw new BadRequestException('Session not found');
}
await session.pushByMessageId(messageId);
if (messageId) {
await session.pushByMessageId(messageId);
} else {
// revert the latest message generated by the assistant
// if messageId is not provided, then we can retry the action
await this.chatSession.revertLatestMessage(sessionId);
session.revertLatestMessage();
}
return session;
}
@@ -129,7 +136,6 @@ export class CopilotController {
@CurrentUser() user: CurrentUser,
@Req() req: Request,
@Param('sessionId') sessionId: string,
@Query('messageId') messageId: string,
@Query() params: Record<string, string | string[]>
): Promise<string> {
const { model } = await this.checkRequest(user.id, sessionId);
@@ -141,6 +147,9 @@ export class CopilotController {
throw new InternalServerErrorException('No provider available');
}
const messageId = Array.isArray(params.messageId)
? params.messageId[0]
: params.messageId;
const session = await this.appendSessionMessage(sessionId, messageId);
try {
@@ -174,7 +183,6 @@ export class CopilotController {
@CurrentUser() user: CurrentUser,
@Req() req: Request,
@Param('sessionId') sessionId: string,
@Query('messageId') messageId: string,
@Query() params: Record<string, string>
): Promise<Observable<ChatEvent>> {
try {
@@ -187,6 +195,9 @@ export class CopilotController {
throw new InternalServerErrorException('No provider available');
}
const messageId = Array.isArray(params.messageId)
? params.messageId[0]
: params.messageId;
const session = await this.appendSessionMessage(sessionId, messageId);
delete params.messageId;
@@ -237,10 +248,12 @@ export class CopilotController {
@CurrentUser() user: CurrentUser,
@Req() req: Request,
@Param('sessionId') sessionId: string,
@Query('messageId') messageId: string,
@Query() params: Record<string, string>
): Promise<Observable<ChatEvent>> {
try {
const messageId = Array.isArray(params.messageId)
? params.messageId[0]
: params.messageId;
const { model, hasAttachment } = await this.checkRequest(
user.id,
sessionId,