mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2025-09-11 16:04:36 +00:00
Workflow CodeGen (#2740)
This commit is contained in:
parent
14bc711240
commit
f6a0ccd32b
10 changed files with 565 additions and 51 deletions
23
skyvern/core/code_generations/run_initializer.py
Normal file
23
skyvern/core/code_generations/run_initializer.py
Normal file
|
@ -0,0 +1,23 @@
|
|||
from typing import Any
|
||||
|
||||
from playwright.async_api import async_playwright
|
||||
|
||||
from skyvern.core.code_generations.skyvern_page import RunContext, SkyvernPage
|
||||
from skyvern.forge.sdk.core import skyvern_context
|
||||
from skyvern.webeye.browser_factory import BrowserContextFactory
|
||||
|
||||
|
||||
# TODO: find a better name for this function
|
||||
async def setup(parameters: dict[str, Any]) -> tuple[SkyvernPage, RunContext]:
|
||||
# set up skyvern context
|
||||
skyvern_context.set(skyvern_context.SkyvernContext())
|
||||
# start playwright
|
||||
pw = await async_playwright().start()
|
||||
(
|
||||
browser_context,
|
||||
_,
|
||||
_,
|
||||
) = await BrowserContextFactory.create_browser_context(playwright=pw)
|
||||
new_page = await browser_context.new_page()
|
||||
skyvern_page = SkyvernPage(page=new_page)
|
||||
return skyvern_page, RunContext(parameters=parameters, page=skyvern_page)
|
Loading…
Add table
Add a link
Reference in a new issue