From b3124aa7ed115db59f53274ba4ebf932073d89e6 Mon Sep 17 00:00:00 2001
From: Michael Hansen <mike@rhasspy.org>
Date: Thu, 25 Apr 2024 13:35:29 -0500
Subject: [PATCH] Update Ollama model names list (#116172)

---
 homeassistant/components/ollama/const.py | 145 ++++++++++++-----------
 1 file changed, 78 insertions(+), 67 deletions(-)

diff --git a/homeassistant/components/ollama/const.py b/homeassistant/components/ollama/const.py
index 853370066dc..e25ae1f0877 100644
--- a/homeassistant/components/ollama/const.py
+++ b/homeassistant/components/ollama/const.py
@@ -81,75 +81,86 @@ DEFAULT_MAX_HISTORY = 20
 MAX_HISTORY_SECONDS = 60 * 60  # 1 hour
 
 MODEL_NAMES = [  # https://ollama.com/library
-    "gemma",
-    "llama2",
-    "mistral",
-    "mixtral",
-    "llava",
-    "neural-chat",
-    "codellama",
-    "dolphin-mixtral",
-    "qwen",
-    "llama2-uncensored",
-    "mistral-openorca",
-    "deepseek-coder",
-    "nous-hermes2",
-    "phi",
-    "orca-mini",
-    "dolphin-mistral",
-    "wizard-vicuna-uncensored",
-    "vicuna",
-    "tinydolphin",
-    "llama2-chinese",
-    "nomic-embed-text",
-    "openhermes",
-    "zephyr",
-    "tinyllama",
-    "openchat",
-    "wizardcoder",
-    "starcoder",
-    "phind-codellama",
-    "starcoder2",
-    "yi",
-    "orca2",
-    "falcon",
-    "wizard-math",
-    "dolphin-phi",
-    "starling-lm",
-    "nous-hermes",
-    "stable-code",
-    "medllama2",
-    "bakllava",
-    "codeup",
-    "wizardlm-uncensored",
-    "solar",
-    "everythinglm",
-    "sqlcoder",
-    "dolphincoder",
-    "nous-hermes2-mixtral",
-    "stable-beluga",
-    "yarn-mistral",
-    "stablelm2",
-    "samantha-mistral",
-    "meditron",
-    "stablelm-zephyr",
-    "magicoder",
-    "yarn-llama2",
-    "llama-pro",
-    "deepseek-llm",
-    "wizard-vicuna",
-    "codebooga",
-    "mistrallite",
-    "all-minilm",
-    "nexusraven",
-    "open-orca-platypus2",
-    "goliath",
-    "notux",
-    "megadolphin",
     "alfred",
-    "xwinlm",
-    "wizardlm",
+    "all-minilm",
+    "bakllava",
+    "codebooga",
+    "codegemma",
+    "codellama",
+    "codeqwen",
+    "codeup",
+    "command-r",
+    "command-r-plus",
+    "dbrx",
+    "deepseek-coder",
+    "deepseek-llm",
+    "dolphin-llama3",
+    "dolphin-mistral",
+    "dolphin-mixtral",
+    "dolphin-phi",
+    "dolphincoder",
     "duckdb-nsql",
+    "everythinglm",
+    "falcon",
+    "gemma",
+    "goliath",
+    "llama-pro",
+    "llama2",
+    "llama2-chinese",
+    "llama2-uncensored",
+    "llama3",
+    "llava",
+    "magicoder",
+    "meditron",
+    "medllama2",
+    "megadolphin",
+    "mistral",
+    "mistral-openorca",
+    "mistrallite",
+    "mixtral",
+    "mxbai-embed-large",
+    "neural-chat",
+    "nexusraven",
+    "nomic-embed-text",
     "notus",
+    "notux",
+    "nous-hermes",
+    "nous-hermes2",
+    "nous-hermes2-mixtral",
+    "open-orca-platypus2",
+    "openchat",
+    "openhermes",
+    "orca-mini",
+    "orca2",
+    "phi",
+    "phi3",
+    "phind-codellama",
+    "qwen",
+    "samantha-mistral",
+    "snowflake-arctic-embed",
+    "solar",
+    "sqlcoder",
+    "stable-beluga",
+    "stable-code",
+    "stablelm-zephyr",
+    "stablelm2",
+    "starcoder",
+    "starcoder2",
+    "starling-lm",
+    "tinydolphin",
+    "tinyllama",
+    "vicuna",
+    "wizard-math",
+    "wizard-vicuna",
+    "wizard-vicuna-uncensored",
+    "wizardcoder",
+    "wizardlm",
+    "wizardlm-uncensored",
+    "wizardlm2",
+    "xwinlm",
+    "yarn-llama2",
+    "yarn-mistral",
+    "yi",
+    "zephyr",
 ]
 DEFAULT_MODEL = "llama2:latest"