fix(StateBasedAgent): catch aggregator execptions

Author: boczekbartek

src/rai_core/rai/agents/base_state_based_agent.py

@@ -136,7 +136,9 @@ def process_aggregator(
                 f"Running aggregator: {aggregator}(source={source}) on {len(aggregator.get_buffer())} messages"
             )
             ts = time.perf_counter()
+
             output = aggregator.get()
+
             self.logger.debug(
                 f'Aggregator "{aggregator}(source={source})" done in {time.perf_counter() - ts:.2f}s'
             )
@@ -150,9 +152,15 @@ def process_aggregator(
                     futures.append(future)
 
             for future in as_completed(futures):
-                source, output = future.result()
-                if output is not None:
-                    self._aggregation_results[source] = output
+                try:
+                    source, output = future.result()
+                except Exception as e:
+                    self.logger.error(f"Aggregator crashed: {e}")
+                    continue
+
+                if output is None:
+                    continue
+                self._aggregation_results[source] = output
 
     def stop(self):
         """Stop the agent's execution loop."""
@@ -164,7 +172,8 @@ def stop(self):
         if self._aggregation_thread is not None:
             self._aggregation_thread.join()
             self._aggregation_thread = None
+        self._stop_event.clear()
         for callback_id in self._registered_callbacks:
             self._connector.unregister_callback(callback_id)
-        self._stop_event.clear()
+        self._connector.shutdown()
         self.logger.info("Agent stopped")