home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-07-31 19:25:12 +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( return GenDataTaskResult(
conversation_id=chat_log.conversation_id, 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, entity_id=TEST_ENTITY_ID,
instructions="Test prompt", instructions="Test prompt",
) )
assert result.data["text"] == "Mock result" assert result.data == "Mock result"
entity = hass.states.get(TEST_ENTITY_ID) entity = hass.states.get(TEST_ENTITY_ID)
assert entity.state == "2025-06-08T16:28:13+00:00" 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, 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", task_name="Test Task",
instructions="Test prompt", instructions="Test prompt",
) )
assert result.data["text"] == "Mock result" assert result.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
@@ -117,7 +117,7 @@ async def test_run_data_task_updates_chat_log(
entity_id=TEST_ENTITY_ID, entity_id=TEST_ENTITY_ID,
instructions="Test prompt", instructions="Test prompt",
) )
assert result.data["text"] == "Mock result" assert result.data == "Mock result"
with ( with (
chat_session.async_get_chat_session(hass, result.conversation_id) as session, chat_session.async_get_chat_session(hass, result.conversation_id) as session,