Spaces:
Configuration error
Configuration error
File size: 1,980 Bytes
0d3af20 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | from typing import Literal
from pydantic import Field
from agency_swarm.tools import BaseTool
from .util.selenium import get_web_driver, set_web_driver
class Scroll(BaseTool):
"""
This tool allows you to scroll the current web page up or down by 1 screen height.
"""
direction: Literal["up", "down"] = Field(
..., description="Direction to scroll."
)
def run(self):
wd = get_web_driver()
height = wd.get_window_size()['height']
# Get the zoom level
zoom_level = wd.execute_script("return document.body.style.zoom || '1';")
zoom_level = float(zoom_level.strip('%')) / 100 if '%' in zoom_level else float(zoom_level)
# Adjust height by zoom level
adjusted_height = height / zoom_level
current_scroll_position = wd.execute_script("return window.pageYOffset;")
total_scroll_height = wd.execute_script("return document.body.scrollHeight;")
result = ""
if self.direction == "up":
if current_scroll_position == 0:
# Reached the top of the page
result = "Reached the top of the page. Cannot scroll up any further.\n"
else:
wd.execute_script(f"window.scrollBy(0, -{adjusted_height});")
result = "Scrolled up by 1 screen height. Make sure to output '[send screenshot]' command to analyze the page after scrolling."
elif self.direction == "down":
if current_scroll_position + adjusted_height >= total_scroll_height:
# Reached the bottom of the page
result = "Reached the bottom of the page. Cannot scroll down any further.\n"
else:
wd.execute_script(f"window.scrollBy(0, {adjusted_height});")
result = "Scrolled down by 1 screen height. Make sure to output '[send screenshot]' command to analyze the page after scrolling."
set_web_driver(wd)
return result
|