youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-06 10:28:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: improve file preview in chat messages by yielding events in task pipeline

Browse Source
This commit is contained in:
Novice
2026-03-11 16:30:04 +08:00
parent 17c9b4dd85
commit f0c82f28cb
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
api/core/app/apps/advanced_chat/generate_task_pipeline.py
View File

@ -692,8 +692,10 @@ class AdvancedChatAppGenerateTaskPipeline(GraphRuntimeStateSupport):
graph_runtime_state=validated_state, graph_runtime_state=validated_state,
) )
yield from self._handle_advanced_chat_message_end_event(
QueueAdvancedChatMessageEndEvent(), graph_runtime_state=validated_state
)
yield workflow_finish_resp yield workflow_finish_resp
self._base_task_pipeline.queue_manager.publish(QueueAdvancedChatMessageEndEvent(), PublishFrom.TASK_PIPELINE)
def _handle_workflow_partial_success_event( def _handle_workflow_partial_success_event(
self, self,
@ -714,6 +716,9 @@ class AdvancedChatAppGenerateTaskPipeline(GraphRuntimeStateSupport):
exceptions_count=event.exceptions_count, exceptions_count=event.exceptions_count,
) )
yield from self._handle_advanced_chat_message_end_event(
QueueAdvancedChatMessageEndEvent(), graph_runtime_state=validated_state
)
yield workflow_finish_resp yield workflow_finish_resp
def _handle_workflow_paused_event( def _handle_workflow_paused_event(
@ -1033,6 +1038,14 @@ class AdvancedChatAppGenerateTaskPipeline(GraphRuntimeStateSupport):
yield from self._handle_workflow_paused_event(event) yield from self._handle_workflow_paused_event(event)
break break
case QueueWorkflowSucceededEvent():
yield from self._handle_workflow_succeeded_event(event, trace_manager=trace_manager)
break
case QueueWorkflowPartialSuccessEvent():
yield from self._handle_workflow_partial_success_event(event, trace_manager=trace_manager)
break
case QueueStopEvent(): case QueueStopEvent():
yield from self._handle_stop_event(event, graph_runtime_state=None, trace_manager=trace_manager) yield from self._handle_stop_event(event, graph_runtime_state=None, trace_manager=trace_manager)
break break