diff options
| author | Mohamed Bassem <me@mbassem.com> | 2025-07-13 10:33:47 +0000 |
|---|---|---|
| committer | Mohamed Bassem <me@mbassem.com> | 2025-07-13 20:44:00 +0000 |
| commit | 4ba3e8047a5b1f160169617187436c09e91662ec (patch) | |
| tree | 6596ef1c5a66840e335289f9c9447ab07671cca2 /apps/workers | |
| parent | d1d5263486f96db578aad918a59007045c3c077f (diff) | |
| download | karakeep-4ba3e8047a5b1f160169617187436c09e91662ec.tar.zst | |
fix: Fix the types of the bookmark types in the db query
Diffstat (limited to 'apps/workers')
| -rw-r--r-- | apps/workers/workers/assetPreprocessingWorker.ts | 10 | ||||
| -rw-r--r-- | apps/workers/workers/inference/tagging.ts | 12 |
2 files changed, 21 insertions, 1 deletions
diff --git a/apps/workers/workers/assetPreprocessingWorker.ts b/apps/workers/workers/assetPreprocessingWorker.ts index c3ecd1e0..f36908cc 100644 --- a/apps/workers/workers/assetPreprocessingWorker.ts +++ b/apps/workers/workers/assetPreprocessingWorker.ts @@ -189,6 +189,11 @@ async function extractAndSaveImageText( bookmark: NonNullable<Awaited<ReturnType<typeof getBookmark>>>, isFixMode: boolean, ): Promise<boolean> { + if (!bookmark.asset) { + throw new Error( + `[assetPreprocessing][${jobId}] Bookmark ${bookmark.id} has no asset attached`, + ); + } { const alreadyHasText = !!bookmark.asset.content; if (alreadyHasText && isFixMode) { @@ -232,6 +237,11 @@ async function extractAndSavePDFText( bookmark: NonNullable<Awaited<ReturnType<typeof getBookmark>>>, isFixMode: boolean, ): Promise<boolean> { + if (!bookmark.asset) { + throw new Error( + `[assetPreprocessing][${jobId}] Bookmark ${bookmark.id} has no asset attached`, + ); + } { const alreadyHasText = !!bookmark.asset.content; if (alreadyHasText && isFixMode) { diff --git a/apps/workers/workers/inference/tagging.ts b/apps/workers/workers/inference/tagging.ts index 3c7b5adb..47419a1f 100644 --- a/apps/workers/workers/inference/tagging.ts +++ b/apps/workers/workers/inference/tagging.ts @@ -118,6 +118,11 @@ async function inferTagsFromImage( inferenceClient: InferenceClient, abortSignal: AbortSignal, ) { + if (!bookmark.asset) { + throw new Error( + `[inference][${jobId}] Bookmark ${bookmark.id} has no asset attached`, + ); + } const { asset, metadata } = await readAsset({ userId: bookmark.userId, assetId: bookmark.asset.assetId, @@ -199,11 +204,16 @@ function containsTagsPlaceholder(prompts: { text: string }[]): boolean { } async function inferTagsFromPDF( - _jobId: string, + jobId: string, bookmark: NonNullable<Awaited<ReturnType<typeof fetchBookmark>>>, inferenceClient: InferenceClient, abortSignal: AbortSignal, ) { + if (!bookmark.asset) { + throw new Error( + `[inference][${jobId}] Bookmark ${bookmark.id} has no asset attached`, + ); + } const prompt = buildTextPrompt( serverConfig.inference.inferredTagLang, await fetchCustomPrompts(bookmark.userId, "text"), |