Use delete models endpoint

transformerlab · Jan 9, 2024 · 7edad89 · 7edad89
1 parent e575abf
commit 7edad89
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/renderer/components/ModelZoo/LocalModels.tsx b/src/renderer/components/ModelZoo/LocalModels.tsx
@@ -317,8 +317,19 @@ export default function LocalModels({
                         &nbsp;
                         <Trash2Icon
                           color="var(--joy-palette-danger-600)"
-                          onClick={() => {
-                            mutate();
+                          onClick={async () => {
+                            if (
+                              confirm(
+                                "Are you sure you want to delete model '" +
+                                  row.model_id +
+                                  "'?"
+                              )
+                            ) {
+                              await fetch(
+                                chatAPI.Endpoints.Models.Delete(row.model_id)
+                              );
+                              mutate();
+                            }
                           }}
                         />
                       </>

diff --git a/src/renderer/lib/transformerlab-api-sdk.ts b/src/renderer/lib/transformerlab-api-sdk.ts
@@ -436,6 +436,7 @@ Endpoints.Models = {
   ModelDetailsFromGallery: (modelId: string) =>
     API_URL() + 'model/gallery/' + convertSlashInUrl(modelId),
   HuggingFaceLogin: () => API_URL() + 'model/login_to_huggingface',
+  Delete: (modelId: string) => API_URL() + 'model/delete?model_id=' + modelId,
 };
 
 Endpoints.Plugins = {