evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Switch to official ollama library, update models (#122471)

Browse source 
* Switch to mainstream ollama library, update models

* Fix mypy error
This commit is contained in:
Michael Hansen 2024-07-23 10:57:54 -05:00 • committed by GitHub
parent d7b0d1a50e
commit 3ba2a0518e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 35 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

41
homeassistant/components/ollama/const.py
View file

@ -86,41 +86,60 @@ MAX_HISTORY_SECONDS = 60 * 60 # 1 hour
MODEL_NAMES = [ # https://ollama.com/library
"alfred",
"all-minilm",
"aya",
"bakllava",
"codebooga",
"codegeex4",
"codegemma",
"codellama",
"codeqwen",
"codestral",
"codeup",
"command-r",
"command-r-plus",
"dbrx",
"deepseek-coder",
"deepseek-coder-v2",
"deepseek-llm",
"deepseek-v2",
"dolphincoder",
"dolphin-llama3",
"dolphin-mistral",
"dolphin-mixtral",
"dolphin-phi",
"dolphincoder",
"duckdb-nsql",
"everythinglm",
"falcon",
"falcon2",
"firefunction-v2",
"gemma",
"gemma2",
"glm4",
"goliath",
"llama-pro",
"granite-code",
"internlm2",
"llama2",
"llama2-chinese",
"llama2-uncensored",
"llama3",
"llama3-chatqa",
"llama3-gradient",
"llama3-groq-tool-use",
"llama-pro",
"llava",
"llava-llama3",
"llava-phi3",
"magicoder",
"mathstral",
"meditron",
"medllama2",
"megadolphin",
"mistral",
"mistral-openorca",
"mistrallite",
"mistral-nemo",
"mistral-openorca",
"mixtral",
"moondream",
"mxbai-embed-large",
"neural-chat",
"nexusraven",
@ -130,36 +149,38 @@ MODEL_NAMES = [ # https://ollama.com/library
"nous-hermes",
"nous-hermes2",
"nous-hermes2-mixtral",
"open-orca-platypus2",
"nuextract",
"openchat",
"openhermes",
"orca-mini",
"open-orca-platypus2",
"orca2",
"orca-mini",
"phi",
"phi3",
"phind-codellama",
"qwen",
"qwen2",
"samantha-mistral",
"snowflake-arctic-embed",
"solar",
"sqlcoder",
"stable-beluga",
"stable-code",
"stablelm-zephyr",
"stablelm2",
"stablelm-zephyr",
"starcoder",
"starcoder2",
"starling-lm",
"tinydolphin",
"tinyllama",
"vicuna",
"wizardcoder",
"wizardlm",
"wizardlm2",
"wizardlm-uncensored",
"wizard-math",
"wizard-vicuna",
"wizard-vicuna-uncensored",
"wizardcoder",
"wizardlm",
"wizardlm-uncensored",
"wizardlm2",
"xwinlm",
"yarn-llama2",
"yarn-mistral",