diff --git a/AutonomousAI/main.py b/AutonomousAI/main.py index 60a5f4a3b..5542fbbe2 100644 --- a/AutonomousAI/main.py +++ b/AutonomousAI/main.py @@ -62,44 +62,46 @@ def print_assistant_thoughts(assistant_reply): except Exception as e: print_to_console("Error: \n", Fore.RED, str(e)) -# Construct the prompt -print_to_console("Welcome to Auto-GPT! ", Fore.GREEN, "Enter the name of your AI and its role below. Entering nothing will load defaults.") +def construct_prompt(): + # Construct the prompt + print_to_console("Welcome to Auto-GPT! ", Fore.GREEN, "Enter the name of your AI and its role below. Entering nothing will load defaults.") -# Get AI Name from User -print_to_console("Name your AI: ", Fore.GREEN, "For example, 'Entrepreneur-GPT'") -ai_name = input("AI Name: ") -if ai_name == "": - ai_name = "Entrepreneur-GPT" + # Get AI Name from User + print_to_console("Name your AI: ", Fore.GREEN, "For example, 'Entrepreneur-GPT'") + ai_name = input("AI Name: ") + if ai_name == "": + ai_name = "Entrepreneur-GPT" -# Get AI Role from User -print_to_console("Describe your AI's role: ", Fore.GREEN, "For example, 'an AI designed to autonomously develop and run businesses with the sole goal of increasing your net worth.'") -ai_role = input(f"{ai_name} is: ") -if ai_role == "": - ai_role = "an AI designed to autonomously develop and run businesses with the sole goal of increasing your net worth." + # Get AI Role from User + print_to_console("Describe your AI's role: ", Fore.GREEN, "For example, 'an AI designed to autonomously develop and run businesses with the sole goal of increasing your net worth.'") + ai_role = input(f"{ai_name} is: ") + if ai_role == "": + ai_role = "an AI designed to autonomously develop and run businesses with the sole goal of increasing your net worth." -# Enter up to 5 goals for the AI -print_to_console("Enter up to 5 goals for your AI: ", Fore.GREEN, "For example: \nIncrease net worth \nGrow Twitter Account \nDevelop and manage multiple businesses autonomously'") -print("Enter nothing to load defaults, enter nothing when finished.", flush=True) -ai_goals = [] -for i in range(5): - ai_goal = input(f"Goal {i+1}: ") - if ai_goal == "": - break - ai_goals.append(ai_goal) -if len(ai_goals) == 0: - ai_goals = ["Increase net worth", "Grow Twitter Account", "Develop and manage multiple businesses autonomously"] + # Enter up to 5 goals for the AI + print_to_console("Enter up to 5 goals for your AI: ", Fore.GREEN, "For example: \nIncrease net worth \nGrow Twitter Account \nDevelop and manage multiple businesses autonomously'") + print("Enter nothing to load defaults, enter nothing when finished.", flush=True) + ai_goals = [] + for i in range(5): + ai_goal = input(f"Goal {i+1}: ") + if ai_goal == "": + break + ai_goals.append(ai_goal) + if len(ai_goals) == 0: + ai_goals = ["Increase net worth", "Grow Twitter Account", "Develop and manage multiple businesses autonomously"] -prompt = data.load_prompt() -prompt_start = """Your decisions must always be made independently without seeking user assistance. Play to your strengths as an LLM and pursue simple strategies with no legal complications.""" + prompt = data.load_prompt() + prompt_start = """Your decisions must always be made independently without seeking user assistance. Play to your strengths as an LLM and pursue simple strategies with no legal complications.""" -# Construct full prompt -full_prompt = f"You are {ai_name}, {ai_role}\n{prompt_start}\n\nGOALS:\n\n" -for i, goal in enumerate(ai_goals): - full_prompt += f"{i+1}. {goal}\n" - -full_prompt += f"\n\n{prompt}" -prompt = full_prompt -print(f"Prompt: {prompt}") + # Construct full prompt + full_prompt = f"You are {ai_name}, {ai_role}\n{prompt_start}\n\nGOALS:\n\n" + for i, goal in enumerate(ai_goals): + full_prompt += f"{i+1}. {goal}\n" + + full_prompt += f"\n\n{prompt}" + return full_prompt + +prompt = construct_prompt() # Initialize variables full_message_history = []