diff --git a/install.sh b/install.sh index 73ac811..2db7871 100644 --- a/install.sh +++ b/install.sh @@ -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 \ No newline at end of file diff --git a/yolo.bat b/yolo.bat new file mode 100644 index 0000000..8eac22f --- /dev/null +++ b/yolo.bat @@ -0,0 +1,3 @@ +@echo off +set /P "prompt=Enter Prompt: " +python.exe C:\Users\kause\Downloads\Commands\yolo.py %prompt% diff --git a/yolo.py b/yolo.py index 0333f45..226ec6e 100755 --- a/yolo.py +++ b/yolo.py @@ -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: