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

fix: handle unexpected fal response (#7158)

Browse Source
This commit is contained in:
darkskygit
2024-06-07 05:41:25 +00:00
parent 17a5900575
commit b75da1f3e0
1 changed files with 10 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
packages/backend/server/src/plugins/copilot/providers/fal.ts
View File

@@ -26,8 +26,8 @@ const FalImageSchema = z
url: z.string(),
seed: z.number().optional(),
content_type: z.string(),
file_name: z.string(),
file_size: z.number(),
file_name: z.string().optional(),
file_size: z.number().optional(),
width: z.number(),
height: z.number(),
})
@@ -200,8 +200,10 @@ export class FalProvider
if (model.startsWith('workflows/')) {
const stream = await falStream(model, { input: prompt });
const result = FalStreamOutputSchema.parse(await stream.done());
return result.output;
const result = FalStreamOutputSchema.safeParse(await stream.done());
if (result.success) return result.data.output;
const errors = JSON.stringify(result.error.errors);
throw new Error(`Unexpected fal response: ${errors}`);
} else {
const response = await fetch(`https://fal.run/fal-ai/${model}`, {
method: 'POST',
@@ -217,7 +219,10 @@ export class FalProvider
}),
signal: options.signal,
});
return FalResponseSchema.parse(await response.json());
const result = FalResponseSchema.safeParse(await response.json());
if (result.success) return result.data;
const errors = JSON.stringify(result.error.errors);
throw new Error(`Unexpected fal response: ${errors}`);
}
}