From 184b66f5ab2ee166651838b7b97c0d91f7f8772a Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Mon, 15 Jan 2024 20:06:29 -0300 Subject: [PATCH] Refactor ReadTheDocsLoaderComponent to update document metadata --- .../components/documentloaders/ReadTheDocsLoader.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/components/documentloaders/ReadTheDocsLoader.py b/src/backend/langflow/components/documentloaders/ReadTheDocsLoader.py index 6f1022696..1c78b7e34 100644 --- a/src/backend/langflow/components/documentloaders/ReadTheDocsLoader.py +++ b/src/backend/langflow/components/documentloaders/ReadTheDocsLoader.py @@ -19,4 +19,11 @@ class ReadTheDocsLoaderComponent(CustomComponent): path: str, metadata: Optional[Dict] = None, ) -> List[Document]: - return ReadTheDocsLoader(path=path, metadata=metadata or {}).load() + documents = ReadTheDocsLoader(path=path).load() + if(metadata): + for document in documents: + if not document.metadata: + document.metadata = metadata + else: + document.metadata.update(metadata) + return documents \ No newline at end of file