schihei/yolo-ai-cmdbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix safety bug (allow to disable it)

Browse Source
This commit is contained in:
wunderwuzzi23
2023-04-04 17:46:51 -07:00
parent 346d714c47
commit 14bc8fd802
1 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
yolo.py
+15 -9
View File
@@ -50,22 +50,24 @@ def get_full_prompt(user_prompt, shell):
return prompt return prompt
def print_usage(): def print_usage():
print("Yolo v0.2 - by @wunderwuzzi23") print("Yolo v0.2.1 - by @wunderwuzzi23")
print() print()
print("Usage: yolo [-a] list the current directory information") print("Usage: yolo [-a] list the current directory information")
print("Argument: -a: Prompt the user before running the command (only useful when safety is off)") print("Argument: -a: Prompt the user before running the command (only useful when safety is off)")
print() print()
yolo_safety_switch = "on"
if config["safety"] != True: #Ensure safety switch has a correct value
yolo_safety_switch = "off" yolo_safety_switch = bool(config["safety"])
# if config["safety"] != True:
# yolo_safety_switch = "off"
print("Current configuration per yolo.yaml:") print("Current configuration per yolo.yaml:")
print("* Model : " + str(config["model"])) print("* Model : " + str(config["model"]))
print("* Temperature : " + str(config["temperature"])) print("* Temperature : " + str(config["temperature"]))
print("* Max. Tokens : " + str(config["max_tokens"])) print("* Max. Tokens : " + str(config["max_tokens"]))
print("* Safety : " + yolo_safety_switch) print("* Safety : " + str(yolo_safety_switch))
def get_os_friendly_name(): def get_os_friendly_name():
@@ -178,16 +180,20 @@ def missing_posix_display():
display = subprocess.check_output("echo $DISPLAY", shell=True) display = subprocess.check_output("echo $DISPLAY", shell=True)
return display == b'\n' return display == b'\n'
def prompt_user_input(response): def prompt_user_input(response):
print("Command: " + colored(response, 'blue')) print("Command: " + colored(response, 'blue'))
if config["safety"] != "off" or ask_flag == True: #print(config["safety"])
if bool(config["safety"]) == True or ask_flag == True:
prompt_text = "Execute command? [Y]es [n]o [m]odify [c]opy to clipboard ==> " prompt_text = "Execute command? [Y]es [n]o [m]odify [c]opy to clipboard ==> "
if os.name == "posix" and missing_posix_display(): if os.name == "posix" and missing_posix_display():
prompt_text = "Execute command? [Y]es [n]o [m]odify ==> " prompt_text = "Execute command? [Y]es [n]o [m]odify ==> "
print(prompt_text, end = '') print(prompt_text, end = '')
user_input = input() user_input = input()
return user_input return user_input
if config["safety"] == False:
return "Y"
def evaluate_input(user_input, command): def evaluate_input(user_input, command):
if user_input.upper() == "Y" or user_input == "": if user_input.upper() == "Y" or user_input == "":
@@ -216,6 +222,6 @@ def evaluate_input(user_input, command):
res_command = call_open_ai(user_prompt) res_command = call_open_ai(user_prompt)
check_for_issue(res_command) check_for_issue(res_command)
check_for_markdown(res_command) check_for_markdown(res_command)
user_iput = prompt_user_input(res_command) user_input = prompt_user_input(res_command)
print() print()
evaluate_input(user_iput, res_command) evaluate_input(user_input, res_command)