| session.commit() | session.commit() | ||||
| yield workflow_finish_resp | 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() | yield self._message_end_to_stream_response() | ||||
| break | break |