pbs proxy debugging (#3409)
This commit is contained in:
@@ -102,14 +102,14 @@ async def _current_viewpoint_screenshot_helper(
|
|||||||
|
|
||||||
|
|
||||||
async def _scrolling_screenshots_helper(
|
async def _scrolling_screenshots_helper(
|
||||||
skyvern_page: SkyvernFrame,
|
page: Page,
|
||||||
url: str | None = None,
|
url: str | None = None,
|
||||||
draw_boxes: bool = False,
|
draw_boxes: bool = False,
|
||||||
max_number: int = SettingsManager.get_settings().MAX_NUM_SCREENSHOTS,
|
max_number: int = SettingsManager.get_settings().MAX_NUM_SCREENSHOTS,
|
||||||
mode: ScreenshotMode = ScreenshotMode.DETAILED,
|
mode: ScreenshotMode = ScreenshotMode.DETAILED,
|
||||||
) -> tuple[list[bytes], list[int]]:
|
) -> tuple[list[bytes], list[int]]:
|
||||||
# page is the main frame and the index must be 0
|
# page is the main frame and the index must be 0
|
||||||
assert isinstance(skyvern_page.frame, Page)
|
skyvern_page = await SkyvernFrame.create_instance(frame=page)
|
||||||
frame = "main.frame"
|
frame = "main.frame"
|
||||||
frame_index = 0
|
frame_index = 0
|
||||||
|
|
||||||
@@ -139,7 +139,7 @@ async def _scrolling_screenshots_helper(
|
|||||||
await skyvern_page.build_tree_from_body(frame_name=frame, frame_index=frame_index)
|
await skyvern_page.build_tree_from_body(frame_name=frame, frame_index=frame_index)
|
||||||
initial_scroll_height = scroll_height
|
initial_scroll_height = scroll_height
|
||||||
|
|
||||||
screenshot = await _current_viewpoint_screenshot_helper(page=skyvern_page.frame, mode=mode)
|
screenshot = await _current_viewpoint_screenshot_helper(page=page, mode=mode)
|
||||||
screenshots.append(screenshot)
|
screenshots.append(screenshot)
|
||||||
positions.append(int(scroll_y_px))
|
positions.append(int(scroll_y_px))
|
||||||
scroll_y_px_old = scroll_y_px
|
scroll_y_px_old = scroll_y_px
|
||||||
@@ -167,7 +167,7 @@ async def _scrolling_screenshots_helper(
|
|||||||
await skyvern_page.build_elements_and_draw_bounding_boxes(frame=frame, frame_index=frame_index)
|
await skyvern_page.build_elements_and_draw_bounding_boxes(frame=frame, frame_index=frame_index)
|
||||||
|
|
||||||
LOG.debug("Page is not scrollable", url=url, num_screenshots=len(screenshots))
|
LOG.debug("Page is not scrollable", url=url, num_screenshots=len(screenshots))
|
||||||
screenshot = await _current_viewpoint_screenshot_helper(page=skyvern_page.frame, mode=mode)
|
screenshot = await _current_viewpoint_screenshot_helper(page=page, mode=mode)
|
||||||
screenshots.append(screenshot)
|
screenshots.append(screenshot)
|
||||||
positions.append(0)
|
positions.append(0)
|
||||||
|
|
||||||
@@ -270,7 +270,7 @@ class SkyvernFrame:
|
|||||||
x, y = await skyvern_frame.get_scroll_x_y()
|
x, y = await skyvern_frame.get_scroll_x_y()
|
||||||
async with asyncio.timeout(timeout):
|
async with asyncio.timeout(timeout):
|
||||||
screenshots, positions = await _scrolling_screenshots_helper(
|
screenshots, positions = await _scrolling_screenshots_helper(
|
||||||
skyvern_page=skyvern_frame, mode=mode, max_number=scrolling_number
|
page=page, mode=mode, max_number=scrolling_number
|
||||||
)
|
)
|
||||||
images = []
|
images = []
|
||||||
|
|
||||||
@@ -324,9 +324,8 @@ class SkyvernFrame:
|
|||||||
if not scroll:
|
if not scroll:
|
||||||
return [await _current_viewpoint_screenshot_helper(page=page, mode=ScreenshotMode.DETAILED)]
|
return [await _current_viewpoint_screenshot_helper(page=page, mode=ScreenshotMode.DETAILED)]
|
||||||
|
|
||||||
skyvern_frame = await SkyvernFrame.create_instance(frame=page)
|
|
||||||
screenshots, _ = await _scrolling_screenshots_helper(
|
screenshots, _ = await _scrolling_screenshots_helper(
|
||||||
skyvern_page=skyvern_frame,
|
page=page,
|
||||||
url=url,
|
url=url,
|
||||||
max_number=max_number,
|
max_number=max_number,
|
||||||
draw_boxes=draw_boxes,
|
draw_boxes=draw_boxes,
|
||||||
|
|||||||
Reference in New Issue
Block a user