diff --git a/custom_components/llama_conversation/const.py b/custom_components/llama_conversation/const.py index 1cad34d..dea023c 100644 --- a/custom_components/llama_conversation/const.py +++ b/custom_components/llama_conversation/const.py @@ -324,7 +324,20 @@ OPTIONS_OVERRIDES = { "command-r": { CONF_PROMPT: DEFAULT_PROMPT_BASE + ICL_EXTRAS, CONF_PROMPT_TEMPLATE: PROMPT_TEMPLATE_COMMAND_R, - } + }, + "stablehome": { + CONF_PROMPT: DEFAULT_PROMPT_BASE_LEGACY, + CONF_PROMPT_TEMPLATE: PROMPT_TEMPLATE_ZEPHYR, + CONF_USE_IN_CONTEXT_LEARNING_EXAMPLES: False, + CONF_SERVICE_CALL_REGEX: FINE_TUNED_SERVICE_CALL_REGEX, + CONF_TOOL_FORMAT: TOOL_FORMAT_MINIMAL, + }, + "tinyhome": { + CONF_PROMPT: DEFAULT_PROMPT_BASE_LEGACY, + CONF_USE_IN_CONTEXT_LEARNING_EXAMPLES: False, + CONF_SERVICE_CALL_REGEX: FINE_TUNED_SERVICE_CALL_REGEX, + CONF_TOOL_FORMAT: TOOL_FORMAT_MINIMAL, + }, } INTEGRATION_VERSION = "0.3.3"