diff --git a/botbotbot/__main__.py b/botbotbot/__main__.py index 72ee44f..88910f6 100644 --- a/botbotbot/__main__.py +++ b/botbotbot/__main__.py @@ -20,6 +20,7 @@ aibot = AIBot(config.get("mistral_api_key"), model="open-mixtral-8x7b") intents = discord.Intents.default() intents.members = True +intents.message_content = True bot = discord.Bot(description=description, intents=intents) @@ -41,7 +42,10 @@ async def reply(message): async def ai_reply(message): print("Running AI") - answer = aibot.answer(message.clean_content) + prompt = message.clean_content + if prompt == "" and message.embeds: + prompt = message.embeds[0].description + answer = aibot.answer(prompt) embed = discord.Embed( description=answer, thumbnail="https://mistral.ai/images/favicon/favicon-32x32.png",