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

refactor: Model input fields of Azure OpenAI component (#5711)

Browse source 
feat: improve model input fields for azure openai component

1. Make api_key field required
2. Convert temperature to SliderInput with range 0-2
3. Add info description to temperature slider
4. Fix code formatting

Co-authored-by: Edwin Jose <edwin.jose@datastax.com>
This commit is contained in:
Vinícios Batista da Silva 2025-01-17 21:58:49 -03:00 • committed by GitHub
commit f65d3aa166
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/backend/base/langflow/components/models/azure_openai.py
View file

@ -2,8 +2,9 @@ from langchain_openai import AzureChatOpenAI
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.field_typing.range_spec import RangeSpec
from langflow.inputs import MessageTextInput from langflow.inputs import MessageTextInput
from langflow.io import DropdownInput, FloatInput, IntInput, SecretStrInput from langflow.io import DropdownInput, IntInput, SecretStrInput, SliderInput
class AzureChatOpenAIComponent(LCModelComponent): class AzureChatOpenAIComponent(LCModelComponent):
@ -35,7 +36,7 @@ class AzureChatOpenAIComponent(LCModelComponent):
required=True, required=True,
), ),
MessageTextInput(name="azure_deployment", display_name="Deployment Name", required=True), MessageTextInput(name="azure_deployment", display_name="Deployment Name", required=True),
SecretStrInput(name="api_key", display_name="API Key"), SecretStrInput(name="api_key", display_name="API Key", required=True),
DropdownInput( DropdownInput(
name="api_version", name="api_version",
display_name="API Version", display_name="API Version",
@ -49,7 +50,13 @@ class AzureChatOpenAIComponent(LCModelComponent):
AZURE_OPENAI_API_VERSIONS[0], AZURE_OPENAI_API_VERSIONS[0],
), ),
), ),
FloatInput(name="temperature", display_name="Temperature", value=0.7), SliderInput(
name="temperature",
display_name="Temperature",
value=0.7,
range_spec=RangeSpec(min=0, max=2, step=0.01),
info="Controls randomness. Lower values are more deterministic, higher values are more creative.",
),
IntInput( IntInput(
name="max_tokens", name="max_tokens",
display_name="Max Tokens", display_name="Max Tokens",