Add hosted Llama models support

black-roland · Dec 10, 2024 · 6d67683 · 6d67683
1 parent a8ac73a
commit 6d67683
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/custom_components/yandexgpt_conversation/config_flow.py b/custom_components/yandexgpt_conversation/config_flow.py
@@ -178,6 +178,8 @@ def yandexgpt_config_option_schema(
         SelectOptionDict(label="YandexGPT Lite", value="yandexgpt-lite/latest"),
         SelectOptionDict(label="YandexGPT Pro", value="yandexgpt/latest"),
         SelectOptionDict(label="YandexGPT Pro 32k", value="yandexgpt-32k/latest"),
+        SelectOptionDict(label="Llama 8b", value="llama-lite/latest"),
+        SelectOptionDict(label="Llama 70b", value="llama/latest"),
     ]
 
     schema.update(