From 37fa441463a78ce8aa1519f67b72296a20965a86 Mon Sep 17 00:00:00 2001 From: Stanislav Novosad Date: Tue, 11 Nov 2025 10:25:36 -0700 Subject: [PATCH] Simplify SkyvernPage _decorate_call (#3959) --- skyvern/core/script_generations/script_skyvern_page.py | 3 ++- skyvern/core/script_generations/skyvern_page.py | 6 ++---- 2 files changed, 4 insertions(+), 5 deletions(-) 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