def cancel(workflow_id: str) -> None: try: workflow_manager = get_management_actor() ray.get(workflow_manager.cancel_workflow.remote(workflow_id)) except ValueError: wf_store = workflow_storage.get_workflow_storage(workflow_id) wf_store.save_workflow_meta(WorkflowMetaData(WorkflowStatus.CANCELED))
def load_workflow_meta(self) -> Optional[WorkflowMetaData]: """Load the metadata of the current workflow. Returns: The metadata of the current workflow. If it doesn't exist, return None. """ try: metadata = asyncio_run(self._get(self._key_workflow_metadata(), True)) return WorkflowMetaData(status=WorkflowStatus(metadata["status"])) except KeyNotFoundError: return None