0xMRTT/matrixai
0xMRTT/matrixai
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add exception handling

Browse Source
This commit is contained in:
0xMRTT 2023-06-08 20:38:15 +02:00
parent c027b2e6cc
commit 0281d1267d
Signed by: 0xMRTT
GPG Key ID: 910B287304120902
1 changed files with 22 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
matrixai/__init__.py
View File

@ -404,11 +404,17 @@ def run():
prompt = " ".join(arg for arg in match.args()) prompt = " ".join(arg for arg in match.args())
response = "".join(Completion.create(prompt)) response = "".join(Completion.create(prompt))
try:
await bot.api.send_markdown_message( await bot.api.send_markdown_message(
room.room_id, f"> {prompt}\n\n{response}" room.room_id, f"> {prompt}\n\n{response}"
) )
except Exception as e:
print(e)
await bot.api.send_markdown_message(
room.room_id, f"> {prompt}\n\n{e}"
)
@bot.listener.on_message_event @bot.listener.on_message_event
async def image(room, message): async def image(room, message):
match = botlib.MessageMatch(room, message, bot, PREFIX) match = botlib.MessageMatch(room, message, bot, PREFIX)
@ -473,12 +479,19 @@ def run():
return filename return filename
filename = await generate_image(prompt, style, ratio, negative) try:
filename = await generate_image(prompt, style, ratio, negative)
await bot.api.send_image_message( await bot.api.send_image_message(
room_id=room.room_id, room_id=room.room_id,
image_filepath=filename) image_filepath=filename)
except Exception as e:
print(e)
await bot.api.send_markdown_message(
room.room_id, f"> {prompt}\n\n{e}"
)
@bot.listener.on_message_event @bot.listener.on_message_event