put loop in in if main

scripts/main.py

@@ -316,27 +316,27 @@ def parse_arguments():
             cfg.memory_backend = chosen
 
 
-# TODO: fill in llm values here
-check_openai_api_key()
-parse_arguments()
-logger.set_level(logging.DEBUG if cfg.debug_mode else logging.INFO)
-ai_name = ""
-prompt = construct_prompt()
-# print(prompt)
-# Initialize variables
-full_message_history = []
-result = None
-next_action_count = 0
-# Make a constant:
-user_input = "Determine which next command to use, and respond using the format specified above:"
-
-# Initialize memory and make sure it is empty.
-# this is particularly important for indexing and referencing pinecone memory
-memory = get_memory(cfg, init=True)
-print('Using memory of type: ' + memory.__class__.__name__)
-
-# Interaction Loop
-while True:
+def main():
+    global ai_name, memory
+    # TODO: fill in llm values here
+    check_openai_api_key()
+    parse_arguments()
+    logger.set_level(logging.DEBUG if cfg.debug_mode else logging.INFO)
+    ai_name = ""
+    prompt = construct_prompt()
+    # print(prompt)
+    # Initialize variables
+    full_message_history = []
+    result = None
+    next_action_count = 0
+    # Make a constant:
+    user_input = "Determine which next command to use, and respond using the format specified above:"
+    # Initialize memory and make sure it is empty.
+    # this is particularly important for indexing and referencing pinecone memory
+    memory = get_memory(cfg, init=True)
+    print('Using memory of type: ' + memory.__class__.__name__)
+    # Interaction Loop
+    while True:
         # Send message to AI, get response
         with Spinner("Thinking... "):
             assistant_reply = chat.chat_with_ai(
@@ -351,7 +351,8 @@ while True:
 
         # Get command name and arguments
         try:
-        command_name, arguments = cmd.get_command(attempt_to_fix_json_by_finding_outermost_brackets(assistant_reply))
+            command_name, arguments = cmd.get_command(
+                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:
@@ -406,7 +407,7 @@ while True:
                 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" ):
+        if command_name is not None and command_name.lower().startswith("error"):
             result = f"Command {command_name} threw the following error: " + arguments
         elif command_name == "human_feedback":
             result = f"Human feedback: {user_input}"
@@ -431,3 +432,7 @@ while True:
                 chat.create_chat_message(
                     "system", "Unable to execute command"))
             logger.typewriter_log("SYSTEM: ", Fore.YELLOW, "Unable to execute command")
+
+
+if __name__ == "__main__":
+    main()