diff --git a/events/ai.py b/events/ai.py index eb4966f..1b4bc98 100644 --- a/events/ai.py +++ b/events/ai.py @@ -62,13 +62,15 @@ class AI(commands.Cog): SquogData.append(SquogUser) SquogUser["messages"].append({"role": "user", "content": message.content}) + try: + SquogResponse = await getResponse(SquogUser["messages"]) - SquogResponse = await getResponse(SquogUser["messages"]) + SquogUser["messages"].append(SquogResponse["choices"][0]["message"]) + await message.reply(SquogResponse["choices"][0]["message"]["content"]) - SquogUser["messages"].append(SquogResponse["choices"][0]["message"]) - await message.reply(SquogResponse["choices"][0]["message"]["content"]) - - await flush() + await flush() + except: + await message.reply("Failed to load a reply.") def setup(bot): bot.add_cog(AI(bot)) \ No newline at end of file