From 0fdef9564545d089dec65c5f68cfd5d5b3ee8a36 Mon Sep 17 00:00:00 2001 From: -LAN- Date: Wed, 21 May 2025 18:22:12 +0800 Subject: [PATCH] feat(workflow_cycle_manager): Add `_get_workflow_execution` Signed-off-by: -LAN- --- api/core/workflow/workflow_cycle_manager.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/api/core/workflow/workflow_cycle_manager.py b/api/core/workflow/workflow_cycle_manager.py index 349163f8ad..bf5aef7a8f 100644 --- a/api/core/workflow/workflow_cycle_manager.py +++ b/api/core/workflow/workflow_cycle_manager.py @@ -904,6 +904,12 @@ class WorkflowCycleManager: return None + def _get_workflow_execution(self, id: str, /) -> WorkflowExecution: + execution = self._workflow_execution_repository.get(id) + if not execution: + raise WorkflowRunNotFoundError(id) + return execution + def _get_workflow_run(self, *, session: Session, workflow_run_id: str) -> WorkflowRun: if self._workflow_run and self._workflow_run.id == workflow_run_id: cached_workflow_run = self._workflow_run