chore: allow LLM to extend timeout (#319)

Author: cuericlee

veadk/tools/builtin_tools/run_code.py

@@ -25,13 +25,16 @@
 logger = get_logger(__name__)
 
 
-def run_code(code: str, language: str, tool_context: ToolContext) -> str:
+def run_code(
+    code: str, language: str, tool_context: ToolContext, timeout: int = 30
+) -> str:
     """Run code in a code sandbox and return the output.
     For C++ code, don't execute it directly, compile and execute via Python; write sources and object files to /tmp.
 
     Args:
         code (str): The code to run.
         language (str): The programming language of the code. Language must be one of the supported languages: python3.
+        timeout (int, optional): The timeout in seconds for the code execution. Defaults to 30.
 
     Returns:
         str: The output of the code execution.
@@ -55,7 +58,7 @@ def run_code(code: str, language: str, tool_context: ToolContext) -> str:
     logger.debug(f"tool_user_session_id: {tool_user_session_id}")
 
     logger.debug(
-        f"Running code in language: {language}, session_id={session_id}, code={code}, tool_id={tool_id}, host={host}, service={service}, region={region}"
+        f"Running code in language: {language}, session_id={session_id}, code={code}, tool_id={tool_id}, host={host}, service={service}, region={region}, timeout={timeout}"
     )
 
     ak = tool_context.state.get("VOLCENGINE_ACCESS_KEY")
@@ -87,7 +90,7 @@ def run_code(code: str, language: str, tool_context: ToolContext) -> str:
             "OperationPayload": json.dumps(
                 {
                     "code": code,
-                    "timeout": 30,
+                    "timeout": timeout,
                     "kernel_name": language,
                 }
             ),