hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-25 02:33:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add from_variable_selector for stream chunk / message event (#8228)

Browse Source
This commit is contained in:
takatost
2024-09-10 22:15:50 +08:00
committed by GitHub
parent fdbbdb706f
commit cee0c51dbb
5 changed files with 23 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
api/core/app/task_pipeline/message_cycle_manage.py
View File

@@ -153,14 +153,21 @@ class MessageCycleManage:
return None
def _message_to_stream_response(self, answer: str, message_id: str) -> MessageStreamResponse:
def _message_to_stream_response(
self, answer: str, message_id: str, from_variable_selector: Optional[list[str]] = None
) -> MessageStreamResponse:
"""
Message to stream response.
:param answer: answer
:param message_id: message id
:return:
"""
return MessageStreamResponse(task_id=self._application_generate_entity.task_id, id=message_id, answer=answer)
return MessageStreamResponse(
task_id=self._application_generate_entity.task_id,
id=message_id,
answer=answer,
from_variable_selector=from_variable_selector,
)
def _message_replace_to_stream_response(self, answer: str) -> MessageReplaceStreamResponse:
"""