black formatting

1 files changed, 410 insertions, 153 deletions

diff --git a/cli.py b/cli.py
index d892139a9..6d0d8dde6 100644
--- a/cli.py
+++ b/cli.py
@@ -3,8 +3,9 @@ try:
     import github
 except ImportError:
     import os
-    os.system('pip3 install click')
-    os.system('pip3 install PyGithub')
+
+    os.system("pip3 install click")
+    os.system("pip3 install PyGithub")
     import click
 
 
@@ -12,69 +13,148 @@ except ImportError:
 def cli():
     pass
 
+
 @cli.command()
 def setup():
     """Installs dependencies needed for your system. Works with Linux, MacOS and Windows WSL."""
     import os
     import subprocess
+
     script_dir = os.path.dirname(os.path.realpath(__file__))
-    setup_script = os.path.join(script_dir, 'setup.sh')
+    setup_script = os.path.join(script_dir, "setup.sh")
     if os.path.exists(setup_script):
         subprocess.Popen([setup_script], cwd=script_dir)
-        click.echo(click.style("🚀 Setup initiated", fg='green'))
+        click.echo(click.style("🚀 Setup initiated", fg="green"))
     else:
-        click.echo(click.style("❌ Error: setup.sh does not exist in the current directory.", fg='red'))
+        click.echo(
+            click.style(
+                "❌ Error: setup.sh does not exist in the current directory.", fg="red"
+            )
+        )
 
     try:
         # Check if GitHub user name is configured
-        user_name = subprocess.check_output(['git', 'config', 'user.name']).decode('utf-8').strip()
-        user_email = subprocess.check_output(['git', 'config', 'user.email']).decode('utf-8').strip()
-        
+        user_name = (
+            subprocess.check_output(["git", "config", "user.name"])
+            .decode("utf-8")
+            .strip()
+        )
+        user_email = (
+            subprocess.check_output(["git", "config", "user.email"])
+            .decode("utf-8")
+            .strip()
+        )
+
         if user_name and user_email:
-            click.echo(click.style(f"✅ GitHub account is configured with username: {user_name} and email: {user_email}", fg='green'))
+            click.echo(
+                click.style(
+                    f"✅ GitHub account is configured with username: {user_name} and email: {user_email}",
+                    fg="green",
+                )
+            )
         else:
-            raise subprocess.CalledProcessError(returncode=1, cmd='git config user.name or user.email')
-            
+            raise subprocess.CalledProcessError(
+                returncode=1, cmd="git config user.name or user.email"
+            )
+
     except subprocess.CalledProcessError:
         # If the GitHub account is not configured, print instructions on how to set it up
-        click.echo(click.style("❌ GitHub account is not configured.", fg='red'))
-        click.echo(click.style("To configure your GitHub account, use the following commands:", fg='red'))
-        click.echo(click.style("  git config --global user.name \"Your GitHub Username\"", fg='red'))
-        click.echo(click.style("  git config --global user.email \"Your GitHub Email\"", fg='red'))
+        click.echo(click.style("❌ GitHub account is not configured.", fg="red"))
+        click.echo(
+            click.style(
+                "To configure your GitHub account, use the following commands:",
+                fg="red",
+            )
+        )
+        click.echo(
+            click.style(
+                '  git config --global user.name "Your GitHub Username"', fg="red"
+            )
+        )
+        click.echo(
+            click.style(
+                '  git config --global user.email "Your GitHub Email"', fg="red"
+            )
+        )
 
     # Check for the existence of the .github_access_token file
-    if os.path.exists('.github_access_token'):
-        with open('.github_access_token', 'r') as file:
+    if os.path.exists(".github_access_token"):
+        with open(".github_access_token", "r") as file:
             github_access_token = file.read().strip()
             if github_access_token:
-                click.echo(click.style("✅ GitHub access token loaded successfully.", fg='green'))
+                click.echo(
+                    click.style(
+                        "✅ GitHub access token loaded successfully.", fg="green"
+                    )
+                )
                 # Check if the token has the required permissions
                 import requests
