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

completed workflow engine main logic

Browse Source
This commit is contained in:
takatost
2024-03-06 17:43:42 +08:00
parent dd50deaa43
commit 7d28fe8ea5
10 changed files with 366 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
api/core/callback_handler/workflow_event_trigger_callback.py
View File

@ -1,5 +1,5 @@
from core.app.app_queue_manager import AppQueueManager, PublishFrom
from core.workflow.callbacks.base_callback import BaseWorkflowCallback
from core.workflow.callbacks.base_workflow_callback import BaseWorkflowCallback
from models.workflow import WorkflowNodeExecution, WorkflowRun
@ -43,3 +43,12 @@ class WorkflowEventTriggerCallback(BaseWorkflowCallback):
workflow_node_execution_id=workflow_node_execution.id,
pub_from=PublishFrom.TASK_PIPELINE
)
def on_text_chunk(self, text: str) -> None:
"""
Publish text chunk
"""
self._queue_manager.publish_text_chunk(
text=text,
pub_from=PublishFrom.TASK_PIPELINE
)