| 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 | ||||
| :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: |