Ensure only pdf files are handled

open-learning-exchange · Sep 30, 2024 · e8508bb · e8508bb
1 parent 3e5ae00
commit e8508bb
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/chatapi/src/utils/chat-helpers.utils.ts b/chatapi/src/utils/chat-helpers.utils.ts
@@ -137,9 +137,11 @@ export async function aiChatNonStream(
       const typedAttachment = attachment as Attachment;
       const contentType = typedAttachment.content_type;
 
-      const file = await fetchFileFromCouchDB(context.resource.id, attachmentName);
-      const text = await extractTextFromDocument(file as Buffer, contentType);
-      context.data += text;
+      if (contentType === 'application/pdf') {
+        const file = await fetchFileFromCouchDB(context.resource.id, attachmentName);
+        const text = await extractTextFromDocument(file as Buffer, contentType);
+        context.data += text;
+      }
     }
   }