vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Disable codeblock (#910)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-10-04 13:15:49 -07:00
committed by GitHub
parent 2c7cbf0eb8
commit f22b8a04aa
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
skyvern/exceptions.py
View File

@@ -13,6 +13,11 @@ class SkyvernHTTPException(SkyvernException):
super().__init__(message) super().__init__(message)
class DisabledBlockExecutionError(SkyvernHTTPException):
def __init__(self, message: str | None = None):
super().__init__(message, status_code=status.HTTP_400_BAD_REQUEST)
class InvalidOpenAIResponseFormat(SkyvernException): class InvalidOpenAIResponseFormat(SkyvernException):
def __init__(self, message: str | None = None): def __init__(self, message: str | None = None):
super().__init__(f"Invalid response format: {message}") super().__init__(f"Invalid response format: {message}")

2
skyvern/forge/sdk/workflow/models/block.py
View File

@@ -22,6 +22,7 @@ from pydantic import BaseModel, Field
from skyvern.config import settings from skyvern.config import settings
from skyvern.exceptions import ( from skyvern.exceptions import (
ContextParameterValueNotFound, ContextParameterValueNotFound,
DisabledBlockExecutionError,
FailedToNavigateToUrl, FailedToNavigateToUrl,
MissingBrowserStatePage, MissingBrowserStatePage,
TaskNotFound, TaskNotFound,
@@ -534,6 +535,7 @@ class CodeBlock(Block):
return self.parameters return self.parameters
async def execute(self, workflow_run_id: str, **kwargs: dict) -> BlockResult: async def execute(self, workflow_run_id: str, **kwargs: dict) -> BlockResult:
raise DisabledBlockExecutionError("CodeBlock is disabled")
# get workflow run context # get workflow run context
workflow_run_context = self.get_workflow_run_context(workflow_run_id) workflow_run_context = self.get_workflow_run_context(workflow_run_id)