mirror of
https://github.com/home-assistant/core.git
synced 2025-11-08 10:29:27 +00:00
20 lines
449 B
Python
20 lines
449 B
Python
"""Tests for the Ollama integration."""
|
|
|
|
from homeassistant.components import ollama
|
|
from homeassistant.helpers import llm
|
|
|
|
TEST_USER_DATA = {
|
|
ollama.CONF_URL: "http://localhost:11434",
|
|
}
|
|
|
|
TEST_OPTIONS = {
|
|
ollama.CONF_PROMPT: llm.DEFAULT_INSTRUCTIONS_PROMPT,
|
|
ollama.CONF_MAX_HISTORY: 2,
|
|
ollama.CONF_MODEL: "test_model:latest",
|
|
}
|
|
|
|
TEST_AI_TASK_OPTIONS = {
|
|
ollama.CONF_MAX_HISTORY: 2,
|
|
ollama.CONF_MODEL: "test_model:latest",
|
|
}
|