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

refactor: Model input fields of Google Generative AI component (#5714)

Browse source 
feat: improve model input fields for google generative ai component

1. Make google_api_key field required
2. Convert temperature to SliderInput with range 0-2
3. Add info description to temperature slider

Co-authored-by: Edwin Jose <edwin.jose@datastax.com>
This commit is contained in:
Vinícios Batista da Silva 2025-01-17 21:58:04 -03:00 • committed by GitHub
commit d9276fdff2
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/google_generative_ai.py
View file

@ -3,7 +3,8 @@ from pydantic.v1 import SecretStr
from langflow.base.models.google_generative_ai_constants import GOOGLE_GENERATIVE_AI_MODELS
from langflow.base.models.model import LCModelComponent
from langflow.field_typing import LanguageModel
from langflow.inputs import DropdownInput, FloatInput, IntInput, SecretStrInput
from langflow.field_typing.range_spec import RangeSpec
from langflow.inputs import DropdownInput, FloatInput, IntInput, SecretStrInput, SliderInput
class GoogleGenerativeAIComponent(LCModelComponent):
@ -28,6 +29,7 @@ class GoogleGenerativeAIComponent(LCModelComponent):
name="google_api_key",
display_name="Google API Key",
info="The Google API Key to use for the Google Generative AI.",
required=True,
),
FloatInput(
name="top_p",
@ -35,7 +37,13 @@ class GoogleGenerativeAIComponent(LCModelComponent):
info="The maximum cumulative probability of tokens to consider when sampling.",
advanced=True,
),
FloatInput(name="temperature", display_name="Temperature", value=0.1),
SliderInput(
name="temperature",
display_name="Temperature",
value=0.1,
range_spec=RangeSpec(min=0, max=2, step=0.01),
info="Controls randomness. Lower values are more deterministic, higher values are more creative.",
),
IntInput(
name="n",
display_name="N",