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

fix(core): copilot client should throw right ai error (#9958)

Browse Source 
[BS-2488](https://linear.app/affine-design/issue/BS-2488/copilot-client-需要正确返回错误类型)
This commit is contained in:
donteatfriedrice
2025-02-05 12:46:12 +00:00
parent 4ed03c9f0e
commit 9cae1c3a28
3 changed files with 94 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

152
packages/frontend/core/src/components/blocksuite/block-suite-editor/ai/copilot-client.ts
View File

@@ -72,57 +72,77 @@ export class CopilotClient {
async createSession(
options: OptionsField<typeof createCopilotSessionMutation>
) {
const res = await this.gql({
query: createCopilotSessionMutation,
variables: {
options,
},
});
return res.createCopilotSession;
try {
const res = await this.gql({
query: createCopilotSessionMutation,
variables: {
options,
},
});
return res.createCopilotSession;
} catch (err) {
throw resolveError(err);
}
}
async updateSession(
options: OptionsField<typeof updateCopilotSessionMutation>
) {
const res = await this.gql({
query: updateCopilotSessionMutation,
variables: {
options,
},
});
return res.updateCopilotSession;
try {
const res = await this.gql({
query: updateCopilotSessionMutation,
variables: {
options,
},
});
return res.updateCopilotSession;
} catch (err) {
throw resolveError(err);
}
}
async forkSession(options: OptionsField<typeof forkCopilotSessionMutation>) {
const res = await this.gql({
query: forkCopilotSessionMutation,
variables: {
options,
},
});
return res.forkCopilotSession;
try {
const res = await this.gql({
query: forkCopilotSessionMutation,
variables: {
options,
},
});
return res.forkCopilotSession;
} catch (err) {
throw resolveError(err);
}
}
async createMessage(
options: OptionsField<typeof createCopilotMessageMutation>
) {
const res = await this.gql({
query: createCopilotMessageMutation,
variables: {
options,
},
});
return res.createCopilotMessage;
try {
const res = await this.gql({
query: createCopilotMessageMutation,
variables: {
options,
},
});
return res.createCopilotMessage;
} catch (err) {
throw resolveError(err);
}
}
async getSessions(workspaceId: string) {
const res = await this.gql({
query: getCopilotSessionsQuery,
variables: {
workspaceId,
},
});
return res.currentUser?.copilot;
try {
const res = await this.gql({
query: getCopilotSessionsQuery,
variables: {
workspaceId,
},
});
return res.currentUser?.copilot;
} catch (err) {
throw resolveError(err);
}
}
async getHistories(
@@ -132,16 +152,20 @@ export class CopilotClient {
typeof getCopilotHistoriesQuery
>['variables']['options']
) {
const res = await this.gql({
query: getCopilotHistoriesQuery,
variables: {
workspaceId,
docId,
options,
},
});
try {
const res = await this.gql({
query: getCopilotHistoriesQuery,
variables: {
workspaceId,
docId,
options,
},
});
return res.currentUser?.copilot?.histories;
return res.currentUser?.copilot?.histories;
} catch (err) {
throw resolveError(err);
}
}
async getHistoryIds(
@@ -151,16 +175,20 @@ export class CopilotClient {
typeof getCopilotHistoriesQuery
>['variables']['options']
) {
const res = await this.gql({
query: getCopilotHistoryIdsQuery,
variables: {
workspaceId,
docId,
options,
},
});
try {
const res = await this.gql({
query: getCopilotHistoryIdsQuery,
variables: {
workspaceId,
docId,
options,
},
});
return res.currentUser?.copilot?.histories;
return res.currentUser?.copilot?.histories;
} catch (err) {
throw resolveError(err);
}
}
async cleanupSessions(input: {
@@ -168,13 +196,17 @@ export class CopilotClient {
docId: string;
sessionIds: string[];
}) {
const res = await this.gql({
query: cleanupCopilotSessionMutation,
variables: {
input,
},
});
return res.cleanupCopilotSession;
try {
const res = await this.gql({
query: cleanupCopilotSessionMutation,
variables: {
input,
},
});
return res.cleanupCopilotSession;
} catch (err) {
throw resolveError(err);
}
}
async chatText({