From b64d43fe22e245ba27935315a6320d8ae3088340 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 21 Jul 2023 09:25:33 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=80=20chore(constants.py):=20update=20?= =?UTF-8?q?import=20statement=20for=20custom=20component=20creator=20in=20?= =?UTF-8?q?vertex=20type=20map=20=F0=9F=94=80=20chore(constants.py):=20upd?= =?UTF-8?q?ate=20vertex=20type=20map=20to=20include=20custom=20component?= =?UTF-8?q?=20vertices=20using=20custom=5Fcomponent=5Fcreator?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/graph/graph/constants.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/graph/graph/constants.py b/src/backend/langflow/graph/graph/constants.py index e928de6ca..5e5c3b709 100644 --- a/src/backend/langflow/graph/graph/constants.py +++ b/src/backend/langflow/graph/graph/constants.py @@ -14,7 +14,7 @@ from langflow.interface.vector_store.base import vectorstore_creator from langflow.interface.wrappers.base import wrapper_creator from langflow.interface.output_parsers.base import output_parser_creator from langflow.interface.retrievers.base import retriever_creator - +from langflow.interface.custom.base import custom_component_creator from typing import Dict, Type @@ -32,6 +32,6 @@ VERTEX_TYPE_MAP: Dict[str, Type[Vertex]] = { **{t: types.DocumentLoaderVertex for t in documentloader_creator.to_list()}, **{t: types.TextSplitterVertex for t in textsplitter_creator.to_list()}, **{t: types.OutputParserVertex for t in output_parser_creator.to_list()}, - **{t: types.CustomComponentVertex for t in tool_creator.to_list()}, + **{t: types.CustomComponentVertex for t in custom_component_creator.to_list()}, **{t: types.RetrieverVertex for t in retriever_creator.to_list()}, }