remove return

Author: willydouhard

backend/chainlit/llama_index/callbacks.py

@@ -198,8 +198,7 @@ def on_event_end(
         else:
             step.output = payload
             self.context.loop.create_task(step.update())
-            return
-
+ 
         self.steps.pop(event_id, None)
 
     def _noop(self, *args, **kwargs):