mirror of
https://github.com/home-assistant/core.git
synced 2025-07-26 06:37:52 +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(
|
@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
|
@websocket_api.async_response
|
||||||
async def websocket_hass_agent_language_scores(
|
async def websocket_hass_agent_language_scores(
|
||||||
@ -349,8 +353,19 @@ async def websocket_hass_agent_language_scores(
|
|||||||
msg: dict[str, Any],
|
msg: dict[str, Any],
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Get support scores per language."""
|
"""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)
|
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)
|
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(
|
async def test_ws_hass_language_scores(
|
||||||
hass: HomeAssistant,
|
hass: HomeAssistant, init_components, hass_ws_client: WebSocketGenerator
|
||||||
init_components,
|
|
||||||
hass_ws_client: WebSocketGenerator,
|
|
||||||
snapshot: SnapshotAssertion,
|
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test getting language support scores."""
|
"""Test getting language support scores."""
|
||||||
client = await hass_ws_client(hass)
|
client = await hass_ws_client(hass)
|
||||||
@ -552,8 +549,47 @@ async def test_ws_hass_language_scores(
|
|||||||
)
|
)
|
||||||
|
|
||||||
msg = await client.receive_json()
|
msg = await client.receive_json()
|
||||||
|
|
||||||
assert msg["success"]
|
assert msg["success"]
|
||||||
|
|
||||||
# Sanity check
|
# 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