2024-11-18 00:43:28 +03:00
|
|
|
"""Constants for the HA text AI integration."""
|
2024-11-19 12:45:26 +03:00
|
|
|
from typing import Final
|
2024-11-18 00:43:28 +03:00
|
|
|
from homeassistant.const import Platform
|
2024-11-14 18:30:03 +03:00
|
|
|
|
2024-11-19 12:45:26 +03:00
|
|
|
# Domain
|
|
|
|
|
DOMAIN: Final = "ha_text_ai"
|
|
|
|
|
PLATFORMS: Final = [Platform.SENSOR]
|
2024-11-14 18:30:03 +03:00
|
|
|
|
2024-11-19 12:45:26 +03:00
|
|
|
# Configuration constants
|
|
|
|
|
CONF_MODEL: Final = "model"
|
|
|
|
|
CONF_TEMPERATURE: Final = "temperature"
|
|
|
|
|
CONF_MAX_TOKENS: Final = "max_tokens"
|
|
|
|
|
CONF_API_ENDPOINT: Final = "api_endpoint"
|
|
|
|
|
CONF_REQUEST_INTERVAL: Final = "request_interval"
|
2024-11-15 00:59:39 +03:00
|
|
|
|
2024-11-19 12:45:26 +03:00
|
|
|
# Default values
|
|
|
|
|
DEFAULT_MODEL: Final = "gpt-3.5-turbo"
|
|
|
|
|
DEFAULT_TEMPERATURE: Final = 0.7
|
|
|
|
|
DEFAULT_MAX_TOKENS: Final = 1000
|
|
|
|
|
DEFAULT_API_ENDPOINT: Final = "https://api.openai.com/v1"
|
|
|
|
|
DEFAULT_REQUEST_INTERVAL: Final = 1.0
|
2024-11-18 00:43:28 +03:00
|
|
|
|
2024-11-19 12:45:26 +03:00
|
|
|
# Parameter constraints
|
|
|
|
|
MIN_TEMPERATURE: Final = 0.0
|
|
|
|
|
MAX_TEMPERATURE: Final = 2.0
|
|
|
|
|
MIN_MAX_TOKENS: Final = 1
|
|
|
|
|
MAX_MAX_TOKENS: Final = 4096
|
|
|
|
|
MIN_REQUEST_INTERVAL: Final = 0.1
|
2024-11-18 02:02:36 +03:00
|
|
|
|
2024-11-19 12:45:26 +03:00
|
|
|
# Service names
|
|
|
|
|
SERVICE_ASK_QUESTION: Final = "ask_question"
|
|
|
|
|
SERVICE_CLEAR_HISTORY: Final = "clear_history"
|
|
|
|
|
SERVICE_GET_HISTORY: Final = "get_history"
|
|
|
|
|
SERVICE_SET_SYSTEM_PROMPT: Final = "set_system_prompt"
|
|
|
|
|
|
|
|
|
|
# Service descriptions
|
|
|
|
|
SERVICE_ASK_QUESTION_DESCRIPTION: Final = "Ask a question to the AI model"
|
|
|
|
|
SERVICE_CLEAR_HISTORY_DESCRIPTION: Final = "Clear conversation history"
|
|
|
|
|
SERVICE_GET_HISTORY_DESCRIPTION: Final = "Get conversation history"
|
|
|
|
|
SERVICE_SET_SYSTEM_PROMPT_DESCRIPTION: Final = "Set system prompt for AI model"
|
|
|
|
|
|
|
|
|
|
# Attribute keys
|
|
|
|
|
ATTR_QUESTION: Final = "question"
|
|
|
|
|
ATTR_RESPONSE: Final = "response"
|
|
|
|
|
ATTR_LAST_UPDATED: Final = "last_updated"
|
|
|
|
|
|
|
|
|
|
# Error messages
|
|
|
|
|
ERROR_INVALID_API_KEY: Final = "invalid_api_key"
|
|
|
|
|
ERROR_CANNOT_CONNECT: Final = "cannot_connect"
|
|
|
|
|
ERROR_UNKNOWN: Final = "unknown_error"
|
|
|
|
|
|
|
|
|
|
# Configuration descriptions
|
|
|
|
|
CONF_MODEL_DESCRIPTION: Final = "AI model to use for responses"
|
|
|
|
|
CONF_TEMPERATURE_DESCRIPTION: Final = "Temperature for response generation (0-2)"
|
|
|
|
|
CONF_MAX_TOKENS_DESCRIPTION: Final = "Maximum tokens in response (1-4096)"
|
|
|
|
|
CONF_API_ENDPOINT_DESCRIPTION: Final = "API endpoint URL"
|
|
|
|
|
CONF_REQUEST_INTERVAL_DESCRIPTION: Final = "Minimum time between API requests (seconds)"
|
|
|
|
|
|
|
|
|
|
# Entity attributes
|
|
|
|
|
ENTITY_NAME: Final = "HA Text AI"
|
|
|
|
|
ENTITY_ICON: Final = "mdi:robot"
|
