fix: clean up experimental flag method

Author: reedham-aws

samcli/commands/build/build_context.py

@@ -281,8 +281,7 @@ def run(self) -> None:
             )
 
             self._check_exclude_warning()
-            self._check_build_method_experimental_flag('rust-cargolambda')
-            self._check_build_method_experimental_flag('python-uv')
+            self._check_build_method_experimental_flag()
 
             for f in self.get_resources_to_build().functions:
                 EventTracker.track_event(EventName.BUILD_FUNCTION_RUNTIME.value, f.runtime)
@@ -697,24 +696,23 @@ def _check_exclude_warning(self) -> None:
         if self._resource_identifier in excludes:
             LOG.warning(self._EXCLUDE_WARNING_MESSAGE)
 
-    def _check_build_method_experimental_flag(self, build_method: str) -> None:
+    def _check_build_method_experimental_flag(self) -> None:
         """
         Prints warning message and confirms if user wants to use beta feature
         """
-        WARNING_MESSAGE = (
-            f'Build method "{build_method}" is a beta feature.\n'
-            'Please confirm if you would like to proceed\n'
-            'You can also enable this beta feature with "sam build --beta-features".'
-        )
+        EXPERIMENTAL_BUILD_METHODS = { "rust-cargolambda": ExperimentalFlag.RustCargoLambda, "python-uv": ExperimentalFlag.UvPackageManager }
+
         resources_to_build = self.get_resources_to_build()
-        is_building_experimental = False
         for function in resources_to_build.functions:
-            if function.metadata and function.metadata.get("BuildMethod", "") == build_method:
-                is_building_experimental = True
-                break
+            if function.metadata and function.metadata.get("BuildMethod", "") in EXPERIMENTAL_BUILD_METHODS:
+                build_method = function.metadata.get("BuildMethod", "")
+                WARNING_MESSAGE = (
+                    f'Build method "{build_method}" is a beta feature.\n'
+                    'Please confirm if you would like to proceed\n'
+                    'You can also enable this beta feature with "sam build --beta-features".'
+                )
 
-        if is_building_experimental:
-            prompt_experimental(ExperimentalFlag.RustCargoLambda if build_method == "rust-cargolambda" else ExperimentalFlag.UvPackageManager, WARNING_MESSAGE)
+                prompt_experimental(EXPERIMENTAL_BUILD_METHODS[build_method], WARNING_MESSAGE)
 
     @property
     def build_in_source(self) -> Optional[bool]: