Disable codeblock (#910)

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

View File

@@ -13,6 +13,11 @@ class SkyvernHTTPException(SkyvernException):
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):
def __init__(self, message: str | None = None):
super().__init__(f"Invalid response format: {message}")

View File

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