From f086f31e79ddd30b2e61ed5388532dc559aa151a Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Wed, 10 Jan 2024 16:43:50 -0300 Subject: [PATCH] Add field type for metadata in ReadTheDocsLoaderComponent --- .../components/documentloaders/ReadTheDocsLoader.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/components/documentloaders/ReadTheDocsLoader.py b/src/backend/langflow/components/documentloaders/ReadTheDocsLoader.py index 901148264..47fa35489 100644 --- a/src/backend/langflow/components/documentloaders/ReadTheDocsLoader.py +++ b/src/backend/langflow/components/documentloaders/ReadTheDocsLoader.py @@ -1,6 +1,6 @@ from langflow import CustomComponent -from langflow.field_typing import Document from typing import Dict, Optional +from langchain_community.document_loaders.readthedocs import ReadTheDocsLoader class ReadTheDocsLoaderComponent(CustomComponent): @@ -9,7 +9,7 @@ class ReadTheDocsLoaderComponent(CustomComponent): def build_config(self): return { - "metadata": {"display_name": "Metadata", "default": {}}, + "metadata": {"display_name": "Metadata", "default": {},"field_type": "dict"}, "path": {"display_name": "Local directory", "required": True}, } @@ -17,5 +17,5 @@ class ReadTheDocsLoaderComponent(CustomComponent): self, path: str, metadata: Optional[Dict] = None, - ) -> Document: - return Document(path=path, metadata=metadata or {}) + ) -> ReadTheDocsLoader: + return ReadTheDocsLoader(path=path, metadata=metadata or {}).load()