From 310886b0ba182cc729a9de99196cd93433acb2f7 Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Mon, 15 Jan 2024 19:49:04 -0300
Subject: [PATCH] Refactor IMSDbLoaderComponent to update document metadata

---
 .../langflow/components/documentloaders/IMSDbLoader.py   | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/backend/langflow/components/documentloaders/IMSDbLoader.py b/src/backend/langflow/components/documentloaders/IMSDbLoader.py
index 99c3b75d3..37c2c542c 100644
--- a/src/backend/langflow/components/documentloaders/IMSDbLoader.py
+++ b/src/backend/langflow/components/documentloaders/IMSDbLoader.py
@@ -20,4 +20,11 @@ class IMSDbLoaderComponent(CustomComponent):
         metadata: Optional[Dict] = None,
         web_path: str = "",
     ) -> Document:
-        return IMSDbLoader(metadata=metadata, web_path=web_path)
+        documents = IMSDbLoader(web_path=web_path).load()
+        if(metadata):
+            for document in documents:
+                if not document.metadata:
+                    document.metadata = metadata
+                else:
+                    document.metadata.update(metadata)
+        return documents