roll out new context tree 100% (#266)

This commit is contained in:
Shuchang Zheng
2024-05-06 20:25:04 -04:00
committed by GitHub
parent bf47f810bb
commit 6473635de4

View File

@@ -1,7 +1,5 @@
from __future__ import annotations
import random
import structlog
from playwright.async_api import Browser, Playwright, async_playwright
@@ -11,9 +9,6 @@ from skyvern.forge.sdk.settings_manager import SettingsManager
from skyvern.forge.sdk.workflow.models.workflow import WorkflowRun
from skyvern.webeye.browser_factory import BrowserContextFactory, BrowserState
# import random
LOG = structlog.get_logger()
@@ -60,8 +55,7 @@ class BrowserManager:
self.pages[task.task_id] = self.pages[task.workflow_run_id]
return self.pages[task.task_id]
# TODO: percentage (50%) to use new context tree
new_ctx = random.choices([False, True], weights=[0.5, 0.5], k=1)[0]
new_ctx = True
LOG.info("Creating browser state for task", task_id=task.task_id, new_ctx=new_ctx)
browser_state = await self._create_browser_state(task.proxy_location, task.url, new_ctx, task.task_id)