diff --git a/src/backend/base/langflow/components/models/aiml.py b/src/backend/base/langflow/components/models/aiml.py index 202f47fd5..dd0613b6a 100644 --- a/src/backend/base/langflow/components/models/aiml.py +++ b/src/backend/base/langflow/components/models/aiml.py @@ -6,7 +6,14 @@ from langflow.base.models.aiml_constants import AimlModels from langflow.base.models.model import LCModelComponent from langflow.field_typing import LanguageModel from langflow.field_typing.range_spec import RangeSpec -from langflow.inputs import DictInput, DropdownInput, FloatInput, IntInput, SecretStrInput, StrInput +from langflow.inputs import ( + DictInput, + DropdownInput, + IntInput, + SecretStrInput, + SliderInput, + StrInput, +) class AIMLModelComponent(LCModelComponent): @@ -46,8 +53,11 @@ class AIMLModelComponent(LCModelComponent): info="The AIML API Key to use for the OpenAI model.", advanced=False, value="AIML_API_KEY", + required=True, + ), + SliderInput( + name="temperature", display_name="Temperature", value=0.1, range_spec=RangeSpec(min=0, max=2, step=0.01) ), - FloatInput(name="temperature", display_name="Temperature", value=0.1), ] @override