Fix gen property

home-assistant-libs · Dec 1, 2023 · babc7a4 · babc7a4
1 parent f206619
commit babc7a4
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/aioshelly/rpc_device/device.py b/aioshelly/rpc_device/device.py
@@ -373,8 +373,11 @@ def shelly(self) -> dict[str, Any]:
 
     @property
     def gen(self) -> int:
-        """Device generation: GEN2 - RPC."""
-        return 2
+        """Device generation: GEN2/3 - RPC."""
+        if self._shelly is None:
+            raise NotInitialized
+
+        return cast(int, self._shelly["gen"])
 
     @property
     def firmware_version(self) -> str: