Update tests.py

tests.py

@@ -264,11 +264,11 @@ def run_code(python_packages:str,filename: str, code: str,start_cmd:str,forever_
 
 
 @mcp.tool()
-def run_code_files(start_cmd:str) -> dict:
-    """(start_cmd:Example-  python /app/code_interpreter/app.py or bash /app/code_interpreter/app.py).The files must be inside the /app/code_interpreter directory."""
+def run_code_files(start_cmd:str,forever_cmd:bool)) -> dict:
+    """(start_cmd:Example-  python /app/code_interpreter/app.py or bash /app/code_interpreter/app.py).The files must be inside the /app/code_interpreter directory.forever_cmd:If True, the command will run indefinitely.Set to True, when runnig a website/server. If False, the command will time out after 300 second and the result will be returned."""
     global files_list
 
-    stdot=run(start_cmd, 300)
+    stdot=run(start_cmd, 300,forever_cmd)
     onlyfiles = glob.glob("/app/code_interpreter/*")
     onlyfiles=list(set(onlyfiles)-set(files_list))
     uploaded_filenames=[]
@@ -283,9 +283,9 @@ def run_code_files(start_cmd:str) -> dict:
 
 
 @mcp.tool()
-def run_shell_command(cmd:str) -> dict:
-    """(cmd:Example- mkdir test.By default , the command is run inside the /app/code_interpreter/ directory.).Remember, the code_interpreter is running on **alpine linux** , so write commands accordingly.Eg-sudo does not work and is not required.."""
-    output=run(cmd, 300)
+def run_shell_command(cmd:str,forever_cmd:bool) -> dict:
+    """(cmd:Example- mkdir test.By default , the command is run inside the /app/code_interpreter/ directory.).Remember, the code_interpreter is running on **alpine linux** , so write commands accordingly.Eg-sudo does not work and is not required.        forever_cmd: If True, the command will run indefinitely.Set to True, when running a website/server. If False, the command will time out after 300 second and the result will be returned."""
+    output=run(cmd, 300,forever_cmd)
     transfer_files()
     return {"output":output}