From 303779e40aafbbd078b8d68d7e736567c10528a0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Vin=C3=ADcios=20Batista=20da=20Silva?= Date: Thu, 16 Jan 2025 15:37:14 -0300 Subject: [PATCH] feat: improve model input fields for Cohere component (#5712) feat: improve model input fields for cohere component 1. Make api_key field required 2. Convert temperature to SliderInput with range 0-2 3. Add info description to temperature slider --- .../base/langflow/components/models/cohere.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) 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]