From f89dcae8e53601acaf33dc454ec0fc5ec91408a3 Mon Sep 17 00:00:00 2001
From: John Cordeiro <john.dalton@weni.ai>
Date: Mon, 8 Apr 2024 18:01:56 -0300
Subject: [PATCH] Fix loader constructor

---
 app/loaders/loaders.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/app/loaders/loaders.py b/app/loaders/loaders.py
index f0ae1c1..4d14af3 100644
--- a/app/loaders/loaders.py
+++ b/app/loaders/loaders.py
@@ -74,7 +74,7 @@ def _get_file(self, file: str):
                 return file_path
         return file
 
-    def __init__(self, file:str) -> None:
+    def __init__(self, file:str, **kwargs) -> None:
         self.file = self._get_file(file)
         self.loader = TextLoader(self.file)
     
@@ -145,7 +145,7 @@ def pdf_loader(file: str) -> Callable:
 
 
 class DocxLoader(DocumentLoader):
-    def __init__(self, file:str) -> None:
+    def __init__(self, file:str, **kwargs) -> None:
         self.loader = Docx2txtLoader(file)
 
     def load(self) -> List[Document]:
@@ -184,7 +184,7 @@ def xlsx_loader(file: str) -> Callable:
 from urllib.request import urlretrieve
 from urllib.parse import urlparse
 class XlsxLoader(DocumentLoader):
-    def __init__(self, file:str) -> None:
+    def __init__(self, file:str, **kwargs) -> None:
         tmp_file, _ = self._get_temp_file(file)
         self.loader = UnstructuredExcelLoader(tmp_file, mode="single")
     
@@ -218,7 +218,7 @@ def _urls(self, urls: Union[List[str], str]):
             return [urls]
         return urls
 
-    def __init__(self, urls: Union[List[str], str]) -> None:
+    def __init__(self, urls: Union[List[str], str], **kwargs) -> None:
         self.urls = self._urls(urls)
         self.loader = UnstructuredURLLoader(urls=self.urls, headers={"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:109.0) Gecko/20100101 Firefox/109.0"})