jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-10 03:19:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow an LLM to see script response values (#131683)

Browse Source
This commit is contained in:
Paulus Schoutsen
2024-11-27 00:51:21 -05:00
committed by GitHub
parent 46fe3dcbf1
commit 7e03100af2
2 changed files with 58 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
tests/helpers/test_llm.py
View File

@@ -656,7 +656,10 @@ async def test_script_tool(
"script": {
"test_script": {
"description": "This is a test script",
"sequence": [],
"sequence": [
{"variables": {"result": {"drinks": 2}}},
{"stop": True, "response_variable": "result"},
],
"fields": {
"beer": {"description": "Number of beers", "required": True},
"wine": {"selector": {"number": {"min": 0, "max": 3}}},
@@ -692,7 +695,7 @@ async def test_script_tool(
api = await llm.async_get_api(hass, "assist", llm_context)
tools = [tool for tool in api.tools if isinstance(tool, llm.ScriptTool)]
assert len(tools) == 1
assert len(tools) == 2
tool = tools[0]
assert tool.name == "test_script"
@@ -719,6 +722,7 @@ async def test_script_tool(
"script_with_no_fields": ("This is another test script", vol.Schema({})),
}
# Test script with response
tool_input = llm.ToolInput(
tool_name="test_script",
tool_args={
@@ -731,26 +735,56 @@ async def test_script_tool(
},
)
with patch("homeassistant.core.ServiceRegistry.async_call") as mock_service_call:
with patch(
"homeassistant.core.ServiceRegistry.async_call",
side_effect=hass.services.async_call,
) as mock_service_call:
response = await api.async_call_tool(tool_input)
mock_service_call.assert_awaited_once_with(
"script",
"turn_on",
"test_script",
{
"entity_id": "script.test_script",
"variables": {
"beer": "3",
"wine": 0,
"where": area.id,
"area_list": [area.id],
"floor": floor.floor_id,
"floor_list": [floor.floor_id],
},
"beer": "3",
"wine": 0,
"where": area.id,
"area_list": [area.id],
"floor": floor.floor_id,
"floor_list": [floor.floor_id],
},
context=context,
blocking=True,
return_response=True,
)
assert response == {"success": True}
assert response == {
"success": True,
"result": {"drinks": 2},
}
# Test script with no response
tool_input = llm.ToolInput(
tool_name="script_with_no_fields",
tool_args={},
)
with patch(
"homeassistant.core.ServiceRegistry.async_call",
side_effect=hass.services.async_call,
) as mock_service_call:
response = await api.async_call_tool(tool_input)
mock_service_call.assert_awaited_once_with(
"script",
"script_with_no_fields",
{},
context=context,
blocking=True,
return_response=True,
)
assert response == {
"success": True,
"result": {},
}
# Test reload script with new parameters
config = {
@@ -782,7 +816,7 @@ async def test_script_tool(
api = await llm.async_get_api(hass, "assist", llm_context)
tools = [tool for tool in api.tools if isinstance(tool, llm.ScriptTool)]
assert len(tools) == 1
assert len(tools) == 2
tool = tools[0]
assert tool.name == "test_script"