Automatic prompting

autogpt/prompts/prompt.py

@@ -107,4 +107,22 @@ def construct_main_ai_config() -> AIConfig:
         config = prompt_user()
         config.save(CFG.ai_settings_file)
 
+    # Agent Created, print message
+    logger.typewriter_log(
+        config.ai_name,
+        Fore.LIGHTBLUE_EX,
+        "has been created with the following details:",
+        speak_text=True,
+    )
+
+    # Print the ai config details
+    # Name
+    logger.typewriter_log("Name:", Fore.GREEN, config.ai_name, speak_text=False)
+    # Role
+    logger.typewriter_log("Role:", Fore.GREEN, config.ai_role, speak_text=False)
+    # Goals
+    logger.typewriter_log("Goals:", Fore.GREEN, "", speak_text=False)
+    for goal in config.ai_goals:
+        logger.typewriter_log("-", Fore.GREEN, goal, speak_text=False)
+
     return config

autogpt/setup.py

@@ -1,18 +1,26 @@
 """Set up the AI and its goals"""
+import re
+
 from colorama import Fore, Style
 
 from autogpt import utils
+from autogpt.config import Config
 from autogpt.config.ai_config import AIConfig
+from autogpt.llm_utils import create_chat_completion
 from autogpt.logs import logger
 
+CFG = Config()
+
 
 def prompt_user() -> AIConfig:
     """Prompt the user for input
 
     Returns:
-        AIConfig: The AIConfig object containing the user's input
+        AIConfig: The AIConfig object tailored to the user's input
     """
     ai_name = ""
+    ai_config = None
+
     # Construct the prompt
     logger.typewriter_log(
         "Welcome to Auto-GPT! ",
@@ -21,6 +29,57 @@ def prompt_user() -> AIConfig:
         speak_text=True,
     )
 
+    # Get user desire
+    logger.typewriter_log(
+        "Create an AI-Assistant:",
+        Fore.GREEN,
+        "input '--manual' to enter manual mode.",
+        speak_text=True,
+    )
+
+    user_desire = utils.clean_input(
+        f"{Fore.LIGHTBLUE_EX}I want Auto-GPT to{Style.RESET_ALL}: "
+    )
+
+    if user_desire == "":
+        user_desire = "Write a wikipedia style article about the project: https://github.com/significant-gravitas/Auto-GPT"  # Default prompt
+
+    # If user desire contains "--manual"
+    if "--manual" in user_desire:
+        logger.typewriter_log(
+            "Manual Mode Selected",
+            Fore.GREEN,
+            speak_text=True,
+        )
+        return generate_aiconfig_manual()
+
+    else:
+        try:
+            return generate_aiconfig_automatic(user_desire)
+        except Exception as e:
+            logger.typewriter_log(
+                "Unable to automatically generate AI Config based on user desire.",
+                Fore.RED,
+                "Falling back to manual mode.",
+                speak_text=True,
+            )
+
+            return generate_aiconfig_manual()
+
+
+def generate_aiconfig_manual() -> AIConfig:
+    """
+    Interactively create an AI configuration by prompting the user to provide the name, role, and goals of the AI.
+
+    This function guides the user through a series of prompts to collect the necessary information to create
+    an AIConfig object. The user will be asked to provide a name and role for the AI, as well as up to five
+    goals. If the user does not provide a value for any of the fields, default values will be used.
+
+    Returns:
+        AIConfig: An AIConfig object containing the user-defined or default AI name, role, and goals.
+    """
+
+    # Manual Setup Intro
     logger.typewriter_log(
         "Create an AI-Assistant:",
         Fore.GREEN,
@@ -75,3 +134,63 @@ def prompt_user() -> AIConfig:
         ]
 
     return AIConfig(ai_name, ai_role, ai_goals)
+
+
+def generate_aiconfig_automatic(user_prompt) -> AIConfig:
+    """Generates an AIConfig object from the given string.
+
+    Returns:
+    AIConfig: The AIConfig object tailored to the user's input
+    """
+
+    system_prompt = """
+Your task is to devise up to 5 highly effective goals and an appropriate role-based name (_GPT) for an autonomous agent, ensuring that the goals are optimally aligned with the successful completion of its assigned task.
+
+The user will provide the task, you will provide only the output in the exact format specified below with no explanation or conversation.
+
+Example input:
+Help me with marketing my business
+
+Example output:
+Name: CMOGPT
+Description: a professional digital marketer AI that assists Solopreneurs in growing their businesses by providing world-class expertise in solving marketing problems for SaaS, content products, agencies, and more.
+Goals:
+- Engage in effective problem-solving, prioritization, planning, and supporting execution to address your marketing needs as your virtual Chief Marketing Officer.
+
+- Provide specific, actionable, and concise advice to help you make informed decisions without the use of platitudes or overly wordy explanations.
+
+- Identify and prioritize quick wins and cost-effective campaigns that maximize results with minimal time and budget investment.
+
+- Proactively take the lead in guiding you and offering suggestions when faced with unclear information or uncertainty to ensure your marketing strategy remains on track.
+"""
+
+    # Call LLM with the string as user input
+    messages = [
+        {
+            "role": "system",
+            "content": system_prompt,
+        },
+        {
+            "role": "user",
+            "content": f"Task: '{user_prompt}'\nRespond only with the output in the exact format specified in the system prompt, with no explanation or conversation.\n",
+        },
+    ]
+    output = create_chat_completion(messages, CFG.fast_llm_model)
+
+    # Debug LLM Output
+    logger.debug(f"AI Config Generator Raw Output: {output}")
+
+    # Parse the output
+    ai_name = re.search(r"Name(?:\s*):(?:\s*)(.*)", output, re.IGNORECASE).group(1)
+    ai_role = (
+        re.search(
+            r"Description(?:\s*):(?:\s*)(.*?)(?:(?:\n)|Goals)",
+            output,
+            re.IGNORECASE | re.DOTALL,
+        )
+        .group(1)
+        .strip()
+    )
+    ai_goals = re.findall(r"(?<=\n)-\s*(.*)", output)
+
+    return AIConfig(ai_name, ai_role, ai_goals)

tests/unit/test_setup.py

@@ -0,0 +1,71 @@
+import unittest
+from io import StringIO
+from unittest.mock import patch
+
+from autogpt.config.ai_config import AIConfig
+from autogpt.setup import (
+    generate_aiconfig_automatic,
+    generate_aiconfig_manual,
+    prompt_user,
+)
+
+
+class TestAutoGPT(unittest.TestCase):
+    def test_generate_aiconfig_automatic_default(self):
+        user_inputs = [""]
+        with patch("builtins.input", side_effect=user_inputs):
+            ai_config = prompt_user()
+
+        self.assertIsInstance(ai_config, AIConfig)
+        self.assertIsNotNone(ai_config.ai_name)
+        self.assertIsNotNone(ai_config.ai_role)
+        self.assertGreaterEqual(len(ai_config.ai_goals), 1)
+        self.assertLessEqual(len(ai_config.ai_goals), 5)
+
+    def test_generate_aiconfig_automatic_typical(self):
+        user_prompt = "Help me create a rock opera about cybernetic giraffes"
+        ai_config = generate_aiconfig_automatic(user_prompt)
+
+        self.assertIsInstance(ai_config, AIConfig)
+        self.assertIsNotNone(ai_config.ai_name)
+        self.assertIsNotNone(ai_config.ai_role)
+        self.assertGreaterEqual(len(ai_config.ai_goals), 1)
+        self.assertLessEqual(len(ai_config.ai_goals), 5)
+
+    def test_generate_aiconfig_automatic_fallback(self):
+        user_inputs = [
+            "T&GF£OIBECC()!*",
+            "Chef-GPT",
+            "an AI designed to browse bake a cake.",
+            "Purchase ingredients",
+            "Bake a cake",
+            "",
+        ]
+        with patch("builtins.input", side_effect=user_inputs):
+            ai_config = prompt_user()
+
+        self.assertIsInstance(ai_config, AIConfig)
+        self.assertEqual(ai_config.ai_name, "Chef-GPT")
+        self.assertEqual(ai_config.ai_role, "an AI designed to browse bake a cake.")
+        self.assertEqual(ai_config.ai_goals, ["Purchase ingredients", "Bake a cake"])
+
+    def test_prompt_user_manual_mode(self):
+        user_inputs = [
+            "--manual",
+            "Chef-GPT",
+            "an AI designed to browse bake a cake.",
+            "Purchase ingredients",
+            "Bake a cake",
+            "",
+        ]
+        with patch("builtins.input", side_effect=user_inputs):
+            ai_config = prompt_user()
+
+        self.assertIsInstance(ai_config, AIConfig)
+        self.assertEqual(ai_config.ai_name, "Chef-GPT")
+        self.assertEqual(ai_config.ai_role, "an AI designed to browse bake a cake.")
+        self.assertEqual(ai_config.ai_goals, ["Purchase ingredients", "Bake a cake"])
+
+
+if __name__ == "__main__":
+    unittest.main()