hass-core/tests/components/ollama/__init__.py

"""Tests for the Ollama integration."""

from homeassistant.components import ollama
from homeassistant.helpers import llm

TEST_USER_DATA = {
    ollama.CONF_URL: "http://localhost:11434",
    ollama.CONF_MODEL: "test model",
}

TEST_OPTIONS = {
    ollama.CONF_PROMPT: llm.DEFAULT_INSTRUCTIONS_PROMPT,
    ollama.CONF_MAX_HISTORY: 2,
}