Timeout

Author: aorwall

moatless/actions/run_python_script.py

@@ -19,7 +19,7 @@ class RunPythonScriptArgs(ActionArguments):
 
     script_path: str = Field(..., description="Path to the Python script to execute")
     args: list[str] = Field(default=[], description="Command line arguments to pass to the script")
-    timeout: int = Field(default=180, description="Timeout in seconds for script execution")
+    timeout: int = Field(default=30, description="Timeout in seconds for script execution")
     max_output_tokens: int = Field(default=2000, description="Maximum number of tokens to return in output")
 
     model_config = ConfigDict(title="RunPythonScript")

moatless/runtime/local.py

@@ -59,7 +59,7 @@ def __init__(
 
         self._install_task = asyncio.create_task(self._run_async_installation(self._install_command))
 
-    async def execute(self, command: str, fail_on_error: bool = False, patch: str | None = None) -> str:
+    async def execute(self, command: str, fail_on_error: bool = False, patch: str | None = None, timeout: int = 600) -> str:
         """Execute a command in the environment."""
         await self._wait_for_install()
         try:
@@ -72,7 +72,7 @@ async def execute(self, command: str, fail_on_error: bool = False, patch: str |
                     stdout, return_code = await self._execute_command(self._install_command)
                     logger.info(f"Installation output: {stdout} {return_code}")
 
-            stdout, return_code = await self._execute_command(command)
+            stdout, return_code = await self._execute_command(command, timeout=timeout)
             logger.info(f"Command {command} returned {return_code}")
             logger.info(f"Output: {stdout}")