feature(web): Add ability to rerun inference on all bookmarks. Fixes 443

1 files changed, 24 insertions, 0 deletions

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>
@@ -68,6 +85,13 @@ export default function AdminActions() {
         </ActionButton>
         <ActionButton
           variant="destructive"
+          loading={isInferencePending}
+          onClick={() => reRunInferenceOnAllBookmarks()}
+        >
+          Regenerate AI Tags for All Bookmarks
+        </ActionButton>
+        <ActionButton
+          variant="destructive"
           loading={isReindexPending}
           onClick={() => reindexBookmarks()}
         >