jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

feat: improve model input fields for Cohere component (#5712)

Browse source 
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
This commit is contained in:
Vinícios Batista da Silva 2025-01-16 15:37:14 -03:00 • committed by GitHub
commit 303779e40a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/backend/base/langflow/components/models/cohere.py
View file

@ -3,7 +3,8 @@ from pydantic.v1 import SecretStr
from langflow.base.models.model import LCModelComponent from langflow.base.models.model import LCModelComponent
from langflow.field_typing import LanguageModel 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): class CohereComponent(LCModelComponent):
@ -21,8 +22,15 @@ class CohereComponent(LCModelComponent):
info="The Cohere API Key to use for the Cohere model.", info="The Cohere API Key to use for the Cohere model.",
advanced=False, advanced=False,
value="COHERE_API_KEY", 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] def build_model(self) -> LanguageModel: # type: ignore[type-var]