Handle keyboard interrupt (ctrl-c.)

This commit is contained in:
AI Christianson 2024-12-19 17:35:35 -05:00
parent 5ad8aeb013
commit c4efac339a
1 changed files with 93 additions and 88 deletions

View File

@ -247,6 +247,7 @@ def run_research_subtasks(base_task: str, config: dict, model, expert_enabled: b
def main():
"""Main entry point for the ra-aid command line tool."""
try:
args = parse_arguments()
expert_enabled, expert_missing = validate_environment(args) # Will exit if main env vars missing
@ -346,5 +347,9 @@ def main():
expert_enabled=expert_enabled
)
except KeyboardInterrupt:
console.print("\n[red]Operation cancelled by user[/red]")
sys.exit(1)
if __name__ == "__main__":
main()