jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

unify moderation and annotation's response behavior in message log of chatflow app with other types of app (#14800)

Browse source
This commit is contained in:
Qun 2025-03-04 09:09:32 +08:00 • committed by GitHub
commit f0fb38fed4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
api/core/app/apps/advanced_chat/generate_task_pipeline.py
View file

@ -582,6 +582,15 @@ class AdvancedChatAppGenerateTaskPipeline:
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