From 2d2fd9a2aef1c9df70157bb0996c17d6fa5a859b Mon Sep 17 00:00:00 2001
From: carlosrcoelho <carlosrodrigo.coelho@gmail.com>
Date: Tue, 9 Jan 2024 19:36:49 -0300
Subject: [PATCH] Update CSVLoader imports and metadata parameter

---
 .../langflow/components/documentloaders/CSVLoader.py     | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/backend/langflow/components/documentloaders/CSVLoader.py b/src/backend/langflow/components/documentloaders/CSVLoader.py
index 64f1f5527..dce3f5708 100644
--- a/src/backend/langflow/components/documentloaders/CSVLoader.py
+++ b/src/backend/langflow/components/documentloaders/CSVLoader.py
@@ -1,8 +1,8 @@
 
-from langchain import CustomComponent
+from langflow import CustomComponent
 from typing import Optional, Dict, List
-from langchain.loaders import CSVLoader
-from langchain.documents import Document
+from langchain_community.document_loaders.csv_loader import CSVLoader
+from langchain.docstore.document import Document
 
 class CSVLoaderComponent(CustomComponent):
     display_name = "CSVLoader"
@@ -15,6 +15,7 @@ class CSVLoaderComponent(CustomComponent):
                 "required": True,
                 "suffixes": [".csv"],
                 "file_types": ["csv"],
+                "field_type": "file",
             },
             "metadata": {
                 "display_name": "Metadata",
@@ -25,6 +26,6 @@ class CSVLoaderComponent(CustomComponent):
     def build(
         self,
         file_path: str,
-        metadata: Optional[Dict[str, str]] = None,
+        metadata: dict
     ) -> List[Document]:
         return CSVLoader(file_path=file_path, metadata=metadata).load()