whoops, missed a snapshot type

Author: saponifi3d

src/sentry/workflow_engine/models/action.py

@@ -3,7 +3,7 @@
 import builtins
 import logging
 from enum import StrEnum
-from typing import TYPE_CHECKING, Any, ClassVar
+from typing import TYPE_CHECKING, ClassVar, TypedDict
 
 from django.db import models
 from django.db.models import Q
@@ -30,6 +30,11 @@
 logger = logging.getLogger(__name__)
 
 
+class ActionSnapshot(TypedDict):
+    id: int
+    type: Action.Type
+
+
 class ActionManager(BaseManager["Action"]):
     def get_queryset(self) -> BaseQuerySet[Action]:
         return (
@@ -112,10 +117,10 @@ class Meta:
             ),
         ]
 
-    def get_snapshot(self) -> dict[str, Any]:
+    def get_snapshot(self) -> ActionSnapshot:
         return {
             "id": self.id,
-            "type": self.type,
+            "type": Action.Type(self.type),
         }
 
     def get_handler(self) -> builtins.type[ActionHandler]: