Simplify SkyvernPage _decorate_call (#3959)

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:

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