From 5281531d6f4aab4605c407d5167dd8e44f237f0d Mon Sep 17 00:00:00 2001
From: MohamedBassem <me@mbassem.com>
Date: Sun, 29 Sep 2024 23:43:10 +0000
Subject: feature(web): Add ability to rerun inference on all bookmarks. Fixes
 443

---
 .../components/dashboard/admin/AdminActions.tsx    | 24 ++++++++++++++++++++++
 1 file changed, 24 insertions(+)

(limited to 'apps/web/components/dashboard/admin/AdminActions.tsx')

diff --git a/apps/web/components/dashboard/admin/AdminActions.tsx b/apps/web/components/dashboard/admin/AdminActions.tsx
index 783f7e76..dfdf65eb 100644
--- a/apps/web/components/dashboard/admin/AdminActions.tsx
+++ b/apps/web/components/dashboard/admin/AdminActions.tsx
@@ -35,6 +35,23 @@ export default function AdminActions() {
       },
     });
 
+  const {
+    mutate: reRunInferenceOnAllBookmarks,
+    isPending: isInferencePending,
+  } = api.admin.reRunInferenceOnAllBookmarks.useMutation({
+    onSuccess: () => {
+      toast({
+        description: "Inference jobs enqueued",
+      });
+    },
+    onError: (e) => {
+      toast({
+        variant: "destructive",
+        description: e.message,
+      });
+    },
+  });
+
   return (
     <div>
       <div className="mb-2 mt-8 text-xl font-medium">Actions</div>
@@ -66,6 +83,13 @@ export default function AdminActions() {
         >
           Recrawl All Links (Without Inference)
         </ActionButton>
+        <ActionButton
+          variant="destructive"
+          loading={isInferencePending}
+          onClick={() => reRunInferenceOnAllBookmarks()}
+        >
+          Regenerate AI Tags for All Bookmarks
+        </ActionButton>
         <ActionButton
           variant="destructive"
           loading={isReindexPending}
-- 
cgit v1.2.3-70-g09d2

