feat: add rebuild machine option

2024-01-12 18:56:02 +08:00
parent 2ff19f53d0
commit b8e478c8ee
2 changed files with 51 additions and 10 deletions
@@ -32,6 +32,7 @@ import {
 import {
  addCustomMachine,
  addMachine,
+  buildMachine,
  deleteMachine,
  disableMachine,
  enableMachine,
@@ -202,14 +203,28 @@ export const columns: ColumnDef<Machine>[] = [
              </DropdownMenuItem>
            )}
            {machine.type === "comfy-deploy-serverless" && (
-              <DropdownMenuItem asChild>
-                <a
-                  target="_blank"
-                  href={machine.endpoint.replace("comfyui-api", "comfyui-app")}
+              <>
+                <DropdownMenuItem asChild>
+                  <a
+                    target="_blank"
+                    href={machine.endpoint.replace(
+                      "comfyui-api",
+                      "comfyui-app"
+                    )}
+                  >
+                    Open ComfyUI
+                  </a>
+                </DropdownMenuItem>
+                <DropdownMenuItem
+                  onClick={() => {
+                    buildMachine({
+                      id: machine.id,
+                    });
+                  }}
                >
-                  Open ComfyUI
-                </a>
-              </DropdownMenuItem>
+                  Rebuild
+                </DropdownMenuItem>
+              </>
            )}
            <DropdownMenuItem onClick={() => setOpen(true)}>
              Edit