mirror of
https://github.com/OMGeeky/gpt-pilot.git
synced 2025-12-29 23:53:35 +01:00
300 lines
11 KiB
Python
300 lines
11 KiB
Python
import subprocess
|
|
import os
|
|
import signal
|
|
import threading
|
|
import queue
|
|
import time
|
|
import uuid
|
|
import platform
|
|
|
|
from fabulous.color import yellow, green, white, red, bold
|
|
from database.database import get_saved_command_run, save_command_run
|
|
from const.function_calls import DEBUG_STEPS_BREAKDOWN
|
|
from helpers.exceptions.TooDeepRecursionError import TooDeepRecursionError
|
|
from helpers.exceptions.TokenLimitError import TokenLimitError
|
|
|
|
from utils.questionary import styled_text
|
|
from const.code_execution import MAX_COMMAND_DEBUG_TRIES, MIN_COMMAND_RUN_TIME, MAX_COMMAND_RUN_TIME, MAX_COMMAND_OUTPUT_LENGTH
|
|
|
|
interrupted = False
|
|
|
|
def enqueue_output(out, q):
|
|
for line in iter(out.readline, ''):
|
|
if interrupted: # Check if the flag is set
|
|
break
|
|
q.put(line)
|
|
out.close()
|
|
|
|
def run_command(command, root_path, q_stdout, q_stderr, pid_container):
|
|
"""
|
|
Execute a command in a subprocess.
|
|
|
|
Args:
|
|
command (str): The command to run.
|
|
root_path (str): The directory in which to run the command.
|
|
q_stdout (Queue): A queue to capture stdout.
|
|
q_stderr (Queue): A queue to capture stderr.
|
|
pid_container (list): A list to store the process ID.
|
|
|
|
Returns:
|
|
subprocess.Popen: The subprocess object.
|
|
"""
|
|
if platform.system() == 'Windows': # Check the operating system
|
|
process = subprocess.Popen(
|
|
command,
|
|
shell=True,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
text=True,
|
|
cwd=root_path
|
|
)
|
|
else:
|
|
process = subprocess.Popen(
|
|
command,
|
|
shell=True,
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
text=True,
|
|
preexec_fn=os.setsid, # Use os.setsid only for Unix-like systems
|
|
cwd=root_path
|
|
)
|
|
|
|
pid_container[0] = process.pid
|
|
t_stdout = threading.Thread(target=enqueue_output, args=(process.stdout, q_stdout))
|
|
t_stderr = threading.Thread(target=enqueue_output, args=(process.stderr, q_stderr))
|
|
t_stdout.daemon = True
|
|
t_stderr.daemon = True
|
|
t_stdout.start()
|
|
t_stderr.start()
|
|
return process
|
|
|
|
def terminate_process(pid):
|
|
if platform.system() == "Windows":
|
|
try:
|
|
subprocess.run(["taskkill", "/F", "/T", "/PID", str(pid)])
|
|
except subprocess.CalledProcessError:
|
|
# Handle any potential errors here
|
|
pass
|
|
else: # Unix-like systems
|
|
try:
|
|
os.killpg(pid, signal.SIGKILL)
|
|
except OSError:
|
|
# Handle any potential errors here
|
|
pass
|
|
|
|
def execute_command(project, command, timeout=None, force=False):
|
|
"""
|
|
Execute a command and capture its output.
|
|
|
|
Args:
|
|
project: The project associated with the command.
|
|
command (str): The command to run.
|
|
timeout (int, optional): The maximum execution time in milliseconds. Default is None.
|
|
force (bool, optional): Whether to execute the command without confirmation. Default is False.
|
|
|
|
Returns:
|
|
str: The command output.
|
|
"""
|
|
if timeout is not None:
|
|
if timeout < 1000:
|
|
timeout *= 1000
|
|
timeout = min(max(timeout, MIN_COMMAND_RUN_TIME), MAX_COMMAND_RUN_TIME)
|
|
|
|
if not force:
|
|
print(yellow(bold(f'\n--------- EXECUTE COMMAND ----------')))
|
|
print(f'Can i execute the command: `' + yellow(bold(command)) + f'` with {timeout}ms timeout?')
|
|
|
|
answer = styled_text(
|
|
project,
|
|
'If yes, just press ENTER'
|
|
)
|
|
|
|
if answer == 'no':
|
|
return '', 'DONE'
|
|
elif answer == 'skip':
|
|
return '', 'DONE'
|
|
|
|
|
|
# TODO when a shell built-in commands (like cd or source) is executed, the output is not captured properly - this will need to be changed at some point
|
|
if "cd " in command or "source " in command:
|
|
command = "bash -c '" + command + "'"
|
|
|
|
|
|
project.command_runs_count += 1
|
|
command_run = get_saved_command_run(project, command)
|
|
if command_run is not None and project.skip_steps:
|
|
# if we do, use it
|
|
project.checkpoints['last_command_run'] = command_run
|
|
print(yellow(f'Restoring command run response id {command_run.id}:\n```\n{command_run.cli_response}```'))
|
|
return command_run.cli_response, None
|
|
|
|
return_value = None
|
|
|
|
q_stderr = queue.Queue()
|
|
q = queue.Queue()
|
|
pid_container = [None]
|
|
process = run_command(command, project.root_path, q, q_stderr, pid_container)
|
|
output = ''
|
|
stderr_output = ''
|
|
start_time = time.time()
|
|
interrupted = False
|
|
|
|
try:
|
|
while True and return_value is None:
|
|
elapsed_time = time.time() - start_time
|
|
if timeout is not None:
|
|
print(white(bold(f'\rt: {round(elapsed_time * 1000)}ms : ')), end='', flush=True)
|
|
|
|
# Check if process has finished
|
|
if process.poll() is not None:
|
|
# Get remaining lines from the queue
|
|
time.sleep(0.1) # TODO this shouldn't be used
|
|
while not q.empty():
|
|
output_line = q.get_nowait()
|
|
if output_line not in output:
|
|
print(green('CLI OUTPUT:') + output_line, end='')
|
|
output += output_line
|
|
break
|
|
|
|
# If timeout is reached, kill the process
|
|
if timeout is not None and elapsed_time * 1000 > timeout:
|
|
raise TimeoutError("Command exceeded the specified timeout.")
|
|
# os.killpg(pid_container[0], signal.SIGKILL)
|
|
# break
|
|
|
|
try:
|
|
line = q.get_nowait()
|
|
except queue.Empty:
|
|
line = None
|
|
|
|
if line:
|
|
output += line
|
|
print(green('CLI OUTPUT:') + line, end='')
|
|
|
|
# Read stderr
|
|
try:
|
|
stderr_line = q_stderr.get_nowait()
|
|
except queue.Empty:
|
|
stderr_line = None
|
|
|
|
if stderr_line:
|
|
stderr_output += stderr_line
|
|
print(red('CLI ERROR:') + stderr_line, end='') # Print with different color for distinction
|
|
|
|
except (KeyboardInterrupt, TimeoutError) as e:
|
|
interrupted = True
|
|
if isinstance(e, KeyboardInterrupt):
|
|
print("\nCTRL+C detected. Stopping command execution...")
|
|
else:
|
|
print("\nTimeout detected. Stopping command execution...")
|
|
|
|
terminate_process(pid_container[0])
|
|
|
|
# stderr_output = ''
|
|
# while not q_stderr.empty():
|
|
# stderr_output += q_stderr.get_nowait()
|
|
|
|
if return_value is None:
|
|
return_value = ''
|
|
if stderr_output != '':
|
|
return_value = 'stderr:\n```\n' + stderr_output[0:MAX_COMMAND_OUTPUT_LENGTH] + '\n```\n'
|
|
return_value += 'stdout:\n```\n' + output[-MAX_COMMAND_OUTPUT_LENGTH:] + '\n```'
|
|
|
|
command_run = save_command_run(project, command, return_value)
|
|
|
|
return return_value, None
|
|
|
|
def build_directory_tree(path, prefix="", ignore=None, is_last=False, files=None, add_descriptions=False):
|
|
"""Build the directory tree structure in tree-like format.
|
|
|
|
Args:
|
|
- path: The starting directory path.
|
|
- prefix: Prefix for the current item, used for recursion.
|
|
- ignore: List of directory names to ignore.
|
|
- is_last: Flag to indicate if the current item is the last in its parent directory.
|
|
|
|
Returns:
|
|
- A string representation of the directory tree.
|
|
"""
|
|
if ignore is None:
|
|
ignore = []
|
|
|
|
if os.path.basename(path) in ignore:
|
|
return ""
|
|
|
|
output = ""
|
|
indent = '| ' if not is_last else ' '
|
|
|
|
if os.path.isdir(path):
|
|
# It's a directory, add its name to the output and then recurse into it
|
|
output += prefix + "|-- " + os.path.basename(path) + ((' - ' + files[os.path.basename(path)].description + ' ' if files and os.path.basename(path) in files and add_descriptions else '')) + "/\n"
|
|
|
|
# List items in the directory
|
|
items = os.listdir(path)
|
|
for index, item in enumerate(items):
|
|
item_path = os.path.join(path, item)
|
|
output += build_directory_tree(item_path, prefix + indent, ignore, index == len(items) - 1, files, add_descriptions)
|
|
|
|
else:
|
|
# It's a file, add its name to the output
|
|
output += prefix + "|-- " + os.path.basename(path) + ((' - ' + files[os.path.basename(path)].description + ' ' if files and os.path.basename(path) in files and add_descriptions else '')) + "\n"
|
|
|
|
return output
|
|
|
|
def execute_command_and_check_cli_response(command, timeout, convo):
|
|
"""
|
|
Execute a command and check its CLI response.
|
|
|
|
Args:
|
|
command (str): The command to run.
|
|
timeout (int): The maximum execution time in milliseconds.
|
|
convo (AgentConvo): The conversation object.
|
|
|
|
Returns:
|
|
tuple: A tuple containing the CLI response and the agent's response.
|
|
"""
|
|
cli_response, response = execute_command(convo.agent.project, command, timeout)
|
|
if response is None:
|
|
response = convo.send_message('dev_ops/ran_command.prompt',
|
|
{ 'cli_response': cli_response, 'command': command })
|
|
return cli_response, response
|
|
|
|
def run_command_until_success(command, timeout, convo, additional_message=None, force=False, return_cli_response=False, is_root_task=False):
|
|
"""
|
|
Run a command until it succeeds or reaches a timeout.
|
|
|
|
Args:
|
|
command (str): The command to run.
|
|
timeout (int): The maximum execution time in milliseconds.
|
|
convo (AgentConvo): The conversation object.
|
|
additional_message (str, optional): Additional message to include in the response.
|
|
force (bool, optional): Whether to execute the command without confirmation. Default is False.
|
|
"""
|
|
cli_response, response = execute_command(convo.agent.project, command, timeout, force)
|
|
if response is None:
|
|
response = convo.send_message('dev_ops/ran_command.prompt',
|
|
{'cli_response': cli_response, 'command': command, 'additional_message': additional_message})
|
|
|
|
if response != 'DONE':
|
|
print(red(f'Got incorrect CLI response:'))
|
|
print(cli_response)
|
|
print(red('-------------------'))
|
|
|
|
reset_branch_id = convo.save_branch()
|
|
while True:
|
|
try:
|
|
# This catch is necessary to return the correct value (cli_response) to continue development function so
|
|
# the developer can debug the appropriate issue
|
|
# this snippet represents the first entry point into debugging recursion because of return_cli_response
|
|
return convo.agent.debugger.debug(convo, {'command': command, 'timeout': timeout})
|
|
except TooDeepRecursionError as e:
|
|
# this is only to put appropriate message in the response after TooDeepRecursionError is raised
|
|
raise TooDeepRecursionError(cli_response) if return_cli_response else e
|
|
except TokenLimitError as e:
|
|
if is_root_task:
|
|
convo.load_branch(reset_branch_id)
|
|
else:
|
|
raise e
|
|
else:
|
|
return { 'success': True, 'cli_response': cli_response }
|