mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2025-09-11 16:04:36 +00:00
23 lines
847 B
Python
23 lines
847 B
Python
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)
|