From a2246be4f5dced7e0d1fce3fd8b65ca00c9335ea Mon Sep 17 00:00:00 2001 From: QuantumGhost Date: Wed, 30 Jul 2025 10:37:36 +0800 Subject: [PATCH] feat(api): Implement checkbox input support in Workflow and Chat App --- .../easy_ui_based_app/variables/manager.py | 25 +++++++++++++++++-- api/core/app/app_config/entities.py | 1 + api/core/app/apps/base_app_generator.py | 5 ++++ 3 files changed, 29 insertions(+), 2 deletions(-) diff --git a/api/core/app/app_config/easy_ui_based_app/variables/manager.py b/api/core/app/app_config/easy_ui_based_app/variables/manager.py index 2f2445a336..6375733448 100644 --- a/api/core/app/app_config/easy_ui_based_app/variables/manager.py +++ b/api/core/app/app_config/easy_ui_based_app/variables/manager.py @@ -3,6 +3,17 @@ import re from core.app.app_config.entities import ExternalDataVariableEntity, VariableEntity, VariableEntityType from core.external_data_tool.factory import ExternalDataToolFactory +_ALLOWED_VARIABLE_ENTITY_TYPE = frozenset( + [ + VariableEntityType.TEXT_INPUT, + VariableEntityType.SELECT, + VariableEntityType.PARAGRAPH, + VariableEntityType.NUMBER, + VariableEntityType.EXTERNAL_DATA_TOOL, + VariableEntityType.CHECKBOX, + ] +) + class BasicVariablesConfigManager: @classmethod @@ -47,6 +58,7 @@ class BasicVariablesConfigManager: VariableEntityType.PARAGRAPH, VariableEntityType.NUMBER, VariableEntityType.SELECT, + VariableEntityType.CHECKBOX, }: variable = variables[variable_type] variable_entities.append( @@ -96,8 +108,17 @@ class BasicVariablesConfigManager: variables = [] for item in config["user_input_form"]: key = list(item.keys())[0] - if key not in {"text-input", "select", "paragraph", "number", "external_data_tool"}: - raise ValueError("Keys in user_input_form list can only be 'text-input', 'paragraph' or 'select'") + # if key not in {"text-input", "select", "paragraph", "number", "external_data_tool"}: + if key not in { + VariableEntityType.TEXT_INPUT, + VariableEntityType.SELECT, + VariableEntityType.PARAGRAPH, + VariableEntityType.NUMBER, + VariableEntityType.EXTERNAL_DATA_TOOL, + VariableEntityType.CHECKBOX, + }: + allowed_keys = ", ".join(i.value for i in _ALLOWED_VARIABLE_ENTITY_TYPE) + raise ValueError(f"Keys in user_input_form list can only be {allowed_keys}") form_item = item[key] if "label" not in form_item: diff --git a/api/core/app/app_config/entities.py b/api/core/app/app_config/entities.py index 75bd2f677a..ffce696a67 100644 --- a/api/core/app/app_config/entities.py +++ b/api/core/app/app_config/entities.py @@ -97,6 +97,7 @@ class VariableEntityType(StrEnum): EXTERNAL_DATA_TOOL = "external_data_tool" FILE = "file" FILE_LIST = "file-list" + CHECKBOX = "checkbox" class VariableEntity(BaseModel): diff --git a/api/core/app/apps/base_app_generator.py b/api/core/app/apps/base_app_generator.py index beece1d77e..a8a6a33bf4 100644 --- a/api/core/app/apps/base_app_generator.py +++ b/api/core/app/apps/base_app_generator.py @@ -144,6 +144,11 @@ class BaseAppGenerator: raise ValueError( f"{variable_entity.variable} in input form must be less than {variable_entity.max_length} files" ) + case VariableEntityType.CHECKBOX: + if not isinstance(value, bool): + raise ValueError(f"{variable_entity.variable} in input form must be a valid boolean value") + case _: + raise AssertionError("this statement should be unreachable.") return value