mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-25 21:36:52 +08:00
9efe6eb776
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: Engel Nyst <enyst@users.noreply.github.com>
81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
import html
|
|
from prompt_toolkit import HTML, print_formatted_text
|
|
|
|
from openhands.sdk.security.confirmation_policy import (
|
|
ConfirmRisky,
|
|
NeverConfirm,
|
|
SecurityRisk,
|
|
)
|
|
from openhands_cli.user_actions.types import ConfirmationResult, UserConfirmation
|
|
from openhands_cli.user_actions.utils import cli_confirm, cli_text_input
|
|
|
|
|
|
def ask_user_confirmation(
|
|
pending_actions: list, using_risk_based_policy: bool = False
|
|
) -> ConfirmationResult:
|
|
"""Ask user to confirm pending actions.
|
|
|
|
Args:
|
|
pending_actions: List of pending actions from the agent
|
|
|
|
Returns:
|
|
ConfirmationResult with decision, optional policy_change, and reason
|
|
"""
|
|
|
|
if not pending_actions:
|
|
return ConfirmationResult(decision=UserConfirmation.ACCEPT)
|
|
|
|
print_formatted_text(
|
|
HTML(
|
|
f'<yellow>🔍 Agent created {len(pending_actions)} action(s) and is waiting for confirmation:</yellow>'
|
|
)
|
|
)
|
|
|
|
for i, action in enumerate(pending_actions, 1):
|
|
tool_name = getattr(action, 'tool_name', '[unknown tool]')
|
|
action_content = (
|
|
str(getattr(action, 'action', ''))[:100].replace('\n', ' ')
|
|
or '[unknown action]'
|
|
)
|
|
print_formatted_text(
|
|
HTML(f'<grey> {i}. {tool_name}: {html.escape(action_content)}...</grey>')
|
|
)
|
|
|
|
question = 'Choose an option:'
|
|
options = [
|
|
'Yes, proceed',
|
|
'Reject',
|
|
"Always proceed (don't ask again)",
|
|
]
|
|
|
|
if not using_risk_based_policy:
|
|
options.append('Auto-confirm LOW/MEDIUM risk, ask for HIGH risk')
|
|
|
|
try:
|
|
index = cli_confirm(question, options, escapable=True)
|
|
except (EOFError, KeyboardInterrupt):
|
|
print_formatted_text(HTML('\n<red>No input received; pausing agent.</red>'))
|
|
return ConfirmationResult(decision=UserConfirmation.DEFER)
|
|
|
|
if index == 0:
|
|
return ConfirmationResult(decision=UserConfirmation.ACCEPT)
|
|
elif index == 1:
|
|
# Handle "Reject" option with optional reason
|
|
try:
|
|
reason = cli_text_input('Reason (and let OpenHands know why): ').strip()
|
|
except (EOFError, KeyboardInterrupt):
|
|
return ConfirmationResult(decision=UserConfirmation.DEFER)
|
|
|
|
return ConfirmationResult(decision=UserConfirmation.REJECT, reason=reason)
|
|
elif index == 2:
|
|
return ConfirmationResult(
|
|
decision=UserConfirmation.ACCEPT, policy_change=NeverConfirm()
|
|
)
|
|
elif index == 3:
|
|
return ConfirmationResult(
|
|
decision=UserConfirmation.ACCEPT,
|
|
policy_change=ConfirmRisky(threshold=SecurityRisk.HIGH),
|
|
)
|
|
|
|
return ConfirmationResult(decision=UserConfirmation.REJECT)
|