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

Remove the dict wrapper from text results

Browse Source
This commit is contained in:
Paulus Schoutsen
2025-06-23 11:42:30 -04:00
parent 455da5b94a
commit 33535ca27a
4 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/components/ai_task/conftest.py
View File

@ -50,7 +50,7 @@ class MockAITaskEntity(AITaskEntity):
)
return GenDataTaskResult(
conversation_id=chat_log.conversation_id,
data={"text": "Mock result"},
data="Mock result",
)

2
tests/components/ai_task/test_entity.py
View File

@ -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"

2
tests/components/ai_task/test_init.py
View File

@ -81,4 +81,4 @@ async def test_generate_data_service(
return_response=True,
)
assert result["data"]["text"] == "Mock result"
assert result["data"] == "Mock result"

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

@ -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,