lzy/sci-gui-agent-benchmark
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Windows URL

Browse Source
This commit is contained in:
FredWuCZ
2024-10-02 12:19:01 +08:00
parent 6bb27d3ddd
commit fd65cf47f6
2 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
desktop_env/desktop_env.py
View File

@@ -36,7 +36,7 @@ class DesktopEnv(gym.Env):
headless: bool = False, headless: bool = False,
require_a11y_tree: bool = True, require_a11y_tree: bool = True,
require_terminal: bool = False, require_terminal: bool = False,
os_type: str = "Ubuntu", os_type: str = "Windows",
): ):
""" """
Args: Args:

15
desktop_env/providers/docker/manager.py
View File

@@ -27,6 +27,7 @@ MAX_RETRY_TIMES = 10
RETRY_INTERVAL = 5 RETRY_INTERVAL = 5
UBUNTU_X86_URL = "https://huggingface.co/datasets/xlangai/ubuntu_osworld/resolve/main/Ubuntu.qcow2" UBUNTU_X86_URL = "https://huggingface.co/datasets/xlangai/ubuntu_osworld/resolve/main/Ubuntu.qcow2"
WINDOWS_X86_URL = "https://huggingface.co/datasets/xlangai/windows_osworld/resolve/main/Windows%2010%20x64%20-%20copy.qcow2"
VMS_DIR = "./docker_vm_data" VMS_DIR = "./docker_vm_data"
# Determine the platform and CPU architecture to decide the correct VM image to download # Determine the platform and CPU architecture to decide the correct VM image to download
@@ -39,21 +40,20 @@ VMS_DIR = "./docker_vm_data"
# URL = UBUNTU_X86_URL # URL = UBUNTU_X86_URL
# else: # else:
# raise Exception("Unsupported platform or architecture") # raise Exception("Unsupported platform or architecture")
URL = UBUNTU_X86_URL
URL = UBUNTU_X86_URL
DOWNLOADED_FILE_NAME = URL.split('/')[-1] DOWNLOADED_FILE_NAME = URL.split('/')[-1]
if platform.system() == 'Windows': if platform.system() == 'Windows':
docker_path = r"C:\Program Files\Docker\Docker" docker_path = r"C:\Program Files\Docker\Docker"
os.environ["PATH"] += os.pathsep + docker_path os.environ["PATH"] += os.pathsep + docker_path
def _download_vm(vms_dir: str): def _download_vm(vms_dir: str, os_type: str):
global URL, DOWNLOADED_FILE_NAME
# Download the virtual machine image # Download the virtual machine image
logger.info("Downloading the virtual machine image...") logger.info("Downloading the virtual machine image...")
downloaded_size = 0 downloaded_size = 0
URL = UBUNTU_X86_URL
DOWNLOADED_FILE_NAME = URL.split('/')[-1]
downloaded_file_name = DOWNLOADED_FILE_NAME downloaded_file_name = DOWNLOADED_FILE_NAME
os.makedirs(vms_dir, exist_ok=True) os.makedirs(vms_dir, exist_ok=True)
@@ -118,6 +118,11 @@ class DockerVMManager(VMManager):
pass pass
def get_vm_path(self, os_type, region): def get_vm_path(self, os_type, region):
if os_type == "Ubuntu":
URL = WINDOWS_X86_URL
elif os_type == "Windows":
URL = WINDOWS_X86_URL
DOWNLOADED_FILE_NAME = URL.split('/')[-1]
if not os.path.exists(os.path.join(VMS_DIR, DOWNLOADED_FILE_NAME)): if not os.path.exists(os.path.join(VMS_DIR, DOWNLOADED_FILE_NAME)):
_download_vm(VMS_DIR) _download_vm(VMS_DIR)
return os.path.join(VMS_DIR, DOWNLOADED_FILE_NAME) return os.path.join(VMS_DIR, DOWNLOADED_FILE_NAME)