mirror of
https://github.com/home-assistant/core.git
synced 2026-07-01 11:46:40 +01:00
Denis Shulyaka
59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
"""Constants for the Anthropic integration."""
|
|
|
|
from enum import StrEnum
|
|
import logging
|
|
|
|
DOMAIN = "anthropic"
|
|
LOGGER = logging.getLogger(__package__)
|
|
|
|
DEFAULT_CONVERSATION_NAME = "Claude conversation"
|
|
DEFAULT_AI_TASK_NAME = "Claude AI Task"
|
|
|
|
CONF_RECOMMENDED = "recommended"
|
|
CONF_CHAT_MODEL = "chat_model"
|
|
CONF_CODE_EXECUTION = "code_execution"
|
|
CONF_MAX_TOKENS = "max_tokens"
|
|
CONF_PROMPT_CACHING = "prompt_caching"
|
|
CONF_THINKING_BUDGET = "thinking_budget"
|
|
CONF_THINKING_EFFORT = "thinking_effort"
|
|
CONF_TOOL_SEARCH = "tool_search"
|
|
CONF_WEB_FETCH = "web_fetch"
|
|
CONF_WEB_FETCH_MAX_USES = "web_fetch_max_uses"
|
|
CONF_WEB_SEARCH = "web_search"
|
|
CONF_WEB_SEARCH_USER_LOCATION = "user_location"
|
|
CONF_WEB_SEARCH_MAX_USES = "web_search_max_uses"
|
|
CONF_WEB_SEARCH_CITY = "city"
|
|
CONF_WEB_SEARCH_REGION = "region"
|
|
CONF_WEB_SEARCH_COUNTRY = "country"
|
|
CONF_WEB_SEARCH_TIMEZONE = "timezone"
|
|
|
|
|
|
class PromptCaching(StrEnum):
|
|
"""Prompt caching options."""
|
|
|
|
OFF = "off"
|
|
PROMPT = "prompt"
|
|
AUTOMATIC = "automatic"
|
|
|
|
|
|
MIN_THINKING_BUDGET = 1024
|
|
|
|
DEFAULT = {
|
|
CONF_CHAT_MODEL: "claude-haiku-4-5",
|
|
CONF_CODE_EXECUTION: False,
|
|
CONF_MAX_TOKENS: 3000,
|
|
CONF_PROMPT_CACHING: PromptCaching.PROMPT.value,
|
|
CONF_THINKING_BUDGET: MIN_THINKING_BUDGET,
|
|
CONF_THINKING_EFFORT: "low",
|
|
CONF_TOOL_SEARCH: False,
|
|
CONF_WEB_FETCH: False,
|
|
CONF_WEB_FETCH_MAX_USES: 5,
|
|
CONF_WEB_SEARCH: False,
|
|
CONF_WEB_SEARCH_USER_LOCATION: False,
|
|
CONF_WEB_SEARCH_MAX_USES: 5,
|
|
}
|
|
|
|
TOOL_SEARCH_UNSUPPORTED_MODELS = [
|
|
"claude-haiku",
|
|
]
|