From 24931545f923dd54aea579009525395a28d2a867 Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Wed, 10 Jan 2024 16:45:31 -0300 Subject: [PATCH] Add Document type to ReadTheDocsLoaderComponent --- .../langflow/components/documentloaders/ReadTheDocsLoader.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/components/documentloaders/ReadTheDocsLoader.py b/src/backend/langflow/components/documentloaders/ReadTheDocsLoader.py index 47fa35489..6f1022696 100644 --- a/src/backend/langflow/components/documentloaders/ReadTheDocsLoader.py +++ b/src/backend/langflow/components/documentloaders/ReadTheDocsLoader.py @@ -1,5 +1,6 @@ from langflow import CustomComponent -from typing import Dict, Optional +from typing import Dict, Optional,List +from langchain_core.documents import Document from langchain_community.document_loaders.readthedocs import ReadTheDocsLoader @@ -17,5 +18,5 @@ class ReadTheDocsLoaderComponent(CustomComponent): self, path: str, metadata: Optional[Dict] = None, - ) -> ReadTheDocsLoader: + ) -> List[Document]: return ReadTheDocsLoader(path=path, metadata=metadata or {}).load()