From 64bcb971ad850516ffbfc2345ffef10e27a29041 Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Mon, 15 Jan 2024 19:38:59 -0300 Subject: [PATCH] Refactor HNLoaderComponent load method to update document metadata --- .../langflow/components/documentloaders/HNLoader.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/components/documentloaders/HNLoader.py b/src/backend/langflow/components/documentloaders/HNLoader.py index 5d3cb768e..d54591d46 100644 --- a/src/backend/langflow/components/documentloaders/HNLoader.py +++ b/src/backend/langflow/components/documentloaders/HNLoader.py @@ -27,7 +27,11 @@ class HNLoaderComponent(CustomComponent): web_path: str, metadata: Optional[Dict] = None, ) -> HNLoader: - # Assuming that there's a specific loader for Hacker News - # as HNloader does not take a web_path argument - # The HackerNewsLoader needs to be defined somewhere in the actual implementation - return HNLoader(metadata=metadata, web_path=web_path) + documents = HNLoader(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