diff options
Diffstat (limited to 'packages')
| -rw-r--r-- | packages/shared/queues.ts | 1 | ||||
| -rw-r--r-- | packages/trpc/routers/admin.ts | 2 |
2 files changed, 3 insertions, 0 deletions
diff --git a/packages/shared/queues.ts b/packages/shared/queues.ts index 6d5fdd5f..6ea89f5e 100644 --- a/packages/shared/queues.ts +++ b/packages/shared/queues.ts @@ -12,6 +12,7 @@ export const queueConnectionDetails = { // Link Crawler export const zCrawlLinkRequestSchema = z.object({ bookmarkId: z.string(), + runInference: z.boolean().optional(), }); export type ZCrawlLinkRequest = z.infer<typeof zCrawlLinkRequestSchema>; diff --git a/packages/trpc/routers/admin.ts b/packages/trpc/routers/admin.ts index 8792f7ed..0a0af173 100644 --- a/packages/trpc/routers/admin.ts +++ b/packages/trpc/routers/admin.ts @@ -100,6 +100,7 @@ export const adminAppRouter = router({ .input( z.object({ crawlStatus: z.enum(["success", "failure", "all"]), + runInference: z.boolean(), }), ) .mutation(async ({ ctx, input }) => { @@ -116,6 +117,7 @@ export const adminAppRouter = router({ bookmarkIds.map((b) => LinkCrawlerQueue.add("crawl", { bookmarkId: b.id, + runInference: input.runInference, }), ), ); |