validate taskV2 step (#1811)

This commit is contained in:
Shuchang Zheng
2025-02-22 00:44:12 -08:00
committed by GitHub
parent 8821b7e150
commit 9a07c0bc6f
3 changed files with 51 additions and 3 deletions

View File

@@ -355,11 +355,21 @@ class UnknownElementTreeFormat(SkyvernException):
super().__init__(f"Unknown element tree format {fmt}")
class StepTerminationError(SkyvernException):
def __init__(self, step_id: str, reason: str) -> None:
class TerminationError(SkyvernException):
def __init__(self, reason: str, step_id: str | None = None, task_id: str | None = None) -> None:
super().__init__(f"Termination error. Reason: {reason}")
class StepTerminationError(TerminationError):
def __init__(self, reason: str, step_id: str | None = None, task_id: str | None = None) -> None:
super().__init__(f"Step {step_id} cannot be executed and task is failed. Reason: {reason}")
class TaskTerminationError(TerminationError):
def __init__(self, reason: str, step_id: str | None = None, task_id: str | None = None) -> None:
super().__init__(f"Task {task_id} failed. Reason: {reason}")
class BlockTerminationError(SkyvernException):
def __init__(self, workflow_run_block_id: str, workflow_run_id: str, reason: str) -> None:
super().__init__(