feat&refactor: add proxy setup functionality and update .gitignore for proxy config file

2025-06-07 11:24:49 +00:00
parent 7d25f902a4
commit 493abdeeab
3 changed files with 61 additions and 1 deletions
--- a/desktop_env/desktop_env.py
+++ b/desktop_env/desktop_env.py
@@ -191,6 +191,10 @@ class DesktopEnv(gym.Env):
            logger.info("Setting up environment...")
            self.setup_controller.setup(self.config)
            logger.info("Environment setup complete.")
+        
+            if task_config.get("proxy", False):
+                # If using proxy, set up the proxy configuration
+                self.setup_controller._proxy_setup()

        observation = self._get_obs()
        return observation