mirror of
https://github.com/home-assistant/core.git
synced 2025-04-24 01:08:12 +00:00
Add language/country args for preferred lang
This commit is contained in:
parent
c2932debde
commit
6a3ea03689
@ -340,7 +340,11 @@ def _get_unmatched_slots(
|
||||
|
||||
|
||||
@websocket_api.websocket_command(
|
||||
{vol.Required("type"): "conversation/agent/homeassistant/language_scores"}
|
||||
{
|
||||
vol.Required("type"): "conversation/agent/homeassistant/language_scores",
|
||||
vol.Optional("language"): str,
|
||||
vol.Optional("country"): str,
|
||||
}
|
||||
)
|
||||
@websocket_api.async_response
|
||||
async def websocket_hass_agent_language_scores(
|
||||
@ -349,8 +353,19 @@ async def websocket_hass_agent_language_scores(
|
||||
msg: dict[str, Any],
|
||||
) -> None:
|
||||
"""Get support scores per language."""
|
||||
language = msg.get("language", hass.config.language)
|
||||
country = msg.get("country", hass.config.country)
|
||||
|
||||
scores = await hass.async_add_executor_job(get_language_scores)
|
||||
result = {lang_key: asdict(lang_scores) for lang_key, lang_scores in scores.items()}
|
||||
matching_langs = language_util.matches(language, scores.keys(), country=country)
|
||||
preferred_lang = matching_langs[0] if matching_langs else language
|
||||
result = {
|
||||
"languages": {
|
||||
lang_key: asdict(lang_scores) for lang_key, lang_scores in scores.items()
|
||||
},
|
||||
"preferred_language": preferred_lang,
|
||||
}
|
||||
|
||||
connection.send_result(msg["id"], result)
|
||||
|
||||
|
||||
|
@ -539,10 +539,7 @@ async def test_ws_hass_agent_debug_sentence_trigger(
|
||||
|
||||
|
||||
async def test_ws_hass_language_scores(
|
||||
hass: HomeAssistant,
|
||||
init_components,
|
||||
hass_ws_client: WebSocketGenerator,
|
||||
snapshot: SnapshotAssertion,
|
||||
hass: HomeAssistant, init_components, hass_ws_client: WebSocketGenerator
|
||||
) -> None:
|
||||
"""Test getting language support scores."""
|
||||
client = await hass_ws_client(hass)
|
||||
@ -552,8 +549,47 @@ async def test_ws_hass_language_scores(
|
||||
)
|
||||
|
||||
msg = await client.receive_json()
|
||||
|
||||
assert msg["success"]
|
||||
|
||||
# Sanity check
|
||||
assert msg["result"]["en_US"] == {"cloud": 3, "focused_local": 2, "full_local": 3}
|
||||
result = msg["result"]
|
||||
assert result["languages"]["en-US"] == {
|
||||
"cloud": 3,
|
||||
"focused_local": 2,
|
||||
"full_local": 3,
|
||||
}
|
||||
|
||||
|
||||
async def test_ws_hass_language_scores_with_filter(
|
||||
hass: HomeAssistant, init_components, hass_ws_client: WebSocketGenerator
|
||||
) -> None:
|
||||
"""Test getting language support scores with language/country filter."""
|
||||
client = await hass_ws_client(hass)
|
||||
|
||||
# Language filter
|
||||
await client.send_json_auto_id(
|
||||
{"type": "conversation/agent/homeassistant/language_scores", "language": "de"}
|
||||
)
|
||||
|
||||
msg = await client.receive_json()
|
||||
assert msg["success"]
|
||||
|
||||
# German should be preferred
|
||||
result = msg["result"]
|
||||
assert result["preferred_language"] == "de-DE"
|
||||
|
||||
# Language/country filter
|
||||
await client.send_json_auto_id(
|
||||
{
|
||||
"type": "conversation/agent/homeassistant/language_scores",
|
||||
"language": "en",
|
||||
"country": "GB",
|
||||
}
|
||||
)
|
||||
|
||||
msg = await client.receive_json()
|
||||
assert msg["success"]
|
||||
|
||||
# GB English should be preferred
|
||||
result = msg["result"]
|
||||
assert result["preferred_language"] == "en-GB"
|
||||
|
Loading…
x
Reference in New Issue
Block a user