First Windows Version

This commit is contained in:
KAuser2094
2023-03-06 10:11:37 +00:00
parent 384784ffe6
commit f5280a1b1b
3 changed files with 14 additions and 14 deletions
+6 -10
View File
@@ -1,16 +1,12 @@
# Installs yolo in the user's home directory
# Installs yolo in the user's CMD directory ( change $CMD to desired file path )
TARGET_DIR=~/yolo-ai-cmdbot
TARGET_DIR=$CMD
TARGET_FULLPATH=$TARGET_DIR/yolo.py
mkdir -p $TARGET_DIR
cp yolo.py prompt.txt $TARGET_DIR
cp yolo.py $TARGET_DIR
cp prompt.txt $TARGET_DIR/yolo_prompt.txt #changed this so it doesnt conflict with possible future prompts
chmod +x $TARGET_FULLPATH
# Creates two aliases for use
alias yolo=$TARGET_FULLPATH
alias computer=$TARGET_FULLPATH
# Add the aliases to the logon scripts
echo "alias yolo=$TARGET_FULLPATH" >> ~/.bash_aliases
echo "alias computer=$TARGET_FULLPATH" >> ~/.bash_aliases
# Adds batch file to System32 so you can call it in powershell
cp yolo.bat C:/Windows/System32
+3
View File
@@ -0,0 +1,3 @@
@echo off
set /P "prompt=Enter Prompt: "
python.exe C:\Users\kause\Downloads\Commands\yolo.py %prompt%
+5 -4
View File
@@ -12,6 +12,8 @@ import subprocess
from termcolor import colored
from colorama import init
import dotenv
# Check if the user globally disabled the safety switch
def get_yolo_safety_switch_config():
@@ -32,7 +34,7 @@ def get_full_prompt(user_prompt, shell):
prompt_path = os.path.dirname(yolo_path)
## Load the prompt and prep it
prompt_file = os.path.join(prompt_path, "prompt.txt")
prompt_file = os.path.join(prompt_path, "yolo_prompt.txt")
pre_prompt = open(prompt_file,"r").read()
pre_prompt = pre_prompt.replace("{shell}", shell)
pre_prompt = pre_prompt.replace("{os}", get_os_friendly_name())
@@ -78,10 +80,9 @@ if __name__ == "__main__":
ask_flag = False # safety switch -a command line argument
yolo = "" # user's answer to safety switch (-a) question y/n
# Two options for the user to specify they openai api key
home_path = os.path.expanduser("~")
# Presumes the folder yolo.py is copied to has a .env
dotenv.load_dotenv(".env")
openai.api_key = os.getenv("OPENAI_API_KEY")
openai.api_key_path = os.path.join(home_path,".openai.apikey")
# Parse arguments and make sure we have at least a single word
if len(sys.argv) < 2: