diff options
Diffstat (limited to 'apps')
| -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"), |