Update model kwargs and temperature values

This commit is contained in:
Rodrigo Nader 2024-03-31 18:17:29 -03:00
commit e0a0220917
5 changed files with 15 additions and 4 deletions

View file

@ -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": {

View file

@ -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,
},
}

View file

@ -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,
},
}

View file

@ -68,7 +68,7 @@ class OpenAIModelComponent(LCModelComponent):
"display_name": "Temperature",
"advanced": False,
"required": False,
"value": 0.7,
"value": 0.1,
},
"stream": {
"display_name": "Stream",

View file

@ -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,
},
}