24
24
get_agent_manager ,
25
25
)
26
26
from .const import DATA_COMPONENT , DATA_DEFAULT_ENTITY
27
- from .default_agent import METADATA_CUSTOM_FILE , METADATA_CUSTOM_SENTENCE , DefaultAgent
27
+ from .default_agent import METADATA_CUSTOM_FILE , METADATA_CUSTOM_SENTENCE
28
28
from .entity import ConversationEntity
29
29
from .models import ConversationInput
30
30
@@ -162,8 +162,7 @@ async def websocket_list_sentences(
162
162
hass : HomeAssistant , connection : websocket_api .ActiveConnection , msg : dict
163
163
) -> None :
164
164
"""List custom registered sentences."""
165
- agent = hass .data .get (DATA_DEFAULT_ENTITY )
166
- assert isinstance (agent , DefaultAgent )
165
+ agent = hass .data [DATA_DEFAULT_ENTITY ]
167
166
168
167
sentences = []
169
168
for trigger_data in agent .trigger_sentences :
@@ -185,8 +184,7 @@ async def websocket_hass_agent_debug(
185
184
hass : HomeAssistant , connection : websocket_api .ActiveConnection , msg : dict
186
185
) -> None :
187
186
"""Return intents that would be matched by the default agent for a list of sentences."""
188
- agent = hass .data .get (DATA_DEFAULT_ENTITY )
189
- assert isinstance (agent , DefaultAgent )
187
+ agent = hass .data [DATA_DEFAULT_ENTITY ]
190
188
191
189
# Return results for each sentence in the same order as the input.
192
190
result_dicts : list [dict [str , Any ] | None ] = []
0 commit comments