| @@ -23,6 +23,7 @@ from core.app.entities.task_entities import ( | |||
| MessageFileStreamResponse, | |||
| MessageReplaceStreamResponse, | |||
| MessageStreamResponse, | |||
| StreamEvent, | |||
| WorkflowTaskState, | |||
| ) | |||
| from core.llm_generator.llm_generator import LLMGenerator | |||
| @@ -180,11 +181,15 @@ class MessageCycleManager: | |||
| :param message_id: message id | |||
| :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( | |||
| task_id=self._application_generate_entity.task_id, | |||
| id=message_id, | |||
| answer=answer, | |||
| from_variable_selector=from_variable_selector, | |||
| event=event_type, | |||
| ) | |||
| def message_replace_to_stream_response(self, answer: str, reason: str = "") -> MessageReplaceStreamResponse: | |||