From 3e55efe0a5abd69afb0ba1acf8bd17d282bfbdcb Mon Sep 17 00:00:00 2001 From: BennyKok Date: Thu, 21 Dec 2023 16:21:06 +0800 Subject: [PATCH] fix(api): return files download url for "files" output --- web/src/app/api/run/route.ts | 21 ++++++++++++++------- 1 file changed, 14 insertions(+), 7 deletions(-) diff --git a/web/src/app/api/run/route.ts b/web/src/app/api/run/route.ts index 6856ab6..f15c31d 100644 --- a/web/src/app/api/run/route.ts +++ b/web/src/app/api/run/route.ts @@ -48,13 +48,20 @@ export async function GET(request: Request) { for (let i = 0; i < run.outputs.length; i++) { const output = run.outputs[i]; - if (output.data?.images === undefined) continue; - - for (let j = 0; j < output.data?.images.length; j++) { - const element = output.data?.images[j]; - element.url = replaceCDNUrl( - `${process.env.SPACES_ENDPOINT}/${process.env.SPACES_BUCKET}/outputs/runs/${run.id}/${element.filename}` - ); + if (output.data?.images !== undefined) { + for (let j = 0; j < output.data?.images.length; j++) { + const element = output.data?.images[j]; + element.url = replaceCDNUrl( + `${process.env.SPACES_ENDPOINT}/${process.env.SPACES_BUCKET}/outputs/runs/${run.id}/${element.filename}` + ); + } + } else if (output.data?.files !== undefined) { + for (let j = 0; j < output.data?.files.length; j++) { + const element = output.data?.files[j]; + element.url = replaceCDNUrl( + `${process.env.SPACES_ENDPOINT}/${process.env.SPACES_BUCKET}/outputs/runs/${run.id}/${element.filename}` + ); + } } } }