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

fix: Ollama model component parameter (#2935)

Browse source 
change params from model to model_name"

Co-authored-by: Sai Kolasani <sai.kolasani@datastax.com>
This commit is contained in:
Sai Kolasani 2024-07-24 16:50:58 -07:00 • committed by GitHub
commit 23b54a63c1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -33,7 +33,7 @@ class ChatOllamaComponent(LCModelComponent):
build_config["mirostat_eta"]["value"] = 0.1
build_config["mirostat_tau"]["value"] = 5
if field_name == "model":
if field_name == "model_name":
base_url_dict = build_config.get("base_url", {})
base_url_load_from_db = base_url_dict.get("load_from_db", False)
base_url_value = base_url_dict.get("value")
@ -41,7 +41,7 @@ class ChatOllamaComponent(LCModelComponent):
base_url_value = self.variables(base_url_value)
elif not base_url_value:
base_url_value = "http://localhost:11434"
build_config["model"]["options"] = self.get_model(base_url_value + "/api/tags")
build_config["model_name"]["options"] = self.get_model(base_url_value + "/api/tags")
if field_name == "keep_alive_flag":
if field_value == "Keep":
@ -77,7 +77,7 @@ class ChatOllamaComponent(LCModelComponent):
DropdownInput(
name="model_name",
display_name="Model Name",
value="llama2",
value="llama3",
info="Refer to https://ollama.ai/library for more models.",
refresh_button=True,
),