Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

pull/91390/head^2
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_FIELDS = {
vol.Required("conversation_engine"): str,
vol.Optional("conversation_engine", default=None): vol.Any(str, None),
vol.Required("language"): str,
vol.Required("name"): str,
vol.Required("stt_engine"): str,
vol.Required("tts_engine"): str,
vol.Optional("stt_engine", default=None): vol.Any(str, None),
vol.Optional("tts_engine", default=None): vol.Any(str, None),
}
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",
},
{
"conversation_engine": "conversation_engine_3",
"conversation_engine": None,
"language": "language_3",
"name": "name_3",
"stt_engine": "stt_engine_3",
"tts_engine": "tts_engine_3",
"stt_engine": None,
"tts_engine": None,
},
]
pipeline_ids = []
@ -91,12 +91,12 @@ async def test_loading_datasets_from_storage(
"tts_engine": "tts_engine_2",
},
{
"conversation_engine": "conversation_engine_3",
"conversation_engine": None,
"id": "01GX8ZWBAQSV1HP3WGJPFWEJ8J",
"language": "language_3",
"name": "name_3",
"stt_engine": "stt_engine_3",
"tts_engine": "tts_engine_3",
"stt_engine": None,
"tts_engine": None,
},
],
"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",
)
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(
hass: HomeAssistant, hass_ws_client: WebSocketGenerator, init_components
@ -808,6 +837,38 @@ async def test_update_pipeline(
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(
hass: HomeAssistant, hass_ws_client: WebSocketGenerator, init_components