diff --git a/skyvern/core/script_generations/script_skyvern_page.py b/skyvern/core/script_generations/script_skyvern_page.py index 934e7869..48ce1967 100644 --- a/skyvern/core/script_generations/script_skyvern_page.py +++ b/skyvern/core/script_generations/script_skyvern_page.py @@ -122,7 +122,6 @@ class ScriptSkyvernPage(SkyvernPage): fn: Callable, action: ActionType, *args: Any, - prompt: str = "", **kwargs: Any, ) -> Any: """ @@ -147,6 +146,8 @@ class ScriptSkyvernPage(SkyvernPage): ActionType.TERMINATE: "🛑", } + prompt = kwargs.get("prompt", "") + # Backward compatibility: use intention if provided and prompt is empty intention = kwargs.get("intention", None) if intention and not prompt: diff --git a/skyvern/core/script_generations/skyvern_page.py b/skyvern/core/script_generations/skyvern_page.py index af69dda6..3ec22c05 100644 --- a/skyvern/core/script_generations/skyvern_page.py +++ b/skyvern/core/script_generations/skyvern_page.py @@ -71,10 +71,9 @@ class SkyvernPage(Page): fn: Callable, action: ActionType, *args: Any, - prompt: str = "", **kwargs: Any, ) -> Any: - return await fn(self, *args, prompt=prompt, **kwargs) + return await fn(self, *args, **kwargs) @staticmethod def action_wrap( @@ -84,10 +83,9 @@ class SkyvernPage(Page): async def wrapper( skyvern_page: SkyvernPage, *args: Any, - prompt: str = "", **kwargs: Any, ) -> Any: - return await skyvern_page._decorate_call(fn, action, *args, prompt=prompt, **kwargs) + return await skyvern_page._decorate_call(fn, action, *args, **kwargs) return wrapper