mirror of
https://github.com/smkrv/ha-text-ai.git
synced 2026-07-21 22:54:00 +08:00
SMKRV
ce0a75f219
- Create utils.py: normalize_name, get_file_hash, safe_log_data - Create providers.py: PROVIDER_REGISTRY with get_default_endpoint, get_default_model, build_auth_headers - Add DEFAULT_ANTHROPIC_MODEL constant (was incorrectly using gpt-4o-mini) - Replace all inline dispatch tables in config_flow.py and __init__.py - Fix circular import: coordinator.py now imports from utils, not config_flow - Fix NameError in error paths: replace bare constant refs with provider functions - Raise ValueError for unknown providers instead of silent OpenAI fallback
258 lines
8.7 KiB
Python
258 lines
8.7 KiB
Python
"""
|
|
Constants for the HA text AI integration.
|
|
|
|
@license: CC BY-NC-SA 4.0 International
|
|
@author: SMKRV
|
|
@github: https://github.com/smkrv/ha-text-ai
|
|
@source: https://github.com/smkrv/ha-text-ai
|
|
"""
|
|
import os
|
|
import json
|
|
from typing import Final
|
|
import voluptuous as vol
|
|
from homeassistant.const import Platform, CONF_API_KEY, CONF_NAME
|
|
from homeassistant.helpers import config_validation as cv
|
|
import logging
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
# Domain and platforms
|
|
DOMAIN: Final = "ha_text_ai"
|
|
PLATFORMS: list[str] = ["sensor"]
|
|
|
|
# Provider configuration
|
|
CONF_API_PROVIDER: Final = "api_provider"
|
|
API_PROVIDER_OPENAI: Final = "openai"
|
|
API_PROVIDER_ANTHROPIC: Final = "anthropic"
|
|
API_PROVIDER_DEEPSEEK: Final = "deepseek"
|
|
API_PROVIDER_GEMINI: Final = "gemini"
|
|
|
|
API_PROVIDERS: Final = [
|
|
API_PROVIDER_OPENAI,
|
|
API_PROVIDER_ANTHROPIC,
|
|
API_PROVIDER_DEEPSEEK,
|
|
API_PROVIDER_GEMINI
|
|
]
|
|
|
|
# Read version from manifest.json
|
|
MANIFEST_PATH = os.path.join(os.path.dirname(__file__), "manifest.json")
|
|
try:
|
|
with open(MANIFEST_PATH) as manifest_file:
|
|
manifest = json.load(manifest_file)
|
|
VERSION = manifest.get("version", "unknown")
|
|
except FileNotFoundError:
|
|
VERSION = "unknown"
|
|
_LOGGER.warning("manifest.json not found at %s", MANIFEST_PATH)
|
|
except json.JSONDecodeError as err:
|
|
VERSION = "unknown"
|
|
_LOGGER.error("Error decoding JSON from manifest.json: %s", err)
|
|
except Exception as err:
|
|
VERSION = "unknown"
|
|
_LOGGER.error("Error reading manifest.json: %s", err)
|
|
|
|
# Default endpoints
|
|
DEFAULT_OPENAI_ENDPOINT: Final = "https://api.openai.com/v1"
|
|
DEFAULT_ANTHROPIC_ENDPOINT: Final = "https://api.anthropic.com"
|
|
DEFAULT_DEEPSEEK_ENDPOINT: Final = "https://api.deepseek.com"
|
|
DEFAULT_GEMINI_ENDPOINT: Final = "https://generativelanguage.googleapis.com/v1beta"
|
|
|
|
# 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"
|
|
CONF_API_TIMEOUT: Final = "api_timeout"
|
|
CONF_INSTANCE: Final = "instance"
|
|
CONF_MAX_HISTORY_SIZE: Final = "max_history_size" # Correct constant name
|
|
CONF_IS_ANTHROPIC: Final = "is_anthropic"
|
|
CONF_CONTEXT_MESSAGES: Final = "context_messages"
|
|
CONF_STRUCTURED_OUTPUT: Final = "structured_output"
|
|
CONF_JSON_SCHEMA: Final = "json_schema"
|
|
|
|
ABSOLUTE_MAX_HISTORY_SIZE = 500
|
|
MAX_ATTRIBUTE_SIZE = 4 * 1024
|
|
MAX_HISTORY_FILE_SIZE = 1 * 1024 * 1024
|
|
ICONS_SUBDOMAIN = "icons"
|
|
|
|
# Default values
|
|
DEFAULT_MODEL: Final = "gpt-4o-mini"
|
|
DEFAULT_ANTHROPIC_MODEL: Final = "claude-3-5-sonnet"
|
|
DEFAULT_DEEPSEEK_MODEL: Final = "deepseek-chat"
|
|
DEFAULT_GEMINI_MODEL: Final = "gemini-2.0-flash"
|
|
DEFAULT_TEMPERATURE: Final = 0.1
|
|
DEFAULT_MAX_TOKENS: Final = 1000
|
|
DEFAULT_REQUEST_INTERVAL: Final = 1.0
|
|
DEFAULT_TIMEOUT: Final = 30
|
|
DEFAULT_API_TIMEOUT: Final = 30
|
|
DEFAULT_MAX_HISTORY: Final = 50
|
|
DEFAULT_NAME: Final = "HA Text AI"
|
|
DEFAULT_NAME_PREFIX = "ha_text_ai"
|
|
DEFAULT_CONTEXT_MESSAGES: Final = 5
|
|
|
|
TRUNCATION_INDICATOR = " ... "
|
|
|
|
# Parameter constraints
|
|
MIN_TEMPERATURE: Final = 0.0
|
|
MAX_TEMPERATURE: Final = 2.0
|
|
MIN_MAX_TOKENS: Final = 1
|
|
MAX_MAX_TOKENS: Final = 100000
|
|
MIN_REQUEST_INTERVAL: Final = 0.1
|
|
MAX_REQUEST_INTERVAL: Final = 60.0
|
|
MIN_API_TIMEOUT: Final = 5
|
|
MAX_API_TIMEOUT: Final = 600
|
|
|
|
# API constants
|
|
API_TIMEOUT: Final = 30 # Legacy constant, use CONF_API_TIMEOUT from config
|
|
API_RETRY_COUNT: Final = 3
|
|
|
|
# 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"
|
|
|
|
# Attribute keys
|
|
ATTR_QUESTION: Final = "question"
|
|
ATTR_RESPONSE: Final = "response"
|
|
ATTR_INSTANCE: Final = "instance"
|
|
ATTR_MODEL: Final = "model"
|
|
ATTR_TEMPERATURE: Final = "temperature"
|
|
ATTR_MAX_TOKENS: Final = "max_tokens"
|
|
ATTR_SYSTEM_PROMPT: Final = "system_prompt"
|
|
ATTR_API_STATUS: Final = "api_status"
|
|
ATTR_ERROR_COUNT: Final = "error_count"
|
|
ATTR_CONVERSATION_HISTORY: Final = "conversation_history"
|
|
|
|
# Sensor attributes
|
|
ATTR_TOTAL_RESPONSES: Final = "total_responses"
|
|
ATTR_TOTAL_ERRORS: Final = "total_errors"
|
|
ATTR_AVG_RESPONSE_TIME: Final = "average_response_time"
|
|
ATTR_LAST_REQUEST_TIME: Final = "last_request_time"
|
|
ATTR_LAST_ERROR: Final = "last_error"
|
|
ATTR_IS_PROCESSING: Final = "is_processing"
|
|
ATTR_IS_RATE_LIMITED: Final = "is_rate_limited"
|
|
ATTR_IS_MAINTENANCE: Final = "is_maintenance"
|
|
ATTR_API_VERSION: Final = "api_version"
|
|
ATTR_ENDPOINT_STATUS: Final = "endpoint_status"
|
|
ATTR_PERFORMANCE_METRICS: Final = "performance_metrics"
|
|
ATTR_HISTORY_SIZE: Final = "history_size"
|
|
ATTR_UPTIME: Final = "uptime"
|
|
ATTR_API_PROVIDER: Final = "api_provider"
|
|
ATTR_METRICS: Final = "metrics"
|
|
ATTR_STATE: Final = "state"
|
|
ATTR_LAST_RESPONSE: Final = "last_response"
|
|
ATTR_ERROR: Final = "error"
|
|
ATTR_TIMESTAMP: Final = "timestamp"
|
|
|
|
# Sensor metrics
|
|
METRIC_TOTAL_TOKENS: Final = "total_tokens"
|
|
METRIC_PROMPT_TOKENS: Final = "prompt_tokens"
|
|
METRIC_COMPLETION_TOKENS: Final = "completion_tokens"
|
|
METRIC_SUCCESSFUL_REQUESTS: Final = "successful_requests"
|
|
METRIC_FAILED_REQUESTS: Final = "failed_requests"
|
|
METRIC_AVERAGE_LATENCY: Final = "average_latency"
|
|
METRIC_MAX_LATENCY: Final = "max_latency"
|
|
METRIC_MIN_LATENCY: Final = "min_latency"
|
|
|
|
# Error messages
|
|
ERROR_INVALID_API_KEY: Final = "invalid_api_key"
|
|
ERROR_CANNOT_CONNECT: Final = "cannot_connect"
|
|
ERROR_UNKNOWN: Final = "unknown_error"
|
|
ERROR_INVALID_MODEL: Final = "invalid_model"
|
|
ERROR_RATE_LIMIT: Final = "rate_limit_exceeded"
|
|
ERROR_CONTEXT_LENGTH: Final = "context_length_exceeded"
|
|
ERROR_API_ERROR: Final = "api_error"
|
|
ERROR_TIMEOUT: Final = "timeout_error"
|
|
ERROR_INVALID_INSTANCE: Final = "invalid_instance"
|
|
ERROR_NAME_EXISTS: Final = "name_exists"
|
|
|
|
# Entity attributes
|
|
ENTITY_ICON: Final = "mdi:robot"
|
|
ENTITY_ICON_ERROR: Final = "mdi:robot-dead"
|
|
ENTITY_ICON_PROCESSING: Final = "mdi:robot-excited"
|
|
|
|
# State attributes
|
|
STATE_READY: Final = "ready"
|
|
STATE_PROCESSING: Final = "processing"
|
|
STATE_ERROR: Final = "error"
|
|
STATE_INITIALIZING: Final = "initializing"
|
|
STATE_MAINTENANCE: Final = "maintenance"
|
|
STATE_RATE_LIMITED: Final = "rate_limited"
|
|
STATE_DISCONNECTED: Final = "disconnected"
|
|
|
|
# Event names
|
|
EVENT_RESPONSE_RECEIVED: Final = f"{DOMAIN}_response_received"
|
|
EVENT_ERROR_OCCURRED: Final = f"{DOMAIN}_error_occurred"
|
|
EVENT_STATE_CHANGED: Final = f"{DOMAIN}_state_changed"
|
|
|
|
# Service schema constants
|
|
SERVICE_SCHEMA_ASK_QUESTION = vol.Schema({
|
|
vol.Required(CONF_INSTANCE): cv.string,
|
|
vol.Required("question"): cv.string,
|
|
vol.Optional("system_prompt"): cv.string,
|
|
vol.Optional("model"): cv.string,
|
|
vol.Optional("temperature"): vol.All(
|
|
vol.Coerce(float),
|
|
vol.Range(min=MIN_TEMPERATURE, max=MAX_TEMPERATURE)
|
|
),
|
|
vol.Optional("max_tokens"): vol.All(
|
|
vol.Coerce(int),
|
|
vol.Range(min=MIN_MAX_TOKENS, max=MAX_MAX_TOKENS)
|
|
),
|
|
vol.Optional("context_messages"): vol.All(
|
|
vol.Coerce(int),
|
|
vol.Range(min=1, max=20)
|
|
),
|
|
vol.Optional(CONF_STRUCTURED_OUTPUT, default=False): cv.boolean,
|
|
vol.Optional(CONF_JSON_SCHEMA): cv.string,
|
|
})
|
|
|
|
SERVICE_SCHEMA_SET_SYSTEM_PROMPT = vol.Schema({
|
|
vol.Required(CONF_INSTANCE): cv.string,
|
|
vol.Required("prompt"): cv.string
|
|
})
|
|
|
|
SERVICE_SCHEMA_GET_HISTORY = vol.Schema({
|
|
vol.Required(CONF_INSTANCE): cv.string,
|
|
vol.Optional("limit", default=10): vol.All(
|
|
vol.Coerce(int),
|
|
vol.Range(min=1, max=100),
|
|
),
|
|
vol.Optional("filter_model"): cv.string
|
|
})
|
|
|
|
# Configuration schema
|
|
CONFIG_SCHEMA = vol.Schema({
|
|
DOMAIN: vol.Schema({
|
|
vol.Required(CONF_NAME): cv.string,
|
|
vol.Required(CONF_API_KEY): cv.string,
|
|
vol.Required(CONF_API_PROVIDER): vol.In(API_PROVIDERS),
|
|
vol.Optional(CONF_MODEL, default=DEFAULT_MODEL): cv.string,
|
|
vol.Optional(CONF_TEMPERATURE, default=DEFAULT_TEMPERATURE): vol.All(
|
|
vol.Coerce(float),
|
|
vol.Range(min=MIN_TEMPERATURE, max=MAX_TEMPERATURE)
|
|
),
|
|
vol.Optional(CONF_MAX_TOKENS, default=DEFAULT_MAX_TOKENS): vol.All(
|
|
vol.Coerce(int),
|
|
vol.Range(min=MIN_MAX_TOKENS, max=MAX_MAX_TOKENS)
|
|
),
|
|
vol.Optional(CONF_API_ENDPOINT): cv.string,
|
|
vol.Optional(CONF_REQUEST_INTERVAL, default=DEFAULT_REQUEST_INTERVAL): vol.All(
|
|
vol.Coerce(float),
|
|
vol.Range(min=MIN_REQUEST_INTERVAL, max=MAX_REQUEST_INTERVAL)
|
|
),
|
|
vol.Optional(CONF_API_TIMEOUT, default=DEFAULT_API_TIMEOUT): vol.All(
|
|
vol.Coerce(int),
|
|
vol.Range(min=MIN_API_TIMEOUT, max=MAX_API_TIMEOUT)
|
|
),
|
|
vol.Optional(CONF_MAX_HISTORY_SIZE, default=DEFAULT_MAX_HISTORY): vol.All( # Correct usage
|
|
vol.Coerce(int),
|
|
vol.Range(min=1, max=100),
|
|
),
|
|
vol.Optional(CONF_CONTEXT_MESSAGES, default=DEFAULT_CONTEXT_MESSAGES): vol.All(
|
|
vol.Coerce(int),
|
|
vol.Range(min=1, max=20)
|
|
)
|
|
})
|
|
}, extra=vol.ALLOW_EXTRA)
|