timeout for interpreter module

Author: lqjack

camel/interpreters/base.py

@@ -19,7 +19,7 @@ class BaseInterpreter(ABC):
     r"""An abstract base class for code interpreters."""
 
     @abstractmethod
-    def run(self, code: str, code_type: str) -> str:
+    def run(self, code: str, code_type: str, timeout: int = None) -> str:
         r"""Executes the given code based on its type.
 
         Args:

camel/interpreters/docker_interpreter.py

@@ -78,11 +78,12 @@ def __init__(
         require_confirm: bool = True,
         print_stdout: bool = False,
         print_stderr: bool = True,
+        default_timeout: int = 60,
     ) -> None:
         self.require_confirm = require_confirm
         self.print_stdout = print_stdout
         self.print_stderr = print_stderr
-
+        self.default_timeout = default_timeout
         # lazy initialization of container
         self._container: Optional[Container] = None
 
@@ -152,6 +153,7 @@ def _run_file_in_container(
         self,
         file: Path,
         code_type: str,
+        timeout: int = None,
     ) -> str:
         code_type = self._check_code_type(code_type)
         commands = shlex.split(
@@ -166,6 +168,7 @@ def _run_file_in_container(
         stdout, stderr = self._container.exec_run(
             commands,
             demux=True,
+            timeout=timeout if timeout is not None else self.default_timeout,
         ).output
 
         if self.print_stdout and stdout:
@@ -184,6 +187,7 @@ def run(
         self,
         code: str,
         code_type: str,
+        timeout: int = None,
     ) -> str:
         r"""Executes the given code in the conatiner attached to the
         interpreter, and captures the stdout and stderr streams.
@@ -228,7 +232,7 @@ def run(
 
         try:
             temp_file_path = self._create_file_in_container(code)
-            result = self._run_file_in_container(temp_file_path, code_type)
+            result = self._run_file_in_container(temp_file_path, code_type,timeout=timeout)
         except docker.errors.APIError as e:
             raise InterpreterError(
                 f"Execution halted due to docker API error: {e.explanation}. "

camel/interpreters/internal_python_interpreter.py

@@ -16,7 +16,7 @@
 import importlib
 import typing
 from typing import Any, ClassVar, Dict, List, Optional
-
+import concurrent.futures
 from camel.interpreters.base import BaseInterpreter
 from camel.interpreters.interpreter_error import InterpreterError
 
@@ -87,15 +87,17 @@ def __init__(
         import_white_list: Optional[List[str]] = None,
         unsafe_mode: bool = False,
         raise_error: bool = False,
+        default_timeout:int = 60
     ) -> None:
         self.action_space = action_space or dict()
         self.state = self.action_space.copy()
         self.fuzz_state: Dict[str, Any] = dict()
         self.import_white_list = import_white_list or list()
         self.raise_error = raise_error
         self.unsafe_mode = unsafe_mode
+        self.default_timeout = default_timeout
 
-    def run(self, code: str, code_type: str) -> str:
+    def run(self, code: str, code_type: str, timeout:int =None) -> str:
         r"""Executes the given code with specified code type in the
         interpreter.
 
@@ -145,7 +147,13 @@ def run(self, code: str, code_type: str) -> str:
 
             return result
         else:
-            return str(self.execute(code))
+            try:
+                with concurrent.futures.ThreadPoolExecutor() as executor:
+                        future = executor.submit(self._execute_ast)
+                        line_result = future.result(timeout=timeout or self.default_timeout)
+                        return str(line_result)
+            except concurrent.futures.TimeoutError:
+                raise InterpreterError("Code execution timed out.")
 
     def update_action_space(self, action_space: Dict[str, Any]) -> None:
         r"""Updates action space for *python* interpreter."""

camel/interpreters/ipython_interpreter.py

@@ -42,11 +42,12 @@ def __init__(
         require_confirm: bool = True,
         print_stdout: bool = False,
         print_stderr: bool = True,
+        default_timeout: int = 60,
     ) -> None:
         self.require_confirm = require_confirm
         self.print_stdout = print_stdout
         self.print_stderr = print_stderr
-
+        self.default_timeout = default_timeout
         self.kernel_manager: Optional[KernelManager] = None
         self.client: Optional[BlockingKernelClient] = None
 
@@ -118,7 +119,7 @@ def _execute(self, code: str, timeout: float) -> str:
         exec_result = "\n".join(outputs)
         return self._clean_ipython_output(exec_result)
 
-    def run(self, code: str, code_type: str) -> str:
+    def run(self, code: str, code_type: str, timeout:int = None) -> str:
         r"""Executes the given code in the Jupyter kernel.
 
         Args:
@@ -138,7 +139,7 @@ def run(self, code: str, code_type: str) -> str:
         if code_type == "bash":
             code = f"%%bash\n({code})"
         try:
-            result = self._execute(code, timeout=TIMEOUT)
+            result = self._execute(code, timeout=timeout or self.default_timeout)
         except Exception as e:
             raise InterpreterError(f"Execution failed: {e!s}")
 

camel/interpreters/subprocess_interpreter.py

@@ -88,6 +88,7 @@ def run_file(
         self,
         file: Path,
         code_type: str,
+        timeout: int = None
     ) -> str:
         r"""Executes a code file in a subprocess and captures its output.