Jayden/AFFiNE-Mirror
0
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-07-21 03:56:23 +08: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
packages/backend/server/tests/utils/copilot.ts
+5 -4
View File
@@ -196,11 +196,12 @@ export async function chatWithText(
app: INestApplication,
userToken: string,
sessionId: string,
messageId: string,
messageId?: string,
prefix = ''
): Promise<string> {
const query = messageId ? `?messageId=${messageId}` : '';
const res = await request(app.getHttpServer())
.get(`/api/copilot/chat/${sessionId}${prefix}?messageId=${messageId}`)
.get(`/api/copilot/chat/${sessionId}${prefix}${query}`)
.auth(userToken, { type: 'bearer' })
.expect(200);
@@ -211,7 +212,7 @@ export async function chatWithTextStream(
app: INestApplication,
userToken: string,
sessionId: string,
messageId: string
messageId?: string
) {
return chatWithText(app, userToken, sessionId, messageId, '/stream');
}
@@ -220,7 +221,7 @@ export async function chatWithImages(
app: INestApplication,
userToken: string,
sessionId: string,
messageId: string
messageId?: string
) {
return chatWithText(app, userToken, sessionId, messageId, '/images');
}