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

refactor: Update OllamaModel to use consistent variable names and improve code readability

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 10:17:08 -03:00
commit c0a8f24e1c
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,8 +1,6 @@
from langchain_community.chat_models import ChatOllama from langchain_community.chat_models import ChatOllama
from langchain_core.language_models.chat_models import BaseChatModel 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.base.models.model import LCModelComponent
from langflow.field_typing import BaseLanguageModel, Text from langflow.field_typing import BaseLanguageModel, Text
from langflow.io import BoolInput, DictInput, DropdownInput, FloatInput, IntInput, Output, StrInput from langflow.io import BoolInput, DictInput, DropdownInput, FloatInput, IntInput, Output, StrInput
@ -19,6 +17,7 @@ class ChatOllamaComponent(LCModelComponent):
display_name="Base URL", display_name="Base URL",
info="Endpoint of the Ollama API. Defaults to 'http://localhost:11434' if not specified.", info="Endpoint of the Ollama API. Defaults to 'http://localhost:11434' if not specified.",
advanced=True, advanced=True,
value="http://localhost:11434",
), ),
StrInput( StrInput(
name="model", name="model",