add additional events to union

Author: dmulcahey

zha/websocket/server/api/model.py

@@ -7,6 +7,7 @@
 from zigpy.types.named import EUI64
 
 from zha.application.model import (
+    ConnectionLostEvent,
     DeviceFullyInitializedEvent,
     DeviceJoinedEvent,
     DeviceLeftEvent,
@@ -22,8 +23,20 @@
 from zha.application.platforms.events import EntityStateChangedEvent
 from zha.model import BaseModel, TypedBaseModel, as_tagged_union
 from zha.websocket.const import APICommands
-from zha.zigbee.cluster_handlers.model import ClusterInfo
-from zha.zigbee.model import ExtendedDeviceInfo, GroupInfo, ZHAEvent
+from zha.zigbee.cluster_handlers.model import (
+    ClusterAttributeUpdatedEvent,
+    ClusterBindEvent,
+    ClusterConfigureReportingEvent,
+    ClusterInfo,
+    LevelChangeEvent,
+)
+from zha.zigbee.cluster_handlers.security import ClusterHandlerStateChangedEvent
+from zha.zigbee.model import (
+    ClusterHandlerConfigurationComplete,
+    ExtendedDeviceInfo,
+    GroupInfo,
+    ZHAEvent,
+)
 
 
 class WebSocketCommand(TypedBaseModel):
@@ -114,12 +127,6 @@ class ErrorResponse(WebSocketCommandResponse):
     command: APICommands
 
 
-class DefaultResponse(WebSocketCommandResponse):
-    """Default command response."""
-
-    command: APICommands
-
-
 class PermitJoiningResponse(WebSocketCommandResponse):
     """Get devices response."""
 
@@ -264,6 +271,13 @@ def get_converted_state(self) -> State:
     | DeviceOfflineEvent
     | DeviceOnlineEvent
     | ZHAEvent
+    | ConnectionLostEvent
+    | ClusterAttributeUpdatedEvent
+    | ClusterBindEvent
+    | ClusterConfigureReportingEvent
+    | LevelChangeEvent
+    | ClusterHandlerStateChangedEvent
+    | ClusterHandlerConfigurationComplete
 )
 
 Messages = CommandResponses | Events