From d24e50950a36df12b7149b66762a231ac1da14d2 Mon Sep 17 00:00:00 2001 From: MohamedBassem Date: Wed, 27 Mar 2024 11:46:29 +0000 Subject: feature: Add support for local models using ollama --- apps/workers/searchWorker.ts | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) (limited to 'apps/workers/searchWorker.ts') diff --git a/apps/workers/searchWorker.ts b/apps/workers/searchWorker.ts index 618e7c89..b24777d7 100644 --- a/apps/workers/searchWorker.ts +++ b/apps/workers/searchWorker.ts @@ -1,16 +1,17 @@ +import type { Job } from "bullmq"; +import { Worker } from "bullmq"; +import { eq } from "drizzle-orm"; + +import type { ZSearchIndexingRequest } from "@hoarder/shared/queues"; import { db } from "@hoarder/db"; +import { bookmarks } from "@hoarder/db/schema"; import logger from "@hoarder/shared/logger"; -import { getSearchIdxClient } from "@hoarder/shared/search"; import { - SearchIndexingQueue, - ZSearchIndexingRequest, queueConnectionDetails, + SearchIndexingQueue, zSearchIndexingRequestSchema, } from "@hoarder/shared/queues"; -import { Job } from "bullmq"; -import { Worker } from "bullmq"; -import { bookmarks } from "@hoarder/db/schema"; -import { eq } from "drizzle-orm"; +import { getSearchIdxClient } from "@hoarder/shared/search"; export class SearchIndexingWorker { static async build() { @@ -25,12 +26,12 @@ export class SearchIndexingWorker { ); worker.on("completed", (job) => { - const jobId = job?.id || "unknown"; + const jobId = job?.id ?? "unknown"; logger.info(`[search][${jobId}] Completed successfully`); }); worker.on("failed", (job, error) => { - const jobId = job?.id || "unknown"; + const jobId = job?.id ?? "unknown"; logger.error(`[search][${jobId}] openai job failed: ${error}`); }); @@ -85,7 +86,7 @@ async function runDelete( } async function runSearchIndexing(job: Job) { - const jobId = job.id || "unknown"; + const jobId = job.id ?? "unknown"; const request = zSearchIndexingRequestSchema.safeParse(job.data); if (!request.success) { -- cgit v1.2.3-70-g09d2