use operator_key

IABTechLab · Jan 31, 2025 · 87814fa · 87814fa
1 parent 9850745
commit 87814fa
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/scripts/aws/ec2.py b/scripts/aws/ec2.py
@@ -92,11 +92,12 @@ def _set_confidential_config(self, secret_identifier: str) -> None:
         def add_defaults(configs: Dict[str, any]) ->  AWSConfidentialComputeConfig:
             """Adds default values to configuration if missing."""
             default_capacity = self.__get_max_capacity()
+            configs.setdefault("operator_key", configs.get("api_token"))
             configs.setdefault("enclave_memory_mb", default_capacity["enclave_memory_mb"])
             configs.setdefault("enclave_cpu_count", default_capacity["enclave_cpu_count"])
             configs.setdefault("debug_mode", False)
-            configs.setdefault("core_api_token", configs.get("api_token", ""))
-            configs.setdefault("optout_api_token", configs.get("api_token", ""))
+            configs.setdefault("core_api_token", configs.get("operator_key", ""))
+            configs.setdefault("optout_api_token", configs.get("operator_key", ""))
             return configs
 
         region = self.__get_current_region()
@@ -258,4 +259,4 @@ def __kill_auxiliaries(self) -> None:
         print("Failed starting up Confidential Compute. Please checks the logs for errors and retry \n", e)
     except Exception as e:
         print("Unexpected failure while starting up Confidential Compute. Please contact UID support team with this log \n ", e)
-
+