-                headers = {'Authorization': f'token {github_access_token}'}
-                response = requests.get('https://api.github.com/user', headers=headers)
+
+                headers = {"Authorization": f"token {github_access_token}"}
+                response = requests.get("https://api.github.com/user", headers=headers)
                 if response.status_code == 200:
-                    scopes = response.headers.get('X-OAuth-Scopes')
-                    if 'public_repo' in scopes or 'repo' in scopes:
-                        click.echo(click.style("✅ GitHub access token has the required permissions.", fg='green'))
+                    scopes = response.headers.get("X-OAuth-Scopes")
+                    if "public_repo" in scopes or "repo" in scopes:
+                        click.echo(
+                            click.style(
+                                "✅ GitHub access token has the required permissions.",
+                                fg="green",
+                            )
+                        )
                     else:
-                        click.echo(click.style("❌ GitHub access token does not have the required permissions. Please ensure it has 'public_repo' or 'repo' scope.", fg='red'))
+                        click.echo(
+                            click.style(
+                                "❌ GitHub access token does not have the required permissions. Please ensure it has 'public_repo' or 'repo' scope.",
+                                fg="red",
+                            )
+                        )
                 else:
-                    click.echo(click.style("❌ Failed to validate GitHub access token. Please ensure it is correct.", fg='red'))
+                    click.echo(
+                        click.style(
+                            "❌ Failed to validate GitHub access token. Please ensure it is correct.",
+                            fg="red",
+                        )
+                    )
             else:
-                click.echo(click.style("❌ GitHub access token file is empty. Please follow the instructions below to set up your GitHub access token.", fg='red'))
+                click.echo(
+                    click.style(
+                        "❌ GitHub access token file is empty. Please follow the instructions below to set up your GitHub access token.",
+                        fg="red",
+                    )
+                )
     else:
         # Create the .github_access_token file if it doesn't exist
-        with open('.github_access_token', 'w') as file:
-            file.write('')
+        with open(".github_access_token", "w") as file:
+            file.write("")
 
         # Instructions to set up GitHub access token
-        click.echo(click.style("❌ To configure your GitHub access token, follow these steps:", fg='red'))
-        click.echo(click.style("\t1. Ensure you are logged into your GitHub account", fg='red'))
-        click.echo(click.style("\t2. Navigate to https://github.com/settings/tokens", fg='red'))
-        click.echo(click.style("\t6. Click on 'Generate new token'.", fg='red'))
-        click.echo(click.style("\t7. Fill out the form to generate a new token. Ensure you select the 'repo' scope.", fg='red'))
-        click.echo(click.style("\t8. Open the '.github_access_token' file in the same directory as this script and paste the token into this file.", fg='red'))
-        click.echo(click.style("\t9. Save the file and run the setup command again.", fg='red'))
+        click.echo(
+            click.style(
+                "❌ To configure your GitHub access token, follow these steps:", fg="red"
+            )
+        )
+        click.echo(
+            click.style("\t1. Ensure you are logged into your GitHub account", fg="red")
+        )
+        click.echo(
+            click.style("\t2. Navigate to https://github.com/settings/tokens", fg="red")
+        )
+        click.echo(click.style("\t6. Click on 'Generate new token'.", fg="red"))
+        click.echo(
+            click.style(
+                "\t7. Fill out the form to generate a new token. Ensure you select the 'repo' scope.",
+                fg="red",
+            )
+        )
+        click.echo(
+            click.style(
+                "\t8. Open the '.github_access_token' file in the same directory as this script and paste the token into this file.",
+                fg="red",
+            )
+        )
+        click.echo(
+            click.style("\t9. Save the file and run the setup command again.", fg="red")
+        )
 
 
 @cli.group()
@@ -82,54 +162,89 @@ def agent():
     """Commands to create, start and stop agents"""
     pass
 
+
 @agent.command()
-@click.argument('agent_name')
+@click.argument("agent_name")
 def create(agent_name):
     """Create's a new agent with the agent name provieded"""
     import os
-    import shutil
     import re
+    import shutil
+
     if not re.match("^[a-zA-Z0-9_-]*$", agent_name):
-        click.echo(click.style(f"😞 Agent name '{agent_name}' is not valid. It should not contain spaces or special characters other than -_", fg='red'))
+        click.echo(
+            click.style(
+                f"😞 Agent name '{agent_name}' is not valid. It should not contain spaces or special characters other than -_",
+                fg="red",
+            )
+        )
         return
     try:
-        new_agent_dir = f'./autogpts/{agent_name}'
-        agent_json_file = f'./arena/{agent_name}.json'
-        
+        new_agent_dir = f"./autogpts/{agent_name}"
+        agent_json_file = f"./arena/{agent_name}.json"
+
         if not os.path.exists(new_agent_dir) and not os.path.exists(agent_json_file):
-            shutil.copytree('./autogpts/forge', new_agent_dir)
-            click.echo(click.style(f"🎉 New agent '{agent_name}' created. The code for your new agent is in: autogpts/{agent_name}", fg='green'))
-            click.echo(click.style(f"🚀 If you would like to enter the arena, run './run arena enter {agent_name}'", fg='yellow'))
+            shutil.copytree("./autogpts/forge", new_agent_dir)
+            click.echo(
+                click.style(
+                    f"🎉 New agent '{agent_name}' created. The code for your new agent is in: autogpts/{agent_name}",
+                    fg="green",
+                )
+            )
+            click.echo(
+                click.style(
+                    f"🚀 If you would like to enter the arena, run './run arena enter {agent_name}'",
+                    fg="yellow",
+                )
+            )
         else:
-            click.echo(click.style(f"😞 Agent '{agent_name}' already exists. Enter a different name for your agent", fg='red'))
+            click.echo(
+                click.style(
+                    f"😞 Agent '{agent_name}' already exists. Enter a different name for your agent",
+                    fg="red",
+                )
+            )
     except Exception as e:
-        click.echo(click.style(f"😢 An error occurred: {e}", fg='red'))
+        click.echo(click.style(f"😢 An error occurred: {e}", fg="red"))
 
 
 @agent.command()
-@click.argument('agent_name')
+@click.argument("agent_name")
 def start(agent_name):
     """Start agent command"""
     import os
     import subprocess
+
     script_dir = os.path.dirname(os.path.realpath(__file__))
-    agent_dir = os.path.join(script_dir, f'autogpts/{agent_name}')
-    run_command = os.path.join(agent_dir, 'run')
+    agent_dir = os.path.join(script_dir, f"autogpts/{agent_name}")
+    run_command = os.path.join(agent_dir, "run")
     if os.path.exists(agent_dir) and os.path.isfile(run_command):
         os.chdir(agent_dir)
         subprocess.Popen(["./run"], cwd=agent_dir)
         click.echo(f"Agent '{agent_name}' started")
     elif not os.path.exists(agent_dir):
-        click.echo(click.style(f"😞 Agent '{agent_name}' does not exist. Please create the agent first.", fg='red'))
+        click.echo(
+            click.style(
+                f"😞 Agent '{agent_name}' does not exist. Please create the agent first.",
+                fg="red",
+            )
+        )
     else:
-        click.echo(click.style(f"😞 Run command does not exist in the agent '{agent_name}' directory.", fg='red'))
+        click.echo(
+            click.style(
+                f"😞 Run command does not exist in the agent '{agent_name}' directory.",
+                fg="red",
+            )
+        )
+
 
 @agent.command()
 def stop():
     """Stop agent command"""
-    import subprocess
     import os
     import signal
+    import subprocess
+
     try:
         pid = int(subprocess.check_output(["lsof", "-t", "-i", ":8000"]))
         os.kill(pid, signal.SIGTERM)
@@ -144,19 +259,24 @@ def stop():
 def list():
     """List agents command"""
     import os
+
     try:
-        agents_dir = './autogpts'
-        agents_list = [d for d in os.listdir(agents_dir) if os.path.isdir(os.path.join(agents_dir, d))]
+        agents_dir = "./autogpts"
+        agents_list = [
+            d
+            for d in os.listdir(agents_dir)
+            if os.path.isdir(os.path.join(agents_dir, d))
+        ]
         if agents_list:
-            click.echo(click.style('Available agents: 🤖', fg='green'))
+            click.echo(click.style("Available agents: 🤖", fg="green"))
             for agent in agents_list:
-                click.echo(click.style(f"\t🐙 {agent}", fg='blue'))
+                click.echo(click.style(f"\t🐙 {agent}", fg="blue"))
         else:
-            click.echo(click.style("No agents found 😞", fg='red'))
+            click.echo(click.style("No agents found 😞", fg="red"))
     except FileNotFoundError:
-        click.echo(click.style("The autogpts directory does not exist 😢", fg='red'))
+        click.echo(click.style("The autogpts directory does not exist 😢", fg="red"))
     except Exception as e:
-        click.echo(click.style(f"An error occurred: {e} 😢", fg='red'))
+        click.echo(click.style(f"An error occurred: {e} 😢", fg="red"))
 
 
 @cli.group()
@@ -164,43 +284,61 @@ def benchmark():
     """Commands to start the benchmark and list tests and categories"""
     pass
 
-@benchmark.command(context_settings=dict(
-    ignore_unknown_options=True,
-))
-@click.argument('agent_name')
-@click.argument('subprocess_args', nargs=-1, type=click.UNPROCESSED)
+
+@benchmark.command(
+    context_settings=dict(
+        ignore_unknown_options=True,
+    )
+)
+@click.argument("agent_name")
+@click.argument("subprocess_args", nargs=-1, type=click.UNPROCESSED)
 def start(agent_name, subprocess_args):
     """Starts the benchmark command"""
     import os
     import subprocess
+
     script_dir = os.path.dirname(os.path.realpath(__file__))
-    agent_dir = os.path.join(script_dir, f'autogpts/{agent_name}')
-    benchmark_script = os.path.join(agent_dir, 'run_benchmark.sh')
+    agent_dir = os.path.join(script_dir, f"autogpts/{agent_name}")
+    benchmark_script = os.path.join(agent_dir, "run_benchmark.sh")
     if os.path.exists(agent_dir) and os.path.isfile(benchmark_script):
         os.chdir(agent_dir)
         subprocess.Popen([benchmark_script, *subprocess_args], cwd=agent_dir)
-        click.echo(click.style(f"🚀 Running benchmark for '{agent_name}' with subprocess arguments: {' '.join(subprocess_args)}", fg='green'))
+        click.echo(
+            click.style(
+                f"🚀 Running benchmark for '{agent_name}' with subprocess arguments: {' '.join(subprocess_args)}",
+                fg="green",
+            )
+        )
     else:
-        click.echo(click.style(f"😞 Agent '{agent_name}' does not exist. Please create the agent first.", fg='red'))
+        click.echo(
+            click.style(
+                f"😞 Agent '{agent_name}' does not exist. Please create the agent first.",
+                fg="red",
+            )
+        )
 
 
-@benchmark.group(name='categories')
+@benchmark.group(name="categories")
 def benchmark_categories():
     """Benchmark categories group command"""
     pass
 
-@benchmark_categories.command(name='list')
+
+@benchmark_categories.command(name="list")
 def benchmark_categories_list():
     """List benchmark categories command"""
-    import os
-    import json
     import glob
+    import json
+    import os
+
     categories = set()
 
     # Get the directory of this file
     this_dir = os.path.dirname(os.path.abspath(__file__))
 
-    glob_path = os.path.join(this_dir, "./benchmark/agbenchmark/challenges/**/[!deprecated]*/data.json")
+    glob_path = os.path.join(
+        this_dir, "./benchmark/agbenchmark/challenges/**/[!deprecated]*/data.json"
+    )
     # Use it as the base for the glob pattern, excluding 'deprecated' directory
     for data_file in glob.glob(glob_path, recursive=True):
         with open(data_file, "r") as f:
@@ -215,30 +353,35 @@ def benchmark_categories_list():
                 continue
 
     if categories:
-        click.echo(click.style('Available categories: 📚', fg='green'))
+        click.echo(click.style("Available categories: 📚", fg="green"))
         for category in categories:
-            click.echo(click.style(f"\t📖 {category}", fg='blue'))
+            click.echo(click.style(f"\t📖 {category}", fg="blue"))
     else:
-        click.echo(click.style("No categories found 😞", fg='red'))
+        click.echo(click.style("No categories found 😞", fg="red"))
+
 
-@benchmark.group(name='tests')
+@benchmark.group(name="tests")
 def benchmark_tests():
     """Benchmark tests group command"""
     pass
 
-@benchmark_tests.command(name='list')
+
+@benchmark_tests.command(name="list")
 def benchmark_tests_list():
     """List benchmark tests command"""
-    import os
-    import json
     import glob
+    import json
+    import os
     import re
+
     tests = {}
 
     # Get the directory of this file
     this_dir = os.path.dirname(os.path.abspath(__file__))
 
-    glob_path = os.path.join(this_dir, "./benchmark/agbenchmark/challenges/**/[!deprecated]*/data.json")
+    glob_path = os.path.join(
+        this_dir, "./benchmark/agbenchmark/challenges/**/[!deprecated]*/data.json"
+    )
     # Use it as the base for the glob pattern, excluding 'deprecated' directory
     for data_file in glob.glob(glob_path, recursive=True):
         with open(data_file, "r") as f:
@@ -258,64 +401,133 @@ def benchmark_tests_list():
                 continue
 
     if tests:
-        click.echo(click.style('Available tests: 📚', fg='green'))
+        click.echo(click.style("Available tests: 📚", fg="green"))
         for category, test_list in tests.items():
-            click.echo(click.style(f"\t📖 {category}", fg='blue'))
+            click.echo(click.style(f"\t📖 {category}", fg="blue"))
             for test in sorted(test_list):
-                test_name = ' '.join(word for word in re.split('([A-Z][a-z]*)', test) if word).replace('_', '').replace('C L I', 'CLI')[5:].replace('  ', ' ')
+                test_name = (
+                    " ".join(word for word in re.split("([A-Z][a-z]*)", test) if word)
+                    .replace("_", "")
+                    .replace("C L I", "CLI")[5:]
+                    .replace("  ", " ")
+                )
                 test_name_padded = f"{test_name:<40}"
-                click.echo(click.style(f"\t\t🔬 {test_name_padded} - {test}", fg='cyan'))
+                click.echo(click.style(f"\t\t🔬 {test_name_padded} - {test}", fg="cyan"))
     else:
-        click.echo(click.style("No tests found 😞", fg='red'))
-        
-@benchmark_tests.command(name='details')
-@click.argument('test_name')
+        click.echo(click.style("No tests found 😞", fg="red"))
+
+
+@benchmark_tests.command(name="details")
+@click.argument("test_name")
 def benchmark_tests_details(test_name):
     """Benchmark test details command"""
-    import os
-    import json
     import glob
+    import json
+    import os
 
     # Get the directory of this file
     this_dir = os.path.dirname(os.path.abspath(__file__))
 
-    glob_path = os.path.join(this_dir, "./benchmark/agbenchmark/challenges/**/[!deprecated]*/data.json")
+    glob_path = os.path.join(
+        this_dir, "./benchmark/agbenchmark/challenges/**/[!deprecated]*/data.json"
+    )
     # Use it as the base for the glob pattern, excluding 'deprecated' directory
     for data_file in glob.glob(glob_path, recursive=True):
         with open(data_file, "r") as f:
             try:
                 data = json.load(f)
                 if data.get("name") == test_name:
-                    click.echo(click.style(f"\n{data.get('name')}\n{'-'*len(data.get('name'))}\n", fg='blue'))
-                    click.echo(click.style(f"\tCategory:  {', '.join(data.get('category'))}", fg='green'))
-                    click.echo(click.style(f"\tTask:  {data.get('task')}", fg='green'))
-                    click.echo(click.style(f"\tDependencies:  {', '.join(data.get('dependencies')) if data.get('dependencies') else 'None'}", fg='green'))
-                    click.echo(click.style(f"\tCutoff:  {data.get('cutoff')}\n", fg='green'))
-                    click.echo(click.style("\tTest Conditions\n\t-------", fg='magenta'))
-                    click.echo(click.style(f"\t\tAnswer: {data.get('ground').get('answer')}", fg='magenta'))
-                    click.echo(click.style(f"\t\tShould Contain: {', '.join(data.get('ground').get('should_contain'))}", fg='magenta'))
-                    click.echo(click.style(f"\t\tShould Not Contain: {', '.join(data.get('ground').get('should_not_contain'))}", fg='magenta'))
-                    click.echo(click.style(f"\t\tFiles: {', '.join(data.get('ground').get('files'))}", fg='magenta'))
-                    click.echo(click.style(f"\t\tEval: {data.get('ground').get('eval').get('type')}\n", fg='magenta'))
-                    click.echo(click.style("\tInfo\n\t-------", fg='yellow'))
-                    click.echo(click.style(f"\t\tDifficulty: {data.get('info').get('difficulty')}", fg='yellow'))
-                    click.echo(click.style(f"\t\tDescription: {data.get('info').get('description')}", fg='yellow'))
-                    click.echo(click.style(f"\t\tSide Effects: {', '.join(data.get('info').get('side_effects'))}", fg='yellow'))
+                    click.echo(
+                        click.style(
+                            f"\n{data.get('name')}\n{'-'*len(data.get('name'))}\n",
+                            fg="blue",
+                        )
+                    )
+                    click.echo(
+                        click.style(
+                            f"\tCategory:  {', '.join(data.get('category'))}",
+                            fg="green",
+                        )
+                    )
+                    click.echo(click.style(f"\tTask:  {data.get('task')}", fg="green"))
+                    click.echo(
+                        click.style(
+                            f"\tDependencies:  {', '.join(data.get('dependencies')) if data.get('dependencies') else 'None'}",
+                            fg="green",
+                        )
+                    )
+                    click.echo(
+                        click.style(f"\tCutoff:  {data.get('cutoff')}\n", fg="green")
+                    )
+                    click.echo(
+                        click.style("\tTest Conditions\n\t-------", fg="magenta")
+                    )
+                    click.echo(
+                        click.style(
+                            f"\t\tAnswer: {data.get('ground').get('answer')}",
+                            fg="magenta",
+                        )
+                    )
+                    click.echo(
+                        click.style(
+                            f"\t\tShould Contain: {', '.join(data.get('ground').get('should_contain'))}",
+                            fg="magenta",
+                        )
+                    )
+                    click.echo(
+                        click.style(
+                            f"\t\tShould Not Contain: {', '.join(data.get('ground').get('should_not_contain'))}",
+                            fg="magenta",
+                        )
+                    )
+                    click.echo(
+                        click.style(
+                            f"\t\tFiles: {', '.join(data.get('ground').get('files'))}",
+                            fg="magenta",
+                        )
+                    )
+                    click.echo(
+                        click.style(
+                            f"\t\tEval: {data.get('ground').get('eval').get('type')}\n",
+                            fg="magenta",
+                        )
+                    )
+                    click.echo(click.style("\tInfo\n\t-------", fg="yellow"))
+                    click.echo(
+                        click.style(
+                            f"\t\tDifficulty: {data.get('info').get('difficulty')}",
+                            fg="yellow",
+                        )
+                    )
+                    click.echo(
+                        click.style(
+                            f"\t\tDescription: {data.get('info').get('description')}",
+                            fg="yellow",
+                        )
+                    )
+                    click.echo(
+                        click.style(
+                            f"\t\tSide Effects: {', '.join(data.get('info').get('side_effects'))}",
+                            fg="yellow",
+                        )
+                    )
                     break
 
-
             except json.JSONDecodeError:
                 print(f"Error: {data_file} is not a valid JSON file.")
                 continue
             except IOError:
                 print(f"IOError: file could not be read: {data_file}")
                 continue
+
+
 @cli.command()
 def frontend():
     """Starts the frontend"""
     import os
-    import subprocess
     import socket
+    import subprocess
+
     try:
         output = subprocess.check_output(["lsof", "-t", "-i", ":8000"])
         if output:
@@ -325,8 +537,8 @@ def frontend():
     except subprocess.CalledProcessError as e:
         click.echo("Error: Unexpected error occurred.")
         return
-    frontend_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'frontend')
-    run_file = os.path.join(frontend_dir, 'run')
+    frontend_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), "frontend")
+    run_file = os.path.join(frontend_dir, "run")
     if os.path.exists(frontend_dir) and os.path.isfile(run_file):
         subprocess.Popen(["./run"], cwd=frontend_dir)
         click.echo("Launching frontend")
@@ -334,49 +546,79 @@ def frontend():
         click.echo("Error: Frontend directory or run file does not exist.")
 
 
-
 @cli.group()
 def arena():
     """Commands to enter the arena"""
     pass
 
+
 @arena.command()
-@click.argument('agent_name')
-@click.option('--branch', default='master', help='Branch to use instead of master')
+@click.argument("agent_name")
+@click.option("--branch", default="master", help="Branch to use instead of master")
 def enter(agent_name, branch):
+    import json
+    import os
     import subprocess
-    from github import Github
     from datetime import datetime
-    import os
-    import json
+
+    from github import Github
+
     # Check if the agent_name directory exists in the autogpts directory
-    agent_dir = f'./autogpts/{agent_name}'
+    agent_dir = f"./autogpts/{agent_name}"
     if not os.path.exists(agent_dir):
-        click.echo(click.style(f"❌ The directory for agent '{agent_name}' does not exist in the autogpts directory.", fg='red'))
-        click.echo(click.style(f"🚀 Run './run agent create {agent_name}' to create the agent.", fg='yellow'))
-        
+        click.echo(
+            click.style(
+                f"❌ The directory for agent '{agent_name}' does not exist in the autogpts directory.",
+                fg="red",
+            )
+        )
+        click.echo(
+            click.style(
+                f"🚀 Run './run agent create {agent_name}' to create the agent.",
+                fg="yellow",
+            )
+        )
+
         return
-    else:    
+    else:
         # Check if the agent has already entered the arena
-        if os.path.exists(f'arena/{agent_name}.json'):
-            click.echo(click.style(f"⚠️  The agent '{agent_name}' has already entered the arena. Use './run arena submit' to update your submission.", fg='yellow'))
+        if os.path.exists(f"arena/{agent_name}.json"):
+            click.echo(
+                click.style(
+                    f"⚠️  The agent '{agent_name}' has already entered the arena. Use './run arena submit' to update your submission.",
+                    fg="yellow",
+                )
+            )
             return
-    
+
     # Check if there are staged changes
-    staged_changes = [line for line in subprocess.check_output(['git', 'status', '--porcelain']).decode('utf-8').split('\n') if line and line[0] in ('A', 'M', 'D', 'R', 'C')]
+    staged_changes = [
+        line
+        for line in subprocess.check_output(["git", "status", "--porcelain"])
+        .decode("utf-8")
+        .split("\n")
+        if line and line[0] in ("A", "M", "D", "R", "C")
+    ]
     if staged_changes:
-        click.echo(click.style(f"❌ There are staged changes. Please commit or stash them and run the command again.", fg='red'))
+        click.echo(
+            click.style(
+                f"❌ There are staged changes. Please commit or stash them and run the command again.",
+                fg="red",
+            )
+        )
         return
 
-
-
     try:
         # Load GitHub access token from file
-        with open('.github_access_token', 'r') as file:
+        with open(".github_access_token", "r") as file:
             github_access_token = file.read().strip()
 
         # Get GitHub repository URL
-        github_repo_url = subprocess.check_output(['git', 'config', '--get', 'remote.origin.url']).decode('utf-8').strip()
+        github_repo_url = (
+            subprocess.check_output(["git", "config", "--get", "remote.origin.url"])
+            .decode("utf-8")
+            .strip()
+        )
 
         # If --branch is passed, use it instead of master
         if branch:
@@ -385,12 +627,16 @@ def enter(agent_name, branch):
             branch_to_use = "master"
 
         # Get the commit hash of HEAD of the branch_to_use
-        commit_hash_to_benchmark = subprocess.check_output(['git', 'rev-parse', branch_to_use]).decode('utf-8').strip()
-        
-        arena_submission_branch = f'arena_submission_{agent_name}'
+        commit_hash_to_benchmark = (
+            subprocess.check_output(["git", "rev-parse", branch_to_use])
+            .decode("utf-8")
+            .strip()
+        )
+
+        arena_submission_branch = f"arena_submission_{agent_name}"
         # Create a new branch called arena_submission_{agent_name}
         # subprocess.check_call(['git', 'checkout', '-b', arena_submission_branch])
-        subprocess.check_call(['git', 'checkout', arena_submission_branch])
+        subprocess.check_call(["git", "checkout", arena_submission_branch])
         # Create a dictionary with the necessary fields
         data = {
             "github_repo_url": github_repo_url,
@@ -403,50 +649,61 @@ def enter(agent_name, branch):
             data["branch_to_benchmark"] = branch
 
         # Create agent directory if it does not exist
-        subprocess.check_call(['mkdir', '-p', 'arena'])
+        subprocess.check_call(["mkdir", "-p", "arena"])
 
         # Create a JSON file with the data
-        with open(f'arena/{agent_name}.json', 'w') as json_file:
+        with open(f"arena/{agent_name}.json", "w") as json_file:
             json.dump(data, json_file, indent=4)
 
         # Create a commit with the specified message
-        subprocess.check_call(['git', 'add', f'arena/{agent_name}.json'])
-        subprocess.check_call(['git', 'commit', '-m', f'{agent_name} entering the arena'])
+        subprocess.check_call(["git", "add", f"arena/{agent_name}.json"])
+        subprocess.check_call(
+            ["git", "commit", "-m", f"{agent_name} entering the arena"]
+        )
 
         # Push the commit
-        subprocess.check_call(['git', 'push', 'origin', arena_submission_branch])
+        subprocess.check_call(["git", "push", "origin", arena_submission_branch])
 
         # Create a PR into the parent repository
         g = Github(github_access_token)
-        repo = g.get_repo(github_repo_url.split(':')[-1].split('.git')[0])
+        repo = g.get_repo(github_repo_url.split(":")[-1].split(".git")[0])
         parent_repo = repo.parent
         if parent_repo:
             pr = parent_repo.create_pull(
-                title=f'{agent_name} entering the arena',
-                body='''**Introduction:** 
+                title=f"{agent_name} entering the arena",
+                body="""**Introduction:** 
 
 **Team Members:** 
 
 **What we are working on:** 
 
-Please replace this text with your own introduction, the names of your team members, and a brief description of your work.''',
-                head=f'arena_submission_{agent_name}',
-
+Please replace this text with your own introduction, the names of your team members, and a brief description of your work.""",
+                head=f"arena_submission_{agent_name}",
                 base=branch_to_use,
             )
-            click.echo(click.style(f"🚀 {agent_name} has entered the arena! Please edit your PR description at the following URL: {pr.html_url}", fg='green'))
+            click.echo(
+                click.style(
+                    f"🚀 {agent_name} has entered the arena! Please edit your PR description at the following URL: {pr.html_url}",
+                    fg="green",
+                )
+            )
         else:
-            click.echo(click.style("❌ This repository does not have a parent repository to sync with.", fg='red'))
+            click.echo(
+                click.style(
+                    "❌ This repository does not have a parent repository to sync with.",
+                    fg="red",
+                )
+            )
             return
 
         # Switch back to the master branch
-        subprocess.check_call(['git', 'checkout', branch_to_use])
-        
+        subprocess.check_call(["git", "checkout", branch_to_use])
+
     except Exception as e:
-        click.echo(click.style(f"❌ An error occurred: {e}", fg='red'))
+        click.echo(click.style(f"❌ An error occurred: {e}", fg="red"))
         # Switch back to the master branch
-        subprocess.check_call(['git', 'checkout', branch_to_use])
+        subprocess.check_call(["git", "checkout", branch_to_use])
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     cli()