Jayden/AFFiNE-Mirror
0
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-08-06 11:59:51 +08: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
packages/backend/server/src/plugins/copilot/providers/fal.ts
+10 -5
View File
@@ -26,8 +26,8 @@ const FalImageSchema = z
url: z.string(), url: z.string(),
seed: z.number().optional(), seed: z.number().optional(),
content_type: z.string(), content_type: z.string(),
file_name: z.string(), file_name: z.string().optional(),
file_size: z.number(), file_size: z.number().optional(),
width: z.number(), width: z.number(),
height: z.number(), height: z.number(),
}) })
@@ -200,8 +200,10 @@ export class FalProvider
if (model.startsWith('workflows/')) { if (model.startsWith('workflows/')) {
const stream = await falStream(model, { input: prompt }); const stream = await falStream(model, { input: prompt });
const result = FalStreamOutputSchema.parse(await stream.done()); const result = FalStreamOutputSchema.safeParse(await stream.done());
return result.output; if (result.success) return result.data.output;
const errors = JSON.stringify(result.error.errors);
throw new Error(`Unexpected fal response: ${errors}`);
} else { } else {
const response = await fetch(`https://fal.run/fal-ai/${model}`, { const response = await fetch(`https://fal.run/fal-ai/${model}`, {
method: 'POST', method: 'POST',
@@ -217,7 +219,10 @@ export class FalProvider
}), }),
signal: options.signal, 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}`);
} }
} }