Jayden/AFFiNE-Mirror
0
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-08-10 05:29:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: adapt new fal response (#7042)

Browse Source
This commit is contained in:
darkskygit
2024-05-23 10:43:12 +00:00
parent f511b02bf9
commit 535254fdf6
1 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/backend/server/src/plugins/copilot/providers/fal.ts
+14 -2
View File
@@ -13,9 +13,16 @@ export type FalConfig = {
apiKey: string; apiKey: string;
}; };
export type FalImage = {
url: string;
seed: number;
file_name: string;
};
export type FalResponse = { export type FalResponse = {
detail: Array<{ msg: string }> | string; detail: Array<{ msg: string }> | string;
images: Array<{ url: string }>; images?: Array<FalImage>;
image?: FalImage;
}; };
type FalPrompt = { type FalPrompt = {
@@ -122,12 +129,17 @@ export class FalProvider
signal: options.signal, signal: options.signal,
}).then(res => res.json())) as FalResponse; }).then(res => res.json())) as FalResponse;
if (!data.images?.length) { if (!data.images?.length && !data.image?.url) {
const error = this.extractError(data); const error = this.extractError(data);
throw new Error( throw new Error(
error ? `Failed to generate image: ${error}` : 'No images generated' error ? `Failed to generate image: ${error}` : 'No images generated'
); );
} }
if (data.image?.url) {
return [data.image.url];
}
return data.images?.map(image => image.url) || []; return data.images?.map(image => image.url) || [];
} }