diff options
| author | MohamedBassem <me@mbassem.com> | 2024-03-27 11:46:29 +0000 |
|---|---|---|
| committer | Mohamed Bassem <me@mbassem.com> | 2024-03-27 11:56:13 +0000 |
| commit | d24e50950a36df12b7149b66762a231ac1da14d2 (patch) | |
| tree | 1e0a9c072bce68c7f4f2bd98c8dd306a02da3cfc /apps/workers/searchWorker.ts | |
| parent | ff00ebca308f445785096611c47beed0c2c46c9c (diff) | |
| download | karakeep-d24e50950a36df12b7149b66762a231ac1da14d2.tar.zst | |
feature: Add support for local models using ollama
Diffstat (limited to 'apps/workers/searchWorker.ts')
| -rw-r--r-- | apps/workers/searchWorker.ts | 21 |
1 files changed, 11 insertions, 10 deletions
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<ZSearchIndexingRequest, void>) { - const jobId = job.id || "unknown"; + const jobId = job.id ?? "unknown"; const request = zSearchIndexingRequestSchema.safeParse(job.data); if (!request.success) { |
