mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-06-27 16:10:30 +08:00
Updated code to work on LAION server
This commit is contained in:
+13
-5
@@ -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!")
|
||||
|
||||
Reference in New Issue
Block a user