refactor: convert to TaskType for better processing

silverback/middlewares.py

@@ -56,18 +56,24 @@ def _create_label(self, message: TaskiqMessage) -> str:
             return message.task_name
 
     def pre_execute(self, message: TaskiqMessage) -> TaskiqMessage:
-        task_type = message.labels.pop("task_type", "<unknown>")
+        if not (task_type := message.labels.pop("task_type")):
+            return message  # Not a silverback task
 
-        # NOTE: Don't compare `str` to `TaskType` using `is`
-        if task_type == TaskType.NEW_BLOCKS:
+        try:
+            task_type = TaskType(task_type)
+        except ValueError:
+            return message  # Not a silverback task
+
+        # Add extra labels for our task to see what their source was
+        if task_type is TaskType.NEW_BLOCKS:
             # NOTE: Necessary because we don't know the exact block class
             message.args[0] = self.provider.network.ecosystem.decode_block(
                 hexbytes_dict(message.args[0])
             )
             message.labels["block_number"] = str(message.args[0].number)
             message.labels["block_hash"] = message.args[0].hash.hex()
 
-        elif task_type == TaskType.EVENT_LOG:
+        elif task_type is TaskType.EVENT_LOG:
             # NOTE: Just in case the user doesn't specify type as `ContractLog`
             message.args[0] = ContractLog.model_validate(message.args[0])
             message.labels["block_number"] = str(message.args[0].block_number)