diff --git a/homeassistant/components/ai_task/const.py b/homeassistant/components/ai_task/const.py index 17dfcca4660..fa8702ed69e 100644 --- a/homeassistant/components/ai_task/const.py +++ b/homeassistant/components/ai_task/const.py @@ -34,6 +34,3 @@ class AITaskEntityFeature(IntFlag): GENERATE_DATA = 1 """Generate data based on instructions.""" - - GENERATE_STRUCTURED_DATA = 2 - """Generate structured data based on instructions.""" diff --git a/homeassistant/components/ai_task/services.yaml b/homeassistant/components/ai_task/services.yaml index eb26541a343..d55b0e60fac 100644 --- a/homeassistant/components/ai_task/services.yaml +++ b/homeassistant/components/ai_task/services.yaml @@ -23,6 +23,3 @@ generate_data: example: '{ "name": { "selector": { "text": }, "description": "Name of the user", "required": "True" } } }, "age": { "selector": { "number": }, "description": "Age of the user" } }' selector: object: - filter: - supported_features: - - ai_task.AITaskEntityFeature.GENERATE_STRUCTURED_DATA diff --git a/homeassistant/components/ai_task/task.py b/homeassistant/components/ai_task/task.py index b401091c6e4..b6defbfad31 100644 --- a/homeassistant/components/ai_task/task.py +++ b/homeassistant/components/ai_task/task.py @@ -37,15 +37,6 @@ async def async_generate_data( f"AI Task entity {entity_id} does not support generating data" ) - if structure is not None: - if ( - AITaskEntityFeature.GENERATE_STRUCTURED_DATA - not in entity.supported_features - ): - raise HomeAssistantError( - f"AI Task entity {entity_id} does not support generating structured data" - ) - return await entity.internal_async_generate_data( GenDataTask( name=task_name, diff --git a/tests/components/ai_task/conftest.py b/tests/components/ai_task/conftest.py index 30f11b91555..e80e70ddaed 100644 --- a/tests/components/ai_task/conftest.py +++ b/tests/components/ai_task/conftest.py @@ -35,9 +35,7 @@ class MockAITaskEntity(AITaskEntity): """Mock AI Task entity for testing.""" _attr_name = "Test Task Entity" - _attr_supported_features = ( - AITaskEntityFeature.GENERATE_DATA | AITaskEntityFeature.GENERATE_STRUCTURED_DATA - ) + _attr_supported_features = AITaskEntityFeature.GENERATE_DATA def __init__(self) -> None: """Initialize the mock entity.""" diff --git a/tests/components/ai_task/test_task.py b/tests/components/ai_task/test_task.py index a74c3f50647..bed760c8a1d 100644 --- a/tests/components/ai_task/test_task.py +++ b/tests/components/ai_task/test_task.py @@ -3,14 +3,13 @@ from freezegun import freeze_time import pytest from syrupy.assertion import SnapshotAssertion -import voluptuous as vol from homeassistant.components.ai_task import AITaskEntityFeature, async_generate_data from homeassistant.components.conversation import async_get_chat_log from homeassistant.const import STATE_UNKNOWN from homeassistant.core import HomeAssistant from homeassistant.exceptions import HomeAssistantError -from homeassistant.helpers import chat_session, selector +from homeassistant.helpers import chat_session from .conftest import TEST_ENTITY_ID, MockAITaskEntity @@ -128,24 +127,3 @@ async def test_run_data_task_updates_chat_log( async_get_chat_log(hass, session) as chat_log, ): assert chat_log.content == snapshot - - -async def test_run_task_structure_unsupported_feature( - hass: HomeAssistant, - init_components: None, - mock_ai_task_entity: MockAITaskEntity, -) -> None: - """Test running a task with an unknown entity.""" - - mock_ai_task_entity.supported_features = AITaskEntityFeature.GENERATE_DATA - with pytest.raises( - HomeAssistantError, - match="AI Task entity ai_task.test_task_entity does not support generating structured data", - ): - await async_generate_data( - hass, - task_name="Test Task", - instructions="Test prompt", - entity_id=TEST_ENTITY_ID, - structure=vol.Schema({vol.Required("name"): selector.TextSelector()}), - )