vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-10 06:15:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Automatically create output parameters (#327)

Browse source
This commit is contained in:
Kerem Yilmaz 2024-05-16 13:08:24 -07:00 • committed by GitHub
parent 72d25cd37d
commit e6cee25416
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 187 additions and 157 deletions
Download patch file Download diff file Expand all files Collapse all files

47
skyvern/forge/sdk/workflow/exceptions.py
View file

@ -1,33 +1,61 @@
from skyvern.exceptions import SkyvernException
from starlette import status
from skyvern.exceptions import SkyvernException, SkyvernHTTPException
class BaseWorkflowException(SkyvernException):
pass
class WorkflowDefinitionHasDuplicateBlockLabels(BaseWorkflowException):
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)}"
f"label. Duplicate label(s): {','.join(duplicate_labels)}",
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
)
class OutputParameterKeyCollisionError(BaseWorkflowException):
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)
super().__init__(
message,
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
)
class WorkflowDefinitionHasDuplicateParameterKeys(BaseWorkflowException):
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)}"
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,
)
@ -36,8 +64,9 @@ class InvalidEmailClientConfiguration(BaseWorkflowException):
super().__init__(f"Email client configuration is invalid. These parameters are missing or invalid: {problems}")
class ContextParameterSourceNotDefined(BaseWorkflowException):
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."
f"Source parameter key {source_key} for context parameter {context_parameter_key} does not exist.",
status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
)