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

2 files changed, 35 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()}
         >
diff --git a/packages/trpc/routers/admin.ts b/packages/trpc/routers/admin.ts
index 14cb4ac9..b3fb2383 100644
--- a/packages/trpc/routers/admin.ts
+++ b/packages/trpc/routers/admin.ts
@@ -132,4 +132,15 @@ export const adminAppRouter = router({
 
     await Promise.all(bookmarkIds.map((b) => triggerSearchReindex(b.id)));
   }),
+  reRunInferenceOnAllBookmarks: adminProcedure.mutation(async ({ ctx }) => {
+    const bookmarkIds = await ctx.db.query.bookmarks.findMany({
+      columns: {
+        id: true,
+      },
+    });
+
+    await Promise.all(
+      bookmarkIds.map((b) => OpenAIQueue.enqueue({ bookmarkId: b.id })),
+    );
+  }),
 });