From 33535ca27a11449eb19d6020aebefe99136349e4 Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <balloob@gmail.com>
Date: Mon, 23 Jun 2025 11:42:30 -0400
Subject: [PATCH] Remove the dict wrapper from text results

---
 tests/components/ai_task/conftest.py    | 2 +-
 tests/components/ai_task/test_entity.py | 2 +-
 tests/components/ai_task/test_init.py   | 2 +-
 tests/components/ai_task/test_task.py   | 4 ++--
 4 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/tests/components/ai_task/conftest.py b/tests/components/ai_task/conftest.py
index 643806c4431..7efbd1ffcdb 100644
--- a/tests/components/ai_task/conftest.py
+++ b/tests/components/ai_task/conftest.py
@@ -50,7 +50,7 @@ class MockAITaskEntity(AITaskEntity):
         )
         return GenDataTaskResult(
             conversation_id=chat_log.conversation_id,
-            data={"text": "Mock result"},
+            data="Mock result",
         )
 
 
diff --git a/tests/components/ai_task/test_entity.py b/tests/components/ai_task/test_entity.py
index 9f9d089dbf9..3ed1c393588 100644
--- a/tests/components/ai_task/test_entity.py
+++ b/tests/components/ai_task/test_entity.py
@@ -29,7 +29,7 @@ async def test_state_generate_data(
         entity_id=TEST_ENTITY_ID,
         instructions="Test prompt",
     )
-    assert result.data["text"] == "Mock result"
+    assert result.data == "Mock result"
 
     entity = hass.states.get(TEST_ENTITY_ID)
     assert entity.state == "2025-06-08T16:28:13+00:00"
diff --git a/tests/components/ai_task/test_init.py b/tests/components/ai_task/test_init.py
index a03284db2a9..fdfaaccd0a4 100644
--- a/tests/components/ai_task/test_init.py
+++ b/tests/components/ai_task/test_init.py
@@ -81,4 +81,4 @@ async def test_generate_data_service(
         return_response=True,
     )
 
-    assert result["data"]["text"] == "Mock result"
+    assert result["data"] == "Mock result"
diff --git a/tests/components/ai_task/test_task.py b/tests/components/ai_task/test_task.py
index 356c6df30d8..c287cf91637 100644
--- a/tests/components/ai_task/test_task.py
+++ b/tests/components/ai_task/test_task.py
@@ -70,7 +70,7 @@ async def test_run_task_preferred_entity(
         task_name="Test Task",
         instructions="Test prompt",
     )
-    assert result.data["text"] == "Mock result"
+    assert result.data == "Mock result"
     state = hass.states.get(TEST_ENTITY_ID)
     assert state is not None
     assert state.state != STATE_UNKNOWN
@@ -117,7 +117,7 @@ async def test_run_data_task_updates_chat_log(
         entity_id=TEST_ENTITY_ID,
         instructions="Test prompt",
     )
-    assert result.data["text"] == "Mock result"
+    assert result.data == "Mock result"
 
     with (
         chat_session.async_get_chat_session(hass, result.conversation_id) as session,