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

try/except in budget and allow adding floats

Browse source
This commit is contained in:
phixxy 2024-03-16 00:38:47 -07:00
parent 3786a1434a
commit f0dccd2233
1 changed files with 19 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

8
cogs/chatgpt.py
View file

@ -124,7 +124,8 @@ class ChatGPT(commands.Cog):
return (flagged, categories, category_scores) return (flagged, categories, category_scores)
@commands.command() @commands.command()
async def budget(self, ctx, command: str, budget: int): async def budget(self, ctx, command: str, budget: float):
try:
if ctx.author.id == self.admin_id: if ctx.author.id == self.admin_id:
if command == "add": if command == "add":
self.dalle_budget += budget self.dalle_budget += budget
@ -137,6 +138,11 @@ class ChatGPT(commands.Cog):
await ctx.send(f"Budget set to {budget}") await ctx.send(f"Budget set to {budget}")
else: else:
await ctx.send(f"The current budget is {self.dalle_budget}") await ctx.send(f"The current budget is {self.dalle_budget}")
else:
await ctx.send(f"The current budget is {self.dalle_budget}")
except Exception as e:
self.logger.exception(f"Budget command failed: {e}")
await ctx.send(f"Budget command failed")
@commands.command( @commands.command(