home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-07-29 18:28:14 +02:00
Code Issues Packages Projects Releases Wiki Activity

Address comments

Browse Source
This commit is contained in:
Paulus Schoutsen
2025-06-24 00:55:42 +00:00
parent 33535ca27a
commit ae55288a26
3 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
homeassistant/components/ai_task/strings.json
View File

@ -2,7 +2,7 @@
"services": { "services": {
"generate_data": { "generate_data": {
"name": "Generate data", "name": "Generate data",
"description": "Use AI to run a task that generates data.", "description": "Uses AI to run a task that generates data.",
"fields": { "fields": {
"task_name": { "task_name": {
"name": "Task name", "name": "Task name",

1
homeassistant/components/ai_task/task.py
View File

@ -70,5 +70,6 @@ class GenDataTaskResult:
def as_dict(self) -> dict[str, Any]: def as_dict(self) -> dict[str, Any]:
"""Return result as a dict.""" """Return result as a dict."""
return { return {
"conversation_id": self.conversation_id,
"data": self.data, "data": self.data,
} }

3
tests/components/ai_task/test_task.py
View File

@ -71,6 +71,9 @@ async def test_run_task_preferred_entity(
instructions="Test prompt", instructions="Test prompt",
) )
assert result.data == "Mock result" assert result.data == "Mock result"
as_dict = result.as_dict()
assert as_dict["conversation_id"] == result.conversation_id
assert as_dict["data"] == "Mock result"
state = hass.states.get(TEST_ENTITY_ID) state = hass.states.get(TEST_ENTITY_ID)
assert state is not None assert state is not None
assert state.state != STATE_UNKNOWN assert state.state != STATE_UNKNOWN