aboutsummaryrefslogtreecommitdiffstats
path: root/apps/workers/openaiWorker.ts
diff options
context:
space:
mode:
authorMohamedBassem <me@mbassem.com>2024-03-30 16:26:16 +0000
committerMohamedBassem <me@mbassem.com>2024-03-30 16:26:16 +0000
commit46b78eaac30be26fe40520e97786563344af8403 (patch)
treec4c0e1ae1d3d21a6f1fbf5f44f68e99243bbb5d3 /apps/workers/openaiWorker.ts
parent853ed13450b3a0d92cba144cc0dfd0696e7c810c (diff)
downloadkarakeep-46b78eaac30be26fe40520e97786563344af8403.tar.zst
format: Add missing lint and format, and format the entire repo
Diffstat (limited to 'apps/workers/openaiWorker.ts')
-rw-r--r--apps/workers/openaiWorker.ts28
1 files changed, 15 insertions, 13 deletions
diff --git a/apps/workers/openaiWorker.ts b/apps/workers/openaiWorker.ts
index ee48d148..bb8015a5 100644
--- a/apps/workers/openaiWorker.ts
+++ b/apps/workers/openaiWorker.ts
@@ -1,7 +1,9 @@
-import { Job, Worker } from "bullmq";
+import type { Job } from "bullmq";
+import { Worker } from "bullmq";
import { and, eq, inArray } from "drizzle-orm";
import { z } from "zod";
+import type { ZOpenAIRequest } from "@hoarder/shared/queues";
import { db } from "@hoarder/db";
import { bookmarks, bookmarkTags, tagsOnBookmarks } from "@hoarder/db/schema";
import { readAsset } from "@hoarder/shared/assetdb";
@@ -10,11 +12,11 @@ import {
OpenAIQueue,
queueConnectionDetails,
SearchIndexingQueue,
- ZOpenAIRequest,
zOpenAIRequestSchema,
} from "@hoarder/shared/queues";
-import { InferenceClient, InferenceClientFactory } from "./inference";
+import type { InferenceClient } from "./inference";
+import { InferenceClientFactory } from "./inference";
const openAIResponseSchema = z.object({
tags: z.array(z.string()),
@@ -41,7 +43,7 @@ async function attemptMarkTaggingStatus(
}
export class OpenAiWorker {
- static async build() {
+ static build() {
logger.info("Starting inference worker ...");
const worker = new Worker<ZOpenAIRequest, void>(
OpenAIQueue.name,
@@ -52,16 +54,16 @@ export class OpenAiWorker {
},
);
- worker.on("completed", async (job): Promise<void> => {
+ worker.on("completed", (job) => {
const jobId = job?.id ?? "unknown";
logger.info(`[inference][${jobId}] Completed successfully`);
- await attemptMarkTaggingStatus(job?.data, "success");
+ attemptMarkTaggingStatus(job?.data, "success");
});
- worker.on("failed", async (job, error): Promise<void> => {
+ worker.on("failed", (job, error) => {
const jobId = job?.id ?? "unknown";
logger.error(`[inference][${jobId}] inference job failed: ${error}`);
- await attemptMarkTaggingStatus(job?.data, "failure");
+ attemptMarkTaggingStatus(job?.data, "failure");
});
return worker;
@@ -90,11 +92,11 @@ function buildPrompt(
bookmark: NonNullable<Awaited<ReturnType<typeof fetchBookmark>>>,
) {
const truncateContent = (content: string) => {
- let words = content.split(" ");
- if (words.length > 1500) {
- words = words.slice(1500);
- content = words.join(" ");
- }
+ let words = content.split(" ");
+ if (words.length > 1500) {
+ words = words.slice(1500);
+ content = words.join(" ");
+ }
return content;
};
if (bookmark.link) {