fix: derive entity_id for helpers when not in WebSocket result

Some helper types (e.g., input_boolean) don't return entity_id in
the WebSocket create result. Derive it from helper_type + result id
so that wait_for_entity_registered and category application work.
Test updated to handle the same fallback.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: kingpanther13

src/ha_mcp/tools/tools_config_helpers.py

@@ -639,6 +639,9 @@ async def ha_config_set_helper(
                 if result.get("success"):
                     helper_data = result.get("result", {})
                     entity_id = helper_data.get("entity_id")
+                    # Some helper types don't return entity_id — derive from result id
+                    if not entity_id and helper_data.get("id"):
+                        entity_id = f"{helper_type}.{helper_data['id']}"
 
                     # Wait for entity to be properly registered before proceeding
                     wait_bool = coerce_bool_param(wait, "wait", default=True)

tests/src/e2e/workflows/categories/test_config_tool_categories.py

@@ -366,12 +366,16 @@ async def test_helper_set_category(self, mcp_client, cleanup_tracker):
             },
         )
         helper_data = assert_mcp_success(helper_result, "Create helper with category")
+        # entity_id may be None for some helper types — derive from helper_data.id
         entity_id = helper_data.get("entity_id")
-        assert entity_id, f"Missing entity_id: {helper_data}"
+        inner_data = helper_data.get("helper_data", {})
+        helper_id = inner_data.get("id")
+        if not entity_id and helper_id:
+            entity_id = f"input_boolean.{helper_id}"
+        assert entity_id, f"Missing entity_id and helper_data.id: {helper_data}"
         cleanup_tracker.track("input_boolean", entity_id)
 
         # Check category was applied (in helper_data sub-dict)
-        inner_data = helper_data.get("helper_data", {})
         assert inner_data.get("category") == category_id, (
             f"Category not set in helper response: {helper_data}"
         )