edpibu/botbotbot
1
Fork 0
Code Issues Releases Activity

Fix AI

Browse source
This commit is contained in:
Edgar P. Burkhart 2024-06-10 09:43:46 +02:00
parent dbe4014421
commit 130013f1e1
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
botbotbot/__main__.py
View file

@ -20,6 +20,7 @@ aibot = AIBot(config.get("mistral_api_key"), model="open-mixtral-8x7b")
intents = discord.Intents.default() intents = discord.Intents.default()
intents.members = True intents.members = True
intents.message_content = True
bot = discord.Bot(description=description, intents=intents) bot = discord.Bot(description=description, intents=intents)
@ -41,7 +42,10 @@ async def reply(message):
async def ai_reply(message): async def ai_reply(message):
print("Running AI") 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( embed = discord.Embed(
description=answer, description=answer,
thumbnail="https://mistral.ai/images/favicon/favicon-32x32.png", thumbnail="https://mistral.ai/images/favicon/favicon-32x32.png",