72fed878b4
* Add ollama conversation agent * Change iot class * Much better default template * Slight adjustment to prompt * Make casing consistent * Switch to ollama Python fork * Add prompt to tests * Rename to "ollama" * Download models in config flow * Update homeassistant/components/ollama/config_flow.py --------- Co-authored-by: Paulus Schoutsen <balloob@gmail.com>
14 lines
348 B
Python
14 lines
348 B
Python
"""Tests for the Ollama integration."""
|
|
|
|
from homeassistant.components import ollama
|
|
from homeassistant.components.ollama.const import DEFAULT_PROMPT
|
|
|
|
TEST_USER_DATA = {
|
|
ollama.CONF_URL: "http://localhost:11434",
|
|
ollama.CONF_MODEL: "test model",
|
|
}
|
|
|
|
TEST_OPTIONS = {
|
|
ollama.CONF_PROMPT: DEFAULT_PROMPT,
|
|
ollama.CONF_MAX_HISTORY: 2,
|
|
}
|