From 587f9295670e0c2fdeffefaa923cc1c902960158 Mon Sep 17 00:00:00 2001 From: Zilong Zhou Date: Fri, 4 Jul 2025 13:31:51 +0800 Subject: [PATCH] fix: proxy setup (#234) --- desktop_env/desktop_env.py | 22 +--------------------- 1 file changed, 1 insertion(+), 21 deletions(-) diff --git a/desktop_env/desktop_env.py b/desktop_env/desktop_env.py index 6eaba44..be817c9 100644 --- a/desktop_env/desktop_env.py +++ b/desktop_env/desktop_env.py @@ -177,28 +177,8 @@ class DesktopEnv(gym.Env): logger.info("Task requires proxy but proxy is disabled at system level, ignoring proxy requirement.") if task_use_proxy != self.current_use_proxy: - logger.info(f"Task proxy requirement changed: {self.current_use_proxy} -> {task_use_proxy}") - - # Close current provider if it exists - if hasattr(self, 'provider') and self.provider: - try: - self.provider.stop_emulator(self.path_to_vm) - except Exception as e: - logger.warning(f"Failed to stop current provider: {e}") - - # Create new provider with appropriate proxy setting + # keep because get_info_from_website depend on this self.current_use_proxy = task_use_proxy - self.manager, self.provider = create_vm_manager_and_provider( - self.provider_name, - self.region, - use_proxy=task_use_proxy - ) - - if task_use_proxy: - logger.info("Using proxy-enabled AWS provider.") - else: - logger.info("Using regular AWS provider.") - # Only revert to snapshot if environment has been used (step/setup) # This optimization is especially important for cloud providers like AWS