From ee63e5432d9226b29acbfd2d0c2fad1ae69373dd Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 3 Aug 2023 14:22:20 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(directory=5Freader.py):=20fi?= =?UTF-8?q?x=20the=20name=20of=20the=20custom=20component=20to=20always=20?= =?UTF-8?q?be=20"CustomComponent"=20for=20consistency=20=F0=9F=90=9B=20fix?= =?UTF-8?q?(directory=5Freader.py):=20add=20the=20output=20types=20of=20th?= =?UTF-8?q?e=20custom=20component=20to=20the=20component=20template=20to?= =?UTF-8?q?=20ensure=20correct=20behavior=20=F0=9F=90=9B=20fix(types.py):?= =?UTF-8?q?=20add=20the=20output=20types=20of=20the=20custom=20component?= =?UTF-8?q?=20to=20the=20component=20template=20to=20ensure=20correct=20be?= =?UTF-8?q?havior?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/custom/directory_reader.py | 5 +++-- src/backend/langflow/interface/types.py | 2 ++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/interface/custom/directory_reader.py b/src/backend/langflow/interface/custom/directory_reader.py index 309577544..3613de5d4 100644 --- a/src/backend/langflow/interface/custom/directory_reader.py +++ b/src/backend/langflow/interface/custom/directory_reader.py @@ -227,12 +227,13 @@ class DirectoryReader: component_name = filename.split(".")[0] # This is the name of the file which will be displayed in the UI # We need to change it from snake_case to CamelCase - camel_case_name = " ".join( + component_name_camelcase = " ".join( word.title() for word in component_name.split("_") ) component_info = { - "name": camel_case_name, + "name": "CustomComponent", + "output_types": [component_name_camelcase], "file": filename, "code": result_content if validation_result else "", "error": "" if validation_result else result_content, diff --git a/src/backend/langflow/interface/types.py b/src/backend/langflow/interface/types.py index a0ad518bd..eb43d95b9 100644 --- a/src/backend/langflow/interface/types.py +++ b/src/backend/langflow/interface/types.py @@ -330,12 +330,14 @@ def build_valid_menu(valid_components): try: component_name = component["name"] component_code = component["code"] + component_output_types = component["output_types"] component_extractor = CustomComponent(code=component_code) component_extractor.is_check_valid() component_template = build_langchain_template_custom_component( component_extractor ) + component_template["output_types"] = component_output_types valid_menu[menu_name][component_name] = component_template