introduce permission checker to task create api (#368)

This commit is contained in:
Kerem Yilmaz
2024-05-26 14:51:53 -07:00
committed by GitHub
parent 5e88d64a11
commit 688a615fb2
7 changed files with 48 additions and 2 deletions

View File

@@ -1,5 +1,6 @@
from playwright.async_api import Page
from skyvern.exceptions import StepTerminationError
from skyvern.forge import app
from skyvern.forge.async_operations import AsyncOperation
from skyvern.forge.sdk.models import Organization, Step, StepStatus
@@ -33,7 +34,7 @@ class AgentFunction:
can_execute = has_valid_task_status and has_valid_step_status and has_no_running_steps
if not can_execute:
raise Exception(f"Cannot execute step. Reasons: {reasons}, Step: {step}")
raise StepTerminationError(step_id=step.step_id, reason="Cannot execute step. Reasons: {reasons}")
def generate_async_operations(
self,