Update ChatOllamaEndpoint.py

This commit is contained in:
yamonkjd 2023-12-26 18:57:40 +09:00 committed by GitHub
commit e9cd339278
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -163,6 +163,7 @@ class ChatOllamaComponent(CustomComponent):
"display_name": "Template",
"field_type": "str",
"info": "Template to use for generating text.",
"advanced": True,
},
}
@ -210,6 +211,7 @@ class ChatOllamaComponent(CustomComponent):
mirostat_eta = None
mirostat_tau = None
# Mapping system settings to their corresponding values
llm_params = {
"base_url": base_url,
"cache": cache,