mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2025-09-03 03:00:11 +00:00
126 lines
5 KiB
Python
126 lines
5 KiB
Python
from starlette import status
|
|
|
|
from skyvern.exceptions import SkyvernException, SkyvernHTTPException
|
|
|
|
|
|
class BaseWorkflowException(SkyvernException):
|
|
pass
|
|
|
|
|
|
class BaseWorkflowHTTPException(SkyvernHTTPException):
|
|
pass
|
|
|
|
|
|
class WorkflowDefinitionHasDuplicateBlockLabels(BaseWorkflowHTTPException):
|
|
def __init__(self, duplicate_labels: set[str]) -> None:
|
|
super().__init__(
|
|
f"WorkflowDefinition has blocks with duplicate labels. Each block needs to have a unique "
|
|
f"label. Duplicate label(s): {','.join(duplicate_labels)}",
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
)
|
|
|
|
|
|
class FailedToCreateWorkflow(BaseWorkflowHTTPException):
|
|
def __init__(self, error_message: str) -> None:
|
|
super().__init__(
|
|
f"Failed to create workflow. Error: {error_message}",
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|
)
|
|
|
|
|
|
class FailedToUpdateWorkflow(BaseWorkflowHTTPException):
|
|
def __init__(self, workflow_permanent_id: str, error_message: str) -> None:
|
|
super().__init__(
|
|
f"Failed to update workflow with ID {workflow_permanent_id}. Error: {error_message}",
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|
)
|
|
|
|
|
|
class OutputParameterKeyCollisionError(BaseWorkflowHTTPException):
|
|
def __init__(self, key: str, retry_count: int | None = None) -> None:
|
|
message = f"Output parameter key {key} already exists in the context manager."
|
|
if retry_count is not None:
|
|
message += f" Retrying {retry_count} more times."
|
|
elif retry_count == 0:
|
|
message += " Max duplicate retries reached, aborting."
|
|
super().__init__(
|
|
message,
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
)
|
|
|
|
|
|
class WorkflowDefinitionHasDuplicateParameterKeys(BaseWorkflowHTTPException):
|
|
def __init__(self, duplicate_keys: set[str]) -> None:
|
|
super().__init__(
|
|
f"WorkflowDefinition has parameters with duplicate keys. Each parameter needs to have a unique "
|
|
f"key. Duplicate key(s): {','.join(duplicate_keys)}",
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
)
|
|
|
|
|
|
class WorkflowDefinitionHasReservedParameterKeys(BaseWorkflowHTTPException):
|
|
def __init__(self, reserved_keys: list[str], parameter_keys: list[str]) -> None:
|
|
super().__init__(
|
|
f"WorkflowDefinition has parameters with reserved keys. User created parameters cannot have the following "
|
|
f"reserved keys: {','.join(reserved_keys)}. Parameter keys: {','.join(parameter_keys)}",
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
)
|
|
|
|
|
|
class InvalidWorkflowDefinition(BaseWorkflowHTTPException):
|
|
def __init__(self, message: str) -> None:
|
|
super().__init__(
|
|
message,
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
)
|
|
|
|
|
|
class InvalidEmailClientConfiguration(BaseWorkflowException):
|
|
def __init__(self, problems: list[str]) -> None:
|
|
super().__init__(f"Email client configuration is invalid. These parameters are missing or invalid: {problems}")
|
|
|
|
|
|
class NoValidEmailRecipient(BaseWorkflowException):
|
|
def __init__(self, recipients: list[str]) -> None:
|
|
super().__init__(f"No valid email recipient found. Recipients: {recipients}")
|
|
|
|
|
|
class ContextParameterSourceNotDefined(BaseWorkflowHTTPException):
|
|
def __init__(self, context_parameter_key: str, source_key: str) -> None:
|
|
super().__init__(
|
|
f"Source parameter key {source_key} for context parameter {context_parameter_key} does not exist.",
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
)
|
|
|
|
|
|
class InvalidFileType(BaseWorkflowHTTPException):
|
|
def __init__(self, file_url: str, file_type: str, error: str) -> None:
|
|
super().__init__(
|
|
f"File URL {file_url} is not a valid {file_type} file. Error: {error}",
|
|
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
|
|
)
|
|
|
|
|
|
class WorkflowParameterMissingRequiredValue(BaseWorkflowHTTPException):
|
|
def __init__(self, workflow_parameter_type: str, workflow_parameter_key: str, required_value: str) -> None:
|
|
super().__init__(
|
|
f"Missing required value for workflow parameter. Workflow parameter type: {workflow_parameter_type}. workflow_parameter_key: {workflow_parameter_key}. Required value: {required_value}",
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
|
|
class InvalidWaitBlockTime(SkyvernException):
|
|
def __init__(self, max_sec: int) -> None:
|
|
super().__init__(f"Invalid wait time for wait block, it should be a number between 0 and {max_sec}.")
|
|
|
|
|
|
class FailedToFormatJinjaStyleParameter(SkyvernException):
|
|
def __init__(self, template: str, msg: str) -> None:
|
|
super().__init__(
|
|
f"Failed to format Jinja style parameter {template}. Please make sure the variable reference is correct. reason: {msg}"
|
|
)
|
|
|
|
|
|
class NoIterableValueFound(SkyvernException):
|
|
def __init__(self) -> None:
|
|
super().__init__("No iterable value found for the loop block")
|