fix: correct safe mode checks for sensitive action blocks

- Add skip_safe_mode_check parameter to HITLReviewHelper to avoid
  checking the wrong safe mode flag for sensitive action blocks
- Simplify SafeModeToggle and FloatingSafeModeToggle by removing
  unnecessary intermediate variables and isHITLStateUndetermined checks

autogpt_platform/backend/backend/blocks/helpers/review.py

@@ -62,6 +62,7 @@ class HITLReviewHelper:
         execution_context: ExecutionContext,
         block_name: str = "Block",
         editable: bool = False,
+        skip_safe_mode_check: bool = False,
     ) -> Optional[ReviewResult]:
         """
         Handle a review request for a block that requires human review.
@@ -76,6 +77,8 @@ class HITLReviewHelper:
             execution_context: Current execution context
             block_name: Name of the block requesting review
             editable: Whether the reviewer can edit the data
+            skip_safe_mode_check: If True, skip the safe mode check (caller already
+                verified). Used by sensitive action blocks that check their own flag.
 
         Returns:
             ReviewResult if review is complete, None if waiting for human input
@@ -84,7 +87,11 @@ class HITLReviewHelper:
             Exception: If review creation or status update fails
         """
         # Skip review if safe mode is disabled - return auto-approved result
-        if not execution_context.human_in_the_loop_safe_mode:
+        # (unless caller already checked and wants to skip this check)
+        if (
+            not skip_safe_mode_check
+            and not execution_context.human_in_the_loop_safe_mode
+        ):
             logger.info(
                 f"Block {block_name} skipping review for node {node_exec_id} - safe mode disabled"
             )
@@ -136,6 +143,7 @@ class HITLReviewHelper:
         execution_context: ExecutionContext,
         block_name: str = "Block",
         editable: bool = False,
+        skip_safe_mode_check: bool = False,
     ) -> Optional[ReviewDecision]:
         """
         Handle a review request and return the decision in a single call.
@@ -150,6 +158,8 @@ class HITLReviewHelper:
             execution_context: Current execution context
             block_name: Name of the block requesting review
             editable: Whether the reviewer can edit the data
+            skip_safe_mode_check: If True, skip the safe mode check (caller already
+                verified). Used by sensitive action blocks that check their own flag.
 
         Returns:
             ReviewDecision if review is complete (approved/rejected),
@@ -165,6 +175,7 @@ class HITLReviewHelper:
             execution_context=execution_context,
             block_name=block_name,
             editable=editable,
+            skip_safe_mode_check=skip_safe_mode_check,
         )
 
         if review_result is None:

autogpt_platform/frontend/src/app/(platform)/build/components/FloatingSafeModeToogle.tsx

@@ -86,7 +86,6 @@ export function FloatingSafeModeToggle({
   const {
     currentHITLSafeMode,
     showHITLToggle,
-    isHITLStateUndetermined,
     handleHITLToggle,
     currentSensitiveActionSafeMode,
     showSensitiveActionToggle,
@@ -99,16 +98,9 @@ export function FloatingSafeModeToggle({
     return null;
   }
 
-  const showHITL = showHITLToggle && !isHITLStateUndetermined;
-  const showSensitive = showSensitiveActionToggle;
-
-  if (!showHITL && !showSensitive) {
-    return null;
-  }
-
   return (
     <div className={cn("fixed z-50 flex flex-col gap-2", className)}>
-      {showHITL && (
+      {showHITLToggle && (
         <SafeModeButton
           isEnabled={currentHITLSafeMode}
           label="Human in the loop block approval"
@@ -119,7 +111,7 @@ export function FloatingSafeModeToggle({
           fullWidth={fullWidth}
         />
       )}
-      {showSensitive && (
+      {showSensitiveActionToggle && (
         <SafeModeButton
           isEnabled={currentSensitiveActionSafeMode}
           label="Sensitive actions blocks approval"

autogpt_platform/frontend/src/app/(platform)/library/agents/[id]/components/NewAgentLibraryView/components/selected-views/SelectedRunView/components/SafeModeToggle.tsx

@@ -69,7 +69,6 @@ export function SafeModeToggle({ graph, className }: Props) {
   const {
     currentHITLSafeMode,
     showHITLToggle,
-    isHITLStateUndetermined,
     handleHITLToggle,
     currentSensitiveActionSafeMode,
     showSensitiveActionToggle,
@@ -78,20 +77,13 @@ export function SafeModeToggle({ graph, className }: Props) {
     shouldShowToggle,
   } = useAgentSafeMode(graph);
 
-  if (!shouldShowToggle || isHITLStateUndetermined) {
-    return null;
-  }
-
-  const showHITL = showHITLToggle && !isHITLStateUndetermined;
-  const showSensitive = showSensitiveActionToggle;
-
-  if (!showHITL && !showSensitive) {
+  if (!shouldShowToggle) {
     return null;
   }
 
   return (
     <div className={cn("flex gap-1", className)}>
-      {showHITL && (
+      {showHITLToggle && (
         <SafeModeIconButton
           isEnabled={currentHITLSafeMode}
           label="Human-in-the-loop"
@@ -101,7 +93,7 @@ export function SafeModeToggle({ graph, className }: Props) {
           isPending={isPending}
         />
       )}
-      {showSensitive && (
+      {showSensitiveActionToggle && (
         <SafeModeIconButton
           isEnabled={currentSensitiveActionSafeMode}
           label="Sensitive actions"