33bde48c9c
* Add AI Task integration * Remove GenTextTaskType * Add AI Task prefs * Add action to LLM task * Remove WS command * Rename result to text for GenTextTaskResult * Apply suggestions from code review Co-authored-by: Allen Porter <allen.porter@gmail.com> * Add supported feature for generate text * Update const.py Co-authored-by: HarvsG <11440490+HarvsG@users.noreply.github.com> * Update homeassistant/components/ai_task/services.yaml Co-authored-by: HarvsG <11440490+HarvsG@users.noreply.github.com> * Use WS API to set preferences * Simplify pref storage * Simplify pref test * Update homeassistant/components/ai_task/services.yaml Co-authored-by: Allen Porter <allen.porter@gmail.com> --------- Co-authored-by: Allen Porter <allen.porter@gmail.com> Co-authored-by: HarvsG <11440490+HarvsG@users.noreply.github.com>
55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
"""HTTP endpoint for AI Task integration."""
|
|
|
|
from typing import Any
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components import websocket_api
|
|
from homeassistant.core import HomeAssistant, callback
|
|
|
|
from .const import DATA_PREFERENCES
|
|
|
|
|
|
@callback
|
|
def async_setup(hass: HomeAssistant) -> None:
|
|
"""Set up the HTTP API for the conversation integration."""
|
|
websocket_api.async_register_command(hass, websocket_get_preferences)
|
|
websocket_api.async_register_command(hass, websocket_set_preferences)
|
|
|
|
|
|
@websocket_api.websocket_command(
|
|
{
|
|
vol.Required("type"): "ai_task/preferences/get",
|
|
}
|
|
)
|
|
@callback
|
|
def websocket_get_preferences(
|
|
hass: HomeAssistant,
|
|
connection: websocket_api.ActiveConnection,
|
|
msg: dict[str, Any],
|
|
) -> None:
|
|
"""Get AI task preferences."""
|
|
preferences = hass.data[DATA_PREFERENCES]
|
|
connection.send_result(msg["id"], preferences.as_dict())
|
|
|
|
|
|
@websocket_api.websocket_command(
|
|
{
|
|
vol.Required("type"): "ai_task/preferences/set",
|
|
vol.Optional("gen_text_entity_id"): vol.Any(str, None),
|
|
}
|
|
)
|
|
@websocket_api.require_admin
|
|
@callback
|
|
def websocket_set_preferences(
|
|
hass: HomeAssistant,
|
|
connection: websocket_api.ActiveConnection,
|
|
msg: dict[str, Any],
|
|
) -> None:
|
|
"""Set AI task preferences."""
|
|
preferences = hass.data[DATA_PREFERENCES]
|
|
msg.pop("type")
|
|
msg_id = msg.pop("id")
|
|
preferences.async_set_preferences(**msg)
|
|
connection.send_result(msg_id, preferences.as_dict())
|