vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Simplify SkyvernPage _decorate_call (#3959)

Browse Source
This commit is contained in:
Stanislav Novosad
2025-11-11 10:25:36 -07:00
committed by GitHub
parent 16c11025da
commit 37fa441463
2 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
skyvern/core/script_generations/script_skyvern_page.py
View File

@@ -122,7 +122,6 @@ class ScriptSkyvernPage(SkyvernPage):
fn: Callable, fn: Callable,
action: ActionType, action: ActionType,
*args: Any, *args: Any,
prompt: str = "",
**kwargs: Any, **kwargs: Any,
) -> Any: ) -> Any:
""" """
@@ -147,6 +146,8 @@ class ScriptSkyvernPage(SkyvernPage):
ActionType.TERMINATE: "🛑", ActionType.TERMINATE: "🛑",
} }
prompt = kwargs.get("prompt", "")
# Backward compatibility: use intention if provided and prompt is empty # Backward compatibility: use intention if provided and prompt is empty
intention = kwargs.get("intention", None) intention = kwargs.get("intention", None)
if intention and not prompt: if intention and not prompt:

6
skyvern/core/script_generations/skyvern_page.py
View File

@@ -71,10 +71,9 @@ class SkyvernPage(Page):
fn: Callable, fn: Callable,
action: ActionType, action: ActionType,
*args: Any, *args: Any,
prompt: str = "",
**kwargs: Any, **kwargs: Any,
) -> Any: ) -> Any:
return await fn(self, *args, prompt=prompt, **kwargs) return await fn(self, *args, **kwargs)
@staticmethod @staticmethod
def action_wrap( def action_wrap(
@@ -84,10 +83,9 @@ class SkyvernPage(Page):
async def wrapper( async def wrapper(
skyvern_page: SkyvernPage, skyvern_page: SkyvernPage,
*args: Any, *args: Any,
prompt: str = "",
**kwargs: Any, **kwargs: Any,
) -> 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 return wrapper