Update code.py

This commit is contained in:
Paillat
2022-12-06 13:49:45 +01:00
parent 6af21f9ae0
commit 32478026ee

View File

@@ -38,6 +38,11 @@ async def setup(ctx, channel: discord.TextChannel, api_key):
except:
await ctx.respond("Invalid api key", ephemeral=True)
return
#if there is not enough tokens remaining, send a message to the user
if openai.Completion.create(engine="davinci", prompt="Hello world", max_tokens=1).headers["X-RateLimit-Remaining"] == 0:
await ctx.respond("Not enough tokens remaining", ephemeral=True)
debug("Not enough tokens remaining")
return
#check if the channel is valid
if channel is None:
await ctx.respond("Invalid channel id", ephemeral=True)