From 431a762421e17035f4e4287081083d3d3bdb2119 Mon Sep 17 00:00:00 2001
From: adlsdztony <zzl0712@outlook.com>
Date: Mon, 26 May 2025 20:37:20 +0800
Subject: [PATCH] feat&fix: add logging for setup function calls and include
 snapshot name in AWS provider configuration

---
 desktop_env/controllers/setup.py | 2 ++
 run_multienv.py                  | 1 +
 2 files changed, 3 insertions(+)

diff --git a/desktop_env/controllers/setup.py b/desktop_env/controllers/setup.py
index ac1c372..04859a0 100644
--- a/desktop_env/controllers/setup.py
+++ b/desktop_env/controllers/setup.py
@@ -61,6 +61,7 @@ class SetupController:
             # protocol
             setup_function: str = "_{:}_setup".format(config_type)
             assert hasattr(self, setup_function), f'Setup controller cannot find init function {setup_function}'
+            logger.info(f"call function {setup_function}")
             getattr(self, setup_function)(**parameters)
 
             logger.info("SETUP: %s(%s)", setup_function, str(parameters))
@@ -229,6 +230,7 @@ class SetupController:
         headers = {"Content-Type": "application/json"}
 
         try:
+            logger.info("REQUEST ADDRESS: %s", self.http_server + "/setup" + "/launch")
             response = requests.post(self.http_server + "/setup" + "/launch", headers=headers, data=payload)
             if response.status_code == 200:
                 logger.info("Command executed successfully: %s", response.text)
diff --git a/run_multienv.py b/run_multienv.py
index 823061b..8399fd3 100644
--- a/run_multienv.py
+++ b/run_multienv.py
@@ -224,6 +224,7 @@ def test(args: argparse.Namespace, test_all_meta: dict) -> None:
 
             provider_name="aws",
             region="us-east-1",
+            snapshot_name="ami-05e7d7bd279ea4f14",
 
             screen_size=(args.screen_width, args.screen_height),
             headless=args.headless,