From 951fb82c7104d94b7f1b4f2532aab2df1a106410 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 19 Jun 2024 20:30:35 -0300 Subject: [PATCH] refactor: Update GroqModel to use consistent variable names and improve code readability --- src/backend/base/langflow/components/models/GroqModel.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/backend/base/langflow/components/models/GroqModel.py b/src/backend/base/langflow/components/models/GroqModel.py index d49a10996..31e445607 100644 --- a/src/backend/base/langflow/components/models/GroqModel.py +++ b/src/backend/base/langflow/components/models/GroqModel.py @@ -93,12 +93,12 @@ class GroqModel(LCModelComponent): stream = self.stream output = ChatGroq( - model_name=model_name, + model=model_name, max_tokens=max_tokens or None, temperature=temperature, - groq_api_base=groq_api_base, + base_url=groq_api_base, n=n or 1, - groq_api_key=SecretStr(groq_api_key), + api_key=SecretStr(groq_api_key), streaming=stream, )