refactor(message.py): use ActionType instead of Enum

src/judas_protocol/__init__.py

@@ -1,6 +1,6 @@
 __version__: str = "0.1.0"
 
 from .message import Message
-from .types import Category, ControlAction
+from .types import ActionType, Category, ControlAction
 
-__all__ = ["__version__", "Message", "Category", "ControlAction"]
+__all__ = ["__version__", "Message", "ActionType", "Category", "ControlAction"]

src/judas_protocol/message.py

@@ -3,10 +3,9 @@ from __future__ import annotations
 
 import json
 import uuid
-from enum import Enum
 from typing import Any, override
 
-from judas_protocol.types import Category, ControlAction
+from judas_protocol.types import ActionType, Category, ControlAction
 
 
 class Message:
@@ -16,7 +15,7 @@ class Message:
         self,
         id_: str | None,
         category: Category,
-        action: Enum,
+        action: ActionType,
         payload: dict[str, Any] | None = None,
         *,
         acknowledged: bool = False,
@@ -27,14 +26,14 @@ class Message:
         Args:
             id_ (str | None): The ID of the message. If None, a new UUID will be generated.
             category (Category): The category of the message.
-            action (Enum): The action of the message.
+            action (ActionType): The action of the message.
             payload (dict[str, Any] | None): The payload of the message.
             acknowledged (bool): Whether the message has been acknowledged. Defaults to False.
             ack_required (bool): Whether the message requires an acknowledgment. Defaults to False.
         """
         self.id: str = id_ or str(uuid.uuid4())
         self.category: Category = category
-        self.action: Enum = action
+        self.action: ActionType = action
         self.payload: dict[str, Any] = payload or {}
 
         self.acknowledged: bool = acknowledged

src/judas_protocol/types.py

@@ -5,6 +5,8 @@ from __future__ import annotations
 
 from enum import Enum
 
+type ActionType = ControlAction
+
 
 class Category(str, Enum):
     CONTROL = "control"