Change Ollama default to llama3.2 (home-assistant#129901)

eduherraiz · Nov 5, 2024 · eb3371b · eb3371b
1 parent e1ef106
commit eb3371b
Showing 1 changed file with 46 additions and 18 deletions.
diff --git a/homeassistant/components/ollama/const.py b/homeassistant/components/ollama/const.py
@@ -24,27 +24,32 @@
 MODEL_NAMES = [  # https://ollama.com/library
     "alfred",
     "all-minilm",
+    "aya-expanse",
     "aya",
     "bakllava",
+    "bespoke-minicheck",
+    "bge-large",
+    "bge-m3",
     "codebooga",
     "codegeex4",
     "codegemma",
     "codellama",
     "codeqwen",
     "codestral",
     "codeup",
-    "command-r",
     "command-r-plus",
+    "command-r",
     "dbrx",
-    "deepseek-coder",
     "deepseek-coder-v2",
+    "deepseek-coder",
     "deepseek-llm",
+    "deepseek-v2.5",
     "deepseek-v2",
-    "dolphincoder",
     "dolphin-llama3",
     "dolphin-mistral",
     "dolphin-mixtral",
     "dolphin-phi",
+    "dolphincoder",
     "duckdb-nsql",
     "everythinglm",
     "falcon",
@@ -55,74 +60,97 @@
     "glm4",
     "goliath",
     "granite-code",
+    "granite3-dense",
+    "granite3-guardian" "granite3-moe",
+    "hermes3",
     "internlm2",
-    "llama2",
+    "llama-guard3",
+    "llama-pro",
     "llama2-chinese",
     "llama2-uncensored",
-    "llama3",
+    "llama2",
     "llama3-chatqa",
     "llama3-gradient",
     "llama3-groq-tool-use",
-    "llama-pro",
-    "llava",
+    "llama3.1",
+    "llama3.2",
+    "llama3",
     "llava-llama3",
     "llava-phi3",
+    "llava",
     "magicoder",
     "mathstral",
     "meditron",
     "medllama2",
     "megadolphin",
-    "mistral",
-    "mistrallite",
+    "minicpm-v",
+    "mistral-large",
     "mistral-nemo",
     "mistral-openorca",
+    "mistral-small",
+    "mistral",
+    "mistrallite",
     "mixtral",
     "moondream",
     "mxbai-embed-large",
+    "nemotron-mini",
+    "nemotron",
     "neural-chat",
     "nexusraven",
     "nomic-embed-text",
     "notus",
     "notux",
     "nous-hermes",
-    "nous-hermes2",
     "nous-hermes2-mixtral",
+    "nous-hermes2",
     "nuextract",
+    "open-orca-platypus2",
     "openchat",
     "openhermes",
-    "open-orca-platypus2",
-    "orca2",
     "orca-mini",
+    "orca2",
+    "paraphrase-multilingual",
     "phi",
+    "phi3.5",
     "phi3",
     "phind-codellama",
     "qwen",
+    "qwen2-math",
+    "qwen2.5-coder",
+    "qwen2.5",
     "qwen2",
+    "reader-lm",
+    "reflection",
     "samantha-mistral",
+    "shieldgemma",
+    "smollm",
+    "smollm2",
     "snowflake-arctic-embed",
+    "solar-pro",
     "solar",
     "sqlcoder",
     "stable-beluga",
     "stable-code",
-    "stablelm2",
     "stablelm-zephyr",
+    "stablelm2",
     "starcoder",
     "starcoder2",
     "starling-lm",
     "tinydolphin",
     "tinyllama",
     "vicuna",
+    "wizard-math",
+    "wizard-vicuna-uncensored",
+    "wizard-vicuna",
     "wizardcoder",
+    "wizardlm-uncensored",
     "wizardlm",
     "wizardlm2",
-    "wizardlm-uncensored",
-    "wizard-math",
-    "wizard-vicuna",
-    "wizard-vicuna-uncensored",
     "xwinlm",
     "yarn-llama2",
     "yarn-mistral",
+    "yi-coder",
     "yi",
     "zephyr",
 ]
-DEFAULT_MODEL = "llama3.1:latest"
+DEFAULT_MODEL = "llama3.2:latest"