From 177e7f5c0fa83ecbae2509f09c3101cb20381c27 Mon Sep 17 00:00:00 2001
From: rkeshwani <rohit.keshwani@gmail.com>
Date: Fri, 11 Aug 2023 00:20:04 +0000
Subject: [PATCH] Add additional optional input parameter for adding additional
 file loaders.

---
 .../nodes/documentloaders/Folder/Folder.ts           | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/packages/components/nodes/documentloaders/Folder/Folder.ts b/packages/components/nodes/documentloaders/Folder/Folder.ts
index 83bffd18..4ffdce91 100644
--- a/packages/components/nodes/documentloaders/Folder/Folder.ts
+++ b/packages/components/nodes/documentloaders/Folder/Folder.ts
@@ -46,6 +46,13 @@ class Folder_DocumentLoaders implements INode {
                 type: 'json',
                 optional: true,
                 additionalParams: true
+            },
+            {
+                label: 'Additional File Loaders',
+                name: 'additionalLoaders',
+                type: 'json',
+                optional: true,
+                additionalParams: true
             }
         ]
     }
@@ -54,6 +61,8 @@ class Folder_DocumentLoaders implements INode {
         const textSplitter = nodeData.inputs?.textSplitter as TextSplitter
         const folderPath = nodeData.inputs?.folderPath as string
         const metadata = nodeData.inputs?.metadata
+        const additionalLoaders = nodeData.inputs?.additionalLoaders
+        const parsedLoaders = additionalLoaders ? ( typeof metadata === 'object' ? additionalLoaders: JSON.parse( additionalLoaders ) ) : []
 
         const loader = new DirectoryLoader(folderPath, {
             '.json': (path) => new JSONLoader(path),
@@ -61,7 +70,8 @@ class Folder_DocumentLoaders implements INode {
             '.csv': (path) => new CSVLoader(path),
             '.docx': (path) => new DocxLoader(path),
             // @ts-ignore
-            '.pdf': (path) => new PDFLoader(path, { pdfjs: () => import('pdf-parse/lib/pdf.js/v1.10.100/build/pdf.js') })
+            '.pdf': (path) => new PDFLoader(path, { pdfjs: () => import('pdf-parse/lib/pdf.js/v1.10.100/build/pdf.js') }),
+            ...parsedLoaders
         })
         let docs = []