From 26a028de2e68402f9515da0a8bfa6d23db8f459e Mon Sep 17 00:00:00 2001 From: Cezar Vasconcelos Date: Mon, 24 Jun 2024 01:52:50 +0000 Subject: [PATCH] fix: Update MistralModel.py with default values for inputs --- .../base/langflow/components/models/MistralModel.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/backend/base/langflow/components/models/MistralModel.py b/src/backend/base/langflow/components/models/MistralModel.py index d9058a4f3..178b73b2c 100644 --- a/src/backend/base/langflow/components/models/MistralModel.py +++ b/src/backend/base/langflow/components/models/MistralModel.py @@ -49,7 +49,7 @@ class MistralAIModelComponent(LCModelComponent): info="The Mistral API Key to use for the Mistral model.", advanced=False, ), - FloatInput(name="temperature", display_name="Temperature", advanced=False, value=0.1), + FloatInput(name="temperature", display_name="Temperature", advanced=False, value=0.5), BoolInput(name="stream", display_name="Stream", info=STREAM_INFO_TEXT, advanced=True), StrInput( name="system_message", @@ -57,10 +57,10 @@ class MistralAIModelComponent(LCModelComponent): info="System message to pass to the model.", advanced=True, ), - IntInput(name="max_retries", display_name="Max Retries", advanced=True), - IntInput(name="timeout", display_name="Timeout", advanced=True), - IntInput(name="max_concurrent_requests", display_name="Max Concurrent Requests", advanced=True), - FloatInput(name="top_p", display_name="Top P", advanced=True), + IntInput(name="max_retries", display_name="Max Retries", advanced=True,value=5), + IntInput(name="timeout", display_name="Timeout", advanced=True, value=60), + IntInput(name="max_concurrent_requests", display_name="Max Concurrent Requests", advanced=True,value=3), + FloatInput(name="top_p", display_name="Top P", advanced=True, value=1), IntInput(name="random_seed", display_name="Random Seed", value=1, advanced=True), BoolInput(name="safe_mode", display_name="Safe Mode", advanced=True), ]