瀏覽代碼

fix(api): fix workflow execution

tags/2.0.0-beta.1
QuantumGhost 2 月之前
父節點
當前提交
0212f0de9f
共有 1 個檔案被更改,包括 3 行新增1 行删除
  1. 3
    1
      api/core/app/apps/advanced_chat/generate_task_pipeline.py

+ 3
- 1
api/core/app/apps/advanced_chat/generate_task_pipeline.py 查看文件

err = self._base_task_pipeline._handle_error(event=event, session=session, message_id=self._message_id) err = self._base_task_pipeline._handle_error(event=event, session=session, message_id=self._message_id)
yield self._base_task_pipeline._error_to_stream_response(err) yield self._base_task_pipeline._error_to_stream_response(err)


def _handle_workflow_started_event(self, **kwargs) -> Generator[StreamResponse, None, None]:
def _handle_workflow_started_event(
self, event: QueueWorkflowStartedEvent, **kwargs
) -> Generator[StreamResponse, None, None]:
"""Handle workflow started events.""" """Handle workflow started events."""
with self._database_session() as session: with self._database_session() as session:
workflow_execution = self._workflow_cycle_manager.handle_workflow_run_start() workflow_execution = self._workflow_cycle_manager.handle_workflow_run_start()

Loading…
取消
儲存