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

---
 packages/trpc/routers/admin.ts | 11 +++++++++++
 1 file changed, 11 insertions(+)

(limited to 'packages')

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 })),
+    );
+  }),
 });
-- 
cgit v1.2.3-70-g09d2