jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

📝 (OllamaModel.py): rename 'stop' variable to 'stop_tokens' for clarity and consistency throughout the codebase

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 20:22:33 -03:00
commit 528845ad98
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/base/langflow/components/models/OllamaModel.py
View file

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