fix: save logs when there is an error in workflow

11 months ago · 064bb1adae
parent 110bb5e5a5
commit 064bb1adae
1 changed files with 2 additions and 0 deletions
--- a/api/core/app/apps/advanced_chat/generate_task_pipeline.py
+++ b/api/core/app/apps/advanced_chat/generate_task_pipeline.py
@ -564,6 +564,8 @@ class AdvancedChatAppGenerateTaskPipeline:
                    err = self._base_task_pipeline._handle_error(
                        event=err_event, session=session, message_id=self._message_id
                    )
+                    self._save_message(session=session, graph_runtime_state=graph_runtime_state)
+                    session.commit()

                yield workflow_finish_resp
                yield self._base_task_pipeline._error_to_stream_response(err)