diff --git a/src/backend/base/langflow/components/models/cohere.py b/src/backend/base/langflow/components/models/cohere.py index e1957d24d..22c43b373 100644 --- a/src/backend/base/langflow/components/models/cohere.py +++ b/src/backend/base/langflow/components/models/cohere.py @@ -3,7 +3,8 @@ from pydantic.v1 import SecretStr from langflow.base.models.model import LCModelComponent from langflow.field_typing import LanguageModel -from langflow.io import FloatInput, SecretStrInput +from langflow.field_typing.range_spec import RangeSpec +from langflow.io import SecretStrInput, SliderInput class CohereComponent(LCModelComponent): @@ -21,8 +22,15 @@ class CohereComponent(LCModelComponent): info="The Cohere API Key to use for the Cohere model.", advanced=False, value="COHERE_API_KEY", + required=True, + ), + SliderInput( + name="temperature", + display_name="Temperature", + value=0.75, + range_spec=RangeSpec(min=0, max=2, step=0.01), + info="Controls randomness. Lower values are more deterministic, higher values are more creative.", ), - FloatInput(name="temperature", display_name="Temperature", value=0.75), ] def build_model(self) -> LanguageModel: # type: ignore[type-var]