code block validation (#2017)

This commit is contained in:
Shuchang Zheng 2025-03-25 02:47:02 -07:00 committed by GitHub
parent 915bcfbf13
commit 54dadcf370
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 4 deletions

View file

@ -30,7 +30,6 @@ from skyvern.config import settings
from skyvern.constants import GET_DOWNLOADED_FILES_TIMEOUT, MAX_UPLOAD_FILE_COUNT
from skyvern.exceptions import (
ContextParameterValueNotFound,
DisabledBlockExecutionError,
MissingBrowserState,
MissingBrowserStatePage,
SkyvernException,
@ -1192,8 +1191,7 @@ async def wrapper():
browser_session_id: str | None = None,
**kwargs: dict,
) -> BlockResult:
if not settings.ENABLE_CODE_BLOCK:
raise DisabledBlockExecutionError("CodeBlock is disabled")
await app.AGENT_FUNCTION.validate_code_block(organization_id=organization_id)
# TODO: only support to use code block to manupilate the browser page
# support browser context in the future