From e0a0220917e4d92a02b265adfed03be346a6123d Mon Sep 17 00:00:00 2001
From: Rodrigo Nader <rodrigosilvanader@gmail.com>
Date: Sun, 31 Mar 2024 18:17:29 -0300
Subject: [PATCH] Update model kwargs and temperature values

---
 .../base/langflow/components/models/AmazonBedrockModel.py   | 5 ++++-
 .../base/langflow/components/models/AzureOpenAIModel.py     | 6 ++++--
 .../langflow/components/models/BaiduQianfanChatModel.py     | 4 ++++
 src/backend/base/langflow/components/models/OpenAIModel.py  | 2 +-
 .../base/langflow/components/models/VertexAiModel.py        | 2 ++
 5 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/src/backend/base/langflow/components/models/AmazonBedrockModel.py b/src/backend/base/langflow/components/models/AmazonBedrockModel.py
index 234924fe2..2fe12bbce 100644
--- a/src/backend/base/langflow/components/models/AmazonBedrockModel.py
+++ b/src/backend/base/langflow/components/models/AmazonBedrockModel.py
@@ -52,7 +52,10 @@ class AmazonBedrockComponent(LCModelComponent):
             "credentials_profile_name": {"display_name": "Credentials Profile Name"},
             "endpoint_url": {"display_name": "Endpoint URL"},
             "region_name": {"display_name": "Region Name"},
-            "model_kwargs": {"display_name": "Model Kwargs"},
+            "model_kwargs": {
+                "display_name": "Model Kwargs",
+                "advanced": True,
+            },
             "cache": {"display_name": "Cache"},
             "input_value": {"display_name": "Input"},
             "system_message": {
diff --git a/src/backend/base/langflow/components/models/AzureOpenAIModel.py b/src/backend/base/langflow/components/models/AzureOpenAIModel.py
index a697d3b68..fb24c24de 100644
--- a/src/backend/base/langflow/components/models/AzureOpenAIModel.py
+++ b/src/backend/base/langflow/components/models/AzureOpenAIModel.py
@@ -73,9 +73,9 @@ class AzureChatOpenAIComponent(LCModelComponent):
             "api_key": {"display_name": "API Key", "required": True, "password": True},
             "temperature": {
                 "display_name": "Temperature",
-                "value": 0.7,
+                "value": 0.1,
                 "field_type": "float",
-                "required": False,
+                "required": True,
             },
             "max_tokens": {
                 "display_name": "Max Tokens",
@@ -90,10 +90,12 @@ class AzureChatOpenAIComponent(LCModelComponent):
             "stream": {
                 "display_name": "Stream",
                 "info": "Stream the response from the model.",
+                "advanced": True,
             },
             "system_message": {
                 "display_name": "System Message",
                 "info": "System message to pass to the model.",
+                "advanced": True,
             },
         }
 
diff --git a/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py b/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py
index 3618b54ba..3d6f900d2 100644
--- a/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py
+++ b/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py
@@ -62,6 +62,7 @@ class QianfanChatEndpointComponent(LCModelComponent):
                 "field_type": "float",
                 "info": "Model params, only supported in ERNIE-Bot and ERNIE-Bot-turbo",
                 "value": 0.8,
+                "advanced": True,
             },
             "temperature": {
                 "display_name": "Temperature",
@@ -74,6 +75,7 @@ class QianfanChatEndpointComponent(LCModelComponent):
                 "field_type": "float",
                 "info": "Model params, only supported in ERNIE-Bot and ERNIE-Bot-turbo",
                 "value": 1.0,
+                "advanced": True,
             },
             "endpoint": {
                 "display_name": "Endpoint",
@@ -84,10 +86,12 @@ class QianfanChatEndpointComponent(LCModelComponent):
             "stream": {
                 "display_name": "Stream",
                 "info": "Stream the response from the model.",
+                "advanced": True,
             },
             "system_message": {
                 "display_name": "System Message",
                 "info": "System message to pass to the model.",
+                "advanced": True,
             },
         }
 
diff --git a/src/backend/base/langflow/components/models/OpenAIModel.py b/src/backend/base/langflow/components/models/OpenAIModel.py
index 5fbd1bd2a..52b5dd57c 100644
--- a/src/backend/base/langflow/components/models/OpenAIModel.py
+++ b/src/backend/base/langflow/components/models/OpenAIModel.py
@@ -68,7 +68,7 @@ class OpenAIModelComponent(LCModelComponent):
                 "display_name": "Temperature",
                 "advanced": False,
                 "required": False,
-                "value": 0.7,
+                "value": 0.1,
             },
             "stream": {
                 "display_name": "Stream",
diff --git a/src/backend/base/langflow/components/models/VertexAiModel.py b/src/backend/base/langflow/components/models/VertexAiModel.py
index 727864f2c..5f778bbbc 100644
--- a/src/backend/base/langflow/components/models/VertexAiModel.py
+++ b/src/backend/base/langflow/components/models/VertexAiModel.py
@@ -78,10 +78,12 @@ class ChatVertexAIComponent(LCModelComponent):
             "stream": {
                 "display_name": "Stream",
                 "info": "Stream the response from the model.",
+                "advanced": True,
             },
             "system_message": {
                 "display_name": "System Message",
                 "info": "System message to pass to the model.",
+                "advanced": True,
             },
         }