Fix register callback

nebula/core/engine.py

@@ -158,14 +158,10 @@ def __init__(
         )
 
         # Register additional callbacks
-        self._event_manager.register_event(
-            (
-                nebula_pb2.FederationMessage,
-                nebula_pb2.FederationMessage.Action.REPUTATION,
-            ),
+        self._event_manager.register_callback(
             self._reputation_callback,
+            # ... add more callbacks here
         )
-        # ... add more callbacks here
 
     @property
     def cm(self):

nebula/core/eventmanager.py

@@ -43,6 +43,14 @@ def _register_default_callbacks(self, default_callbacks):
             else:
                 raise ValueError("The callback must be decorated with @event_handler.")
 
+    def register_callback(self, callback):
+        """Registers a callback for an event."""
+        handler_info = getattr(callback, "_event_handler", None)
+        if handler_info is not None:
+            self.register_event(handler_info, callback)
+        else:
+            raise ValueError("The callback must be decorated with @event_handler.")
+
     def register_event(self, handler_info, callback):
         """Records a callback for a specific event."""
         if callable(callback):