From f65d3aa1668ef2569c4a1e605562b13894294375 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Vin=C3=ADcios=20Batista=20da=20Silva?= Date: Fri, 17 Jan 2025 21:58:49 -0300 Subject: [PATCH] refactor: Model input fields of Azure OpenAI component (#5711) 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 --- .../base/langflow/components/models/azure_openai.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/components/models/azure_openai.py b/src/backend/base/langflow/components/models/azure_openai.py index 726620ce8..a6aa5b047 100644 --- a/src/backend/base/langflow/components/models/azure_openai.py +++ b/src/backend/base/langflow/components/models/azure_openai.py @@ -2,8 +2,9 @@ from langchain_openai import AzureChatOpenAI from langflow.base.models.model import LCModelComponent from langflow.field_typing import LanguageModel +from langflow.field_typing.range_spec import RangeSpec from langflow.inputs import MessageTextInput -from langflow.io import DropdownInput, FloatInput, IntInput, SecretStrInput +from langflow.io import DropdownInput, IntInput, SecretStrInput, SliderInput class AzureChatOpenAIComponent(LCModelComponent): @@ -35,7 +36,7 @@ class AzureChatOpenAIComponent(LCModelComponent): 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( name="api_version", display_name="API Version", @@ -49,7 +50,13 @@ class AzureChatOpenAIComponent(LCModelComponent): 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( name="max_tokens", display_name="Max Tokens",