[FIX] Type Alias error

custom_components/mcp_bridge/__init__.py

@@ -1,20 +1,11 @@
-"""MCP Bridge integration for Home Assistant.
+"""MCP Bridge integration for Home Assistant."""
-
-This integration provides services to expose filtered entities and scripts
-to MCP (Model Context Protocol) servers for AI agent control.
-"""
 from __future__ import annotations
 
 import logging
 from typing import Any
 
 from homeassistant.const import CONF_ENTITY_ID
-from homeassistant.core import (
+from homeassistant.core import HomeAssistant, ServiceCall, SupportsResponse
-    HomeAssistant,
-    ServiceCall,
-    ServiceResponse,
-    SupportsResponse,
-)
 from homeassistant.helpers import entity_registry as er
 from homeassistant.helpers import area_registry as ar
 from homeassistant.helpers.typing import ConfigType
@@ -26,15 +17,13 @@ MCP_ACCESSIBLE_LABEL = "mcp_accessible"
 
 
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
-    """Set up the MCP Bridge integration."""
     _LOGGER.info("Setting up MCP Bridge integration")
 
-    async def get_exposed_entities(call: ServiceCall) -> ServiceResponse:
+    async def get_exposed_entities(call: ServiceCall) -> dict[str, Any]:
-        """Return entities exposed to conversation."""
         entity_reg = er.async_get(hass)
         area_reg = ar.async_get(hass)
 
-        exposed_entities: list[dict[str, Any]] = []
+        entities: list[dict[str, Any]] = []
 
         for state in hass.states.async_all():
             entity_entry = entity_reg.async_get(state.entity_id)
@@ -52,7 +41,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
                 if area:
                     area_name = area.name
 
-            exposed_entities.append(
+            entities.append(
                 {
                     "entity_id": state.entity_id,
                     "state": state.state,
@@ -71,17 +60,14 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
                 }
             )
 
-        return ServiceResponse(
+        return {
-            {
+            "entities": entities,
-                "entities": exposed_entities,
+            "count": len(entities),
-                "count": len(exposed_entities),
+        }
-            }
-        )
 
-    async def get_exposed_scripts(call: ServiceCall) -> ServiceResponse:
+    async def get_exposed_scripts(call: ServiceCall) -> dict[str, Any]:
-        """Return scripts marked as MCP-accessible."""
         entity_reg = er.async_get(hass)
-        exposed_scripts: list[dict[str, Any]] = []
+        scripts: list[dict[str, Any]] = []
 
         for entity_id in hass.states.async_entity_ids("script"):
             entity_entry = entity_reg.async_get(entity_id)
@@ -95,7 +81,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
             if not state:
                 continue
 
-            exposed_scripts.append(
+            scripts.append(
                 {
                     "entity_id": entity_id,
                     "friendly_name": state.attributes.get(
@@ -106,32 +92,22 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
                 }
             )
 
-        return ServiceResponse(
+        return {
-            {
+            "scripts": scripts,
-                "scripts": exposed_scripts,
+            "count": len(scripts),
-                "count": len(exposed_scripts),
+        }
-            }
-        )
 
-    async def get_entity_metadata(call: ServiceCall) -> ServiceResponse:
+    async def get_entity_metadata(call: ServiceCall) -> dict[str, Any]:
-        """Return detailed metadata for a single entity."""
         entity_id = call.data.get(CONF_ENTITY_ID)
-
         if not entity_id:
-            return ServiceResponse(
+            return {"error": "entity_id is required"}
-                {"error": "entity_id is required"},
-                success=False,
-            )
 
         entity_reg = er.async_get(hass)
         area_reg = ar.async_get(hass)
 
         state = hass.states.get(entity_id)
         if not state:
-            return ServiceResponse(
+            return {"error": f"Entity {entity_id} not found"}
-                {"error": f"Entity {entity_id} not found"},
-                success=False,
-            )
 
         entity_entry = entity_reg.async_get(entity_id)
 
@@ -141,7 +117,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
             if area:
                 area_name = area.name
 
-        metadata: dict[str, Any] = {
+        data: dict[str, Any] = {
             "entity_id": entity_id,
             "state": state.state,
             "friendly_name": state.attributes.get(
@@ -159,12 +135,12 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
         }
 
         if entity_entry:
-            metadata["labels"] = list(entity_entry.labels)
+            data["labels"] = list(entity_entry.labels)
-            metadata["is_exposed_to_conversation"] = entity_entry.options.get(
+            data["is_exposed_to_conversation"] = entity_entry.options.get(
                 "conversation", {}
             ).get("should_expose", False)
 
-        return ServiceResponse(metadata)
+        return data
 
     hass.services.async_register(
         DOMAIN,