From 55d27409b736928c04eb0ca4bea905a5422ed1b7 Mon Sep 17 00:00:00 2001
From: Feng Wang <wang6922@126.com>
Date: Mon, 23 Dec 2024 17:03:09 +0800
Subject: [PATCH] fix the deleted public data class

---
 custom_components/xiaomi_home/miot/miot_mips.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/custom_components/xiaomi_home/miot/miot_mips.py b/custom_components/xiaomi_home/miot/miot_mips.py
index 626b020..d2bebde 100644
--- a/custom_components/xiaomi_home/miot/miot_mips.py
+++ b/custom_components/xiaomi_home/miot/miot_mips.py
@@ -196,6 +196,17 @@ class MIoTDeviceState(Enum):
     OFFLINE = auto()
     ONLINE = auto()
 
+@dataclass
+class MipsDeviceState:
+    """MIoT Pub/Sub device state."""
+    did: str | None = None
+    """handler
+    str: did
+    MIoTDeviceState: online/offline/disable
+    Any: ctx
+    """
+    handler: Callable[[str, MIoTDeviceState, Any], None] | None = None
+    handler_ctx: Any = None
 
 class _MipsClient(ABC):
     """MIoT Pub/Sub client."""