Update tests.py

tests.py

@@ -139,7 +139,7 @@ chat_id = "5075390513"
 from requests_futures.sessions import FuturesSession
 session = FuturesSession()
 
-def run(cmd, timeout_sec,forever_cmd=False):
+def run(cmd, timeout_sec,forever_cmd):
     t=time.time()
     child = pexpect.spawn("bash")
     output=""
@@ -159,7 +159,7 @@ def run(cmd, timeout_sec,forever_cmd=False):
             output=output.replace("END","")
             child.close()
             break
-        if forever_cmd:
+        if "true" in forever_cmd:
             break
     return output
 
@@ -221,7 +221,7 @@ def create_code_files(filename: str, code: str) -> dict:
     return {"info":"task completed. The referenced code files were created successfully. "}
     
 @mcp.tool()
-def run_code(python_packages:str,filename: str, code: str,start_cmd:str,forever_cmd:bool) -> dict:
+def run_code(python_packages:str,filename: str, code: str,start_cmd:str,forever_cmd:str) -> dict:
     """  
     Execute code in a controlled environment with package installation and file handling.
     Args:
@@ -231,7 +231,7 @@ def run_code(python_packages:str,filename: str, code: str,start_cmd:str,forever_
         code:           Full code to write to the file.
         start_cmd:      Command to execute the file (e.g., "python /app/code_interpreter/app.py" 
                         or "bash /app/code_interpreter/app.py").
-        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.
+        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.
     Notes:
         - All user-uploaded files are in /app/code_interpreter/.
         - After execution, embed a download link (or display images/gifs/videos directly in markdown format) in your response.
@@ -249,7 +249,7 @@ def run_code(python_packages:str,filename: str, code: str,start_cmd:str,forever_
     f.write(code)
     f.close()
     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=[]
@@ -264,8 +264,8 @@ def run_code(python_packages:str,filename: str, code: str,start_cmd:str,forever_
 
 
 @mcp.tool()
-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."""
+def run_code_files(start_cmd:str,forever_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.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."""
     global files_list
 
     stdot=run(start_cmd, 300,forever_cmd)
@@ -283,8 +283,8 @@ def run_code_files(start_cmd:str,forever_cmd:bool) -> dict:
 
 
 @mcp.tool()
-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."""
+def run_shell_command(cmd:str,forever_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.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}
@@ -417,4 +417,4 @@ def deepthinking3(query:str,info:str) -> dict:
 
 if __name__ == "__main__":
     # Initialize and run the server
-    mcp.run(transport='stdio')
+    mcp.run(transport='stdio')