diff --git a/bot/bot.py b/bot/bot.py index 2509138a..6f7ede73 100644 --- a/bot/bot.py +++ b/bot/bot.py @@ -27,7 +27,9 @@ headers = {"X-API-Key": API_SERVER_KEY} # For testing only. TEST_GUILD = os.getenv("TEST_GUILD") - +TEST_GUILD_LAION = os.getenv("TEST_GUILD_LAION") +# TEST_GUILD = False +guild_ids = [TEST_GUILD, TEST_GUILD_LAION] # Initiate the client and command tree to create slash commands. class OpenChatGPTClient(discord.Client): @@ -39,10 +41,16 @@ class OpenChatGPTClient(discord.Client): if TEST_GUILD: # When testing the bot it's handy to run in a single server (called a # Guide in the API). This is relatively fast. - guild = discord.Object(id=TEST_GUILD) - self.tree.copy_global_to(guild=guild) - await self.tree.sync(guild=guild) - else: + for guild_id in guild_ids: + guild = discord.Object(id=guild_id) + self.tree.copy_global_to(guild=guild) + await self.tree.sync(guild=guild) + + + # guild = discord.Object(id=TEST_GUILD) + # self.tree.copy_global_to(guild=guild) + # await self.tree.sync(guild=guild) + else: # This can take up to an hour for the commands to be registered. await self.tree.sync() logger.debug("Ready!")