From ae55288a261553301df9299cb4086218971685a7 Mon Sep 17 00:00:00 2001 From: Paulus Schoutsen Date: Tue, 24 Jun 2025 00:55:42 +0000 Subject: [PATCH] Address comments --- homeassistant/components/ai_task/strings.json | 2 +- homeassistant/components/ai_task/task.py | 1 + tests/components/ai_task/test_task.py | 3 +++ 3 files changed, 5 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/ai_task/strings.json b/homeassistant/components/ai_task/strings.json index 2b93c3a9111..877174de681 100644 --- a/homeassistant/components/ai_task/strings.json +++ b/homeassistant/components/ai_task/strings.json @@ -2,7 +2,7 @@ "services": { "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": { "task_name": { "name": "Task name", diff --git a/homeassistant/components/ai_task/task.py b/homeassistant/components/ai_task/task.py index 0e5832be12c..2e546897602 100644 --- a/homeassistant/components/ai_task/task.py +++ b/homeassistant/components/ai_task/task.py @@ -70,5 +70,6 @@ class GenDataTaskResult: def as_dict(self) -> dict[str, Any]: """Return result as a dict.""" return { + "conversation_id": self.conversation_id, "data": self.data, } diff --git a/tests/components/ai_task/test_task.py b/tests/components/ai_task/test_task.py index c287cf91637..bed760c8a1d 100644 --- a/tests/components/ai_task/test_task.py +++ b/tests/components/ai_task/test_task.py @@ -71,6 +71,9 @@ async def test_run_task_preferred_entity( instructions="Test prompt", ) 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) assert state is not None assert state.state != STATE_UNKNOWN