First Windows Version
This commit is contained in:
+6
-10
@@ -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
|
TARGET_FULLPATH=$TARGET_DIR/yolo.py
|
||||||
|
|
||||||
mkdir -p $TARGET_DIR
|
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
|
chmod +x $TARGET_FULLPATH
|
||||||
|
|
||||||
# Creates two aliases for use
|
# Adds batch file to System32 so you can call it in powershell
|
||||||
alias yolo=$TARGET_FULLPATH
|
cp yolo.bat C:/Windows/System32
|
||||||
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
|
|
||||||
@@ -0,0 +1,3 @@
|
|||||||
|
@echo off
|
||||||
|
set /P "prompt=Enter Prompt: "
|
||||||
|
python.exe C:\Users\kause\Downloads\Commands\yolo.py %prompt%
|
||||||
@@ -12,6 +12,8 @@ import subprocess
|
|||||||
from termcolor import colored
|
from termcolor import colored
|
||||||
from colorama import init
|
from colorama import init
|
||||||
|
|
||||||
|
import dotenv
|
||||||
|
|
||||||
# Check if the user globally disabled the safety switch
|
# Check if the user globally disabled the safety switch
|
||||||
def get_yolo_safety_switch_config():
|
def get_yolo_safety_switch_config():
|
||||||
|
|
||||||
@@ -32,7 +34,7 @@ def get_full_prompt(user_prompt, shell):
|
|||||||
prompt_path = os.path.dirname(yolo_path)
|
prompt_path = os.path.dirname(yolo_path)
|
||||||
|
|
||||||
## Load the prompt and prep it
|
## 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 = open(prompt_file,"r").read()
|
||||||
pre_prompt = pre_prompt.replace("{shell}", shell)
|
pre_prompt = pre_prompt.replace("{shell}", shell)
|
||||||
pre_prompt = pre_prompt.replace("{os}", get_os_friendly_name())
|
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
|
ask_flag = False # safety switch -a command line argument
|
||||||
yolo = "" # user's answer to safety switch (-a) question y/n
|
yolo = "" # user's answer to safety switch (-a) question y/n
|
||||||
|
|
||||||
# Two options for the user to specify they openai api key
|
# Presumes the folder yolo.py is copied to has a .env
|
||||||
home_path = os.path.expanduser("~")
|
dotenv.load_dotenv(".env")
|
||||||
openai.api_key = os.getenv("OPENAI_API_KEY")
|
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
|
# Parse arguments and make sure we have at least a single word
|
||||||
if len(sys.argv) < 2:
|
if len(sys.argv) < 2:
|
||||||
|
|||||||
Reference in New Issue
Block a user