feat&fix: add proxy support in setup and remove hardcoded proxy from example

2025-07-01 13:47:16 +00:00
parent 64f47d1a32
commit 2f9ae1f3ad
3 changed files with 7 additions and 3 deletions
--- a/desktop_env/desktop_env.py
+++ b/desktop_env/desktop_env.py
@@ -203,7 +203,7 @@ class DesktopEnv(gym.Env):
                self._set_task_info(task_config)
                self.setup_controller.reset_cache_dir(self.cache_dir)
                logger.info("Setting up environment...")
-                success = self.setup_controller.setup(self.config)
+                success = self.setup_controller.setup(self.config, task_config.get("proxy", False) and self.enable_proxy)
                if success:
                    break
                else: