Add timeout to browser use

Author: whoisarpit

patchwork/steps/BrowserUse/BrowserUse.py

@@ -99,10 +99,16 @@ def run(self) -> dict:
 
         # Run the agent in an event loop
         loop = asyncio.new_event_loop()
-        self.history = loop.run_until_complete(agent.run())
-        loop.run_until_complete(browser_context.close())
-        loop.run_until_complete(browser_context.browser.close())
-        loop.close()
+        timeout = self.inputs.get("timeout", 600)  # default timeout of 10 minutes
+        try:
+            self.history = loop.run_until_complete(asyncio.wait_for(agent.run(), timeout=timeout))
+        except asyncio.TimeoutError:
+            logger.error(f"Agent timed out after {timeout} seconds")
+            raise
+        finally:
+            loop.run_until_complete(browser_context.close())
+            loop.run_until_complete(browser_context.browser.close())
+            loop.close()
 
         # Format results as JSON if schema provided
         if "example_json" in self.inputs:

patchwork/steps/BrowserUse/typed.py

@@ -18,6 +18,7 @@ class BrowserUseInputs(__BrowserUseInputsRequired, total=False):
     initial_actions: Optional[List[Dict[str, Dict[str, Any]]]]
     downloads_path: Optional[str]
     use_vision: Optional[bool]
+    timeout: Optional[int]  # optional timeout in seconds, defaults to 600 if not provided
 
 
 class BrowserUseOutputs(TypedDict):