diff --git a/homeassistant/helpers/selector.py b/homeassistant/helpers/selector.py index 322cfe34042..438998aafb8 100644 --- a/homeassistant/helpers/selector.py +++ b/homeassistant/helpers/selector.py @@ -1020,11 +1020,15 @@ class MediaSelector(Selector[MediaSelectorConfig]): selector_type = "media" - CONFIG_SCHEMA = BASE_SELECTOR_CONFIG_SCHEMA + CONFIG_SCHEMA = BASE_SELECTOR_CONFIG_SCHEMA.extend( + { + vol.Optional("accept"): [str], + } + ) DATA_SCHEMA = vol.Schema( { - # Although marked as optional in frontend, this field is required - vol.Required("entity_id"): cv.entity_id_or_uuid, + # If accept is set, the entity_id field will not be present + vol.Optional("entity_id"): cv.entity_id_or_uuid, # Although marked as optional in frontend, this field is required vol.Required("media_content_id"): str, # Although marked as optional in frontend, this field is required diff --git a/tests/helpers/test_llm.py b/tests/helpers/test_llm.py index 98dee920bd9..b6894505534 100644 --- a/tests/helpers/test_llm.py +++ b/tests/helpers/test_llm.py @@ -1125,7 +1125,7 @@ async def test_selector_serializer( "media_content_type": {"type": "string"}, "metadata": {"type": "object", "additionalProperties": True}, }, - "required": ["entity_id", "media_content_id", "media_content_type"], + "required": ["media_content_id", "media_content_type"], } assert selector_serializer(selector.NumberSelector({"mode": "box"})) == { "type": "number" diff --git a/tests/helpers/test_selector.py b/tests/helpers/test_selector.py index 51ee467b029..97c02bdc837 100644 --- a/tests/helpers/test_selector.py +++ b/tests/helpers/test_selector.py @@ -817,6 +817,23 @@ def test_theme_selector_schema(schema, valid_selections, invalid_selections) -> ), (None, "abc", {}), ), + ( + { + "accept": ["image/*"], + }, + ( + { + "media_content_id": "abc", + "media_content_type": "def", + }, + { + "media_content_id": "abc", + "media_content_type": "def", + "metadata": {}, + }, + ), + (None, "abc", {}), + ), ], ) def test_media_selector_schema(schema, valid_selections, invalid_selections) -> None: