formatting

Author: Swiftyos

autogpt/__main__.py

@@ -1,14 +1,14 @@
-import autogpt.commands as cmd
 import json
 import traceback
 from tkinter.ttk import Style
 
 from colorama import Fore
 
 import autogpt.chat
+import autogpt.commands as cmd
+import autogpt.speak
 from autogpt.config import Config
 from autogpt.logger import logger
-import autogpt.speak
 from autogpt.spinner import Spinner
 
 
@@ -24,13 +24,16 @@ class Agent:
         user_input: The user input.
 
     """
-    def __init__(self,
-                 ai_name,
-                 memory,
-                 full_message_history,
-                 next_action_count,
-                 prompt,
-                 user_input):
+
+    def __init__(
+        self,
+        ai_name,
+        memory,
+        full_message_history,
+        next_action_count,
+        prompt,
+        user_input,
+    ):
         self.ai_name = ai_name
         self.memory = memory
         self.full_message_history = full_message_history
@@ -43,10 +46,16 @@ def start_interaction_loop(self):
         cfg = Config()
         loop_count = 0
         while True:
-             # Discontinue if continuous limit is reached
+            # Discontinue if continuous limit is reached
             loop_count += 1
-            if cfg.continuous_mode and cfg.continuous_limit > 0 and loop_count > cfg.continuous_limit:
-                logger.typewriter_log("Continuous Limit Reached: ", Fore.YELLOW, f"{cfg.continuous_limit}")
+            if (
+                cfg.continuous_mode
+                and cfg.continuous_limit > 0
+                and loop_count > cfg.continuous_limit
+            ):
+                logger.typewriter_log(
+                    "Continuous Limit Reached: ", Fore.YELLOW, f"{cfg.continuous_limit}"
+                )
                 break
 
             # Send message to AI, get response
@@ -56,15 +65,17 @@ def start_interaction_loop(self):
                     self.user_input,
                     self.full_message_history,
                     self.memory,
-                    cfg.fast_token_limit)  # TODO: This hardcodes the model to use GPT3.5. Make this an argument
+                    cfg.fast_token_limit,
+                )  # TODO: This hardcodes the model to use GPT3.5. Make this an argument
 
             # Print Assistant thoughts
             print_assistant_thoughts(assistant_reply)
 
             # Get command name and arguments
             try:
                 command_name, arguments = cmd.get_command(
-                    attempt_to_fix_json_by_finding_outermost_brackets(assistant_reply))
+                    attempt_to_fix_json_by_finding_outermost_brackets(assistant_reply)
+                )
                 if cfg.speak_mode:
                     speak.say_text(f"I want to execute {command_name}")
             except Exception as e:
@@ -78,21 +89,29 @@ def start_interaction_loop(self):
                 logger.typewriter_log(
                     "NEXT ACTION: ",
                     Fore.CYAN,
-                    f"COMMAND = {Fore.CYAN}{command_name}{Style.RESET_ALL}  ARGUMENTS = {Fore.CYAN}{arguments}{Style.RESET_ALL}")
+                    f"COMMAND = {Fore.CYAN}{command_name}{Style.RESET_ALL}  ARGUMENTS = {Fore.CYAN}{arguments}{Style.RESET_ALL}",
+                )
                 print(
                     f"Enter 'y' to authorise command, 'y -N' to run N continuous commands, 'n' to exit program, or enter feedback for {self.ai_name}...",
-                    flush=True)
+                    flush=True,
+                )
                 while True:
-                    console_input = utils.clean_input(Fore.MAGENTA + "Input:" + Style.RESET_ALL)
+                    console_input = utils.clean_input(
+                        Fore.MAGENTA + "Input:" + Style.RESET_ALL
+                    )
                     if console_input.lower().rstrip() == "y":
                         self.user_input = "GENERATE NEXT COMMAND JSON"
                         break
                     elif console_input.lower().startswith("y -"):
                         try:
-                            self.next_action_count = abs(int(console_input.split(" ")[1]))
+                            self.next_action_count = abs(
+                                int(console_input.split(" ")[1])
+                            )
                             self.user_input = "GENERATE NEXT COMMAND JSON"
                         except ValueError:
-                            print("Invalid input format. Please enter 'y -n' where n is the number of continuous tasks.")
+                            print(
+                                "Invalid input format. Please enter 'y -n' where n is the number of continuous tasks."
+                            )
                             continue
                         break
                     elif console_input.lower() == "n":
@@ -107,7 +126,8 @@ def start_interaction_loop(self):
                     logger.typewriter_log(
                         "-=-=-=-=-=-=-= COMMAND AUTHORISED BY USER -=-=-=-=-=-=-=",
                         Fore.MAGENTA,
-                        "")
+                        "",
+                    )
                 elif self.user_input == "EXIT":
                     print("Exiting...", flush=True)
                     break
@@ -116,54 +136,68 @@ def start_interaction_loop(self):
                 logger.typewriter_log(
                     "NEXT ACTION: ",
                     Fore.CYAN,
-                    f"COMMAND = {Fore.CYAN}{command_name}{Style.RESET_ALL}  ARGUMENTS = {Fore.CYAN}{arguments}{Style.RESET_ALL}")
+                    f"COMMAND = {Fore.CYAN}{command_name}{Style.RESET_ALL}  ARGUMENTS = {Fore.CYAN}{arguments}{Style.RESET_ALL}",
+                )
 
             # Execute command
             if command_name is not None and command_name.lower().startswith("error"):
-                result = f"Command {command_name} threw the following error: " + arguments
+                result = (
+                    f"Command {command_name} threw the following error: " + arguments
+                )
             elif command_name == "human_feedback":
                 result = f"Human feedback: {self.user_input}"
             else:
                 result = f"Command {command_name} returned: {cmd.execute_command(command_name, arguments)}"
                 if self.next_action_count > 0:
                     self.next_action_count -= 1
 
-            memory_to_add = f"Assistant Reply: {assistant_reply} " \
-                            f"\nResult: {result} " \
-                            f"\nHuman Feedback: {self.user_input} "
+            memory_to_add = (
+                f"Assistant Reply: {assistant_reply} "
+                f"\nResult: {result} "
+                f"\nHuman Feedback: {self.user_input} "
+            )
 
             self.memory.add(memory_to_add)
 
             # Check if there's a result from the command append it to the message
             # history
             if result is not None:
-                self.full_message_history.append(chat.create_chat_message("system", result))
+                self.full_message_history.append(
+                    chat.create_chat_message("system", result)
+                )
                 logger.typewriter_log("SYSTEM: ", Fore.YELLOW, result)
             else:
                 self.full_message_history.append(
-                    chat.create_chat_message(
-                        "system", "Unable to execute command"))
-                logger.typewriter_log("SYSTEM: ", Fore.YELLOW, "Unable to execute command")
+                    chat.create_chat_message("system", "Unable to execute command")
+                )
+                logger.typewriter_log(
+                    "SYSTEM: ", Fore.YELLOW, "Unable to execute command"
+                )
 
 
 def attempt_to_fix_json_by_finding_outermost_brackets(json_string):
     cfg = Config()
     if cfg.speak_mode and cfg.debug_mode:
-      speak.say_text("I have received an invalid JSON response from the OpenAI API. Trying to fix it now.")
+        speak.say_text(
+            "I have received an invalid JSON response from the OpenAI API. Trying to fix it now."
+        )
     logger.typewriter_log("Attempting to fix JSON by finding outermost brackets\n")
 
     try:
         # Use regex to search for JSON objects
         import regex
+
         json_pattern = regex.compile(r"\{(?:[^{}]|(?R))*\}")
         json_match = json_pattern.search(json_string)
 
         if json_match:
             # Extract the valid JSON object from the string
             json_string = json_match.group(0)
-            logger.typewriter_log(title="Apparently json was fixed.", title_color=Fore.GREEN)
+            logger.typewriter_log(
+                title="Apparently json was fixed.", title_color=Fore.GREEN
+            )
             if cfg.speak_mode and cfg.debug_mode:
-               speak.say_text("Apparently json was fixed.")
+                speak.say_text("Apparently json was fixed.")
         else:
             raise ValueError("No valid JSON object found")
 
@@ -187,7 +221,9 @@ def print_assistant_thoughts(assistant_reply):
             assistant_reply_json = fix_and_parse_json(assistant_reply)
         except json.JSONDecodeError as e:
             logger.error("Error: Invalid JSON in assistant thoughts\n", assistant_reply)
-            assistant_reply_json = attempt_to_fix_json_by_finding_outermost_brackets(assistant_reply)
+            assistant_reply_json = attempt_to_fix_json_by_finding_outermost_brackets(
+                assistant_reply
+            )
             assistant_reply_json = fix_and_parse_json(assistant_reply_json)
 
         # Check if assistant_reply_json is a string and attempt to parse it into a JSON object
@@ -196,7 +232,11 @@ def print_assistant_thoughts(assistant_reply):
                 assistant_reply_json = json.loads(assistant_reply_json)
             except json.JSONDecodeError as e:
                 logger.error("Error: Invalid JSON\n", assistant_reply)
-                assistant_reply_json = attempt_to_fix_json_by_finding_outermost_brackets(assistant_reply_json)
+                assistant_reply_json = (
+                    attempt_to_fix_json_by_finding_outermost_brackets(
+                        assistant_reply_json
+                    )
+                )
 
         assistant_thoughts_reasoning = None
         assistant_thoughts_plan = None
@@ -211,7 +251,9 @@ def print_assistant_thoughts(assistant_reply):
             assistant_thoughts_criticism = assistant_thoughts.get("criticism")
             assistant_thoughts_speak = assistant_thoughts.get("speak")
 
-        logger.typewriter_log(f"{ai_name.upper()} THOUGHTS:", Fore.YELLOW, assistant_thoughts_text)
+        logger.typewriter_log(
+            f"{ai_name.upper()} THOUGHTS:", Fore.YELLOW, assistant_thoughts_text
+        )
         logger.typewriter_log("REASONING:", Fore.YELLOW, assistant_thoughts_reasoning)
 
         if assistant_thoughts_plan:
@@ -223,7 +265,7 @@ def print_assistant_thoughts(assistant_reply):
                 assistant_thoughts_plan = str(assistant_thoughts_plan)
 
             # Split the input_string using the newline character and dashes
-            lines = assistant_thoughts_plan.split('\n')
+            lines = assistant_thoughts_plan.split("\n")
             for line in lines:
                 line = line.lstrip("- ")
                 logger.typewriter_log("- ", Fore.GREEN, line.strip())
@@ -237,7 +279,9 @@ def print_assistant_thoughts(assistant_reply):
     except json.decoder.JSONDecodeError as e:
         logger.error("Error: Invalid JSON\n", assistant_reply)
         if cfg.speak_mode:
-            speak.say_text("I have received an invalid JSON response from the OpenAI API. I cannot ignore this response.")
+            speak.say_text(
+                "I have received an invalid JSON response from the OpenAI API. I cannot ignore this response."
+            )
 
     # All other errors, return "Error: + error message"
     except Exception as e:

autogpt/agent.py

@@ -12,7 +12,9 @@ def create_agent(task, prompt, model):
     global next_key
     global agents
 
-    messages = [{"role": "user", "content": prompt}, ]
+    messages = [
+        {"role": "user", "content": prompt},
+    ]
 
     # Start GPT instance
     agent_reply = create_chat_completion(

autogpt/agent_manager.py

@@ -1,5 +1,7 @@
-import yaml
 import os
+
+import yaml
+
 from autogpt.prompt import get_prompt
 
 
@@ -13,7 +15,9 @@ class AIConfig:
         ai_goals (list): The list of objectives the AI is supposed to complete.
     """
 
-    def __init__(self, ai_name: str="", ai_role: str="", ai_goals: list=[]) -> None:
+    def __init__(
+        self, ai_name: str = "", ai_role: str = "", ai_goals: list = []
+    ) -> None:
         """
         Initialize a class instance
 
@@ -30,10 +34,10 @@ def __init__(self, ai_name: str="", ai_role: str="", ai_goals: list=[]) -> None:
         self.ai_goals = ai_goals
 
     # Soon this will go in a folder where it remembers more stuff about the run(s)
-    SAVE_FILE = os.path.join(os.path.dirname(__file__), '..', 'ai_settings.yaml')
+    SAVE_FILE = os.path.join(os.path.dirname(__file__), "..", "ai_settings.yaml")
 
     @classmethod
-    def load(cls: object, config_file: str=SAVE_FILE) -> object:
+    def load(cls: object, config_file: str = SAVE_FILE) -> object:
         """
         Returns class object with parameters (ai_name, ai_role, ai_goals) loaded from yaml file if yaml file exists,
         else returns class with no parameters.
@@ -47,7 +51,7 @@ def load(cls: object, config_file: str=SAVE_FILE) -> object:
         """
 
         try:
-            with open(config_file, encoding='utf-8') as file:
+            with open(config_file, encoding="utf-8") as file:
                 config_params = yaml.load(file, Loader=yaml.FullLoader)
         except FileNotFoundError:
             config_params = {}
@@ -58,7 +62,7 @@ def load(cls: object, config_file: str=SAVE_FILE) -> object:
 
         return cls(ai_name, ai_role, ai_goals)
 
-    def save(self, config_file: str=SAVE_FILE) -> None:
+    def save(self, config_file: str = SAVE_FILE) -> None:
         """
         Saves the class parameters to the specified file yaml file path as a yaml file.
 
@@ -69,8 +73,12 @@ def save(self, config_file: str=SAVE_FILE) -> None:
             None
         """
 
-        config = {"ai_name": self.ai_name, "ai_role": self.ai_role, "ai_goals": self.ai_goals}
-        with open(config_file, "w",  encoding='utf-8') as file:
+        config = {
+            "ai_name": self.ai_name,
+            "ai_role": self.ai_role,
+            "ai_goals": self.ai_goals,
+        }
+        with open(config_file, "w", encoding="utf-8") as file:
             yaml.dump(config, file, allow_unicode=True)
 
     def construct_full_prompt(self) -> str:
@@ -87,7 +95,9 @@ def construct_full_prompt(self) -> str:
         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 {self.ai_name}, {self.ai_role}\n{prompt_start}\n\nGOALS:\n\n"
+        full_prompt = (
+            f"You are {self.ai_name}, {self.ai_role}\n{prompt_start}\n\nGOALS:\n\n"
+        )
         for i, goal in enumerate(self.ai_goals):
             full_prompt += f"{i+1}. {goal}\n"
 

autogpt/ai_config.py

@@ -1,7 +1,9 @@
-from typing import List
 import json
-from autogpt.config import Config
+from typing import List
+
 from autogpt.call_ai_function import call_ai_function
+from autogpt.config import Config
+
 cfg = Config()