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

fix(server): better copilot error handle (#10509)

Browse Source
This commit is contained in:
darkskygit
2025-02-28 11:01:58 +00:00
parent 7227b7f8f6
commit 6b76037e39
2 changed files with 81 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

66
packages/backend/server/src/plugins/copilot/providers/perplexity.ts
View File

@@ -21,15 +21,24 @@ export type PerplexityConfig = {
endpoint?: string;
};
const PerplexityErrorSchema = z.object({
detail: z.array(
z.object({
loc: z.array(z.string()),
msg: z.string(),
const PerplexityErrorSchema = z.union([
z.object({
detail: z.array(
z.object({
loc: z.array(z.string()),
msg: z.string(),
type: z.string(),
})
),
}),
z.object({
error: z.object({
message: z.string(),
type: z.string(),
})
),
});
code: z.number(),
}),
}),
]);
const PerplexityDataSchema = z.object({
citations: z.array(z.string()),
@@ -50,6 +59,8 @@ const PerplexityDataSchema = z.object({
const PerplexitySchema = z.union([PerplexityDataSchema, PerplexityErrorSchema]);
type PerplexityError = z.infer<typeof PerplexityErrorSchema>;
export class CitationParser {
private readonly SQUARE_BRACKET_OPEN = '[';
@@ -214,12 +225,8 @@ export class PerplexityProvider implements CopilotTextToTextProvider {
params
);
const data = PerplexitySchema.parse(await response.json());
if ('detail' in data) {
throw new CopilotProviderSideError({
provider: this.type,
kind: 'unexpected_response',
message: data.detail[0].msg || 'Unexpected perplexity response',
});
if ('detail' in data || 'error' in data) {
throw this.convertError(data);
} else {
const citationParser = new CitationParser();
const { content } = data.choices[0].message;
@@ -264,9 +271,9 @@ export class PerplexityProvider implements CopilotTextToTextProvider {
this.config.endpoint || 'https://api.perplexity.ai/chat/completions',
params
);
if (response.body) {
const errorHandler = this.convertError;
if (response.ok && response.body) {
const citationParser = new CitationParser();
const provider = this.type;
const eventStream = response.body
.pipeThrough(new TextDecoderStream())
.pipeThrough(new EventSourceParserStream())
@@ -280,13 +287,8 @@ export class PerplexityProvider implements CopilotTextToTextProvider {
const json = JSON.parse(chunk.data);
if (json) {
const data = PerplexitySchema.parse(json);
if ('detail' in data) {
throw new CopilotProviderSideError({
provider,
kind: 'unexpected_response',
message:
data.detail[0].msg || 'Unexpected perplexity response',
});
if ('detail' in data || 'error' in data) {
throw errorHandler(data);
}
const { content } = data.choices[0].delta;
const { citations } = data;
@@ -331,6 +333,24 @@ export class PerplexityProvider implements CopilotTextToTextProvider {
}
}
private convertError(e: PerplexityError) {
function getErrMessage(e: PerplexityError) {
let err = 'Unexpected perplexity response';
if ('detail' in e) {
err = e.detail[0].msg || err;
} else if ('error' in e) {
err = e.error.message || err;
}
return err;
}
throw new CopilotProviderSideError({
provider: this.type,
kind: 'unexpected_response',
message: getErrMessage(e),
});
}
private handleError(e: any) {
if (e instanceof CopilotProviderSideError) {
return e;