From 98e3f4bdaa890915a3fbe33a3189124b860dacee Mon Sep 17 00:00:00 2001 From: ogabrielluiz Date: Tue, 18 Jun 2024 22:41:10 -0300 Subject: [PATCH] refactor: Update langflow custom components to use specific input classes The code changes update the langflow custom components to use specific input classes, such as IntInput and TextInput, instead of the generic field types. This refactor improves the clarity and type safety of the code. --- tests/data/component_multiple_outputs.py | 7 ++++--- tests/data/component_nested_call.py | 7 ++++--- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/tests/data/component_multiple_outputs.py b/tests/data/component_multiple_outputs.py index 5cdff8773..e0a5f6846 100644 --- a/tests/data/component_multiple_outputs.py +++ b/tests/data/component_multiple_outputs.py @@ -1,11 +1,12 @@ from langflow.custom import Component -from langflow.template.field.base import Input, Output +from langflow.inputs.inputs import IntInput, TextInput +from langflow.template.field.base import Output class MultipleOutputsComponent(Component): inputs = [ - Input(display_name="Input", name="input", field_type=str), - Input(display_name="Number", name="number", field_type=int), + TextInput(display_name="Input", name="input"), + IntInput(display_name="Number", name="number"), ] outputs = [ Output(display_name="Certain Output", name="certain_output", method="certain_output"), diff --git a/tests/data/component_nested_call.py b/tests/data/component_nested_call.py index 6cf64944a..8dbc33cc4 100644 --- a/tests/data/component_nested_call.py +++ b/tests/data/component_nested_call.py @@ -1,13 +1,14 @@ from random import randint from langflow.custom import Component -from langflow.template.field.base import Input, Output +from langflow.inputs.inputs import IntInput, TextInput +from langflow.template.field.base import Output class MultipleOutputsComponent(Component): inputs = [ - Input(display_name="Input", name="input", field_type=str), - Input(display_name="Number", name="number", field_type=int), + TextInput(display_name="Input", name="input"), + IntInput(display_name="Number", name="number"), ] outputs = [ Output(display_name="Certain Output", name="certain_output", method="certain_output"),