enable auto-teardown of instances with lab-sdk

Author: deep1401

src/renderer/components/Experiment/Tasks/EditTaskModal.tsx

@@ -7,14 +7,15 @@ import Button from '@mui/joy/Button';
 import FormControl from '@mui/joy/FormControl';
 import FormLabel from '@mui/joy/FormLabel';
 import Input from '@mui/joy/Input';
-import Textarea from '@mui/joy/Textarea';
 import {
   FormHelperText,
   ModalClose,
   ModalDialog,
   Sheet,
   Stack,
   Typography,
+  Checkbox,
+  Alert,
 } from '@mui/joy';
 import { FolderIcon } from 'lucide-react';
 import { Editor } from '@monaco-editor/react';
@@ -57,6 +58,7 @@ export default function EditTaskModal({
   const [accelerators, setAccelerators] = React.useState('');
   const [numNodes, setNumNodes] = React.useState('');
   const [setup, setSetup] = React.useState('');
+  const [shutdownAfterCompletion, setShutdownAfterCompletion] = React.useState(false);
   const [saving, setSaving] = React.useState(false);
 
   const setupEditorRef = useRef<any>(null);
@@ -74,6 +76,7 @@ export default function EditTaskModal({
     setAccelerators(cfg.accelerators != null ? String(cfg.accelerators) : '');
     setNumNodes(cfg.num_nodes != null ? String(cfg.num_nodes) : '');
     setSetup(cfg.setup != null ? String(cfg.setup) : '');
+    setShutdownAfterCompletion(cfg.shutdown_after_completion || false);
   }, [task]);
 
   // Keep Monaco editors in sync if the state changes after mount
@@ -156,6 +159,7 @@ export default function EditTaskModal({
       accelerators: accelerators || undefined,
       num_nodes: numNodes ? parseInt(numNodes, 10) : undefined,
       setup: setupValue || undefined,
+      shutdown_after_completion: shutdownAfterCompletion,
     } as any;
 
     const body = {
@@ -363,6 +367,26 @@ export default function EditTaskModal({
               </FormHelperText>
             </FormControl>
 
+            <FormControl sx={{ mt: 2 }}>
+              <Checkbox
+                checked={shutdownAfterCompletion}
+                onChange={(e) => setShutdownAfterCompletion(e.target.checked)}
+                label="Shutdown machine after execution"
+              />
+              <FormHelperText>
+                When enabled, the remote instance will be automatically shut
+                down after the task completes.
+              </FormHelperText>
+            </FormControl>
+
+            {shutdownAfterCompletion && (
+              <Alert color="warning" sx={{ mt: 1 }}>
+                <strong>Warning:</strong> This will destroy all data on the
+                remote machine that is not logged or saved elsewhere. Make sure
+                to save any important results before the task completes.
+              </Alert>
+            )}
+
             {/* Show uploaded directory indicator if present */}
             {task &&
               (() => {

src/renderer/components/Experiment/Tasks/Tasks.tsx

@@ -270,7 +270,10 @@ export default function Tasks() {
     if (cfg.uploaded_dir_path)
       formData.append('uploaded_dir_path', String(cfg.uploaded_dir_path));
     if (cfg.shutdown_after_completion !== undefined)
-      formData.append('shutdown_after_completion', String(cfg.shutdown_after_completion));
+      formData.append(
+        'shutdown_after_completion',
+        String(cfg.shutdown_after_completion),
+      );
 
     return formData;
   };