From b8e9a78d6f5e2d595dea948c10d15759e3cd9abc Mon Sep 17 00:00:00 2001
From: Michael Hansen <mike@rhasspy.org>
Date: Wed, 27 Mar 2024 17:19:34 -0500
Subject: [PATCH] Add more Ollama models (#114339)

Add more models
---
 homeassistant/components/ollama/const.py | 41 ++++++++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/homeassistant/components/ollama/const.py b/homeassistant/components/ollama/const.py
index 59f1888cfc7..853370066dc 100644
--- a/homeassistant/components/ollama/const.py
+++ b/homeassistant/components/ollama/const.py
@@ -110,5 +110,46 @@ MODEL_NAMES = [  # https://ollama.com/library
     "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",
+    "duckdb-nsql",
+    "notus",
 ]
 DEFAULT_MODEL = "llama2:latest"