Updated code to work on LAION server

This commit is contained in:
Birger Moell
2022-12-14 21:55:33 +01:00
parent 0ff400c947
commit f04e0efd49
+13 -5
View File
@@ -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!")