From 08358514b4884919b337b9ae5585086de07b55ca Mon Sep 17 00:00:00 2001 From: Erik Montnemery Date: Thu, 20 Feb 2025 09:14:17 +0100 Subject: [PATCH] Adjust config entry state checks in mcp_server (#138913) --- homeassistant/components/mcp_server/http.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/homeassistant/components/mcp_server/http.py b/homeassistant/components/mcp_server/http.py index 433d978cef7..bc8fdbd56c8 100644 --- a/homeassistant/components/mcp_server/http.py +++ b/homeassistant/components/mcp_server/http.py @@ -25,7 +25,6 @@ from mcp import types from homeassistant.components import conversation from homeassistant.components.http import KEY_HASS, HomeAssistantView -from homeassistant.config_entries import ConfigEntryState from homeassistant.const import CONF_LLM_HASS_API from homeassistant.core import HomeAssistant, callback from homeassistant.helpers import llm @@ -56,11 +55,9 @@ def async_get_config_entry(hass: HomeAssistant) -> MCPServerConfigEntry: Will raise an HTTP error if the expected configuration is not present. """ - config_entries: list[MCPServerConfigEntry] = [ - config_entry - for config_entry in hass.config_entries.async_entries(DOMAIN) - if config_entry.state == ConfigEntryState.LOADED - ] + config_entries: list[MCPServerConfigEntry] = ( + hass.config_entries.async_loaded_entries(DOMAIN) + ) if not config_entries: raise HTTPNotFound(text="Model Context Protocol server is not configured") if len(config_entries) > 1: