jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 13:17:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make tts/stt/conversation optional on pipeline (#91555)

Browse Source
This commit is contained in:
Bram Kragten 2023-04-17 20:54:04 +02:00 committed by GitHub
parent afc9e4303a
commit e3ff7d048a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 70 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
homeassistant/components/assist_pipeline/pipeline.py
View File

@ -40,11 +40,11 @@ STORAGE_KEY = f"{DOMAIN}.pipelines"
STORAGE_VERSION = 1 STORAGE_VERSION = 1
STORAGE_FIELDS = { STORAGE_FIELDS = {
vol.Required("conversation_engine"): str, vol.Optional("conversation_engine", default=None): vol.Any(str, None),
vol.Required("language"): str, vol.Required("language"): str,
vol.Required("name"): str, vol.Required("name"): str,
vol.Required("stt_engine"): str, vol.Optional("stt_engine", default=None): vol.Any(str, None),
vol.Required("tts_engine"): str, vol.Optional("tts_engine", default=None): vol.Any(str, None),
} }
STORED_PIPELINE_RUNS = 10 STORED_PIPELINE_RUNS = 10

12
tests/components/assist_pipeline/test_pipeline.py
View File

@ -34,11 +34,11 @@ async def test_load_datasets(hass: HomeAssistant, init_components) -> None:
"tts_engine": "tts_engine_2", "tts_engine": "tts_engine_2",
}, },
{ {
"conversation_engine": "conversation_engine_3", "conversation_engine": None,
"language": "language_3", "language": "language_3",
"name": "name_3", "name": "name_3",
"stt_engine": "stt_engine_3", "stt_engine": None,
"tts_engine": "tts_engine_3", "tts_engine": None,
}, },
] ]
pipeline_ids = [] pipeline_ids = []
@ -91,12 +91,12 @@ async def test_loading_datasets_from_storage(
"tts_engine": "tts_engine_2", "tts_engine": "tts_engine_2",
}, },
{ {
"conversation_engine": "conversation_engine_3", "conversation_engine": None,
"id": "01GX8ZWBAQSV1HP3WGJPFWEJ8J", "id": "01GX8ZWBAQSV1HP3WGJPFWEJ8J",
"language": "language_3", "language": "language_3",
"name": "name_3", "name": "name_3",
"stt_engine": "stt_engine_3", "stt_engine": None,
"tts_engine": "tts_engine_3", "tts_engine": None,
}, },
], ],
"preferred_item": "01GX8ZWBAQYWNB1XV3EXEZ75DY", "preferred_item": "01GX8ZWBAQYWNB1XV3EXEZ75DY",

61
tests/components/assist_pipeline/test_websocket.py
View File

@ -613,6 +613,35 @@ async def test_add_pipeline(
tts_engine="test_tts_engine", tts_engine="test_tts_engine",
) )
await client.send_json_auto_id(
{
"type": "assist_pipeline/pipeline/create",
"language": "test_language",
"name": "test_name",
}
)
msg = await client.receive_json()
assert msg["success"]
assert msg["result"] == {
"conversation_engine": None,
"id": ANY,
"language": "test_language",
"name": "test_name",
"stt_engine": None,
"tts_engine": None,
}
assert len(pipeline_store.data) == 2
pipeline = pipeline_store.data[msg["result"]["id"]]
assert pipeline == Pipeline(
conversation_engine=None,
id=msg["result"]["id"],
language="test_language",
name="test_name",
stt_engine=None,
tts_engine=None,
)
async def test_delete_pipeline( async def test_delete_pipeline(
hass: HomeAssistant, hass_ws_client: WebSocketGenerator, init_components hass: HomeAssistant, hass_ws_client: WebSocketGenerator, init_components
@ -808,6 +837,38 @@ async def test_update_pipeline(
tts_engine="new_tts_engine", tts_engine="new_tts_engine",
) )
await client.send_json_auto_id(
{
"type": "assist_pipeline/pipeline/update",
"conversation_engine": None,
"language": "new_language",
"name": "new_name",
"pipeline_id": pipeline_id,
"stt_engine": None,
"tts_engine": None,
}
)
msg = await client.receive_json()
assert msg["success"]
assert msg["result"] == {
"conversation_engine": None,
"id": pipeline_id,
"language": "new_language",
"name": "new_name",
"stt_engine": None,
"tts_engine": None,
}
pipeline = pipeline_store.data[pipeline_id]
assert pipeline == Pipeline(
conversation_engine=None,
id=pipeline_id,
language="new_language",
name="new_name",
stt_engine=None,
tts_engine=None,
)
async def test_set_preferred_pipeline( async def test_set_preferred_pipeline(
hass: HomeAssistant, hass_ws_client: WebSocketGenerator, init_components hass: HomeAssistant, hass_ws_client: WebSocketGenerator, init_components