ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-01 03:31:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix upload progress for datasets in the ui

Browse Source
This commit is contained in:
Jaret Burkett
2025-07-26 09:07:30 -06:00
parent 523c159579
commit 3e14a674ac
1 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
ui/src/app/api/datasets/upload/route.ts
View File

@@ -22,19 +22,21 @@ export async function POST(request: NextRequest) {
const uploadDir = join(datasetsPath, datasetName); const uploadDir = join(datasetsPath, datasetName);
await mkdir(uploadDir, { recursive: true }); await mkdir(uploadDir, { recursive: true });
const savedFiles = await Promise.all( const savedFiles: string[] = [];
files.map(async (file: any) => {
const bytes = await file.arrayBuffer(); // Process files sequentially to avoid overwhelming the system
const buffer = Buffer.from(bytes); for (let i = 0; i < files.length; i++) {
const file = files[i] as any;
const bytes = await file.arrayBuffer();
const buffer = Buffer.from(bytes);
// Clean filename and ensure it's unique // Clean filename and ensure it's unique
const fileName = file.name.replace(/[^a-zA-Z0-9.-]/g, '_'); const fileName = file.name.replace(/[^a-zA-Z0-9.-]/g, '_');
const filePath = join(uploadDir, fileName); const filePath = join(uploadDir, fileName);
await writeFile(filePath, buffer); await writeFile(filePath, buffer);
return fileName; savedFiles.push(fileName);
}), }
);
return NextResponse.json({ return NextResponse.json({
message: 'Files uploaded successfully', message: 'Files uploaded successfully',