fix: registered events before adding to agent

Author: yarikdevcom

agents-core/vision_agents/core/events/manager.py

@@ -409,7 +409,7 @@ def _prepare_event(self, event):
 
         # Validate event is registered (handles both BaseEvent and generated protobuf events)
         if hasattr(event, "type") and event.type in self._events:
-            # logger.info(f"Received event {_truncate_event_for_logging(event)}")
+            logger.debug(f"Received event {_truncate_event_for_logging(event)}")
             return event
         elif self._ignore_unknown_events:
             logger.debug(f"Event not registered {_truncate_event_for_logging(event)}")

agents-core/vision_agents/core/profiling/base.py

@@ -12,15 +12,16 @@ class Profiler:
     def __init__(self, output_path='./profile.html'):
         self.output_path = output_path
         self.events = EventManager()
+        self.events.register_events_from_module(events)
         self.profiler = pyinstrument.Profiler()
         self.events.subscribe(self.on_start)
         self.events.subscribe(self.on_finish)
 
-    def on_start(self, event: events.AgentInitEvent):
+    async def on_start(self, event: events.AgentInitEvent):
         logger.info("Profiler started.")
         self.profiler.start()
 
-    def on_finish(self, event: events.AgentFinishEvent):
+    async def on_finish(self, event: events.AgentFinishEvent):
         self.profiler.stop()
         logger.info(f"Profiler stopped. Time file saved at: {self.output_path}")
         with open(self.output_path, 'w') as f: