| @@ -582,6 +582,15 @@ class AdvancedChatAppGenerateTaskPipeline: | |||
| session.commit() | |||
| yield workflow_finish_resp | |||
| elif event.stopped_by in ( | |||
| QueueStopEvent.StopBy.INPUT_MODERATION, | |||
| QueueStopEvent.StopBy.ANNOTATION_REPLY, | |||
| ): | |||
| # When hitting input-moderation or annotation-reply, the workflow will not start | |||
| with Session(db.engine, expire_on_commit=False) as session: | |||
| # Save message | |||
| self._save_message(session=session) | |||
| session.commit() | |||
| yield self._message_end_to_stream_response() | |||
| break | |||