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

using message_file event instead of message event (#22988)

Browse source
This commit is contained in:
znn 2025-08-08 07:20:20 +05:30 • committed by GitHub
commit 084dcd1a50
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
api/core/app/task_pipeline/message_cycle_manager.py
View file

@ -23,6 +23,7 @@ from core.app.entities.task_entities import (
MessageFileStreamResponse, MessageFileStreamResponse,
MessageReplaceStreamResponse, MessageReplaceStreamResponse,
MessageStreamResponse, MessageStreamResponse,
StreamEvent,
WorkflowTaskState, WorkflowTaskState,
) )
from core.llm_generator.llm_generator import LLMGenerator from core.llm_generator.llm_generator import LLMGenerator
@ -180,11 +181,15 @@ class MessageCycleManager:
:param message_id: message id :param message_id: message id
:return: :return:
""" """
message_file = db.session.query(MessageFile).filter(MessageFile.id == message_id).first()
event_type = StreamEvent.MESSAGE_FILE if message_file else StreamEvent.MESSAGE
return MessageStreamResponse( return MessageStreamResponse(
task_id=self._application_generate_entity.task_id, task_id=self._application_generate_entity.task_id,
id=message_id, id=message_id,
answer=answer, answer=answer,
from_variable_selector=from_variable_selector, from_variable_selector=from_variable_selector,
event=event_type,
) )
def message_replace_to_stream_response(self, answer: str, reason: str = "") -> MessageReplaceStreamResponse: def message_replace_to_stream_response(self, answer: str, reason: str = "") -> MessageReplaceStreamResponse: