diff --git a/src/templates/n8nTemplate.ts b/src/templates/n8nTemplate.ts index f0fb2e8..0ad93af 100644 --- a/src/templates/n8nTemplate.ts +++ b/src/templates/n8nTemplate.ts @@ -17,6 +17,16 @@ export abstract class N8nImageGenerateTemplate extends ImageGenerateTemplate { "image_url": imageUrl } } + }).then(res => res.data).then(res => { + if (res.status) { + const data = res.data?.data; + if(!data) throw new Error(`结果有误`) + if (Array.isArray(data)) { + return data; + } + return data; + } + throw new Error(res.msg) }) } } @@ -48,6 +58,16 @@ export abstract class N8nVideoGenerateTemplate extends VideoGenerateTemplate { "aspect_ratio": this.aspectRatio } } + }).then(res => res.data).then(res => { + if (res.status) { + const data = res.data?.data; + if(!data) throw new Error(`结果有误`) + if (Array.isArray(data)) { + return data; + } + return data; + } + throw new Error(res.msg) }) } }