From 528845ad98e7e4e0eb41cf472644227abf81acc0 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 19 Jun 2024 20:22:33 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=93=9D=20(OllamaModel.py):=20rename=20'st?= =?UTF-8?q?op'=20variable=20to=20'stop=5Ftokens'=20for=20clarity=20and=20c?= =?UTF-8?q?onsistency=20throughout=20the=20codebase?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/base/langflow/components/models/OllamaModel.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/components/models/OllamaModel.py b/src/backend/base/langflow/components/models/OllamaModel.py index 4173b9e13..2204ddb2b 100644 --- a/src/backend/base/langflow/components/models/OllamaModel.py +++ b/src/backend/base/langflow/components/models/OllamaModel.py @@ -1,5 +1,7 @@ from langchain_community.chat_models import ChatOllama from langchain_core.language_models.chat_models import BaseChatModel + + from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.models.model import LCModelComponent from langflow.field_typing import BaseLanguageModel, Text @@ -128,7 +130,7 @@ class ChatOllamaComponent(LCModelComponent): advanced=True, ), StrInput( - name="stop", + name="stop_tokens", display_name="Stop Tokens", info="Comma-separated list of tokens to signal the model to stop generating text.", advanced=True, @@ -207,7 +209,7 @@ class ChatOllamaComponent(LCModelComponent): "repeat_last_n": self.repeat_last_n or None, "repeat_penalty": self.repeat_penalty or None, "temperature": self.temperature or None, - "stop": self.stop.split(",") if self.stop else None, + "stop": self.stop_tokens.split(",") if self.stop_tokens else None, "system": self.system, "template": self.template, "tfs_z": self.tfs_z or None,