detect experimental models in integration

acon96 · Jun 16, 2024 · 9a6cde3 · 9a6cde3
1 parent b7edf6a
commit 9a6cde3
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/custom_components/llama_conversation/const.py b/custom_components/llama_conversation/const.py
@@ -324,7 +324,20 @@
     "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"