Return to the user prompt when an error occurs

e2b-dev · Jan 19, 2025 · dc846cf · dc846cf
1 parent 7407c20
commit dc846cf
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -1,5 +1,6 @@
 from os_computer_use.streaming import Sandbox, DisplayClient
 from os_computer_use.sandbox_agent import SandboxAgent
+from os_computer_use.utils import print_colored
 import asyncio
 import argparse
 
@@ -43,6 +44,9 @@ async def start(user_input=None, output_dir=None):
                     user_input = None
                 except KeyboardInterrupt:
                     user_input = None
+                except Exception as e:
+                    print_colored(f"An error occurred: {e}", "red")
+                    user_input = None
 
     finally:
         if client: