From 7fac3c7f3879ee49c4ea8bd7086cb0a68a60c2e3 Mon Sep 17 00:00:00 2001
From: Oliver Walters <oliver.henry.walters@gmail.com>
Date: Sat, 23 May 2026 05:51:15 +0000
Subject: [PATCH] Handle case where task returns immediately

---
 .../src/tables/build/BuildOutputTable.tsx     | 24 ++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

diff --git a/src/frontend/src/tables/build/BuildOutputTable.tsx b/src/frontend/src/tables/build/BuildOutputTable.tsx
index a0676aa8af..6367564acb 100644
--- a/src/frontend/src/tables/build/BuildOutputTable.tsx
+++ b/src/frontend/src/tables/build/BuildOutputTable.tsx
@@ -385,7 +385,13 @@ export default function BuildOutputTable({
     outputs: selectedOutputs,
     hasTrackedItems: hasTrackedItems,
     onFormSuccess: (response: any) => {
-      setCompleteTaskId(response.task_id);
+      if (response.task_id) {
+        setCompleteTaskId(response.task_id);
+      } else {
+        // If no task ID is returned, immediately refresh the table and build data
+        table.refreshTable(true);
+        refreshBuild();
+      }
     }
   });
 
@@ -393,7 +399,13 @@ export default function BuildOutputTable({
     build: build,
     outputs: selectedOutputs,
     onFormSuccess: (response: any) => {
-      setScrapTaskId(response.task_id);
+      if (response.task_id) {
+        setScrapTaskId(response.task_id);
+      } else {
+        // If no task ID is returned, immediately refresh the table and build data
+        table.refreshTable(true);
+        refreshBuild();
+      }
     }
   });
 
@@ -401,7 +413,13 @@ export default function BuildOutputTable({
     build: build,
     outputs: selectedOutputs,
     onFormSuccess: (response: any) => {
-      setDeleteTaskId(response.task_id);
+      if (response.task_id) {
+        setDeleteTaskId(response.task_id);
+      } else {
+        // If no task ID is returned, immediately refresh the table and build data
+        table.refreshTable(true);
+        refreshBuild();
+      }
     }
   });