From e587ecc485594a2cb45d41941cd63ef6e5d5e6e0 Mon Sep 17 00:00:00 2001
From: Ettore Di Giacinto <mudler@localai.io>
Date: Tue, 31 Mar 2026 17:20:53 +0000
Subject: [PATCH] chore(ui): allow to unload forcefully

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
---
 core/http/react-ui/src/pages/Nodes.jsx | 34 ++++++++++++++++++++++----
 1 file changed, 29 insertions(+), 5 deletions(-)

diff --git a/core/http/react-ui/src/pages/Nodes.jsx b/core/http/react-ui/src/pages/Nodes.jsx
index 1ecd5cf57..9d3c44872 100644
--- a/core/http/react-ui/src/pages/Nodes.jsx
+++ b/core/http/react-ui/src/pages/Nodes.jsx
@@ -261,6 +261,7 @@ export default function Nodes() {
   const [nodeModels, setNodeModels] = useState({})
   const [nodeBackends, setNodeBackends] = useState({})
   const [confirmDelete, setConfirmDelete] = useState(null)
+  const [confirmUnload, setConfirmUnload] = useState(null)
   const [showTips, setShowTips] = useState(false)
   const [activeTab, setActiveTab] = useState('backend') // 'backend', 'agent', or 'scheduling'
   const [schedulingConfigs, setSchedulingConfigs] = useState([])
@@ -819,13 +820,15 @@ export default function Nodes() {
                                         <td style={{ textAlign: 'right' }}>
                                           <button
                                             className="btn btn-danger btn-sm"
-                                            disabled={m.in_flight > 0}
-                                            title={m.in_flight > 0 ? 'Cannot unload while serving requests' : 'Unload model'}
+                                            title={m.in_flight > 0 ? 'Unload model (has in-flight requests)' : 'Unload model'}
                                             onClick={(e) => {
                                               e.stopPropagation()
-                                              if (confirm(`Unload "${m.model_name}" from ${node.name}?`)) {
-                                                handleUnloadModel(node.id, m.model_name)
-                                              }
+                                              setConfirmUnload({
+                                                nodeId: node.id,
+                                                nodeName: node.name,
+                                                modelName: m.model_name,
+                                                inFlight: m.in_flight ?? 0,
+                                              })
                                             }}
                                           >
                                             <i className="fas fa-stop" />
@@ -1057,6 +1060,27 @@ export default function Nodes() {
         onConfirm={() => confirmDelete && handleDelete(confirmDelete.id)}
         onCancel={() => setConfirmDelete(null)}
       />
+
+      <ConfirmDialog
+        open={!!confirmUnload}
+        title="Unload Model"
+        message={
+          confirmUnload
+            ? confirmUnload.inFlight > 0
+              ? `"${confirmUnload.modelName}" on ${confirmUnload.nodeName} currently has ${confirmUnload.inFlight} in-flight request(s). Unloading will interrupt them. Continue?`
+              : `Unload "${confirmUnload.modelName}" from ${confirmUnload.nodeName}?`
+            : ''
+        }
+        confirmLabel="Unload"
+        danger={confirmUnload?.inFlight > 0}
+        onConfirm={() => {
+          if (confirmUnload) {
+            handleUnloadModel(confirmUnload.nodeId, confirmUnload.modelName)
+          }
+          setConfirmUnload(null)
+        }}
+        onCancel={() => setConfirmUnload(null)}
+      />
     </div>
   )
 }