phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

allow admin to increase dalle budget with commands

Browse source
This commit is contained in:
phixxy 2024-03-16 00:35:18 -07:00
parent 6e74ac1fd5
commit 3786a1434a
1 changed files with 15 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

15
cogs/chatgpt.py
View file

@ -122,6 +122,21 @@ class ChatGPT(commands.Cog):
categories = response_data['results'][0]['categories']
category_scores = response_data['results'][0]['category_scores']
return (flagged, categories, category_scores)
@commands.command()
async def budget(self, ctx, command: str, budget: int):
if ctx.author.id == self.admin_id:
if command == "add":
self.dalle_budget += budget
await ctx.send(f"Budget increased by {budget}")
elif command == "remove":
self.dalle_budget -= budget
await ctx.send(f"Budget decreased by {budget}")
elif command == "set":
self.dalle_budget = budget
await ctx.send(f"Budget set to {budget}")
else:
await ctx.send(f"The current budget is {self.dalle_budget}")
@commands.command(